1 files changed, 122 insertions, 0 deletions
diff --git a/src/caelestia/utils/theme.py b/src/caelestia/utils/theme.py
new file mode 100644
index 0000000..d205fb1
--- /dev/null
+++ b/src/caelestia/utils/theme.py
@@ -0,0 +1,122 @@
+import subprocess
+import tempfile
+from pathlib import Path
+
+from caelestia.utils.paths import config_dir, templates_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 gen_replace(colours: dict[str, str], template: Path, hash: bool = False) -> str:
+    template = template.read_text()
+    for name, colour in colours.items():
+        template = template.replace(f"{{{{ ${name} }}}}", f"#{colour}" if hash else colour)
+    return template
+
+
+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_discord(scss: str) -> None:
+    with tempfile.TemporaryDirectory("w") as tmp_dir:
+        (Path(tmp_dir) / "_colours.scss").write_text(scss)
+        conf = subprocess.check_output(["sass", "-I", tmp_dir, templates_dir / "discord.scss"], text=True)
+
+    for client in "Equicord", "Vencord", "BetterDiscord", "equicord", "vesktop", "legcord":
+        try_write(config_dir / client / "themes/caelestia.theme.css", conf)
+
+
+def apply_spicetify(colours: dict[str, str], mode: str) -> None:
+    template = gen_replace(colours, templates_dir / f"spicetify-{mode}.ini")
+    try_write(config_dir / "spicetify/Themes/caelestia/color.ini", template)
+
+
+def apply_fuzzel(colours: dict[str, str]) -> None:
+    template = gen_replace(colours, templates_dir / "fuzzel.ini")
+    try_write(config_dir / "fuzzel/fuzzel.ini", template)
+
+
+def apply_btop(colours: dict[str, str]) -> None:
+    template = gen_replace(colours, templates_dir / "btop.theme", hash=True)
+    try_write(config_dir / "btop/themes/caelestia.theme", template)
+    subprocess.run(["killall", "-USR2", "btop"])
+
+
+def apply_colours(colours: dict[str, str], mode: str) -> None:
+    apply_terms(gen_sequences(colours))
+    apply_hypr(gen_conf(colours))  # FIXME: LAGGY
+    apply_discord(gen_scss(colours))
+    apply_spicetify(colours, mode)
+    apply_fuzzel(colours)
+    apply_btop(colours)