"""Small Linux integration fixes for the pinned community WebUI.

Keep these transformations explicit and fail the image build if upstream moves
the expected code.  That prevents a future upstream update from silently
producing a half-patched console.
"""
from pathlib import Path
import sys


root = Path(sys.argv[1])
app_js = root / "static" / "app.js"
index_html = root / "static" / "index.html"
server_py = root / "server.py"


def replace_once(path: Path, old: str, new: str) -> None:
    text = path.read_text(encoding="utf-8")
    if text.count(old) != 1:
        raise RuntimeError(f"expected exactly one patch marker in {path}: {old[:80]!r}")
    path.write_text(text.replace(old, new, 1), encoding="utf-8")


# The published UI calls refreshArt() during startup, but does not define it.
# Athena does not install the Windows-only stable-diffusion.cpp helper, so show
# that state honestly and keep the optional controls inert instead of crashing
# the whole page.
art_marker = '''  $("artDirAdd").addEventListener("click", function () {
'''
art_fix = '''  function refreshArt() {
    return api("/api/art/status").then(function (data) {
      STATE.art = data;
      var installed = !!data.installed;
      var supported = !!data.supported;
      $("artState").textContent = data.busy || (installed ? "installed" :
        (supported ? "not installed" : "not available on Linux"));
      $("artState").dataset.s = data.busy ? "busy" : (installed ? "ready" : "missing");
      $("artInstall").disabled = !!data.busy || !supported;
      $("artInstall").textContent = installed ? "Reinstall" : "Install";
      $("setArtAuto").checked = !!data.auto;
      $("setArtAuto").disabled = !installed;
      $("setArtModel").innerHTML = (data.models || []).length
        ? data.models.map(function (m) {
            return '<option value="' + escape(m.path) + '">' + escape(m.name || m.file) + '</option>';
          }).join("")
        : '<option value="">no art model available</option>';
      if (data.selected) $("setArtModel").value = data.selected;
      $("setArtModel").disabled = !installed;
      $("artDir").disabled = !supported;
      $("artDirAdd").disabled = !supported;
      $("artDirs").textContent = supported
        ? ((data.dirs || []).length ? "Also scanning: " + data.dirs.join("  ·  ") : "No extra folders configured.")
        : "The community release only provides the cover-art installer for Windows; song generation is unaffected.";
      $("artCatalog").innerHTML = "";
      return data;
    }).catch(function (error) {
      $("artState").textContent = "unavailable";
      $("artInstall").disabled = true;
      $("artDirs").textContent = error.message;
    });
  }

  $("artDirAdd").addEventListener("click", function () {
'''
replace_once(app_js, art_marker, art_fix)

# YuE2 natively supports an empty lyric string for true instrumentals.  The
# community server rejected that valid request even though the browser did not.
replace_once(
    server_py,
    '''    if not body.style.strip() or not body.lyrics.strip():
        raise HTTPException(400, "A style prompt and lyrics are both required")
''',
    '''    if not body.style.strip():
        raise HTTPException(400, "A style prompt is required")
''',
)

replace_once(
    index_html,
    '''<span class="label">Lyrics <em>section tags on their own line</em></span>''',
    '''<span class="label">Lyrics <em>section tags on their own line · leave empty for a true instrumental</em></span>''',
)