fix: retry and throttle GitHub API requests

package.json

@@ -23,6 +23,8 @@
   "dependencies": {
     "@actions/core": "^1.10.1",
     "@actions/github": "^6.0.0",
+    "@octokit/plugin-throttling": "^8.1.2",
+    "@octokit/plugin-retry": "^6.0.1",
     "joi": "^17.11.0",
     "js-yaml": "^4.1.0"
   },

src/index.js

@@ -2,6 +2,7 @@ import core from '@actions/core';
 import github from '@actions/github';
 import yaml from 'js-yaml';
 
+import {getClient} from './utils.js';
 import {configSchema, actionSchema} from './schema.js';
 import {
   addDiscussionCommentQuery,
@@ -19,7 +20,7 @@ import {
 async function run() {
   try {
     const config = getConfig();
-    const client = github.getOctokit(config['github-token']);
+    const client = getClient(config['github-token']);
 
     const actions = await getActionConfig(client, config['config-path']);
 

src/utils.js

@@ -0,0 +1,37 @@
+import core from '@actions/core';
+import github from '@actions/github';
+import {retry} from '@octokit/plugin-retry';
+import {throttling} from '@octokit/plugin-throttling';
+
+function getClient(token) {
+  const requestRetries = 3;
+
+  const rateLimitCallback = function (
+    retryAfter,
+    options,
+    octokit,
+    retryCount
+  ) {
+    core.info(
+      `Request quota exhausted for request ${options.method} ${options.url}`
+    );
+
+    if (retryCount < requestRetries) {
+      core.info(`Retrying after ${retryAfter} seconds`);
+
+      return true;
+    }
+  };
+
+  const options = {
+    request: {retries: requestRetries},
+    throttle: {
+      onSecondaryRateLimit: rateLimitCallback,
+      onRateLimit: rateLimitCallback
+    }
+  };
+
+  return github.getOctokit(token, options, retry, throttling);
+}
+
+export {getClient};