[DownstreamMerge] Bug 2026461: 4-4-22 merge

.github/actions/retest-action/entrypoint.sh

@@ -2,16 +2,61 @@
 
 set -ex
 
+##############################
+# Prerequisites check
+##############################
+
 if ! jq -e '.issue.pull_request' ${GITHUB_EVENT_PATH}; then
     echo "Not a PR... Exiting."
     exit 0
 fi
 
-if [ "$(jq -r '.comment.body' ${GITHUB_EVENT_PATH})" != "/retest" ]; then
-    echo "Nothing to do... Exiting."
+COMMENT_BODY=$(jq -r '.comment.body' ${GITHUB_EVENT_PATH})
+if [ "${COMMENT_BODY}" != "/retest" ] &&
+  [ "${COMMENT_BODY}" != "/retest-failed" ] &&
+  [ "${COMMENT_BODY}" != "/cancel" ] &&
+  [ "${COMMENT_BODY}" != "/help" ]; then
+    echo "Unknown action. Nothing to do... Exiting."
     exit 0
 fi
 
+##############################
+# functions section
+##############################
+
+send_reaction() {
+  local REACTION_SYMBOL="$1"
+  local REACTION_URL="$(jq -r '.comment.url' ${GITHUB_EVENT_PATH})/reactions"
+  curl --request POST \
+    --url "${REACTION_URL}" \
+    --header "authorization: Bearer ${GITHUB_TOKEN}" \
+    --header "accept: application/vnd.github.squirrel-girl-preview+json" \
+    --header "content-type: application/json" \
+    --data '{ "content" : "'${REACTION_SYMBOL}'" }'
+}
+
+send_comment() {
+  local COMMENT="$1"
+  local COMMENTS_URL=$(jq -r '.issue.comments_url' ${GITHUB_EVENT_PATH})
+  curl --request POST \
+      --url "${COMMENTS_URL}" \
+      --header "authorization: Bearer ${GITHUB_TOKEN}" \
+      --header "accept: application/vnd.github.squirrel-girl-preview+json" \
+      --header "content-type: application/json" \
+      --data '{ "body" : "'"${COMMENT}"'" }'
+}
+
+##############################
+# logic section
+##############################
+
+ACTION="${COMMENT_BODY}"
+
+if [ "$ACTION" == "/help" ]; then
+	send_comment "Supported operations are /retest, /retest-failed, /cancel"
+	exit 0
+fi
+
 PR_URL=$(jq -r '.issue.pull_request.url' ${GITHUB_EVENT_PATH})
 
 curl --request GET \
@@ -27,19 +72,38 @@ curl --request GET \
     --header "authorization: Bearer ${GITHUB_TOKEN}" \
     --header "content-type: application/json" | jq '.workflow_runs | max_by(.run_number)' > run.json
 
-RERUN_URL=$(jq -r '.rerun_url' run.json)
+ACTION_URL=""
+if [ "$ACTION" == "/retest" ]; then
+  ACTION_URL=$(jq -r '.rerun_url' run.json)
+elif [ "$ACTION" == "/retest-failed" ]; then
+  # New feature, rerun failed jobs:
+  # https://docs.github.com/en/rest/reference/actions#re-run-failed-jobs-from-a-workflow-run
+  RERUN_URL=$(jq -r '.rerun_url' run.json)
+  ACTION_URL=${RERUN_URL}-failed-jobs
+elif [ "$ACTION" == "/cancel" ]; then
+  ACTION_URL=$(jq -r '.cancel_url' run.json)
+else
+  echo "Something went wrong, unsupported action"
+  exit 0
+fi
 
-curl --request POST \
-    --url "${RERUN_URL}" \
+# Execute the action.
+# Store the response code in a variable.
+# Store the answer in file .action-response.json.
+RESPONSE_CODE=$(curl --write-out '%{http_code}' --silent --output .action-response.json --request POST \
+    --url "${ACTION_URL}" \
     --header "authorization: Bearer ${GITHUB_TOKEN}" \
-    --header "content-type: application/json"
+    --header "content-type: application/json")
 
 
-REACTION_URL="$(jq -r '.comment.url' ${GITHUB_EVENT_PATH})/reactions"
+REACTION_SYMBOL="rocket"
+if ! echo ${RESPONSE_CODE} | egrep -q '^2'; then
+  REACTION_SYMBOL="confused"
+fi
+send_reaction "${REACTION_SYMBOL}"
 
-curl --request POST \
-    --url "${REACTION_URL}" \
-    --header "authorization: Bearer ${GITHUB_TOKEN}" \
-    --header "accept: application/vnd.github.squirrel-girl-preview+json" \
-    --header "content-type: application/json" \
-    --data '{ "content" : "rocket" }'
+# In case we received a non 2xx response code, relay the error message as a comment.
+if ! echo ${RESPONSE_CODE} | egrep -q '^2'; then
+  RESPONSE_MESSAGE=$(jq -r '.message' .action-response.json)
+  send_comment "Oops, something went wrong:\n~~~\n${RESPONSE_MESSAGE}\n~~~\n"
+fi