#!/usr/bin/env python3
# scripts/generate-pr.py — generate PR title and description from branch diff

import subprocess
import sys
import anthropic

def get_branch_info():
    branch = subprocess.check_output(['git', 'rev-parse', '--abbrev-ref', 'HEAD'],
                                     text=True).strip()
    base = 'main'

    # Get commits since branching from main
    commits = subprocess.check_output(
        ['git', 'log', f'{base}...HEAD', '--oneline'],
        text=True
    ).strip()

    # Get the diff
    diff_stat = subprocess.check_output(
        ['git', 'diff', f'{base}...HEAD', '--stat'],
        text=True
    ).strip()

    # Get full diff (truncated for large PRs)
    diff = subprocess.check_output(
        ['git', 'diff', f'{base}...HEAD', '--unified=2'],
        text=True
    )[:15000]  # Max 15K chars

    return {
        'branch': branch,
        'commits': commits,
        'diff_stat': diff_stat,
        'diff': diff
    }

def generate_pr_description(info: dict) -> str:
    client = anthropic.Anthropic()

    response = client.messages.create(
        model='claude-sonnet-4-5',
        max_tokens=1024,
        messages=[{
            'role': 'user',
            'content': f"""Generate a GitHub pull request title and description.

Branch: {info['branch']}

Commits:
{info['commits']}

Files changed:
{info['diff_stat']}

Code diff (sample):
{info['diff'][:8000]}

Output format:
TITLE: <concise title under 72 chars>

## What changed
<2-4 bullet points describing the changes>

## Why
<1-2 sentences on the motivation/problem solved>

## Testing
<how to verify this works>

## Notes for reviewer
<anything specific to check or context that helps review>"""
        }]
    )

    return response.content[0].text

info = get_branch_info()
description = generate_pr_description(info)
print(description)

# Optionally: create the PR with gh CLI
if '--create' in sys.argv:
    lines = description.strip().split('\n')
    title = lines[0].replace('TITLE: ', '').strip()
    body = '\n'.join(lines[2:])

    subprocess.run([
        'gh', 'pr', 'create',
        '--title', title,
        '--body', body
    ])

#!/usr/bin/env python3
# scripts/branch-from-issue.py <github-issue-url>

import sys
import subprocess
import anthropic
import re
from urllib.request import urlopen, Request
import json
import os

def get_issue_title(url: str) -> str:
    # Parse owner/repo/issue_number from URL
    match = re.match(r'https://github.com/([^/]+)/([^/]+)/issues/(\d+)', url)
    if not match:
        return None

    owner, repo, number = match.groups()

    headers = {}
    if token := os.getenv('GITHUB_TOKEN'):
        headers['Authorization'] = f'Bearer {token}'

    req = Request(
        f'https://api.github.com/repos/{owner}/{repo}/issues/{number}',
        headers=headers
    )
    data = json.loads(urlopen(req).read())
    return data.get('title', '')

def generate_branch_name(title: str, issue_number: str) -> str:
    client = anthropic.Anthropic()
    response = client.messages.create(
        model='claude-haiku-4-5',
        max_tokens=64,
        messages=[{
            'role': 'user',
            'content': f"""Convert this issue title to a git branch name.

Issue #{issue_number}: {title}

Rules:
- Format: feature/123-short-description OR fix/123-short-description OR chore/123-description
- Use 'fix' for bug reports, 'feature' for new functionality, 'chore' for maintenance
- Max 60 chars total
- Use hyphens, lowercase only, no special chars
- Keep only essential words (remove articles, prepositions)

Output only the branch name, nothing else."""
        }]
    )
    return response.content[0].text.strip()

if len(sys.argv) < 2:
    print("Usage: branch-from-issue.py <github-issue-url>")
    sys.exit(1)

url = sys.argv[1]
title = get_issue_title(url)
if not title:
    print("Could not fetch issue")
    sys.exit(1)

issue_number = url.split('/')[-1]
branch_name = generate_branch_name(title, issue_number)

print(f"Creating branch: {branch_name}")
subprocess.run(['git', 'checkout', '-b', branch_name])