#!/usr/bin/env bash

set -e

if [ $# -lt 1 ]; then
    echo "Usage: get-pr-comments PR_NUMBER [OWNER/REPO]"
    echo "Example: get-pr-comments 123"
    echo "Example: get-pr-comments 123 EveryInc/cora"
    exit 1
fi

PR_NUMBER=$1

if [ -n "$2" ]; then
    OWNER=$(echo "$2" | cut -d/ -f1)
    REPO=$(echo "$2" | cut -d/ -f2)
else
    OWNER=$(gh repo view --json owner -q .owner.login 2>/dev/null)
    REPO=$(gh repo view --json name -q .name 2>/dev/null)
fi

if [ -z "$OWNER" ] || [ -z "$REPO" ]; then
    echo "Error: Could not detect repository. Pass OWNER/REPO as second argument."
    exit 1
fi

# Fetch review threads, regular PR comments, and review bodies in one query.
# Output is a JSON object with three keys:
#   review_threads - unresolved, non-outdated inline code review threads
#   pr_comments    - top-level PR conversation comments (excludes PR author)
#   review_bodies  - review submissions with non-empty body text (excludes PR author)
gh api graphql -f owner="$OWNER" -f repo="$REPO" -F pr="$PR_NUMBER" -f query='
query FetchPRFeedback($owner: String!, $repo: String!, $pr: Int!) {
  repository(owner: $owner, name: $repo) {
    pullRequest(number: $pr) {
      author { login }
      reviewThreads(first: 100) {
        edges {
          node {
            id
            isResolved
            isOutdated
            path
            line
            comments(first: 50) {
              nodes {
                id
                author { login }
                body
                createdAt
                url
              }
            }
          }
        }
      }
      comments(first: 100) {
        nodes {
          id
          author { login }
          body
          createdAt
          url
        }
      }
      reviews(first: 50) {
        nodes {
          id
          author { login }
          body
          state
          createdAt
          url
        }
      }
    }
  }
}' | jq '.data.repository.pullRequest as $pr | {
  review_threads: [$pr.reviewThreads.edges[]
    | select(.node.isResolved == false and .node.isOutdated == false)],
  pr_comments: [$pr.comments.nodes[]
    | select(.author.login != $pr.author.login)
    | select(.body | test("^\\s*$") | not)],
  review_bodies: [$pr.reviews.nodes[]
    | select(.body != null and .body != "")
    | select(.author.login != $pr.author.login)]
}'
