build_zip.py

"""Build distribution ZIPs for Chrome Web Store / Firefox AMO / Edge Add-ons.

All three stores accept the same zip — the only thing Firefox is stricter
about is no inline <script>/<style> (handled by extracting options.js +
options.css) and explicit "permissions" in manifest (handled by adding
"storage" to permissions array).

Output:
  growth-math-v<version>.zip   — runtime only, suitable for all 3 stores

Excludes:
  - generate_*.py, render_*.py, test_*.py, build_zip.py, _retry_hero.py
    (Python tooling that never needs to ship to users)
  - promo/, README.md, SUBMISSION.md, .gitignore
    (developer-only assets and docs)
  - icon.svg
    (PNG icons listed in manifest are what stores read; SVG is source)

Run:
  python extension/build_zip.py
"""
import json
import os
import sys
import zipfile
from pathlib import Path

EXT_DIR = Path(__file__).parent
ROOT = EXT_DIR.parent


def main():
    manifest_path = EXT_DIR / 'manifest.json'
    manifest = json.loads(manifest_path.read_text(encoding='utf-8'))
    version = manifest['version']

    out = ROOT / f'growth-math-v{version}.zip'

    # Whitelist — exactly what users need at runtime
    runtime_files = [
        'manifest.json',
        'popup.html',
        'popup.css',
        'popup.js',
        'options.html',
        'options.css',
        'options.js',
        '_locales/en/messages.json',
        '_locales/ru/messages.json',
        'icons/icon-16.png',
        'icons/icon-32.png',
        'icons/icon-48.png',
        'icons/icon-128.png',
    ]

    missing = []
    with zipfile.ZipFile(out, 'w', zipfile.ZIP_DEFLATED, compresslevel=9) as zf:
        for rel in runtime_files:
            src = EXT_DIR / rel
            if not src.exists():
                missing.append(rel)
                continue
            zf.write(src, arcname=rel)

    if missing:
        sys.stderr.write(f'[ERROR] Missing files (will reject by stores):\n')
        for m in missing:
            sys.stderr.write(f'  - {m}\n')
        sys.exit(1)

    size_kb = out.stat().st_size / 1024
    sys.stderr.write(f'\n[OK] Built {out.name}\n')
    sys.stderr.write(f'  path:    {out}\n')
    sys.stderr.write(f'  size:    {size_kb:.1f} KB\n')
    sys.stderr.write(f'  version: {version}\n')
    sys.stderr.write(f'  files:   {len(runtime_files)}\n')

    # Print contents for verification
    sys.stderr.write('\n  Contents:\n')
    with zipfile.ZipFile(out) as zf:
        for n in zf.namelist():
            info = zf.getinfo(n)
            sys.stderr.write(f'    {info.file_size:>7,} B  {n}\n')


if __name__ == '__main__':
    main()