generators.py

"""aiseed web creator — サイトマップ・フィード・QRコード生成"""

import json
from pathlib import Path
from datetime import datetime

import requests


def generate_sitemap(output_dir: Path, site_url: str, pages: list, blog_posts: list):
    """sitemap.xml と robots.txt を output_dir に生成"""
    if not site_url:
        return
    site_url = site_url.rstrip("/")
    now = datetime.now().strftime("%Y-%m-%d")
    urls = []
    for p in pages:
        fn = p.get("filename", "index.html")
        loc = f"{site_url}/{fn}" if fn != "index.html" else site_url + "/"
        urls.append(f'  <url><loc>{loc}</loc><lastmod>{now}</lastmod><priority>{"1.0" if fn=="index.html" else "0.8"}</priority></url>')
    for post in blog_posts:
        if post.get("status") != "publish":
            continue
        slug = post.get("slug", "")
        if slug:
            urls.append(f'  <url><loc>{site_url}/blog/{slug}.html</loc><lastmod>{post.get("date", now)}</lastmod><priority>0.6</priority></url>')
    sitemap = '<?xml version="1.0" encoding="UTF-8"?>\n<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">\n' + "\n".join(urls) + "\n</urlset>\n"
    (output_dir / "sitemap.xml").write_text(sitemap, "utf-8")
    robots = f"User-agent: *\nAllow: /\nSitemap: {site_url}/sitemap.xml\n"
    (output_dir / "robots.txt").write_text(robots, "utf-8")


def generate_feed(output_dir: Path, site_url: str, data: dict):
    """サイトの構造化データを feed.json として出力。
    aiseed ハブAPIへの同期やサードパーティ連携に使用。
    Returns: feed dict"""
    if not site_url:
        return {}
    site_url = site_url.rstrip("/")
    now = datetime.now().strftime("%Y-%m-%dT%H:%M:%S+09:00")

    site = data.get("site", {})
    feed = {
        "_aiseed": {
            "version": "1.0",
            "generated": now,
            "generator": "aiseed-web-cowork",
        },
        "site": {
            "url": site_url,
            "name": site.get("name", ""),
            "description": site.get("description", ""),
            "language": "ja",
        },
        "pages": [],
        "events": [],
        "blog": [],
        "business": {},
    }

    for p in data.get("pages", []):
        fn = p.get("filename", "index.html")
        feed["pages"].append({
            "url": f"{site_url}/{fn}" if fn != "index.html" else f"{site_url}/",
            "title": p.get("title", ""),
            "description": p.get("description", ""),
        })

    for ev in data.get("events", []):
        if ev.get("status") not in ("公開", "満員"):
            continue
        feed["events"].append({
            "title": ev.get("title", ""),
            "slug": ev.get("slug", ""),
            "url": f"{site_url}/events/{ev.get('slug', '')}.html",
            "date": ev.get("date", ""),
            "date_end": ev.get("date_end", ""),
            "time_start": ev.get("time_start", ""),
            "time_end": ev.get("time_end", ""),
            "category": ev.get("category", ""),
            "location": ev.get("location", ""),
            "location_url": ev.get("location_url", ""),
            "description": ev.get("description", "")[:500],
            "capacity": ev.get("capacity", ""),
            "price": ev.get("price", ""),
            "booking_url": ev.get("booking_url", ""),
            "status": ev.get("status", ""),
            "thumbnail": f"{site_url}/{ev['thumbnail']}" if ev.get("thumbnail") else "",
        })

    for post in data.get("blog", []):
        if post.get("status") != "publish":
            continue
        feed["blog"].append({
            "title": post.get("title", ""),
            "slug": post.get("slug", ""),
            "url": f"{site_url}/blog/{post.get('slug', '')}.html",
            "date": post.get("date", ""),
            "category": post.get("category", ""),
            "tags": post.get("tags", ""),
            "excerpt": post.get("excerpt", "")[:300],
            "thumbnail": f"{site_url}/{post['thumbnail']}" if post.get("thumbnail") else "",
        })

    for key in ("business", "shop", "farm", "company", "profile"):
        if key in data and isinstance(data[key], dict):
            feed["business"].update(data[key])
            break

    feed = {k: v for k, v in feed.items() if v}
    feed["_aiseed"] = {
        "version": "1.0",
        "generated": now,
        "generator": "aiseed-web-cowork",
    }

    (output_dir / "feed.json").write_text(
        json.dumps(feed, ensure_ascii=False, indent=2), "utf-8")
    return feed


def sync_to_hub(feed: dict, hub_url: str, site_token: str, log_fn=None):
    """feed.json の内容を aiseed ハブAPIに同期。
    Returns: (success: bool, message: str)"""
    if not hub_url or not site_token:
        return False, "ハブURLまたはサイトトークンが未設定です"
    try:
        headers = {
            "Authorization": f"Bearer {site_token}",
            "Content-Type": "application/json",
        }
        r = requests.post(
            f"{hub_url.rstrip('/')}/api/v1/sites/sync",
            headers=headers,
            json=feed,
            timeout=30,
        )
        if r.ok:
            msg = f"ハブに同期しました（{len(feed.get('events', []))}イベント, {len(feed.get('blog', []))}記事）"
            if log_fn:
                log_fn(msg)
            return True, msg
        else:
            msg = f"ハブ同期エラー: {r.status_code} {r.text[:200]}"
            if log_fn:
                log_fn(msg)
            return False, msg
    except Exception as ex:
        msg = f"ハブ接続エラー: {ex}"
        if log_fn:
            log_fn(msg)
        return False, msg


def generate_qr_codes(output_dir: Path, site_url: str, site_name: str = ""):
    """商品・値札・チラシ用のQRコードを生成。
    output_dir/qr/ にサイズ別のPNG + 印刷用SVGを保存。
    Returns: 生成されたファイルのリスト [(path, label), ...]"""
    if not site_url:
        return []
    site_url = site_url.rstrip("/")
    qr_dir = output_dir / "qr"
    qr_dir.mkdir(parents=True, exist_ok=True)
    results = []

    try:
        import qrcode
        import qrcode.image.svg
    except ImportError:
        return []

    sizes = [
        ("product",  "商品ラベル用（小）",   6, 2),
        ("price",    "値札用（中）",         10, 3),
        ("flyer",    "チラシ用（大）",       16, 4),
        ("poster",   "ポスター用（特大）",   24, 4),
    ]

    for name, label, box_size, border in sizes:
        qr = qrcode.QRCode(
            version=None,
            error_correction=qrcode.constants.ERROR_CORRECT_H,
            box_size=box_size,
            border=border,
        )
        qr.add_data(site_url)
        qr.make(fit=True)

        try:
            from PIL import Image
            img = qr.make_image(fill_color="black", back_color="white").convert("RGB")
            png_path = qr_dir / f"qr-{name}.png"
            img.save(png_path, "PNG")
            results.append((png_path, label))
        except ImportError:
            pass

    qr = qrcode.QRCode(
        version=None,
        error_correction=qrcode.constants.ERROR_CORRECT_H,
        box_size=10,
        border=4,
    )
    qr.add_data(site_url)
    qr.make(fit=True)
    svg_img = qr.make_image(image_factory=qrcode.image.svg.SvgPathImage)
    svg_path = qr_dir / "qr-print.svg"
    svg_img.save(str(svg_path))
    results.append((svg_path, "印刷用SVG（任意サイズに拡大可）"))

    return results