saner

Author: Azathothas

Github/Runners/manage_linux.sh

@@ -133,6 +133,34 @@ generate_token() {
 export -f generate_token
 #------------------------------------------------------------------------------------#
 
+#------------------------------------------------------------------------------------#
+##Remove Runner Upon Completion
+remove_runner() {
+  generate_token
+  echo -e "\n[+] Removing Runners ...\n"
+  #Remove Offline Runners
+  if [ -n "${GITHUB_REPOSITORY}" ]; then
+     OFFLINE_RUNNERS="$(curl -qfsSL "https://api.github.com/repos/${GITHUB_OWNER}/${GITHUB_REPOSITORY}/actions/runners?per_page=100" -H "Authorization: Bearer ${GITHUB_PERSONAL_TOKEN}" -H 'Accept: application/vnd.github+json' | jq -r '.runners[] | select(.status == "offline") | .id')" && export OFFLINE_RUNNERS="${OFFLINE_RUNNERS}"
+     readarray -t OFFLINE_RUNNERS_ID <<< "${OFFLINE_RUNNERS}"
+     for R_ID in "${OFFLINE_RUNNERS_ID[@]}"; do
+         curl -qfsSL -X 'DELETE' "https://api.github.com/repos/${GITHUB_OWNER}/${GITHUB_REPOSITORY}/actions/runners/${R_ID}" -H "Authorization: Bearer ${GITHUB_PERSONAL_TOKEN}" -H 'Accept: application/vnd.github+json'
+     done
+  elif [ -n "${GITHUB_ORG}" ]; then
+     OFFLINE_RUNNERS="$(curl -qfsSL "https://api.github.com/orgs/${GITHUB_OWNER}/actions/runners?per_page=100" -H "Authorization: Bearer ${GITHUB_PERSONAL_TOKEN}" -H 'Accept: application/vnd.github+json' | jq -r '.runners[] | select(.status == "offline") | .id')" && export OFFLINE_RUNNERS="${OFFLINE_RUNNERS}"
+     readarray -t OFFLINE_RUNNERS_ID <<< "${OFFLINE_RUNNERS}"
+     for R_ID in "${OFFLINE_RUNNERS_ID[@]}"; do
+         curl -qfsSL -X 'DELETE' "https://api.github.com/orgs/${GITHUB_OWNER}/actions/runners/${R_ID}" -H "Authorization: Bearer ${GITHUB_PERSONAL_TOKEN}" -H 'Accept: application/vnd.github+json'
+     done
+  fi
+  #Remove Self  
+  "/runner-init/config.sh" remove --unattended --token "${RUNNER_TOKEN}"
+  #Cleanup
+  unset API_RESPONSE AUTH_URL OFFLINE_RUNNERS_ID REG_URL RUNNERS_ID R_ID RUNNER_ID RUNNER_LABELS RUNNER_TOKEN
+  kill -9 $$
+}
+export -f remove_runner
+#------------------------------------------------------------------------------------#
+
 #------------------------------------------------------------------------------------#
 ##Register Runner
 cd "/runner-init"
@@ -145,6 +173,7 @@ if [ -n "${RUNNER_TOKEN}" ]; then
    echo "[+] (LABELS: ${RUNNER_LABELS})"
    echo "[+] (TOKEN: ${RUNNER_TOKEN})"
    echo "[+] (URL: ${REG_URL})"
+ ##Run  
    "/runner-init/config.sh" \
      --name "${RUNNER_ID}" \
      --labels "${RUNNER_LABELS}" \
@@ -153,6 +182,13 @@ if [ -n "${RUNNER_TOKEN}" ]; then
      --unattended \
      --replace \
      --ephemeral
+ ##Write Status
+   [[ -f "/tmp/GHA_CI_STATUS" && -w "/tmp/GHA_CI_STATUS" ]] && echo "" > "/tmp/GHA_CI_STATUS"
+   if [[ -d "/tmp" && -w "/tmp" ]]; then
+     echo "EXITED" | tee "/tmp/GHA_CI_STATUS"
+   fi
+ ##Remove  
+   remove_runner
 else
    echo "[-] Failed to Generate Token..."
    echo -e "\n[+] GITHUB_PERSONAL_TOKEN: ${GITHUB_PERSONAL_TOKEN}\n"
@@ -162,33 +198,6 @@ fi
 #------------------------------------------------------------------------------------#
 
 #------------------------------------------------------------------------------------#
-##Remove Runner Upon Completion
-remove_runner() {
-  generate_token
-  echo -e "\n[+] Removing Runners ...\n"
-  #Remove Offline Runners
-  if [ -n "${GITHUB_REPOSITORY}" ]; then
-     OFFLINE_RUNNERS="$(curl -qfsSL "https://api.github.com/repos/${GITHUB_OWNER}/${GITHUB_REPOSITORY}/actions/runners?per_page=100" -H "Authorization: Bearer ${GITHUB_PERSONAL_TOKEN}" -H 'Accept: application/vnd.github+json' | jq -r '.runners[] | select(.status == "offline") | .id')" && export OFFLINE_RUNNERS="${OFFLINE_RUNNERS}"
-     readarray -t OFFLINE_RUNNERS_ID <<< "${OFFLINE_RUNNERS}"
-     for R_ID in "${OFFLINE_RUNNERS_ID[@]}"; do
-         curl -qfsSL -X 'DELETE' "https://api.github.com/repos/${GITHUB_OWNER}/${GITHUB_REPOSITORY}/actions/runners/${R_ID}" -H "Authorization: Bearer ${GITHUB_PERSONAL_TOKEN}" -H 'Accept: application/vnd.github+json'
-     done
-  elif [ -n "${GITHUB_ORG}" ]; then
-     OFFLINE_RUNNERS="$(curl -qfsSL "https://api.github.com/orgs/${GITHUB_OWNER}/actions/runners?per_page=100" -H "Authorization: Bearer ${GITHUB_PERSONAL_TOKEN}" -H 'Accept: application/vnd.github+json' | jq -r '.runners[] | select(.status == "offline") | .id')" && export OFFLINE_RUNNERS="${OFFLINE_RUNNERS}"
-     readarray -t OFFLINE_RUNNERS_ID <<< "${OFFLINE_RUNNERS}"
-     for R_ID in "${OFFLINE_RUNNERS_ID[@]}"; do
-         curl -qfsSL -X 'DELETE' "https://api.github.com/orgs/${GITHUB_OWNER}/actions/runners/${R_ID}" -H "Authorization: Bearer ${GITHUB_PERSONAL_TOKEN}" -H 'Accept: application/vnd.github+json'
-     done
-  fi
-  #Remove Self  
-  "/runner-init/config.sh" remove --unattended --token "${RUNNER_TOKEN}"
-  #Write Status
-  [[ -d "/tmp" && -w "/tmp" ]] && echo "EXITED" | tee "/tmp/GHA_CI_STATUS"
-  #Cleanup
-  unset API_RESPONSE AUTH_URL OFFLINE_RUNNERS_ID REG_URL RUNNERS_ID R_ID RUNNER_ID RUNNER_LABELS RUNNER_TOKEN
-  kill -9 $$
-}
-export -f remove_runner
 #exit if ctrl + c
 trap 'remove_runner; exit 130' SIGINT
 #exit if kill|pkill -9