Enable GitHub API request throttling 🏁

package.json

@@ -29,6 +29,7 @@
   "dependencies": {
     "@actions/core": "^1.10.0",
     "@octokit/core": "^5.0.0",
+    "@octokit/plugin-throttling": "^8.1.3",
     "@octokit/rest": "^20.0.0",
     "@octokit/types": "^12.0.0",
     "@probot/octokit-plugin-config": "^2.0.0",

src/octokit.ts

@@ -1,12 +1,33 @@
+import { info, warning } from '@actions/core';
 import { Octokit } from '@octokit/core';
 import { config } from '@probot/octokit-plugin-config';
+import { throttling } from '@octokit/plugin-throttling';
 
-const CustomOctokit = Octokit.plugin(config);
+const CustomOctokit = Octokit.plugin(config, throttling);
 
 export type CustomOctokit = InstanceType<typeof CustomOctokit>;
 
 export function getOctokit(token: string) {
   return new CustomOctokit({
     auth: token,
+    throttle: {
+      onRateLimit: (retryAfter, options, _octokit, retryCount) => {
+        warning(
+          `Request quota exhausted for request ${options.method} ${options.url}`
+        );
+
+        // Retry once after hitting a rate limit error, then give up
+        if (retryCount < 1) {
+          info(`Retrying after ${retryAfter} seconds!`);
+          return true;
+        }
+      },
+      onSecondaryRateLimit: (_retryAfter, options, _octokit) => {
+        // When a secondary rate limit is hit, don't retry
+        warning(
+          `SecondaryRateLimit detected for request ${options.method} ${options.url}`
+        );
+      },
+    },
   });
 }

yarn.lock

@@ -574,6 +574,18 @@ __metadata:
   languageName: node
   linkType: hard
 
+"@octokit/plugin-throttling@npm:^8.1.3":
+  version: 8.1.3
+  resolution: "@octokit/plugin-throttling@npm:8.1.3"
+  dependencies:
+    "@octokit/types": ^12.2.0
+    bottleneck: ^2.15.3
+  peerDependencies:
+    "@octokit/core": ^5.0.0
+  checksum: 98963ef2eab825015702b1ca1ef4ccbda0c009242e93001144e51014d3b53d3ecb1282b67488680c7f5f4e805d0c3af4020ad673079fff92c6353f04903a9a64
+  languageName: node
+  linkType: hard
+
 "@octokit/request-error@npm:^2.0.5, @octokit/request-error@npm:^2.1.0":
   version: 2.1.0
   resolution: "@octokit/request-error@npm:2.1.0"
@@ -635,7 +647,7 @@ __metadata:
   languageName: node
   linkType: hard
 
-"@octokit/types@npm:^12.0.0":
+"@octokit/types@npm:^12.0.0, @octokit/types@npm:^12.2.0":
   version: 12.4.0
   resolution: "@octokit/types@npm:12.4.0"
   dependencies:
@@ -1190,6 +1202,13 @@ __metadata:
   languageName: node
   linkType: hard
 
+"bottleneck@npm:^2.15.3":
+  version: 2.19.5
+  resolution: "bottleneck@npm:2.19.5"
+  checksum: c5eef1bbea12cef1f1405e7306e7d24860568b0f7ac5eeab706a86762b3fc65ef6d1c641c8a166e4db90f412fc5c948fc5ce8008a8cd3d28c7212ef9c3482bda
+  languageName: node
+  linkType: hard
+
 "bplist-parser@npm:^0.2.0":
   version: 0.2.0
   resolution: "bplist-parser@npm:0.2.0"
@@ -3621,6 +3640,7 @@ __metadata:
     "@actions/core": ^1.10.0
     "@actions/github": ^5.1.1
     "@octokit/core": ^5.0.0
+    "@octokit/plugin-throttling": ^8.1.3
     "@octokit/rest": ^20.0.0
     "@octokit/types": ^12.0.0
     "@probot/octokit-plugin-config": ^2.0.0