Add throttling plugin for Octokit

package.json

@@ -16,6 +16,7 @@
   "dependencies": {
     "@actions/core": "1.2.6",
     "@actions/github": "4.0.0",
+    "@octokit/plugin-throttling": "3.3.1",
     "fast-glob": "3.2.4"
   },
   "devDependencies": {

src/octokit.ts

@@ -0,0 +1,56 @@
+import { Octokit } from '@octokit/core';
+import { throttling } from '@octokit/plugin-throttling';
+
+import { GitHub, getOctokitOptions } from '@actions/github/lib/utils';
+
+const OctokitConstructor = GitHub.plugin(throttling);
+
+const maxRetryCount = 1;
+
+interface RequestOptions {
+	method: string;
+	url: string;
+	request: RequestInfo;
+}
+
+interface RequestInfo {
+	retryCount: number;
+}
+
+export function createOctokit(
+	accessToken: string
+): InstanceType<typeof GitHub> {
+	const octokit = new OctokitConstructor(
+		getOctokitOptions(accessToken, {
+			throttle: { onRateLimit, onAbuseLimit },
+		})
+	);
+
+	return octokit;
+}
+
+function onRateLimit(
+	retryAfter: number,
+	options: RequestOptions,
+	octokit: Octokit
+): boolean {
+	octokit.log.warn(
+		`Request quota exhausted for request ${options.method} ${options.url}`
+	);
+	if (options.request.retryCount < maxRetryCount) {
+		octokit.log.info(`Retrying after ${retryAfter} seconds!`);
+		return true;
+	}
+
+	return false;
+}
+
+function onAbuseLimit(
+	_: number,
+	options: RequestOptions,
+	octokit: Octokit
+): void {
+	octokit.log.warn(
+		`Abuse detected for request ${options.method} ${options.url}`
+	);
+}

src/update.ts

@@ -3,9 +3,11 @@
 import { readFile, existsSync } from 'fs';
 import { promisify } from 'util';
 
-import { getOctokit, context } from '@actions/github';
+import { GitHub } from '@actions/github/lib/utils';
+import { context } from '@actions/github';
 
 import { UpdaterOptions, isNotNull } from './util';
+import { createOctokit } from './octokit';
 
 const readFileAsync = promisify(readFile);
 
@@ -27,12 +29,12 @@ interface UpdateResult {
 }
 
 export class Updater {
-	private octokit: ReturnType<typeof getOctokit>;
+	private octokit: InstanceType<typeof GitHub>;
 	private message: string;
 	private defaultBranch: string | null;
 
 	constructor(options: UpdaterOptions) {
-		this.octokit = getOctokit(options.token);
+		this.octokit = createOctokit(options.token);
 
 		this.message = options.message;
 		this.defaultBranch = options.branch || null;