gh auth status || { echo "Error: Run 'gh auth login' first"; exit 1; }

# Validate inputs before shell use
validate_branch() {
    [[ "$1" =~ ^[a-zA-Z0-9/_-]+$ ]] || { echo "Invalid branch name: $1"; exit 1; }
}

validate_number() {
    [[ "$1" =~ ^[0-9]+$ ]] || { echo "Invalid number: $1"; exit 1; }
}

validate_username() {
    [[ "$1" =~ ^[a-zA-Z0-9_-]+$ ]] || { echo "Invalid username: $1"; exit 1; }
}

# Create label if it doesn't exist
ensure_label() {
    local label="$1"
    local color="${2:-ededed}"
    gh label list --search "$label" --json name --jq '.[].name' 2>/dev/null | grep -qx "$label" || \
        gh label create "$label" --color "$color" 2>/dev/null || true
}

# Ensure required label exists
ensure_label "claude-code" "7C3AED"

BRANCH=$(git branch --show-current)
validate_branch "$BRANCH"

# Get PR info in single API call
PR_INFO=$(gh pr view "$BRANCH" --json number,title,state,reviewDecision,reviews,body 2>/dev/null) || {
    echo "No PR found for branch: $BRANCH"
    echo "Create a PR first with /gw-work"
    exit 1
}

PR_NUM=$(echo "$PR_INFO" | jq -r '.number')
validate_number "$PR_NUM"

PR_TITLE=$(echo "$PR_INFO" | jq -r '.title')
REVIEW_DECISION=$(echo "$PR_INFO" | jq -r '.reviewDecision // "NONE"')

echo "PR #${PR_NUM}: $PR_TITLE"
echo "Review status: $REVIEW_DECISION"

case "$REVIEW_DECISION" in
    "APPROVED")
        echo ""
        echo "PR approved! Ready to merge."
        # Continue to Step 3a: Handle Approved
        ;;
    "CHANGES_REQUESTED")
        echo ""
        echo "Changes requested. Let's address the feedback."
        # Continue to Step 3b: Handle Changes Requested
        ;;
    "REVIEW_REQUIRED"|"NONE"|"")
        echo ""
        echo "Awaiting review. No action needed yet."
        echo ""
        echo "Options:"
        echo "  - Wait for reviewer feedback"
        echo "  - Request review: gh pr edit $PR_NUM --add-reviewer <username>"
        exit 0
        ;;
    *)
        echo "Review state: $REVIEW_DECISION"
        ;;
esac

# Confirm merge
echo "Ready to merge PR #${PR_NUM}"
echo "Merge method: squash (combines all commits)"
echo ""

# Merge PR
gh pr merge "$PR_NUM" --squash --delete-branch || {
    echo "Error: Merge failed. Check for conflicts or CI status."
    exit 1
}

echo "PR merged successfully!"

# Get issue from PR body (looks for "Closes #N")
PR_BODY=$(echo "$PR_INFO" | jq -r '.body')
ISSUE=$(echo "$PR_BODY" | grep -oE 'Closes #[0-9]+' | grep -oE '[0-9]+' | head -1)

if [ -n "$ISSUE" ]; then
    validate_number "$ISSUE"
    echo "Linked issue: #$ISSUE"
fi

echo ""
echo "Any learnings to capture from this work? (y/n)"
# If user says yes:

# Generate solution doc path
SLUG=$(echo "$PR_TITLE" | sed 's/^feat: //' | tr '[:upper:]' '[:lower:]' | sed 's/[^a-z0-9]/-/g' | sed 's/--*/-/g' | cut -c1-40)
DATE=$(date +%Y-%m-%d)
SOLUTION_PATH="docs/solutions/${SLUG}.md"

# {Title} Learnings

**PR:** #{PR_NUM} | **Issue:** #{ISSUE} | **Date:** {DATE}

## Problem

{What challenge did we face?}

## Solution

{How did we solve it?}

## Gotchas

- {Important thing we learned}
- {Edge case we discovered}

## References

- PR: https://github.com/{owner}/{repo}/pull/{PR_NUM}
- Issue: https://github.com/{owner}/{repo}/issues/{ISSUE}

# Link learning from issue comment
if [ -n "$ISSUE" ]; then
    gh issue comment "$ISSUE" --body "Learnings captured: \`$SOLUTION_PATH\`"
fi

if [ -n "$ISSUE" ]; then
    gh issue close "$ISSUE" || {
        echo "Warning: Could not close issue #$ISSUE"
    }
    echo "Closed issue #$ISSUE"
fi

# Get latest review comments
LATEST_REVIEW=$(echo "$PR_INFO" | jq -r '.reviews | map(select(.state == "CHANGES_REQUESTED")) | last')
REVIEW_BODY=$(echo "$LATEST_REVIEW" | jq -r '.body // "No comment body"')
REVIEWER=$(echo "$LATEST_REVIEW" | jq -r '.author.login')

# Validate reviewer username before using
validate_username "$REVIEWER"

echo "Feedback from @$REVIEWER:"
echo "---"
echo "$REVIEW_BODY"
echo "---"

# Also check for inline comments
COMMENTS=$(gh pr view "$PR_NUM" --json comments --jq '.comments[].body' 2>/dev/null)
if [ -n "$COMMENTS" ]; then
    echo ""
    echo "PR comments:"
    echo "$COMMENTS"
fi

git add <changed-files>
git commit -m "fix: address review feedback

- {Change 1}
- {Change 2}

Requested by @$REVIEWER in PR #${PR_NUM}"

git push

gh pr edit "$PR_NUM" --add-reviewer "$REVIEWER" || {
    echo "Warning: Could not request re-review from @$REVIEWER"
}

echo ""
echo "Changes pushed and re-review requested from @$REVIEWER"

PR #${PR_NUM} merged successfully!

Issue #${ISSUE} closed
Branch deleted
Learnings: {SOLUTION_PATH or "none captured"}

Workflow complete!

Changes pushed to PR #${PR_NUM}

Re-review requested from @{REVIEWER}

Next: Wait for review, then run /gw-review again