import json
import subprocess
from pathlib import Path

from caelestia.utils.paths import config_dir


def gen_conf(colours: dict[str, str]) -> str:
    conf = ""
    for name, colour in colours.items():
        conf += f"${name} = {colour}\n"
    return conf


def gen_scss(colours: dict[str, str]) -> str:
    scss = ""
    for name, colour in colours.items():
        scss += f"${name}: {colour};\n"
    return scss


def c2s(c: str, *i: list[int]) -> str:
    """Hex to ANSI sequence (e.g. ffffff, 11 -> \x1b]11;rgb:ff/ff/ff\x1b\\)"""
    return f"\x1b]{';'.join(map(str, i))};rgb:{c[0:2]}/{c[2:4]}/{c[4:6]}\x1b\\"


def gen_sequences(colours: dict[str, str]) -> str:
    """
    10: foreground
    11: background
    12: cursor
    17: selection
    4:
        0 - 7: normal colours
        8 - 15: bright colours
        16+: 256 colours
    """
    return (
        c2s(colours["onSurface"], 10)
        + c2s(colours["surface"], 11)
        + c2s(colours["secondary"], 12)
        + c2s(colours["secondary"], 17)
        + c2s(colours["surfaceContainer"], 4, 0)
        + c2s(colours["red"], 4, 1)
        + c2s(colours["green"], 4, 2)
        + c2s(colours["yellow"], 4, 3)
        + c2s(colours["blue"], 4, 4)
        + c2s(colours["pink"], 4, 5)
        + c2s(colours["teal"], 4, 6)
        + c2s(colours["onSurfaceVariant"], 4, 7)
        + c2s(colours["surfaceContainer"], 4, 8)
        + c2s(colours["red"], 4, 9)
        + c2s(colours["green"], 4, 10)
        + c2s(colours["yellow"], 4, 11)
        + c2s(colours["blue"], 4, 12)
        + c2s(colours["pink"], 4, 13)
        + c2s(colours["teal"], 4, 14)
        + c2s(colours["onSurfaceVariant"], 4, 15)
        + c2s(colours["primary"], 4, 16)
        + c2s(colours["secondary"], 4, 17)
        + c2s(colours["tertiary"], 4, 18)
    )


def try_write(path: Path, content: str) -> None:
    try:
        path.write_text(content)
    except FileNotFoundError:
        pass


def apply_terms(sequences: str) -> None:
    pts_path = Path("/dev/pts")
    for pt in pts_path.iterdir():
        if pt.name.isdigit():
            with pt.open("a") as f:
                f.write(sequences)


def apply_hypr(conf: str) -> None:
    try_write(config_dir / "hypr/scheme/current.conf", conf)


def apply_colours(colours: dict[str, str]) -> None:
    apply_terms(gen_sequences(colours))
    apply_hypr(gen_conf(colours))