from __future__ import annotations

import json
import re
import urllib.request
from pathlib import Path


ROOT = Path(__file__).resolve().parent.parent
TEMPLATE_DIR = ROOT / "scripts" / "project_rules"
API_BASE = "http://127.0.0.1:8000"


def fetch(path: str):
    with urllib.request.urlopen(f"{API_BASE}{path}") as response:
        return json.load(response)


def render(template: str, values: dict[str, str]) -> str:
    for key, value in values.items():
        template = template.replace("{" + key + "}", value)
    return template


def repo_topic_id(repo: dict, topics: list[dict]) -> str:
    if repo.get("topic_id"):
        return repo["topic_id"]
    match = next((t for t in topics if t.get("domain_slug") == repo.get("domain_slug")), None)
    return match["id"] if match else "(none)"


def wp_prefix(repo_slug: str) -> str:
    first = repo_slug.split("-", 1)[0].upper()
    return f"{first}-WP"


def brief_domain(path: Path) -> str | None:
    brief = path / ".custodian-brief.md"
    if not brief.exists():
        return None
    match = re.search(r"^\*\*Domain:\*\*\s+(\S+)\s*$", brief.read_text(encoding="utf-8"), re.MULTILINE)
    return match.group(1) if match else None


def choose_repos(repos: list[dict]) -> list[dict]:
    by_path: dict[str, list[dict]] = {}
    for repo in repos:
        local_path = repo.get("local_path") or ""
        path = Path(local_path)
        if not local_path.startswith("/home/worsch/") or not path.exists():
            continue
        by_path.setdefault(str(path), []).append(repo)

    chosen: list[dict] = []
    for local_path, candidates in sorted(by_path.items()):
        path = Path(local_path)
        domain = brief_domain(path)
        if domain:
            domain_matches = [r for r in candidates if r.get("domain_slug") == domain]
            if domain_matches:
                candidates = domain_matches
        active = [r for r in candidates if r.get("status") == "active"]
        chosen.append(active[0] if active else candidates[0])
    return chosen


def main() -> None:
    repos = fetch("/repos/")
    topics = fetch("/topics/?status=active")

    agents_template = (TEMPLATE_DIR / "agents-codex.template").read_text(encoding="utf-8")
    claude_template = (TEMPLATE_DIR / "claude-md.template").read_text(encoding="utf-8")
    scope_template = (TEMPLATE_DIR / "scope.template").read_text(encoding="utf-8")
    rule_names = [
        "repo-identity",
        "session-protocol",
        "first-session",
        "workplan-convention",
        "stack-and-commands",
        "architecture",
        "repo-boundary",
        "agents",
    ]
    rule_templates = {
        name: (TEMPLATE_DIR / f"{name}.template").read_text(encoding="utf-8")
        for name in rule_names
    }

    updated: list[str] = []
    for repo in choose_repos(repos):
        path = Path(repo["local_path"])
        repo_slug = repo["slug"]
        project_name = repo.get("name") or path.name
        description = repo.get("description") or f"{project_name} - (fill in purpose)"
        values = {
            "PROJECT_NAME": project_name,
            "PROJECT_DESCRIPTION": description,
            "DOMAIN": repo.get("domain_slug") or "",
            "TOPIC_ID": repo_topic_id(repo, topics),
            "REPO_SLUG": repo_slug,
            "WP_PREFIX": wp_prefix(repo_slug),
        }

        (path / "AGENTS.md").write_text(render(agents_template, values), encoding="utf-8")
        (path / "CLAUDE.md").write_text(render(claude_template, values), encoding="utf-8")
        scope_path = path / "SCOPE.md"
        if not scope_path.exists():
            scope_path.write_text(render(scope_template, values), encoding="utf-8")

        rules_dir = path / ".claude" / "rules"
        rules_dir.mkdir(parents=True, exist_ok=True)
        for name, template in rule_templates.items():
            (rules_dir / f"{name}.md").write_text(render(template, values), encoding="utf-8")

        updated.append(f"{repo_slug}\t{path}")

    print(f"Updated {len(updated)} local repo(s):")
    for line in updated:
        print(line)


if __name__ == "__main__":
    main()