octokit on demand

src/GithubClient.js

@@ -22,16 +22,20 @@ const owner_repo = (opts) => {
   return { owner, repo };
 };
 
+const octokit = (token) => {
+  if (!token) throw new Error('token not found');
+
+  return github.getOctokit(token);
+};
+
 class GithubClient {
   constructor(opts = {}) {
     const { repo = this.env_repo(), token = this.env_token() } = opts;
 
     if (!repo) throw new Error('repo not found');
-    if (!token) throw new Error('token not found');
 
     this.repo = repo.endsWith('/') ? strip_last_chars(repo, 1) : repo;
     this.token = token;
-    this.octokit = github.getOctokit(token);
   }
 
   env_repo() {
@@ -67,7 +71,9 @@ class GithubClient {
   async comment_create(opts = {}) {
     const { report: body, commit_sha = this.env_head_sha() } = opts;
 
-    const { url: commit_url } = await this.octokit.repos.createCommitComment({
+    const { url: commit_url } = await octokit(
+      this.token
+    ).repos.createCommitComment({
       ...owner_repo({ uri: this.repo }),
       body,
       commit_sha
@@ -88,7 +94,7 @@ class GithubClient {
       status = 'completed'
     } = opts;
 
-    return await this.octokit.checks.create({
+    return await octokit(this.token).checks.create({
       ...owner_repo({ uri: this.repo }),
       head_sha,
       started_at,
@@ -106,11 +112,12 @@ class GithubClient {
 
   async runner_token() {
     const { owner, repo } = owner_repo({ uri: this.repo });
+    const { actions } = octokit(this.token);
 
     if (typeof repo !== 'undefined') {
       const {
         data: { token }
-      } = await this.octokit.actions.createRegistrationTokenForRepo({
+      } = await actions.createRegistrationTokenForRepo({
         owner,
         repo
       });
@@ -120,7 +127,7 @@ class GithubClient {
 
     const {
       data: { token }
-    } = await this.octokit.actions.createRegistrationTokenForOrg({
+    } = await actions.createRegistrationTokenForOrg({
       org: owner
     });
 

src/GithubClient.test.js

@@ -15,8 +15,9 @@ describe('Non Enviromental tests', () => {
     expect(gh_client.token).toBe(TOKEN);
 
     const { owner, repo } = gh_client.owner_repo();
-    expect(owner).toBe('DavidGOrtega');
-    expect(repo).toBe('3_tensorboard');
+    const parts = GITHUB_REPOSITORY.split('/');
+    expect(owner).toBe(parts[0]);
+    expect(repo).toBe(parts[1]);
   });
 
   test('Comment', async () => {
@@ -76,8 +77,9 @@ describe('Enviromental tests', () => {
     expect(gh_client.token).toBe(TOKEN);
 
     const { owner, repo } = gh_client.owner_repo();
-    expect(owner).toBe('DavidGOrtega');
-    expect(repo).toBe('3_tensorboard');
+    const parts = GITHUB_REPOSITORY.split('/');
+    expect(owner).toBe(parts[0]);
+    expect(repo).toBe(parts[1]);
   });
 
   test('Comment', async () => {

src/cml.js

@@ -7,7 +7,6 @@ const get_client = (opts) => {
   if (!driver) throw new Error('driver not set');
 
   if (driver === 'github') return new GithubClient({ repo, token });
-
   if (driver === 'gitlab') return new GitlabClient({ repo, token });
 
   throw new Error('driver unknown!');