#!/usr/bin/env python3
"""Prepare a new Invidious release.

This automates the release procedure documented at
https://github.com/iv-org/documentation/blob/master/docs/create-new-invidious-release.md

What it does (all of it, so the GitHub workflow only has to run this script):
  1. Computes the next version `<MAJOR>.<YYYYMMDD>.<PATCH>` from shard.yml.
  2. Collects EVERY pull request merged since the previous release, via `gh`.
  3. Asks an LLM (OpenRouter API) to write only the changelog *prose*
     (wrap-up + categorised sections), using recent CHANGELOG.md as a style
     guide. The exhaustive PR list is generated deterministically so no PR is
     ever missed.
  4. Bumps shard.yml and prepends the new section to CHANGELOG.md.
  5. Creates the `release-vX.Y.Z` branch, commits, pushes, and opens a PR to
     master with the post-merge checklist (tagging + publishing stay manual,
     as the docs require maintainer review and forbid squash-merges/retagging).

The LLM is called through any OpenAI-compatible Chat Completions endpoint,
configured with the standard `OPENAI_API_KEY` / `OPENAI_BASE_URL` environment
variables (as used by the official OpenAI SDK). It defaults to OpenRouter, but
point it at OpenAI, a local server, etc. by setting `OPENAI_BASE_URL`.

Manual usage:
  export OPENAI_API_KEY=sk-or-...              # OpenRouter key by default
  ./scripts/create-release.py                  # full run: edits, commit, push, PR
  ./scripts/create-release.py --dry-run        # only edit files locally
  ./scripts/create-release.py --major 3        # bump major (breaking changes)
  ./scripts/create-release.py --patch 1        # 2nd release on the same day
  ./scripts/create-release.py --model x/y      # override the model
  # Use vanilla OpenAI instead of OpenRouter:
  OPENAI_BASE_URL=https://api.openai.com/v1 \
    ./scripts/create-release.py --model gpt-4o

Requirements: python3, git, and the GitHub CLI (`gh`) authenticated (or
GH_TOKEN set). Only OPENAI_API_KEY is mandatory (skippable with --dry-run
combined with --pr-json for offline testing).
"""

import argparse
import datetime
import json
import os
import re
import subprocess
import sys
import urllib.error
import urllib.request

DEFAULT_BASE_URL = "https://openrouter.ai/api/v1"
DEFAULT_MODEL = "deepseek/deepseek-v4-pro"
REPO_URL = "https://github.com/iv-org/invidious"
CHANGELOG_FILE = "CHANGELOG.md"
SHARD_FILE = "shard.yml"


def die(msg):
    prefix = "::error::" if os.environ.get("GITHUB_ACTIONS") else "error: "
    print(f"{prefix}{msg}", file=sys.stderr)
    sys.exit(1)


def run(cmd, capture=False, check=True):
    """Run a subprocess command (list of args)."""
    print(f"$ {' '.join(cmd)}", file=sys.stderr)
    result = subprocess.run(
        cmd, check=False, text=True,
        stdout=subprocess.PIPE if capture else None,
        stderr=subprocess.PIPE if capture else None,
    )
    if check and result.returncode != 0:
        detail = (result.stderr or "").strip() if capture else ""
        die(f"command failed ({result.returncode}): {' '.join(cmd)}\n{detail}")
    return (result.stdout or "").strip() if capture else ""


# --------------------------------------------------------------------------- #
# Version + git helpers
# --------------------------------------------------------------------------- #
def read_current_version():
    with open(SHARD_FILE, encoding="utf-8") as fh:
        for line in fh:
            m = re.match(r"^version:\s*(\S+)", line)
            if m:
                return m.group(1)
    die(f"no 'version:' line found in {SHARD_FILE}")


def compute_version(args):
    current = read_current_version()  # e.g. 2.20260626.0-dev
    print(f"Current shard.yml version: {current}", file=sys.stderr)
    major = str(args.major) if args.major is not None else current.split(".", 1)[0]
    date = datetime.datetime.now(datetime.timezone.utc).strftime("%Y%m%d")
    version = f"{major}.{date}.{args.patch}"
    tag = f"v{version}"
    existing = run(["git", "tag", "--list", tag], capture=True)
    if existing:
        die(f"tag {tag} already exists; pass --patch {args.patch + 1}")
    return version, tag


def previous_tag():
    tags = run(["git", "tag", "--list", "v*", "--sort=-creatordate"],
               capture=True)
    return tags.splitlines()[0].strip() if tags else ""


def tag_date(tag):
    return run(["git", "log", "-1", "--format=%cI", tag], capture=True)


# --------------------------------------------------------------------------- #
# Pull request collection
# --------------------------------------------------------------------------- #
def collect_prs(repo, prev_tag, pr_json_path):
    if pr_json_path:
        with open(pr_json_path, encoding="utf-8") as fh:
            prs = json.load(fh)
    else:
        cmd = ["gh", "pr", "list", "--repo", repo, "--base", "master",
               "--state", "merged", "--limit", "1000",
               "--json", "number,title,url,author,mergedAt"]
        if prev_tag:
            since = tag_date(prev_tag)
            print(f"Listing PRs merged after {since} ({prev_tag})",
                  file=sys.stderr)
            cmd += ["--jq", f'[.[] | select(.mergedAt > "{since}")]']
        else:
            print("No previous tag; listing all merged PRs.", file=sys.stderr)
        out = run(cmd, capture=True)
        prs = json.loads(out) if out else []
    prs.sort(key=lambda p: p.get("mergedAt") or "", reverse=True)
    print(f"Collected {len(prs)} merged PR(s).", file=sys.stderr)
    return prs


def format_author(author):
    """Match the CHANGELOG convention (`gh` returns bots as `app/dependabot`,
    but the changelog uses `dependabot[bot]`)."""
    login = (author or {}).get("login") or "ghost"
    if login.startswith("app/"):
        return f"{login[len('app/'):]}[bot]"
    return login


def format_pr_list(prs):
    lines = []
    for pr in prs:
        author = format_author(pr.get("author"))
        title = (pr.get("title") or "").strip()
        lines.append(f"* {title} ({REPO_URL}/pull/{pr.get('number')}, "
                     f"by @{author})")
    return "\n".join(lines)


# --------------------------------------------------------------------------- #
# Changelog generation
# --------------------------------------------------------------------------- #
def extract_style_reference(changelog, max_chars=6000):
    matches = list(re.finditer(r"^## v\S+.*$", changelog, flags=re.MULTILINE))
    for i, m in enumerate(matches):
        if "(future)" in m.group(0):
            continue
        start = m.start()
        end = matches[i + 1].start() if i + 1 < len(matches) else len(changelog)
        section = changelog[start:end].strip()
        section = re.split(r"^### Full list of pull requests", section,
                           flags=re.MULTILINE)[0].strip()
        return section[:max_chars]
    return ""


def build_prompt(prs, style_ref, version):
    pr_bullets = "\n".join(
        f"- #{p.get('number')}: {(p.get('title') or '').strip()}" for p in prs)
    return f"""You are writing the changelog prose for the Invidious release v{version}.

Below is an example of a PAST release entry from CHANGELOG.md. Match its \
Markdown structure, tone and heading style EXACTLY (a "### Wrap-up" narrative \
of 2-3 short paragraphs, then "### New features & important changes" with \
"#### For Users", "#### For instance owners" and "#### For developers" \
sub-sections, then a "### Bugs fixed" section when relevant).

--- STYLE EXAMPLE START ---
{style_ref}
--- STYLE EXAMPLE END ---

Here is the complete list of pull requests merged since the last release, \
which you must summarise (reference PRs inline as "(#1234)"). Sort content by \
user impact, with new features and breaking changes first:

{pr_bullets}

Rules:
- Output ONLY the Markdown body starting at "### Wrap-up". Do NOT include the \
"## v{version}" header.
- Do NOT include the exhaustive "### Full list of pull requests merged since \
the last release" section; it is appended separately.
- Do NOT invent PRs or changes that are not in the list above.
- Keep it concise and factual."""


def call_llm(base_url, api_key, model, prompt):
    url = base_url.rstrip("/") + "/chat/completions"
    payload = {
        "model": model,
        "messages": [
            {"role": "system", "content": "You are a meticulous release "
             "manager for the Invidious project who writes clear, accurate "
             "changelogs in Markdown."},
            {"role": "user", "content": prompt},
        ],
        "temperature": 0.3,
    }
    req = urllib.request.Request(
        url, data=json.dumps(payload).encode("utf-8"),
        headers={
            "Authorization": f"Bearer {api_key}",
            "Content-Type": "application/json",
            # OpenRouter-specific attribution headers; ignored by other APIs.
            "HTTP-Referer": REPO_URL,
            "X-Title": "Invidious release changelog",
        }, method="POST")
    try:
        with urllib.request.urlopen(req, timeout=180) as resp:
            body = json.loads(resp.read().decode("utf-8"))
    except urllib.error.HTTPError as exc:
        die(f"LLM API HTTP {exc.code}: {exc.read().decode('utf-8', 'replace')}")
    except urllib.error.URLError as exc:
        die(f"LLM API request failed: {exc}")
    try:
        content = body["choices"][0]["message"]["content"].strip()
    except (KeyError, IndexError):
        die(f"unexpected LLM API response: {json.dumps(body)[:1000]}")
    if not content:
        die("LLM API returned empty content")
    return content


def generate_prose(prs, version, base_url, api_key, model):
    if not prs:
        return ("### Wrap-up\n\nNo pull requests were merged since the "
                "previous release.")
    with open(CHANGELOG_FILE, encoding="utf-8") as fh:
        changelog = fh.read()
    prompt = build_prompt(prs, extract_style_reference(changelog), version)
    return call_llm(base_url, api_key, model, prompt)


# --------------------------------------------------------------------------- #
# File edits
# --------------------------------------------------------------------------- #
def update_changelog(version, entry_body):
    with open(CHANGELOG_FILE, encoding="utf-8") as fh:
        content = fh.read()
    section = f"## v{version}\n\n{entry_body}\n"
    future = re.search(r"^## v\S*\s*\(future\)\s*$", content, flags=re.MULTILINE)
    if future:
        at = future.end()
        rest = content[at:].lstrip("\n")
        content = content[:at] + "\n\n" + section + "\n" + rest
    else:
        title = re.search(r"^# CHANGELOG\s*$", content, flags=re.MULTILINE)
        at = title.end() if title else 0
        content = (content[:at] + "\n\n" + section + "\n"
                   + content[at:].lstrip("\n"))
    with open(CHANGELOG_FILE, "w", encoding="utf-8") as fh:
        fh.write(content)


def update_shard(version):
    with open(SHARD_FILE, encoding="utf-8") as fh:
        content = fh.read()
    content, n = re.subn(r"^version:\s*.*$", f"version: {version}", content,
                         count=1, flags=re.MULTILINE)
    if n != 1:
        die(f"could not update 'version:' in {SHARD_FILE}")
    with open(SHARD_FILE, "w", encoding="utf-8") as fh:
        fh.write(content)


# --------------------------------------------------------------------------- #
# Git / GitHub
# --------------------------------------------------------------------------- #
def detect_repo(explicit):
    if explicit:
        return explicit
    env = os.environ.get("GITHUB_REPOSITORY")
    if env:
        return env
    url = run(["git", "remote", "get-url", "origin"], capture=True, check=False)
    m = re.search(r"github\.com[:/](.+?)(?:\.git)?$", url)
    return m.group(1) if m else "iv-org/invidious"


def create_pull_request(repo, tag, branch, model, summary):
    if run(["git", "config", "user.name"], capture=True, check=False) == "":
        run(["git", "config", "user.name", "github-actions[bot]"])
        run(["git", "config", "user.email",
             "41898282+github-actions[bot]@users.noreply.github.com"])
    run(["git", "checkout", "-b", branch])
    run(["git", "add", CHANGELOG_FILE, SHARD_FILE])
    run(["git", "commit", "-m", f"Release {tag}"])
    run(["git", "push", "-u", "origin", branch])

    body = f"""Automated release preparation for **{tag}**.

Bumps `{SHARD_FILE}` and updates `{CHANGELOG_FILE}` (prose generated by the \
`{model}` model; the PR list is generated deterministically so every merged \
PR is included).

## Before merging
- [ ] Review the generated changelog for accuracy
- [ ] Confirm the version number and major bump (if any)

## After merging (maintainer, per the release docs)
- Do **not** squash-merge; keep the release commit distinct.
- Tag the merge commit: `git tag -as {tag}` using the changelog summary as \
the annotation, then `git push origin {tag}`.
- Run the "Prepare for next release" step (append `-dev` in {SHARD_FILE}, add \
a new `(future)` changelog header).
- Create the GitHub release from the tag with the changelog summary + PR list.

Refer to the documentation for more details:
https://github.com/iv-org/documentation/blob/master/docs/create-new-invidious-release.md

### Proposed tag annotation (changelog summary)
```markdown
{summary}
```
"""
    run(["gh", "pr", "create", "--repo", repo, "--base", "master",
         "--head", branch, "--title", f"Release {tag}", "--body", body])


# --------------------------------------------------------------------------- #
def main():
    parser = argparse.ArgumentParser(description="Prepare an Invidious release.")
    parser.add_argument("--major", type=int, default=None,
                        help="major version (default: keep current)")
    parser.add_argument("--patch", type=int, default=0,
                        help="patch number (default: 0)")
    parser.add_argument("--model", default=os.environ.get(
        "LLM_MODEL", DEFAULT_MODEL),
        help=f"model name (default: {DEFAULT_MODEL})")
    parser.add_argument("--base-url", default=os.environ.get(
        "OPENAI_BASE_URL", DEFAULT_BASE_URL),
        help=f"OpenAI-compatible API base URL (default: {DEFAULT_BASE_URL})")
    parser.add_argument("--repo", default=None,
                        help="owner/repo (default: autodetect)")
    parser.add_argument("--pr-json", default=None,
                        help="read PR list from a JSON file instead of gh")
    parser.add_argument("--dry-run", action="store_true",
                        help="edit files only; no branch/commit/push/PR")
    args = parser.parse_args()

    api_key = os.environ.get("OPENAI_API_KEY")
    if not api_key:
        die("OPENAI_API_KEY is not set")

    repo = detect_repo(args.repo)
    version, tag = compute_version(args)
    branch = f"release-{tag}"
    prev = previous_tag()
    print(f"Preparing {tag} (previous: {prev or '<none>'}) on {repo}",
          file=sys.stderr)

    prs = collect_prs(repo, prev, args.pr_json)
    prose = generate_prose(prs, version, args.base_url, api_key, args.model)

    entry_body = prose
    pr_list = format_pr_list(prs)
    if pr_list:
        entry_body += ("\n\n### Full list of pull requests merged since the "
                       "last release (newest first)\n\n" + pr_list)

    update_changelog(version, entry_body)
    update_shard(version)
    print(f"Updated {CHANGELOG_FILE} and {SHARD_FILE} for {tag}.",
          file=sys.stderr)

    if args.dry_run:
        print(f"\n--- Dry run complete. Review the diff, then:\n"
              f"  git checkout -b {branch}\n"
              f"  git add {CHANGELOG_FILE} {SHARD_FILE}\n"
              f"  git commit -m 'Release {tag}'\n"
              f"  git push -u origin {branch}\n"
              f"  gh pr create --base master --head {branch} "
              f"--title 'Release {tag}'", file=sys.stderr)
        return

    create_pull_request(repo, tag, branch, args.model, prose)
    print(f"\nOpened release PR for {tag}.", file=sys.stderr)


if __name__ == "__main__":
    main()