"""
OnlyLegs - Setup
Runs when the app detects that there is no user directory
"""
import os
import re
import platformdirs
import yaml


APPLICATION_ROOT = platformdirs.user_config_dir("onlylegs")
REQUIRED_DIRS = {
    "root": APPLICATION_ROOT,
    "instance": os.path.join(APPLICATION_ROOT, "instance"),
    "media": os.path.join(APPLICATION_ROOT, "media"),
    "uploads": os.path.join(APPLICATION_ROOT, "media", "uploads"),
    "cache": os.path.join(APPLICATION_ROOT, "media", "cache"),
    "pfp": os.path.join(APPLICATION_ROOT, "media", "pfp"),
    "banner": os.path.join(APPLICATION_ROOT, "media", "banner"),
}

EMAIL_REGEX = re.compile(r"\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b")
USERNAME_REGEX = re.compile(r"\b[A-Za-z0-9._%+-]+\b")

config = {
    # Version of the config file
    "version": "0.1.7",
    # Not really used much, but good to have for future use
    "admin": {
        "username": "admin",
        "email": "admin@example.com",
    },
    "upload": {
        "allowed-extensions": {
            "jpg": "jpeg",
            "jpeg": "jpeg",
            "png": "png",
            "webp": "webp",
        },
        # Max size in MB
        "max-size": 69,
        # Max images to load per page
        "max-load": 50,
    },
    "website": {
        # Website name and motto
        # Also CSS styling, hue is the color offset for hsl
        "name": "OnlyLegs",
        "motto": "A gallery built for fast and simple image management!",
        "styling": {
            "force": False,
            "hue": "69",
            "saturation": "25%",
            "rad": "0.4rem",
        },
    },
}


def check_dirs():
    for directory in REQUIRED_DIRS.values():
        if not os.path.exists(directory):
            os.makedirs(directory)
            print("Created directory at:", directory)
        print("User directory already exists at:", directory)


def check_env():
    if os.path.exists(os.path.join(APPLICATION_ROOT, ".env")):
        print("Environment file already exists at:", APPLICATION_ROOT)
        return

    env_conf = {
        "FLASK_SECRET": os.urandom(32).hex(),
    }

    with open(
        os.path.join(APPLICATION_ROOT, ".env"), encoding="utf-8", mode="w+"
    ) as file:
        for key, value in env_conf.items():
            file.write(key + "=" + value + "\n")

    print(
        "####################################################",
        "# A NEW KEY WAS GENERATED FOR YOU! PLEASE NOTE     #",
        "# DOWN THE FLASK_SECRET KEY LOCATED IN YOUR        #",
        "# ~/.config/onlylegs/.env FOLDER! LOOSING THIS KEY #",
        "# WILL RESULT IN YOU BEING UNABLE TO LOG IN!       #",
        "####################################################",
        sep="\n",
    )


def check_conf():
    config_file = os.path.join(APPLICATION_ROOT, "conf.yml")
    if os.path.exists(config_file):
        print("Config file already exists at:", APPLICATION_ROOT)
        return

    cant_continue = True
    username = "admin"
    email = "admin@example.com"

    print("No config file found, please enter the following information:")
    while cant_continue:
        username = input("Admin username: ").strip()
        email = input("Admin email: ").strip()

        if not username or not USERNAME_REGEX.match(username):
            print("Username is invalid!")
            continue
        if not email or not EMAIL_REGEX.match(email):
            print("Email is invalid!")
            continue

        # Check if user is happy with the values
        is_correct = input("Is this correct? (Y/n): ").lower().strip()
        if is_correct == "y" or not is_correct:
            cant_continue = False

    config["admin"]["username"] = username
    config["admin"]["email"] = email

    with open(config_file, encoding="utf-8", mode="w+") as file:
        yaml.dump(config, file, default_flow_style=False)

    print(
        "####################################################",
        "# A NEW CONFIG HAS BEEN GENERATED AT:              #",
        "# ~/.config/onlylegs/conf.yml                      #",
        "####################################################",
        sep="\n",
    )