#!/usr/bin/env python3
"""DW store kit: build.py (DRAFT). Renders the store pages for one tenant into an output folder.

READ-ONLY: it only calls public endpoints on the tenant's own domain
  POST /store/products/ (list, read), POST /store/products/categories/ (list), POST /files/ (read, PUBLIC files only)
and GETs the page named in site_chrome.chrome_from to copy the brand's <header> and <footer>.
It never sends a session or brand code and never writes to the app. Upload the output with /files (see AGENT.md).

Usage: python3 build.py CONFIG.json OUT_DIR [--base https://example.com]
"""
import html, json, re, sys, urllib.request, urllib.error
from pathlib import Path

e = lambda s: html.escape(str(s if s is not None else ''), quote=True)
KIT = Path(__file__).resolve().parent.parent / 'templates'


def post(base, path, body):
    req = urllib.request.Request(base + path, data=json.dumps(body).encode(), method='POST',
                                 headers={'Content-Type': 'application/json', 'User-Agent': 'DWStoreKit/0.1'})
    try:
        with urllib.request.urlopen(req, timeout=60) as r:
            return json.loads(r.read())
    except urllib.error.HTTPError as x:
        return json.loads(x.read() or b'{}')


def get(base, path):
    req = urllib.request.Request(base + path, headers={'User-Agent': 'DWStoreKit/0.1'})
    with urllib.request.urlopen(req, timeout=60) as r:
        return r.read().decode('utf-8', 'replace')


def strip_loader(s):
    # The app injects one inline <script> line that loads /_dw/edit.js; drop it from copied chrome.
    return re.sub(r'(?m)^<script>[^\n]*/_dw/edit\.js[^\n]*</script>\n?', '', s)


def main():
    if len(sys.argv) < 3:
        sys.exit(__doc__)
    cfg = json.load(open(sys.argv[1]))
    out = Path(sys.argv[2])
    base = sys.argv[sys.argv.index('--base') + 1] if '--base' in sys.argv else 'https://' + cfg['canonical_host']
    chrome = cfg.get('site_chrome', {})

    cats = post(base, '/store/products/categories/', {'action': 'list'}).get('categories', [])
    if cfg.get('company_id'):
        cats = [c for c in cats if c['company_id'] == cfg['company_id']]
    cats.sort(key=lambda c: (c['position'], c['name']))
    lst = post(base, '/store/products/', {'action': 'list', 'limit': 100}).get('products', [])
    prods = {}
    for p in lst:
        r = post(base, '/store/products/', {'action': 'read', 'slug': p['slug']})
        if r.get('ok'):
            prods[p['slug']] = r['product']
    cfg_prod = {p['slug']: p for p in cfg.get('products', [])}

    url_cache = {}

    def file_url(fid):
        # PUBLIC /files read needs no session and returns file_name + file_path (and the bytes, which we drop).
        if fid not in url_cache:
            f = post(base, '/files/', {'action': 'read', 'id': fid}).get('file') or {}
            p = (f.get('file_path') or '').strip('/')
            url_cache[fid] = ('/' + (p + '/' if p else '') + f['file_name']) if f.get('file_name') else None
        return url_cache[fid]

    def images(p):
        outl = []
        for i in sorted(p.get('images', []), key=lambda i: i['position']):
            desc = i.get('description') or {}
            u = desc.get('src') if isinstance(desc, dict) else None
            u = u or file_url(i['file_id'])
            if u:
                outl.append((u, (desc.get('alt') if isinstance(desc, dict) else None) or p['name']))
        return outl or [('/assets/store-placeholder.jpg', p['name'])]

    def variant(p):
        v = [x for x in p['variants'] if x['variant_id'] == p['default_variant_id']] or p['variants']
        return v[0] if v else None

    def sold(p):
        v = variant(p)
        return v is not None and v['stock'] is not None and v['stock'] <= 0

    page = strip_loader(get(base, chrome.get('chrome_from', '/index.html')))
    hdr = re.search(r'<header\b.*?</header>', page, re.S)
    ftr = re.search(r'<footer\b.*?</footer>', page, re.S)
    absol = lambda s: re.sub(r'(href|src)="(?!https?:|mailto:|tel:|/|#)([^"]+)"', lambda m: f'{m.group(1)}="/{m.group(2)}"', s)
    header = absol(hdr.group(0)) if hdr else ''
    footer = absol(ftr.group(0)) if ftr else ''

    js_cfg = {k: cfg.get(k) for k in ('brand_name', 'contact_email', 'company_id')}
    co = cfg.get('checkout', {})
    js_cfg.update({'payment_mode': co.get('payment_mode', 'offline'), 'record_pending_payment': bool(co.get('record_pending_payment')),
                   'payment_method': co.get('payment_method'), 'payment_note': co.get('payment_note')})
    common = {
        'BRAND_NAME': e(cfg['brand_name']), 'CANONICAL_HOST': e(cfg['canonical_host']), 'CONTACT_EMAIL': e(cfg.get('contact_email', '')),
        'HEAD_EXTRA': chrome.get('head_extra', ''), 'SITE_HEADER': header, 'SITE_FOOTER': footer,
        'STORE_CONFIG_JSON': json.dumps(js_cfg).replace('</', '<\\/'),
    }

    def fill(tpl, vals):
        s = (KIT / 'store' / tpl).read_text()
        for k, v in {**common, **vals}.items():
            s = s.replace('{{' + k + '}}', v)
        left = sorted(set(re.findall(r'\{\{([A-Z0-9_]+)\}\}', s)))
        if left:
            raise SystemExit(f'{tpl}: unfilled placeholders {left}')
        return s

    def anchor(c):
        d = c.get('description') or {}
        return (d.get('anchor') if isinstance(d, dict) else None) or re.sub(r'[^a-z0-9]+', '-', c['name'].lower()).strip('-')

    def tagline(p):
        return cfg_prod.get(p['slug'], {}).get('tagline', '')

    def card(p):
        im = images(p)[0]
        flag = '<span class="dw-flag" data-stock-flag>Sold out</span>' if sold(p) else '<span class="dw-flag" data-stock-flag style="display:none">Sold out</span>'
        return (f'<a class="dw-card" href="/store/{e(p["slug"])}.html" data-slug="{e(p["slug"])}"><div class="ph">{flag}'
                f'<img src="{e(im[0])}" alt="{e(im[1])}" loading="lazy" width="800" height="1000"></div><h3>{e(p["name"])}</h3>'
                f'<p class="tag">{e(tagline(p))}</p><div class="dw-row"><span class="dw-price" data-price>${e(p["price"])}</span>'
                f'<span class="dw-badge{" out" if sold(p) else ""}" data-stock-badge>{"Sold out" if sold(p) else "In stock"}</span></div></a>')

    bycat = {c['category_id']: [] for c in cats}
    for p in sorted(prods.values(), key=lambda p: p['name']):
        for c in p['categories']:
            if c['category_id'] in bycat:
                bycat[c['category_id']].append(p)
    sections = []
    for i, c in enumerate(cats):
        d = c.get('description') if isinstance(c.get('description'), dict) else {}
        sections.append(f'<section class="dw-section{" alt" if i % 2 else ""}" id="{e(anchor(c))}"><div class="dw-wrap"><h2>{e(d.get("title") or c["name"])}</h2>'
                        f'<p class="dw-note">{e(d.get("summary", ""))}</p><div class="dw-grid">{"".join(card(p) for p in bycat[c["category_id"]])}</div></div></section>')
    uncategorised = [p for p in prods.values() if not any(c['category_id'] in bycat for c in p['categories'])]
    if uncategorised:
        sections.append('<section class="dw-section" id="more"><div class="dw-wrap"><h2>More</h2><div class="dw-grid">' + ''.join(card(p) for p in uncategorised) + '</div></div></section>')

    (out / 'store').mkdir(parents=True, exist_ok=True)
    (out / 'store/index.html').write_text(fill('index.html', {
        'STORE_DESCRIPTION': e(cfg.get('store_description', f'Shop {cfg["brand_name"]}.')),
        'STORE_HEADING': e(cfg.get('store_heading', 'Shop')), 'STORE_INTRO': e(cfg.get('store_intro', '')),
        'CATEGORY_NAV': ''.join(f'<a href="#{e(anchor(c))}">{e(c["name"])}</a>' for c in cats),
        'CATEGORY_SECTIONS': '\n'.join(sections)}))
    catmap = {c['category_id']: c for c in cats}
    for slug, p in prods.items():
        im = images(p)
        c = next((catmap[x['category_id']] for x in p['categories'] if x['category_id'] in catmap), {'name': 'Shop', 'description': {'anchor': ''}})
        desc = next((x['description'] for x in p['descriptions'] if x['used_location'] == 'main'), '')
        plain = re.sub(r'\s+', ' ', re.sub('<[^>]+>', ' ', desc)).strip()
        thumbs = ''
        if len(im) > 1:
            thumbs = '<div class="thumbs">' + ''.join(f'<button type="button" data-src="{e(u)}" aria-label="Photo {k + 1} of {len(im)}" aria-current="{"true" if k == 0 else "false"}"><img src="{e(u)}" alt="" loading="lazy" width="120" height="120"></button>' for k, (u, _) in enumerate(im)) + '</div>'
        v = variant(p)
        ld = {'@context': 'https://schema.org', '@type': 'Product', 'name': p['name'], 'image': ['https://' + cfg['canonical_host'] + u for u, _ in im],
              'description': plain[:500], 'offers': {'@type': 'Offer', 'price': p['price'], 'priceCurrency': 'USD',
              'availability': 'https://schema.org/' + ('OutOfStock' if sold(p) else 'InStock'), 'url': f'https://{cfg["canonical_host"]}/store/{slug}.html'}}
        (out / f'store/{slug}.html').write_text(fill('product.html', {
            'PRODUCT_NAME': e(p['name']), 'PRODUCT_SLUG': e(slug), 'PRODUCT_META': e((p['name'] + ' — ' + (tagline(p) or plain))[:300]),
            'PRODUCT_IMAGE_1': e(im[0][0]), 'PRODUCT_IMAGE_1_ALT': e(im[0][1]), 'PRODUCT_THUMBS': thumbs,
            'PRODUCT_JSONLD': json.dumps(ld).replace('</', '<\\/'), 'CATEGORY_ANCHOR': e(anchor(c)), 'CATEGORY_NAME': e(c['name']),
            'PRODUCT_TAGLINE': e(tagline(p)), 'PRODUCT_PRICE': e(v['price'] if v else p['price']),
            'SOLD_CLASS': ' out' if sold(p) else '', 'STOCK_TEXT': 'Sold out' if sold(p) else 'In stock',
            'DEFAULT_VARIANT_ID': e(p['default_variant_id'] or ''), 'SOLD_ARIA': 'true' if sold(p) else 'false',
            'BUY_TEXT': 'Sold out' if sold(p) else 'Add to cart', 'PRODUCT_NOTICE': cfg.get('product_notice_html', ''),
            'PRODUCT_DESCRIPTION_HTML': desc}))  # staff-authored HTML from /store/product/description, published as-is
    for name in ('cart.html', 'checkout.html', 'account.html'):
        (out / 'store' / name).write_text(fill(name, {}))
    for sub in ('js', 'css'):
        (out / 'assets' / sub).mkdir(parents=True, exist_ok=True)
        for f in (KIT / 'assets' / sub).iterdir():
            (out / 'assets' / sub / f.name).write_text(f.read_text())
    print(f'built {len(prods)} product pages, {len(cats)} categories, cart/checkout/account into {out}')


if __name__ == '__main__':
    main()