475 lines
14 KiB
Nix
475 lines
14 KiB
Nix
{ pkgs }:
|
|
|
|
pkgs.writeText "theme-builder.py" ''
|
|
from colorsys import rgb_to_hls, hls_to_rgb
|
|
from collections import Counter
|
|
from pathlib import Path
|
|
from PIL import Image
|
|
from pynvim import attach
|
|
import subprocess
|
|
import math
|
|
import os
|
|
import sys
|
|
|
|
STATE_DIR = Path(os.path.expanduser("~/.local/state/sysflake"))
|
|
MULT = 2.5
|
|
|
|
GRAYSCALE = {
|
|
"base00": "1a1a1a",
|
|
"base01": "2a2a2a",
|
|
"base02": "3a3a3a",
|
|
"base03": "5a5a5a",
|
|
"base04": "8a8a8a",
|
|
"base05": "b0b0b0",
|
|
"base06": "d0d0d0",
|
|
"base07": "e8e8e8",
|
|
"base08": "4a4a4a",
|
|
"base09": "555555",
|
|
"base0A": "606060",
|
|
"base0B": "6b6b6b",
|
|
"base0C": "767676",
|
|
"base0D": "818181",
|
|
"base0E": None, # accent slot
|
|
"base0F": "8c8c8c",
|
|
}
|
|
|
|
KITTY_MAP = {
|
|
"background": "00", "foreground": "05", "cursor": "05",
|
|
"selection_background": "02", "selection_foreground": "05",
|
|
"color0": "00", "color1": "08", "color2": "0E", "color3": "0A",
|
|
"color4": "0D", "color5": "0E", "color6": "0E", "color7": "05",
|
|
"color8": "03", "color9": "08", "color10": "0E", "color11": "0A",
|
|
"color12": "0D", "color13": "0E", "color14": "0E", "color15": "07",
|
|
}
|
|
|
|
|
|
def extract_accent(image_path):
|
|
img = Image.open(image_path).resize((200, 200)).quantize(colors=32).convert("RGB")
|
|
pixels = list(img.get_flattened_data())
|
|
counts = Counter(pixels)
|
|
total = len(pixels)
|
|
|
|
def score(color):
|
|
_, l, s = rgb_to_hls(color[0] / 255, color[1] / 255, color[2] / 255)
|
|
vibrancy = s * (1 - abs(l - 0.5) * 2)
|
|
frequency = counts[color] / total
|
|
return vibrancy * frequency
|
|
|
|
best = max(counts.keys(), key=score)
|
|
h, l, s = rgb_to_hls(best[0] / 255, best[1] / 255, best[2] / 255)
|
|
|
|
new_s = min(max(s * MULT, 0.6), 0.7)
|
|
|
|
# Blue hues are perceptually darker — gradually boost lightness floor
|
|
blue_center = 0.63
|
|
blue_width = 0.09
|
|
blue_factor = max(0, 1 - abs(h - blue_center) / blue_width)
|
|
min_l = 0.45 + 0.2 * blue_factor
|
|
|
|
new_l = max(min_l, min(0.7, l))
|
|
r, g, b = hls_to_rgb(h, new_l, new_s)
|
|
return f"{int(r * 255):02x}{int(g * 255):02x}{int(b * 255):02x}"
|
|
|
|
|
|
def build_scheme(accent):
|
|
scheme = dict(GRAYSCALE)
|
|
scheme["base0E"] = accent
|
|
return scheme
|
|
|
|
|
|
def write_scheme(scheme):
|
|
path = STATE_DIR / "scheme.txt"
|
|
with open(path, "w") as f:
|
|
for key, val in scheme.items():
|
|
f.write(f"{key} {val}\n")
|
|
|
|
|
|
def write_kitty(scheme):
|
|
path = STATE_DIR / "kitty-colors.conf"
|
|
with open(path, "w") as f:
|
|
for name, base in KITTY_MAP.items():
|
|
f.write(f"{name} #{scheme['base' + base]}\n")
|
|
|
|
|
|
VESKTOP_DIR = Path(os.path.expanduser("~/.config/vesktop/themes"))
|
|
|
|
|
|
def hex_to_oklch(hex_color):
|
|
"""Convert hex color to oklch (lightness, chroma, hue) values."""
|
|
r = int(hex_color[0:2], 16) / 255
|
|
g = int(hex_color[2:4], 16) / 255
|
|
b = int(hex_color[4:6], 16) / 255
|
|
|
|
# sRGB to linear
|
|
def to_linear(c):
|
|
return c / 12.92 if c <= 0.04045 else ((c + 0.055) / 1.055) ** 2.4
|
|
|
|
lr, lg, lb = to_linear(r), to_linear(g), to_linear(b)
|
|
|
|
# Linear RGB to OKLab
|
|
l_ = 0.4122214708 * lr + 0.5363325363 * lg + 0.0514459929 * lb
|
|
m_ = 0.2119034982 * lr + 0.6806995451 * lg + 0.1073969566 * lb
|
|
s_ = 0.0883024619 * lr + 0.2817188376 * lg + 0.6299787005 * lb
|
|
|
|
l_ = l_ ** (1/3) if l_ >= 0 else -((-l_) ** (1/3))
|
|
m_ = m_ ** (1/3) if m_ >= 0 else -((-m_) ** (1/3))
|
|
s_ = s_ ** (1/3) if s_ >= 0 else -((-s_) ** (1/3))
|
|
|
|
L = 0.2104542553 * l_ + 0.7936177850 * m_ - 0.0040720468 * s_
|
|
a = 1.9779984951 * l_ - 2.4285922050 * m_ + 0.4505937099 * s_
|
|
bv = 0.0259040371 * l_ + 0.7827717662 * m_ - 0.8086757660 * s_
|
|
|
|
C = math.sqrt(a ** 2 + bv ** 2)
|
|
H = math.degrees(math.atan2(bv, a)) % 360
|
|
|
|
return L, C, H
|
|
|
|
|
|
def write_vesktop(scheme):
|
|
VESKTOP_DIR.mkdir(parents=True, exist_ok=True)
|
|
accent = scheme["base0E"]
|
|
_, chroma, hue = hex_to_oklch(accent)
|
|
|
|
path = VESKTOP_DIR / "system24-dynamic.css"
|
|
with open(path, "w") as f:
|
|
f.write("""\
|
|
/**
|
|
* system24 dynamic theme — generated by theme-engine
|
|
*/
|
|
|
|
@import url('https://refact0r.github.io/system24/build/system24.css');
|
|
|
|
:root {
|
|
--colors: on;
|
|
|
|
/* override purple (accent) with extracted color */
|
|
""")
|
|
for i, lightness in enumerate([75, 70, 65, 60, 55], start=1):
|
|
f.write(f" --purple-{i}: oklch({lightness}% {chroma:.4f} {hue:.1f});\n")
|
|
|
|
f.write("}\n")
|
|
|
|
|
|
def write_nvim(scheme):
|
|
accent = scheme["base0E"]
|
|
path = STATE_DIR / "nvim-colors.lua"
|
|
with open(path, "w") as f:
|
|
f.write("require('base16-colorscheme').setup({\n")
|
|
for key, val in scheme.items():
|
|
f.write(f" {key} = '#{val}',\n")
|
|
f.write("})\n\n")
|
|
f.write(f"local accent = '#{accent}'\n")
|
|
f.write(f"local light_gray = '#{scheme['base05']}'\n")
|
|
for group in ["Statement", "Conditional", "Repeat", "Macro", "Function", "Exception",
|
|
"TSMethod", "@lsp.type.method", "@lsp.type.function", "@lsp.type.macro",
|
|
"DiagnosticUnderlineHint", "Boolean"]:
|
|
f.write(f"vim.api.nvim_set_hl(0, '{group}', {{ fg = accent }})\n")
|
|
for group in ["@lsp.type.variable", "@lsp.type.parameter", "@lsp.type.struct", "@lsp.type.class",
|
|
"@lsp.type.selfTypeKeyword", "Identifier"]:
|
|
f.write(f"vim.api.nvim_set_hl(0, '{group}', {{ fg = light_gray }})\n")
|
|
for group in ["@lsp.type.enumMember", "@lsp.type.enum", "Number", "Integer"]:
|
|
f.write(f"vim.api.nvim_set_hl(0, '{group}', {{ fg = '#{scheme['base04']}' }})\n")
|
|
|
|
|
|
def write_swaync(scheme):
|
|
accent = scheme["base0E"]
|
|
path = STATE_DIR / "swaync-colors.css"
|
|
with open(path, "w") as f:
|
|
f.write(f"""\
|
|
@define-color base #{scheme['base00']};
|
|
@define-color mantle #{scheme['base01']};
|
|
@define-color crust #{scheme['base00']};
|
|
@define-color text #{scheme['base05']};
|
|
@define-color subtext0 #{scheme['base04']};
|
|
@define-color subtext1 #{scheme['base05']};
|
|
@define-color surface0 #{scheme['base01']};
|
|
@define-color surface1 #{scheme['base02']};
|
|
@define-color surface2 #{scheme['base03']};
|
|
@define-color overlay0 #{scheme['base03']};
|
|
@define-color overlay1 #{scheme['base04']};
|
|
@define-color overlay2 #{scheme['base04']};
|
|
@define-color lavender #{accent};
|
|
""")
|
|
|
|
|
|
EZA_DIR = Path(os.path.expanduser("~/.config/eza"))
|
|
|
|
|
|
def write_eza(scheme):
|
|
EZA_DIR.mkdir(parents=True, exist_ok=True)
|
|
accent = f'"#{scheme["base0E"]}"'
|
|
file_color = f'"#{scheme["base05"]}"'
|
|
dim = f'"#{scheme["base04"]}"'
|
|
dimmer = f'"#{scheme["base03"]}"'
|
|
dark = f'"#{scheme["base02"]}"'
|
|
|
|
path = EZA_DIR / "theme.yml"
|
|
with open(path, "w") as f:
|
|
f.write(f"""\
|
|
filekinds:
|
|
normal:
|
|
foreground: {file_color}
|
|
directory:
|
|
foreground: {accent}
|
|
is_bold: true
|
|
is_underline: true
|
|
symlink:
|
|
foreground: {dim}
|
|
pipe:
|
|
foreground: {dimmer}
|
|
block_device:
|
|
foreground: {dim}
|
|
char_device:
|
|
foreground: {dim}
|
|
socket:
|
|
foreground: {dimmer}
|
|
special:
|
|
foreground: {dim}
|
|
executable:
|
|
foreground: {accent}
|
|
is_bold: true
|
|
mount_point:
|
|
foreground: {accent}
|
|
|
|
perms:
|
|
user_read:
|
|
foreground: {file_color}
|
|
user_write:
|
|
foreground: {dim}
|
|
user_execute_file:
|
|
foreground: {accent}
|
|
user_execute_other:
|
|
foreground: {accent}
|
|
group_read:
|
|
foreground: {dim}
|
|
group_write:
|
|
foreground: {dimmer}
|
|
group_execute:
|
|
foreground: {dim}
|
|
other_read:
|
|
foreground: {dimmer}
|
|
other_write:
|
|
foreground: {dimmer}
|
|
other_execute:
|
|
foreground: {dimmer}
|
|
special_user_file:
|
|
foreground: {accent}
|
|
special_other:
|
|
foreground: {dimmer}
|
|
attribute:
|
|
foreground: {dimmer}
|
|
|
|
size:
|
|
number_byte:
|
|
foreground: {file_color}
|
|
number_kilo:
|
|
foreground: {dim}
|
|
number_mega:
|
|
foreground: {dim}
|
|
number_giga:
|
|
foreground: {accent}
|
|
number_huge:
|
|
foreground: {accent}
|
|
unit_byte:
|
|
foreground: {dimmer}
|
|
unit_kilo:
|
|
foreground: {dimmer}
|
|
unit_mega:
|
|
foreground: {dimmer}
|
|
unit_giga:
|
|
foreground: {dimmer}
|
|
unit_huge:
|
|
foreground: {dimmer}
|
|
|
|
users:
|
|
user_you:
|
|
foreground: {file_color}
|
|
user_root:
|
|
foreground: {accent}
|
|
user_other:
|
|
foreground: {dim}
|
|
group_yours:
|
|
foreground: {dim}
|
|
group_other:
|
|
foreground: {dimmer}
|
|
group_root:
|
|
foreground: {accent}
|
|
|
|
links:
|
|
normal:
|
|
foreground: {dim}
|
|
multi_link_file:
|
|
foreground: {accent}
|
|
|
|
git:
|
|
new:
|
|
foreground: {accent}
|
|
modified:
|
|
foreground: {dim}
|
|
deleted:
|
|
foreground: {dimmer}
|
|
renamed:
|
|
foreground: {dim}
|
|
typechange:
|
|
foreground: {dim}
|
|
ignored:
|
|
foreground: {dark}
|
|
conflicted:
|
|
foreground: {accent}
|
|
|
|
git_repo:
|
|
branch_main:
|
|
foreground: {file_color}
|
|
branch_other:
|
|
foreground: {dim}
|
|
git_clean:
|
|
foreground: {accent}
|
|
git_dirty:
|
|
foreground: {dim}
|
|
|
|
file_type:
|
|
image:
|
|
foreground: {dim}
|
|
video:
|
|
foreground: {dim}
|
|
music:
|
|
foreground: {dim}
|
|
lossless:
|
|
foreground: {dim}
|
|
crypto:
|
|
foreground: {dimmer}
|
|
document:
|
|
foreground: {file_color}
|
|
compressed:
|
|
foreground: {dimmer}
|
|
temp:
|
|
foreground: {dark}
|
|
compiled:
|
|
foreground: {dimmer}
|
|
build:
|
|
foreground: {dimmer}
|
|
source:
|
|
foreground: {file_color}
|
|
|
|
punctuation:
|
|
foreground: {dimmer}
|
|
date:
|
|
foreground: {dim}
|
|
inode:
|
|
foreground: {dimmer}
|
|
blocks:
|
|
foreground: {dimmer}
|
|
header:
|
|
foreground: {file_color}
|
|
octal:
|
|
foreground: {dimmer}
|
|
flags:
|
|
foreground: {dim}
|
|
|
|
symlink_path:
|
|
foreground: {dim}
|
|
control_char:
|
|
foreground: {dimmer}
|
|
broken_symlink:
|
|
foreground: {accent}
|
|
broken_path_overlay:
|
|
foreground: {dark}
|
|
""")
|
|
|
|
|
|
def write_waybar(scheme):
|
|
accent = scheme["base0E"]
|
|
bg = scheme["base00"]
|
|
fg = scheme["base05"]
|
|
path = STATE_DIR / "waybar-colors.css"
|
|
with open(path, "w") as f:
|
|
f.write(f"@define-color accent #{accent};\n")
|
|
f.write(f"@define-color bg-dark #{bg};\n")
|
|
f.write(f"@define-color fg-text #{fg};\n")
|
|
|
|
|
|
def write_hyprland(scheme):
|
|
accent = scheme["base0E"]
|
|
inactive = scheme["base03"]
|
|
path = STATE_DIR / "hyprland-colors.conf"
|
|
with open(path, "w") as f:
|
|
f.write(f"general:col.active_border = rgba({accent}ff)\n")
|
|
f.write(f"general:col.inactive_border = rgba({inactive}ff)\n")
|
|
|
|
|
|
def reload_apps():
|
|
print("Reloading applications with new theme...")
|
|
subprocess.run(["pkill", "-SIGUSR1", "kitty"], capture_output=True)
|
|
print("Sent reload signal to Kitty.")
|
|
|
|
nvim_colors = STATE_DIR / "nvim-colors.lua"
|
|
for sock in Path("/run/user/1000").glob("nvim.*.0"):
|
|
try:
|
|
nvim = attach('socket', path=str(sock))
|
|
nvim.command(f"luafile {nvim_colors}")
|
|
print(f"Sent reload command to Neovim instance at {sock}.")
|
|
except Exception as e:
|
|
print(f"Failed to connect to Neovim instance at {sock}: {e}")
|
|
|
|
# Live-update hyprland borders
|
|
scheme = {}
|
|
for line in open(STATE_DIR / "scheme.txt"):
|
|
k, v = line.split()
|
|
scheme[k] = v
|
|
subprocess.run(["hyprctl", "keyword", "general:col.active_border", f"rgba({scheme['base0E']}ff)"], capture_output=True)
|
|
subprocess.run(["hyprctl", "keyword", "general:col.inactive_border", f"rgba({scheme['base03']}ff)"], capture_output=True)
|
|
print("Updated Hyprland border colors.")
|
|
|
|
# Reload swaync styles
|
|
subprocess.run(["swaync-client", "-rs"], capture_output=True)
|
|
print("Sent reload signal to Swaync.")
|
|
|
|
# Reload waybar
|
|
subprocess.run(["pkill", "-SIGUSR2", "waybar"], capture_output=True)
|
|
print("Sent reload signal to Waybar.")
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) < 2:
|
|
print("Usage: theme-engine.py <image-path>", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
STATE_DIR.mkdir(parents=True, exist_ok=True)
|
|
|
|
accent = extract_accent(sys.argv[1])
|
|
scheme = build_scheme(accent)
|
|
print(f"Extracted accent color: #{accent}")
|
|
write_scheme(scheme)
|
|
print("Scheme written to state directory.")
|
|
write_kitty(scheme)
|
|
print("Kitty config written.")
|
|
write_vesktop(scheme)
|
|
print("Vesktop theme written.")
|
|
write_nvim(scheme)
|
|
print("Neovim colors written.")
|
|
write_hyprland(scheme)
|
|
print("Hyprland colors written.")
|
|
write_swaync(scheme)
|
|
print("Swaync colors written.")
|
|
write_eza(scheme)
|
|
print("Eza theme written.")
|
|
write_waybar(scheme)
|
|
print("Waybar colors written.")
|
|
reload_apps()
|
|
print("Sent reload signals to applications.")
|
|
|
|
print(f"Accent: #{accent}")
|
|
print(f"Scheme written to {STATE_DIR / 'scheme.txt'}")
|
|
print(f"Kitty config written to {STATE_DIR / 'kitty-colors.conf'}")
|
|
print(f"Vesktop theme written to {VESKTOP_DIR / 'system24-dynamic.css'}")
|
|
print(f"Neovim colors written to {STATE_DIR / 'nvim-colors.lua'}")
|
|
print(f"Hyprland colors written to {STATE_DIR / 'hyprland-colors.conf'}")
|
|
print(f"Swaync colors written to {STATE_DIR / 'swaync-colors.css'}")
|
|
print(f"Eza theme written to {EZA_DIR / 'theme.yml'}")
|
|
print(f"Waybar colors written to {STATE_DIR / 'waybar-colors.css'}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
''
|