import io
from PIL import Image, ImageDraw, ImageOps, ImageEnhance
from .base import BaseGenerator

class CommonGenerator(BaseGenerator):
    def build_template(self, image_data: bytes, element: str) -> Image.Image:
        r_color = self.ELEMENT_COLORS.get(element, (156, 163, 175))
        
        canvas = Image.new("RGBA", (self.CANVAS_W, self.CANVAS_H), (0, 0, 0, 0))

        char_raw = Image.open(io.BytesIO(image_data)).convert("RGBA")
        if char_raw.size != (self.CARD_W, self.CARD_H):
            char_base = ImageOps.fit(char_raw, (self.CARD_W, self.CARD_H), centering=(0.5, 0.15))
        else:
            char_base = char_raw
            
        char_base = ImageEnhance.Contrast(char_base).enhance(1.05)

        banner_crop = char_base.crop((0, self.CARD_H - self.BANNER_H, self.CARD_W, self.CARD_H))
        gradient = Image.new("RGBA", banner_crop.size, (20, 20, 25, 200))
        banner_img = Image.alpha_composite(banner_crop, gradient)
        
        banner_draw = ImageDraw.Draw(banner_img)
        banner_draw.line([(0, 0), (self.CARD_W, 0)], fill=r_color + (255,), width=2)
        banner_img.putalpha(self.create_bottom_rounded_mask(banner_img.size, self.CORNER_RAD))

        char_base.putalpha(self.create_rounded_mask((self.CARD_W, self.CARD_H), self.CORNER_RAD))
        canvas.paste(char_base, (self.OFFSET_X, self.OFFSET_Y), char_base)
        canvas.paste(banner_img, (self.OFFSET_X, self.OFFSET_Y + self.CARD_H - self.BANNER_H), banner_img)

        ui_layer = Image.new("RGBA", (self.CANVAS_W, self.CANVAS_H), (0, 0, 0, 0))
        uid = ImageDraw.Draw(ui_layer)
        uid.rounded_rectangle(
            (self.OFFSET_X, self.OFFSET_Y, self.OFFSET_X + self.CARD_W, self.OFFSET_Y + self.CARD_H), 
            radius=self.CORNER_RAD, 
            outline=r_color + (200,), 
            width=2
        )
        canvas.alpha_composite(ui_layer)

        return canvas