import logging
from typing import Optional

from PIL import Image

from .types import Text, Pos
from . import img_factory
from . import meme_db
from . import utils

logger = logging.getLogger("meme_otron")

right_wmark = Text("Made with meme-otron")
right_wmark.position = Pos.SE
right_wmark.fill = (128, 128, 128, 128)
right_wmark.font_size = 0.02
right_wmark.x_range = [0.005, 0.995]
right_wmark.y_range = [0.005, 0.995]

left_wmark = Text()
left_wmark.position = Pos.SW
left_wmark.fill = (128, 128, 128, 128)
left_wmark.font_size = 0.02
left_wmark.x_range = [0.005, 0.995]
left_wmark.y_range = [0.005, 0.995]


def compute(*args: str, left_wmark_text: Optional[Text] = None, debug: bool = False) -> Optional[Image.Image]:
    if len(args) < 1:
        return None

    parts = utils.split_arguments(args, "-")
    images = []
    for part in parts:
        images += [compute_part(*part, debug=debug)]

    output_image = img_factory.compose_image(images)

    watermarks = [right_wmark]
    if left_wmark_text is not None:
        left_wmark.text = left_wmark_text
        watermarks += [left_wmark]
    output_image = img_factory.apply_texts(output_image, watermarks, debug=debug)

    return output_image


def compute_part(*args: str, debug: bool = False) -> Optional[Image.Image]:
    meme_id = args[0]
    meme = meme_db.get_meme(meme_id)
    if meme is None:
        logger.warning(f"Meme template '{meme_id}' not found")
        return None
    if len(args) > 1:
        c = 0
        for i in range(len(meme.texts)):
            if meme.texts[i].text_ref is None:
                if c < len(args) - 1:
                    meme.texts[i].text = args[c + 1].replace("\\n", "\n")
                else:
                    meme.texts[i].text = ""
                c += 1
            else:
                meme.texts[i].text = meme.texts[meme.texts[i].text_ref].text
    return img_factory.build_image(meme.template, meme.texts, debug=debug)