"""
Theme file handling.
"""

import logging
import os
import random
import sys

from .settings import CACHE_DIR, CONF_DIR, MODULE_DIR
from . import util
from . import colors


def list_out():
    """List all themes in a pretty format."""
    dark_themes = [theme.name.replace(".json", "") for theme in list_themes()]
    light_themes = [
        theme.name.replace(".json", "") for theme in list_themes(dark=False)
    ]
    user_themes = [
        theme.name.replace(".json", "") for theme in list_themes_user()
    ]

    try:
        last_used_theme = util.read_file(
            os.path.join(CACHE_DIR, "last_used_theme")
        )[0].replace(".json", "")
    except FileNotFoundError:
        last_used_theme = ""

    if user_themes:
        print("\033[1;32mUser Themes\033[0m:")
        print(
            " -",
            "\n - ".join(
                t + " (last used)" if t == last_used_theme else t
                for t in sorted(user_themes)
            ),
        )

    print("\033[1;32mDark Themes\033[0m:")
    print(
        " -",
        "\n - ".join(
            t + " (last used)" if t == last_used_theme else t
            for t in sorted(dark_themes)
        ),
    )

    print("\033[1;32mLight Themes\033[0m:")
    print(
        " -",
        "\n - ".join(
            t + " (last used)" if t == last_used_theme else t
            for t in sorted(light_themes)
        ),
    )

    print("\033[1;32mExtra\033[0m:")
    print(" - random (select a random dark theme)")
    print(" - random_dark (select a random dark theme)")
    print(" - random_light (select a random light theme)")
    print(" - random_user (select a random user theme)")


def list_themes(dark=True):
    """List all installed theme files."""
    dark = "dark" if dark else "light"
    themes = os.scandir(os.path.join(MODULE_DIR, "colorschemes", dark))
    return [t for t in themes if os.path.isfile(t.path)]


def list_themes_user():
    """List user theme files."""
    themes = [
        *os.scandir(os.path.join(CONF_DIR, "colorschemes/dark/")),
        *os.scandir(os.path.join(CONF_DIR, "colorschemes/light/")),
    ]
    return [t for t in themes if os.path.isfile(t.path)]


def terminal_sexy_to_wal(data):
    """Convert terminal.sexy json schema to wal."""
    data["colors"] = {}
    data["special"] = {
        "foreground": data["foreground"],
        "background": data["background"],
        "cursor": data["color"][9],
    }

    for i, color in enumerate(data["color"]):
        data["colors"]["color%s" % i] = color

    return data


def parse(theme_file):
    """Parse the theme file."""
    data = util.read_file_json(theme_file)

    if "checksum" not in data:
        data["checksum"] = "None"

    if "wallpaper" not in data:
        data["wallpaper"] = "None"

    if util.Color.passed_alpha_num or "alpha" not in data:
        data["alpha"] = util.Color.alpha_num
    else:
        util.Color.alpha_num = data["alpha"]

    # Terminal.sexy format.
    if "color" in data:
        data = terminal_sexy_to_wal(data)

    return data


def get_random_theme(dark=True):
    """Get a random theme file."""
    themes = [theme.path for theme in list_themes(dark)]
    random.shuffle(themes)
    return themes[0]


def get_random_theme_user():
    """Get a random theme file from user theme directories."""
    themes = [theme.path for theme in list_themes_user()]
    random.shuffle(themes)
    return themes[0]


def file(input_file, light=False, **kwargs):
    """Import colorscheme from json file."""
    if "c16" in kwargs:
        cols16 = kwargs["c16"]
    else:
        cols16 = False

    util.create_dir(os.path.join(CONF_DIR, "colorschemes/light/"))
    util.create_dir(os.path.join(CONF_DIR, "colorschemes/dark/"))

    theme_name = ".".join((input_file, "json"))
    bri = "light" if light else "dark"

    user_theme_file = os.path.join(CONF_DIR, "colorschemes", bri, theme_name)
    theme_file = os.path.join(MODULE_DIR, "colorschemes", bri, theme_name)

    # Find the theme file.
    if input_file in ("random", "random_dark"):
        theme_file = get_random_theme()

    elif input_file == "random_light":
        theme_file = get_random_theme(light)

    elif input_file == "random_user":
        theme_file = get_random_theme_user()

    elif os.path.isfile(user_theme_file):
        theme_file = user_theme_file

    elif os.path.isfile(input_file):
        theme_file = input_file

    # Parse the theme file.
    if os.path.isfile(theme_file):
        logging.info(
            "Set theme to \033[1;37m%s\033[0m.", os.path.basename(theme_file)
        )
        util.save_file(
            os.path.basename(theme_file),
            os.path.join(CACHE_DIR, "last_used_theme"),
        )
        r_theme = parse(theme_file)
        if cols16:
            if r_theme["colors"]["color1"] == r_theme["colors"]["color9"]:
                logging.info("requested theme uses 9 shades, converting to 16")
                colors.shade_16(r_theme["colors"], light, cols16)
        return r_theme

    logging.error("No %s colorscheme file found.", bri)
    logging.error("Try adding   '-l' to set light themes.")
    logging.error("Try removing '-l' to set dark themes.")
    sys.exit(1)


def save(colors, theme_name, light=False):
    """Save colors to a theme file."""
    theme_file = theme_name + ".json"
    theme_path = os.path.join(
        CONF_DIR, "colorschemes", "light" if light else "dark", theme_file
    )
    util.save_file_json(colors, theme_path)
