#!/usr/bin/env python3

"""Recolour the SVG sources to the Darkmatter palette.

Two jobs, because the two SVG sets are quite different:

xfwm4/assets/*.svg are clean and in sync with their PNGs. Upstream draws the

window buttons as macOS-style traffic lights -- a coloured disc on a dark rim,

with the glyph appearing only on hover. Darkmatter keeps that geometry but

drops the hues: the disc is grey at rest and takes the accent on hover. Which

colour holds the disc depends on the button's state, so the map is per file.

The same source colour also serves two roles at different opacities -- a wash

disc at 0.1 and a dot glyph at 0.75 -- so fills are matched together with the

fill-opacity beside them.

assets/*.svg are only build sources; GTK reads the PNGs. Several of them still

carry elementary's blues, never re-rendered when Nordic forked, so they are

already out of step with their own PNGs. Those get exact-match substitution

only: a colour the palette does not name is left exactly as it was rather than

guessed at.

"""

import glob

import os

import re

import sys

sys.path.insert(0, os.path.join(os.path.dirname(__file__), "lib"))

import darkmatter as dm

# --- xfwm4 -----------------------------------------------------------------

FRAME = {

    "#353c4a": dm.DARK,      # titlebar and frame background

    "#232831": dm.SHADOW,    # rim under the disc

    "#231d2b": dm.BG,        # glyph drawn on the disc

}

# The disc colour by button state. Upstream spends a different hue on each

# button; Darkmatter spends brightness instead.

DISC = {

    "active":   dm.DIM,

    "inactive": dm.BORDER_HI,

    "prelight": dm.SUBTLE,

    "pressed":  dm.FG,

}

# ...except close, which is the one button worth the accent.

DISC_CLOSE = {

    "active":   dm.DIM,

    "inactive": dm.BORDER_HI,

    "prelight": dm.ORANGE,

    "pressed":  dm.ORANGE_D1,

}

# Source colours that carry the disc, whatever their hue upstream.

DISC_SRC = {"#bf616a", "#ebcb8b", "#a3be8c", "#8fbcbb", "#3b4252"}

# #eceff1 is a wash at low opacity and a dot glyph at high opacity.

WASH_MAX_OPACITY = 0.5

def xfwm_map(path):

    name = os.path.basename(path)[:-4]

    state = name.rsplit("-", 1)[-1]

    if state == "shaded":                       # e.g. title-1-active-shaded

        state = name.split("-")[-2]

    state = state if state in DISC else "active"

    disc = (DISC_CLOSE if name.startswith("close") else DISC)[state]

    def sub(m):

        style = m.group(1)

        fill = re.search(r"(?<![-\w])fill:(#[0-9a-fA-F]{6})", style)

        if not fill:

            return m.group(0)

        color = fill.group(1).lower()

        opac = re.search(r"fill-opacity:([0-9.]+)", style)

        opac = float(opac.group(1)) if opac else 1.0

        if color in FRAME:

            new_color = FRAME[color]

        elif color == "#eceff1":

            new_color = dm.FG if opac <= WASH_MAX_OPACITY else disc

        elif color in DISC_SRC:

            new_color = disc

        else:

            return m.group(0)                   # #ffffff highlights, etc.

        s, e = fill.span(1)

        return 'style="%s"' % (style[:s] + new_color + style[e:])

    text = open(path).read()

    out = re.sub(r'style="([^"]*)"', sub, text)

    if out != text:

        open(path, "w").write(out)

        return True

    return False

# --- gtk assets ------------------------------------------------------------

EXACT = {

    "#353c4a": dm.DARK,

    "#232831": dm.BG,

    "#3b4252": dm.BASE,

    "#5d6a83": "#4a494b",

    "#8fbcbb": dm.ORANGE,

    "#ebcb8b": dm.CREAM,

    "#bf616a": dm.RED,

    "#a3be8c": dm.TEAL,

}

CONTROL = {

    "close":              dm.DIM,

    "close_prelight":     dm.ORANGE,

    "close_pressed":      dm.ORANGE_D1,

    "close_unfocused":    dm.BORDER_HI,

    "maximize":           dm.DIM,

    "maximize_prelight":  dm.SUBTLE,

    "maximize_pressed":   dm.FG,

    "maximize_unfocused": dm.BORDER_HI,

    "min":                dm.DIM,

    "min_prelight":       dm.SUBTLE,

    "min_pressed":        dm.FG,

    "min_unfocused":      dm.BORDER_HI,

}

CONTROL_DISC = {

    "close": "#bf616a", "close_prelight": "#bf616a", "close_pressed": "#d52735",

    "close_unfocused": "#3b4252",

    "maximize": "#a3be8c", "maximize_prelight": "#a3be8c",

    "maximize_pressed": "#13c11e", "maximize_unfocused": "#d4d4d4",

    "min": "#ebcb8b", "min_prelight": "#ebcb8b",

    "min_pressed": "#fac536", "min_unfocused": "#d4d4d4",

}

CONTROL_RIM = {

    "#232831": dm.SHADOW, "#9f1d2b": dm.SHADOW, "#975914": dm.SHADOW,

    "#0b7407": dm.SHADOW, "#b4b4b4": dm.SHADOW,

}

def asset_map(path):

    stem = os.path.basename(path)[:-4].replace("@2", "")

    table = dict(EXACT)

    if stem in CONTROL:

        table.update(CONTROL_RIM)

        table[CONTROL_DISC[stem]] = CONTROL[stem]

    text = open(path).read()

    out = re.sub(

        r"#[0-9a-fA-F]{6}",

        lambda m: table.get(m.group(0).lower(), m.group(0)),

        text,

    )

    if out != text:

        open(path, "w").write(out)

        return True

    return False

def main():

    n = sum(xfwm_map(p) for p in sorted(glob.glob("xfwm4/assets/*.svg")))

    print("xfwm4:  recoloured %d SVGs" % n)

    n = sum(asset_map(p) for p in sorted(glob.glob("assets/*.svg")))

    print("assets: recoloured %d SVGs" % n)

if __name__ == "__main__":

    main()