Introduce queries helper

Author: khiga8

action.yml

@@ -9,6 +9,7 @@ runs:
     - name: Runs alt text check and adds comment
       run: |
           source ${{ github.action_path }}/flag-alt-text.sh
+          source ${{ github.action_path }}/queries.sh
 
           if [ ${{ github.event.comment }} ]; then
             content=$COMMENT
@@ -19,7 +20,12 @@ runs:
             elif ${{ github.event.discussion.id != '' }}; then
               type=discussion_comment
               discussion_node_id='${{ github.event.discussion.node_id }}'
-              reply_to_id='${{ github.event.comment.node_id }}'
+              comment_node_id='${{ github.event.comment.node_id }}'
+              if ${{ github.event.comment.parent_id != '' }}; then
+                reply_to_id=$(getDiscussionReplyToId $comment_node_id)
+              else
+                reply_to_id=$comment_node_id
+              fi
             else
               type=issue_comment
               issue_url=${{ github.event.issue.html_url }}
@@ -62,25 +68,9 @@ runs:
             elif [[ $type = issue_comment ]] || [[ $type = issue_description ]]; then
               gh issue comment $issue_url --body "$message"
             elif [[ $type = discussion_description ]]; then
-              gh api graphql -F discussionId="$discussion_node_id" -F body="$message" -f query='
-                mutation($discussionId: ID!, $body: String!) {
-                  addDiscussionComment(input: {discussionId: $discussionId, body: $body}) {
-                    comment {
-                      id
-                    }
-                  }
-                }
-              '
+              addDiscussionComment $discussion_node_id "$message"
             elif [[ $type = discussion_comment ]]; then
-              gh api graphql -F discussionId="$discussion_node_id" -F replyToId="$reply_to_id" -F body="$message" -f query='
-                mutation($discussionId: ID!, , $replyToId: ID, $body: String!) {
-                  addDiscussionComment(input: {discussionId: $discussionId, replyToId: $replyToId, body: $body}) {
-                    comment {
-                      id
-                    }
-                  }
-                }
-              '
+              addDiscussionCommentAsReply $discussion_node_id $reply_to_id "$message"
             fi
           fi
       shell: bash

queries.sh

@@ -0,0 +1,48 @@
+#!/bin/bash
+
+# Given a node_id for a Discussion Comment with a parent comment, return the parent comment's node ID.
+function getDiscussionReplyToId() {
+  local NODE_ID=$1
+  local REPLY_TO_DATA=$(gh api graphql -f query='
+  query($nodeId: ID!) {
+    node(id: $nodeId) {
+      ... on DiscussionComment {
+        replyTo {
+          id
+        }
+      }
+    }
+  }' -F nodeId=$NODE_ID)
+  echo $REPLY_TO_DATA | jq -r '.data.node.replyTo.id'
+}
+
+# Adds a top-level discussion comment given a discussion node ID and a message.
+function addDiscussionComment() {
+  local DISCUSSION_NODE_ID=$1
+  local MESSAGE=$2
+  gh api graphql -F discussionId="$discussion_node_id" -F body="$message" -f query='
+    mutation($discussionId: ID!, $body: String!) {
+      addDiscussionComment(input: {discussionId: $discussionId, body: $body}) {
+        comment {
+          id
+        }
+      }
+    }
+  '
+}
+
+# Adds a Discussion Comment as a reply to an existing discussion comment given a discussion node ID, discussion comment node ID, and a message.
+function addDiscussionCommentAsReply() {
+  local DISCUSSION_NODE_ID=$1
+  local REPLY_TO_ID=$2
+  local MESSAGE=$3
+  gh api graphql -F discussionId="$DISCUSSION_NODE_ID" -F replyToId="$REPLY_TO_ID" -F body="$MESSAGE" -f query='
+    mutation($discussionId: ID!, , $replyToId: ID, $body: String!) {
+        addDiscussionComment(input: {discussionId: $discussionId, replyToId: $replyToId, body: $body}) {
+        comment {
+            id
+        }
+      }
+    }
+  '
+}