mrva/vscode-codeql
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Use octokit retry module

Browse Source
This commit is contained in:
Robert
2022-07-01 11:19:49 +00:00
committed by GitHub
parent f7aa0a5ae5
commit 825054a271
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extensions/ql-vscode/src/authentication.ts
View File

@@ -1,5 +1,6 @@
import * as vscode from 'vscode'; import * as vscode from 'vscode';
import * as Octokit from '@octokit/rest'; import * as Octokit from '@octokit/rest';
import { retry } from '@octokit/plugin-retry';
const GITHUB_AUTH_PROVIDER_ID = 'github'; const GITHUB_AUTH_PROVIDER_ID = 'github';
@@ -51,14 +52,15 @@ export class Credentials {
private async createOctokit(createIfNone: boolean, overrideToken?: string): Promise<Octokit.Octokit | undefined> { private async createOctokit(createIfNone: boolean, overrideToken?: string): Promise<Octokit.Octokit | undefined> {
if (overrideToken) { if (overrideToken) {
return new Octokit.Octokit({ auth: overrideToken }); return new Octokit.Octokit({ auth: overrideToken, retry });
} }
const session = await vscode.authentication.getSession(GITHUB_AUTH_PROVIDER_ID, SCOPES, { createIfNone }); const session = await vscode.authentication.getSession(GITHUB_AUTH_PROVIDER_ID, SCOPES, { createIfNone });
if (session) { if (session) {
return new Octokit.Octokit({ return new Octokit.Octokit({
auth: session.accessToken auth: session.accessToken,
retry
}); });
} else { } else {
return undefined; return undefined;