from typing import Optional, Union
from PIL import Image, ImageDraw, ImageFont, ImageFilter

def text_draw_default(
    draw: ImageDraw.Draw,
    position: tuple,
    font: str,
    font_size: int,
    txt: str,
    txt_color: Optional[Union[str, tuple]] = None,
    variation: Optional[str] = None,
    only_size = False
):

    font = ImageFont.truetype(f"font/{font}", font_size)

    if variation: font.set_variation_by_name(variation)
    if not txt_color: txt_color = (255, 255, 255)  # Default to white
    if isinstance(txt_color, str): txt_color = hex2rgb(txt_color)

    txt_bbox = font.getbbox(txt)
    txt_size = (txt_bbox[2] - txt_bbox[0], txt_bbox[3] - txt_bbox[1])
    if not only_size:
        draw.text(position, txt, fill=txt_color, font=font)
    return txt_size

def text_draw_center(
        draw: ImageDraw.Draw,
        position: tuple,
        font: str,
        font_size: int,
        txt: str,
        txt_color: Optional[Union[str, tuple[int, int, int]]] = None,
        variation: Optional[str] = None,
        only_size = False
):
    font = ImageFont.truetype(f"font/{font}", font_size)

    # Apply variation if specified
    if variation: font.set_variation_by_name(variation)

    # Set default text color to white if not provided
    if not txt_color: txt_color = (255, 255, 255)  # Default to white
    if isinstance(txt_color, str): txt_color = hex2rgb(txt_color)

    # Calculate text size
    txt_bbox = font.getbbox(txt)
    txt_width = txt_bbox[2] - txt_bbox[0]
    txt_height = txt_bbox[3] - txt_bbox[1]

    # Calculate top-left position to center the text
    top_left_x = position[0] - txt_width / 2
    top_left_y = position[1] - txt_height / 2

    # Draw the text
    if not only_size:
        draw.text((top_left_x, top_left_y), txt, fill=txt_color, font=font)

    return int(txt_width), int(txt_height)

def text_draw_right(
    draw: ImageDraw.Draw,
    position: tuple,
    font: str,
    font_size: int,
    txt: str,
    txt_color: Optional[Union[str, tuple]] = None,
    variation: Optional[str] = None,
    only_size = False
):

    font = ImageFont.truetype(f"font/{font}", font_size)

    if variation: font.set_variation_by_name(variation)
    if not txt_color: txt_color = (255, 255, 255)  # Default to white
    if isinstance(txt_color, str): txt_color = hex2rgb(txt_color)

    txt_bbox = font.getbbox(txt)
    txt_size = (txt_bbox[2] - txt_bbox[0], txt_bbox[3] - txt_bbox[1])
    if not only_size:
        draw.text((position[0]-txt_size[0], position[1]), txt, fill=txt_color, font=font)
    return txt_size


def hex2rgb(hex_color):
    hex_color = hex_color.lstrip('#')
    return tuple(int(hex_color[i:i + 2], 16) for i in (0, 2, 4))