from os import makedirs
from os.path import join
from random import choice
from sys import argv
from urllib.request import urlretrieve

from PIL import Image, ImageDraw, ImageFont

zodiac_signs = [
    "Belier",
    "Cancer",
    "Balance",
    "Capricorne",
    "Taureau",
    "Lion",
    "Scorpion",
    "Verseau",
    "Gemeaux",
    "Vierge",
    "Sagittaire",
    "Poissons",
]


def generate_horoscope() -> dict[str, dict[str, str]]:
    """Generate horoscope predictions"""
    horoscope = {}
    for sign in zodiac_signs:
        love = choice(["+++", "++", "+", "-", "- -"])
        work = choice(["+++", "++", "+", "-", "- -"])
        horoscope[sign] = {"love": love, "work": work}
    return horoscope


def get_path(image: str):
    """Return the path of the image"""
    return join("images", f"{image.lower()}.png")


def create_horoscope_image(horoscope: dict[str, dict[str, str]]):
    """Generate horoscope images"""
    img_width, img_height = 1200, 425
    image = Image.new("RGBA", (img_width, img_height), color=(255, 255, 255, 0))
    draw = ImageDraw.Draw(image)

    size_emoji = (30, 30)

    font_sign = ImageFont.load_default(24)
    font_prediction = ImageFont.load_default(40)

    x, y = 10, 10
    for sign, prediction in horoscope.items():
        sign_image = Image.open(get_path(sign)).convert("RGBA")
        sign_image.thumbnail((100, 105))
        image.paste(sign_image, (x + 20, y), sign_image)

        # Sign text
        text_img = Image.new("RGBA", (135, 24), (255, 255, 255, 0))
        text_draw = ImageDraw.Draw(text_img)
        text_draw.text((0, 0), f"{sign:^15}", font=font_sign, fill="black")
        text_draw = text_img.rotate(90, expand=True)
        image.paste(text_draw, (x - 10, y - 10), text_draw)

        # Love prediction
        love = Image.open(join("images", "love.png")).convert("RGBA")
        love.thumbnail(size_emoji)
        image.paste(love, (x + 125, y + 20), love)
        draw.text(
            (x + 160, y + 10),
            prediction["love"],
            fill="black",
            font=font_prediction,
        )

        # Work prediction
        work = Image.open(join("images", "work.png")).convert("RGBA")
        work.thumbnail(size_emoji)
        image.paste(work, (x + 125, y + 60), work)
        draw.text(
            (x + 160, y + 50),
            prediction["work"],
            fill="black",
            font=font_prediction,
        )

        x += 320
        if x > 1200:
            x = 10
            y += 150

    return image


def download_images():
    """Download images from source"""
    url = "https://git.mylloon.fr/Anri/Horoscope/raw/branch/main/images/"

    makedirs("images", exist_ok=True)
    for el in zodiac_signs + ["love", "work"]:
        urlretrieve(url + el.lower() + ".png", get_path(el))


if __name__ == "__main__":
    if len(argv) > 2 or (len(argv) == 2 and argv[1] != "online"):
        exit(1)

    if len(argv) == 2:
        download_images()

    # Generate new horoscope
    new_horoscope = generate_horoscope()

    # Create and save the image
    horoscope_image = create_horoscope_image(new_horoscope)
    horoscope_image.save("nouvel_horoscope.png")

    print("Nouvel horoscope généré et sauvegardé sous 'nouvel_horoscope.png'")