feat: theme hypr and terminals

1 files changed, 86 insertions, 0 deletions

diff --git a/src/caelestia/utils/theme.py b/src/caelestia/utils/theme.py
new file mode 100644
index 0000000..7774472
--- /dev/null
+++ b/src/caelestia/utils/theme.py
@@ -0,0 +1,86 @@
+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))