#!/usr/bin/env python3
"""
LayerNX © – LNX Search: SearXNG beim Image-Build auf LayerNX umbauen.

    python3 lnx_brand.py <searx-paket> <lnx-ordner>

* Theme: lnx-search.css wird an das SearXNG-Theme angehängt (Wiki-Stil, Schriften, Farben)
* Startseite: Logo „LayerNX Search“, angemeldet stattdessen eine Begrüßung („Guten Abend, Moritz“) + Dienste
* Logo: eigenes SVG-Logo (lnx-logo.html / lnx-mark.html) statt SearXNG-Logo
* Kopfzeile: Konto-Link (accounts.layernx.de), Cookie-Hinweis, Footer mit LayerNX-Domains
* Dateien: Schriften, Win7-Symbole, Skripte nach static/themes/simple/lnx/
* Favicons/App-Icons: LayerNX
* vorkomprimierte .gz-Dateien werden neu erzeugt, veraltete .br entfernt

„Powered by SearXNG“ und der Link zum Original-Quellcode bleiben (AGPL); zusätzlich verlinkt der
Footer den Quellcode dieser Version (rechtliches.layernx.de/credits#lnx-search).
Findet das Skript eine Stelle in einer neueren SearXNG-Version nicht, gibt es eine Warnung aus.
"""
from __future__ import annotations

import base64
import gzip
import re
import shutil
import sys
from pathlib import Path

warnings = 0


def warn(msg: str) -> None:
    global warnings
    warnings += 1
    print(f"[lnx-brand] WARNUNG: {msg}", file=sys.stderr)


def recompress(path: Path) -> None:
    """Whitenoise liefert vorkomprimierte Varianten aus – die müssen zur Datei passen."""
    br = path.with_name(path.name + ".br")
    if br.exists():
        br.unlink()
    gz = path.with_name(path.name + ".gz")
    if path.suffix in (".css", ".js", ".svg", ".html", ".json"):
        gz.write_bytes(gzip.compress(path.read_bytes(), 9, mtime=0))
    elif gz.exists():
        gz.unlink()


def replace_in(path: Path, old: str, new: str, count: int = 1, regex: bool = False) -> None:
    text = path.read_text(encoding="utf-8")
    if regex:
        out, n = re.subn(old, new, text, count=count, flags=re.S)
        if not n:
            warn(f"Muster '{old[:50]}' nicht gefunden in {path.name}")
            return
    else:
        if old not in text:
            warn(f"'{old[:50]}' nicht gefunden in {path.name}")
            return
        out = text.replace(old, new, count)
    path.write_text(out, encoding="utf-8")


def copy_tree(src: Path, dst: Path) -> int:
    n = 0
    if not src.is_dir():
        warn(f"{src} fehlt")
        return 0
    for f in src.rglob("*"):
        if f.is_file():
            t = dst / f.relative_to(src)
            t.parent.mkdir(parents=True, exist_ok=True)
            shutil.copyfile(f, t)
            recompress(t)
            n += 1
    return n


FOOTER = """<footer class="lnx-foot">
    <span><a href="https://moritzsoft.de" rel="noopener">moritzsoft.de</a> · <a href="https://layernx.de">layernx.de</a> · <a href="https://layernx.systems">layernx.systems</a></span>
    <span>© LayerNX · <a href="https://rechtliches.layernx.de">Rechtliches</a> · <a href="https://rechtliches.layernx.de/#datenschutz">Datenschutz</a> · <a href="#" data-consent-open>Cookies</a></span>
    <p class="sx">{{ _('Powered by') }} <a href="{{ url_for('info', pagename='about') }}">SearXNG</a> {{ searxng_version }} · <a href="{{ searxng_git_url }}">{{ _('Source code') }}</a> · <a href="https://rechtliches.layernx.de/credits#lnx-search">Quellcode dieser Version &amp; Credits</a>{% if enable_metrics %} · <a href="{{ url_for('stats') }}">{{ _('Engine stats') }}</a>{% endif %}</p>
  </footer>"""

HEAD_SCRIPTS = """
  <script src="{{ url_for('static', filename='lnx/lnx-common.js') }}" defer></script>
  <script src="{{ url_for('static', filename='lnx/js/lnx-consent.js') }}" data-legal="https://rechtliches.layernx.de/#cookies" defer></script>
  <script src="{{ url_for('static', filename='lnx/js/lnx-notice.js') }}" defer></script>"""

ACCOUNT_LINK = """<a id="lnxAccount" class="link_on_top_account" href="https://accounts.layernx.de/login"><img src="{{ url_for('static', filename='lnx/icons/account.svg') }}" alt="" width="18" height="18"><span>Anmelden</span></a>
    </nav>"""


def main(searx: Path, lnx: Path) -> int:
    static = searx / "static" / "themes" / "simple"
    img = static / "img"
    tpl = searx / "templates" / "simple"
    brand = lnx / "brand"
    out = static / "lnx"

    # 1. Theme-CSS anhängen
    css = (lnx / "lnx-search.css").read_text(encoding="utf-8")
    for name in ("sxng-ltr.min.css", "sxng-rtl.min.css"):
        p = static / name
        if p.exists():
            p.write_text(p.read_text(encoding="utf-8") + "\n" + css, encoding="utf-8")
            recompress(p)
        else:
            warn(f"{name} fehlt")

    # 2. Eigene Dateien: Skripte, Kritzeleien, Schriften, Symbole, Cookie-Hinweis
    n = copy_tree(lnx / "static", out)
    n += copy_tree(lnx / "fonts", out / "fonts")
    n += copy_tree(lnx / "icons", out / "icons")
    n += copy_tree(lnx / "consent" / "js", out / "js")
    n += copy_tree(lnx / "consent" / "css", out / "css")
    print(f"[lnx-brand] {n} Dateien nach static/themes/simple/lnx kopiert")

    # 3. Favicons & App-Icons
    fav192 = (brand / "favicon-192.png").read_bytes()
    fav_svg = ('<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 192 192">'
               f'<image width="192" height="192" href="data:image/png;base64,{base64.b64encode(fav192).decode()}"/></svg>')
    for name, data in {"favicon.png": fav192, "favicon.svg": fav_svg.encode(), "192.png": fav192,
                       "512.png": (brand / "nx.png").read_bytes(), "searxng.png": (brand / "layernx-wordmark.png").read_bytes(),
                       "searxng.svg": (brand / "layernx-logo.svg").read_bytes()}.items():
        p = img / name
        p.write_bytes(data)
        recompress(p)

    # 4. Templates: Startseite, Logos
    for name in ("index.html", "lnx-logo.html", "lnx-mark.html"):
        shutil.copyfile(lnx / "templates" / name, tpl / name)
    # Logo neben dem Suchfeld (wird „without context“ eingebunden -> reines SVG)
    shutil.copyfile(lnx / "templates" / "lnx-mark.html", tpl / "searxng-wordmark.min.svg")
    replace_in(tpl / "search.html", "<span hidden>SearXNG</span>", "<span hidden>LayerNX Search</span>")
    replace_in(tpl / "page_with_header.html", r'<img class="logo"[^>]*>', "{% include 'simple/lnx-logo.html' %}", regex=True)

    # 5. base.html: Meta, Skripte, Konto-Link, Footer
    base = tpl / "base.html"
    replace_in(base, r'<meta name="description" content="[^"]*">',
               '<meta name="description" content="LayerNX Search – die Metasuche von LayerNX, ohne Tracking.">', regex=True)
    replace_in(base, r"\{% endblock %\}\{\{ instance_name \}\}</title>", "{% endblock %}LayerNX Search</title>", regex=True)
    replace_in(base, r'<meta name="keywords" content="[^"]*">',
               '<meta name="keywords" content="LNX Search, LayerNX, Suche, Metasuche, Startseite">', regex=True)
    replace_in(base, r"(<script type=\"module\" src=\"\{\{ url_for\('static', filename='sxng-core\.min\.js'\) \}\}\"[^>]*></script>)",
               r"\1" + HEAD_SCRIPTS.replace("\\", "\\\\"), regex=True)
    replace_in(base, "    </nav>\n    {% block header %}", "    " + ACCOUNT_LINK + "\n    {% block header %}")
    replace_in(base, r"<footer>.*?</footer>", FOOTER.replace("\\", "\\\\"), regex=True)

    print(f"[lnx-brand] LayerNX-Theme angewendet ({warnings} Warnung(en))")
    return 0


if __name__ == "__main__":
    sys.exit(main(Path(sys.argv[1]), Path(sys.argv[2])))