#!/bin/bash
# pr-metadata.sh - Get PR metadata as JSON on GitHub or Gitea
# Usage: pr-metadata.sh -n <pr_number> [-o <output_file>]

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# shellcheck disable=SC1091
source "$SCRIPT_DIR/detect-platform.sh"

# Parse arguments
PR_NUMBER=""
OUTPUT_FILE=""

while [[ $# -gt 0 ]]; do
    case $1 in
        -n|--number)
            PR_NUMBER="$2"
            shift 2
            ;;
        -o|--output)
            OUTPUT_FILE="$2"
            shift 2
            ;;
        -h|--help)
            echo "Usage: pr-metadata.sh -n <pr_number> [-o <output_file>]"
            echo ""
            echo "Options:"
            echo "  -n, --number   PR number (required)"
            echo "  -o, --output   Output file (optional, prints to stdout if omitted)"
            echo "  -h, --help     Show this help"
            exit 0
            ;;
        *)
            echo "Unknown option: $1"
            exit 1
            ;;
    esac
done

if [[ -z "$PR_NUMBER" ]]; then
    echo "Error: PR number is required (-n)" >&2
    exit 1
fi

detect_platform > /dev/null

if [[ "$PLATFORM" == "github" ]]; then
    METADATA=$(gh pr view "$PR_NUMBER" --json number,title,body,state,author,headRefName,baseRefName,files,labels,assignees,milestone,createdAt,updatedAt,url,isDraft)

    if [[ -n "$OUTPUT_FILE" ]]; then
        echo "$METADATA" > "$OUTPUT_FILE"
    else
        echo "$METADATA"
    fi
elif [[ "$PLATFORM" == "gitea" ]]; then
    OWNER=$(get_repo_owner)
    REPO=$(get_repo_name)
    HOST=$(get_remote_host) || {
        echo "Error: Cannot determine host from remote URL" >&2
        exit 1
    }

    API_URL="https://${HOST}/api/v1/repos/${OWNER}/${REPO}/pulls/${PR_NUMBER}"

    GITEA_API_TOKEN=$(get_gitea_token "$HOST" || true)

    RESPONSE_FILE=$(mktemp)
    trap 'rm -f "$RESPONSE_FILE"' EXIT
    if [[ -n "$GITEA_API_TOKEN" ]]; then
        HTTP_CODE=$(curl -sS -H "Authorization: token $GITEA_API_TOKEN" -o "$RESPONSE_FILE" -w '%{http_code}' "$API_URL")
    else
        HTTP_CODE=$(curl -sS -o "$RESPONSE_FILE" -w '%{http_code}' "$API_URL")
    fi
    RAW=$(cat "$RESPONSE_FILE")
    rm -f "$RESPONSE_FILE"
    trap - EXIT

    if [[ ! "$HTTP_CODE" =~ ^2 ]]; then
        echo "Error: Gitea PR metadata request failed for ${OWNER}/${REPO}#${PR_NUMBER} (HTTP ${HTTP_CODE})" >&2
        exit 1
    fi

    # Normalize Gitea response to match our expected schema
    METADATA=$(echo "$RAW" | python3 -c "
import json, sys
data = json.load(sys.stdin)
if 'message' in data and not data.get('number'):
    raise SystemExit('Error: Gitea PR metadata response did not contain PR data')
head = data.get('head') or {}
head_ref = head.get('ref') or ''
head_label = head.get('label') or ''
# Gitea can report closed/merged PR heads as refs/pull/<n>/head; callers need
# the source branch name equivalent to GitHub headRefName, so prefer label then.
if head_ref.startswith('refs/pull/') and head_label:
    head_ref = head_label
normalized = {
    'number': data.get('number'),
    'title': data.get('title'),
    'body': data.get('body', ''),
    'state': data.get('state'),
    'author': data.get('user', {}).get('login', ''),
    'headRefName': head_ref,
    'baseRefName': data.get('base', {}).get('ref', ''),
    'labels': [l.get('name', '') for l in data.get('labels', [])],
    'assignees': [a.get('login', '') for a in data.get('assignees', [])],
    'milestone': data.get('milestone', {}).get('title', '') if data.get('milestone') else '',
    'createdAt': data.get('created_at', ''),
    'updatedAt': data.get('updated_at', ''),
    'url': data.get('html_url', ''),
    'isDraft': data.get('draft', False),
    'mergeable': data.get('mergeable'),
    'diffUrl': data.get('diff_url', ''),
}
json.dump(normalized, sys.stdout, indent=2)
")

    if [[ -n "$OUTPUT_FILE" ]]; then
        echo "$METADATA" > "$OUTPUT_FILE"
    else
        echo "$METADATA"
    fi
else
    echo "Error: Unknown platform" >&2
    exit 1
fi