#!/usr/bin/env python3
"""Build a manual-install release ZIP for Owen Graphite v3.

Bundles src/ via bundle_v3.py, promotes the result to theme.css, and packages
the standard release assets into dist/Owen-Graphite-<version>.zip.
"""

from __future__ import annotations

import argparse
import importlib.util
import json
import shutil
import sys
import zipfile
from pathlib import Path


ROOT = Path(__file__).resolve().parents[2]
DEFAULT_FILES = [
    "theme.css",
    "manifest.json",
    "README.md",
    "CHANGELOG.md",
    "LICENSE",
    "dev/WIKI/MAP/map-info-classification.md",
    "dev/WIKI/MAP/theme-css-risk-map.html",
    "dev/WIKI/MAP/theme-css-risk-map.json",
    "dev/WIKI/MAP/selector-provenance.json",
    "screenshots/light.png",
    "screenshots/dark.png",
    "screenshots/readme/workspace-chrome-connected-glass.svg",
    "screenshots/readme/top-tabs-liquid-glass.svg",
    "screenshots/readme/workspace-writing-surface.jpg",
    "screenshots/readme/style-settings-report-options.jpg",
    "screenshots/readme/pdf-customer-delivery-feature.png",
    "screenshots/readme/owen-editor-toolbar-settings.jpg",
    "screenshots/readme/file-explorer-type-badges.svg",
    "screenshots/readme/sponsor-coffee.svg",
]


def version() -> str:
    return json.loads((ROOT / "manifest.json").read_text(encoding="utf-8"))["version"]


def bundle_v3() -> Path:
    script = ROOT / "dev" / "scripts" / "bundle_v3.py"
    spec = importlib.util.spec_from_file_location("bundle_v3", script)
    if spec is None or spec.loader is None:
        raise RuntimeError("unable to load dev/scripts/bundle_v3.py")
    module = importlib.util.module_from_spec(spec)
    sys.modules[spec.name] = module
    spec.loader.exec_module(module)
    result = module.main()
    if result:
        raise RuntimeError("bundle_v3.py failed")
    return ROOT / "dist" / "theme-v3.css"


def build_src_map() -> None:
    script = ROOT / "dev" / "scripts" / "build_src_map.py"
    spec = importlib.util.spec_from_file_location("build_src_map", script)
    if spec is None or spec.loader is None:
        raise RuntimeError("unable to load dev/scripts/build_src_map.py")
    module = importlib.util.module_from_spec(spec)
    sys.modules[spec.name] = module
    spec.loader.exec_module(module)
    result = module.main()
    if result:
        raise RuntimeError("build_src_map.py failed")


def audit_pdf_header_footer() -> None:
    script = ROOT / "dev" / "scripts" / "audit_pdf_header_footer.py"
    spec = importlib.util.spec_from_file_location("audit_pdf_header_footer", script)
    if spec is None or spec.loader is None:
        raise RuntimeError("unable to load dev/scripts/audit_pdf_header_footer.py")
    module = importlib.util.module_from_spec(spec)
    sys.modules[spec.name] = module
    spec.loader.exec_module(module)
    result = module.main()
    if result:
        raise RuntimeError("PDF header/footer audit failed")


def promote_to_theme_css(bundle: Path) -> None:
    target = ROOT / "theme.css"
    if not bundle.is_file():
        raise FileNotFoundError(f"bundle missing: {bundle}")
    shutil.copy2(bundle, target)
    print(f"OK: promoted {bundle.relative_to(ROOT)} -> theme.css")


def build(output_dir: Path, skip_bundle: bool = False) -> Path:
    if not skip_bundle:
        bundle = bundle_v3()
        promote_to_theme_css(bundle)
    build_src_map()
    audit_pdf_header_footer()
    release_version = version()
    output_dir.mkdir(parents=True, exist_ok=True)
    zip_path = output_dir / f"Owen-Graphite-{release_version}.zip"
    if zip_path.exists():
        zip_path.unlink()

    with zipfile.ZipFile(zip_path, "w", compression=zipfile.ZIP_DEFLATED) as archive:
        for rel in DEFAULT_FILES:
            source = ROOT / rel
            if not source.is_file() or source.stat().st_size == 0:
                raise FileNotFoundError(f"missing release asset: {rel}")
            archive.write(source, Path("Owen Graphite") / rel)

    print(f"OK: built {zip_path} ({zip_path.stat().st_size // 1024} KB)")
    for old_zip in output_dir.glob("Owen-Graphite-*.zip"):
        if old_zip != zip_path:
            old_zip.unlink()
            print(f"OK: removed old release ZIP {old_zip}")
    return zip_path


def main() -> int:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--output-dir", type=Path, default=ROOT / "dist", help="Directory for generated release ZIP.")
    parser.add_argument("--skip-bundle", action="store_true", help="Reuse existing theme.css; skip bundle + promote.")
    args = parser.parse_args()
    build(args.output_dir, skip_bundle=args.skip_bundle)
    return 0


if __name__ == "__main__":
    raise SystemExit(main())