Save youtube refresh token in config for re-use

Author: hlxid

nodecg-io-youtube/extension/index.ts

@@ -1,12 +1,15 @@
 import { NodeCG } from "nodecg/types/server";
 import { Result, emptySuccess, success, error, ServiceBundle } from "nodecg-io-core";
 import { google, youtube_v3 } from "googleapis";
+import type { Credentials } from "google-auth-library/build/src/auth/credentials";
+import type { OAuth2Client } from "google-auth-library/build/src/auth/oauth2client";
 import * as express from "express";
 import opn = require("open");
 
 interface YoutubeServiceConfig {
     clientID: string;
     clientSecret: string;
+    refreshToken?: string;
 }
 
 export type YoutubeServiceClient = youtube_v3.Youtube;
@@ -26,6 +29,36 @@ class YoutubeService extends ServiceBundle<YoutubeServiceConfig, YoutubeServiceC
             clientSecret: config.clientSecret,
             redirectUri: "http://localhost:9090/nodecg-io-youtube/oauth2callback",
         });
+
+        if (config.refreshToken) {
+            this.nodecg.log.info("Re-using saved refresh token.");
+            auth.setCredentials({
+                refresh_token: config.refreshToken,
+            });
+        } else {
+            this.nodecg.log.info("No refresh token found. Starting auth flow to get one...");
+            auth.setCredentials(await this.initialAuth(auth));
+            if (auth.credentials.refresh_token) {
+                config.refreshToken = auth.credentials.refresh_token;
+            }
+        }
+
+        // Save refresh tokens so they can be used next time to get a access token again
+        auth.on("tokens", (tokens) => {
+            if (tokens.refresh_token) {
+                config.refreshToken = tokens.refresh_token;
+            }
+        });
+
+        const client = new youtube_v3.Youtube({ auth });
+        return success(client);
+    }
+
+    stopClient(_client: YoutubeServiceClient): void {
+        // Cannot stop client
+    }
+
+    private initialAuth(auth: OAuth2Client): Promise<Credentials> {
         const authUrl = auth.generateAuthUrl({
             access_type: "offline",
             scope: "https://www.googleapis.com/auth/youtube",
@@ -41,9 +74,7 @@ class YoutubeService extends ServiceBundle<YoutubeServiceConfig, YoutubeServiceC
                     res.end("<script>window.close()</script>");
                     // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
                     const { tokens } = await auth.getToken(params.code!.toString());
-                    auth.credentials = tokens;
-                    const client = new youtube_v3.Youtube({ auth });
-                    resolve(success(client));
+                    resolve(tokens);
                 } catch (e) {
                     reject(error(e));
                 }
@@ -53,8 +84,4 @@ class YoutubeService extends ServiceBundle<YoutubeServiceConfig, YoutubeServiceC
             opn(authUrl, { wait: false }).then((cp) => cp.unref());
         });
     }
-
-    stopClient(_client: YoutubeServiceClient): void {
-        // Cannot stop client
-    }
 }

nodecg-io-youtube/youtube-schema.json

@@ -10,6 +10,10 @@
         "clientSecret": {
             "type": "string",
             "description": "The oauth client secret https://console.cloud.google.com/apis/credentials/oauthclient"
+        },
+        "refreshToken": {
+            "type": "string",
+            "description": "Token that allows the client to refresh access tokens. This is set automatically after first login, you don't need to set it."
         }
     },
     "required": ["clientID", "clientSecret"]