From 606cd07ef79b551ede260a3407787825a8386abd Mon Sep 17 00:00:00 2001
From: Zoe Roux <zoe.roux@sdg.moe>
Date: Sat, 5 Mar 2022 11:26:37 +0100
Subject: [PATCH] Handling token refresh for spotify and google

---
 worker/src/models/pipeline.ts  |  2 +-
 worker/src/services/spotify.ts | 17 ++++++++++++++++-
 worker/src/services/youtube.ts | 16 +++++++++++++---
 3 files changed, 30 insertions(+), 5 deletions(-)

diff --git a/worker/src/models/pipeline.ts b/worker/src/models/pipeline.ts
index b24ffe5..cf8705c 100644
--- a/worker/src/models/pipeline.ts
+++ b/worker/src/models/pipeline.ts
@@ -96,7 +96,7 @@ export class Pipeline {
 export class Token {
 	accessToken: string;
 	refreshToken: string;
-	expiresIn: string;
+	expiresAt: string;
 };
 
 export class Reaction {
diff --git a/worker/src/services/spotify.ts b/worker/src/services/spotify.ts
index ce1944e..de27122 100644
--- a/worker/src/services/spotify.ts
+++ b/worker/src/services/spotify.ts
@@ -7,12 +7,14 @@ import { Pipeline, PipelineEnv, PipelineType, ReactionType, ServiceType } from "
 @service(ServiceType.Spotify)
 export class Spotify extends BaseService {
 
+	private _pipeline: Pipeline;
 	private _spotify;
 
 	constructor(pipeline: Pipeline) {
 		super();
 		if (!("Spotify" in pipeline.userData))
 			throw new Error("User is not authenticated via Spotify");
+		this._pipeline = pipeline;
 		this._spotify = new SpotifyWebApi({
 			accessToken: pipeline.userData["Spotify"].accessToken,
 			refreshToken: pipeline.userData["Spotify"].refreshToken,
@@ -22,7 +24,20 @@ export class Spotify extends BaseService {
 	}
 
 	private async _refreshIfNeeded(): Promise<void> {
-		
+		if (Date.parse(this._pipeline.userData["Spotify"].expiresAt) >= Date.now() + 100_000)
+			return;
+		const ret = await this._spotify.refreshAccessToken();
+		fetch(`${process.env["WORKER_API_URL"]}/spotify/${this._pipeline.userId}?WORKER_API_KEY=${process.env["WORKER_API_KEY"]}`, {
+			method: "POST",
+			headers: {
+				"Content-Type": "application/json",
+			},
+			body: JSON.stringify({
+				accessToken: ret.body.access_token,
+				refreshToken: ret.body.refresh_token,
+				expiresAt: new Date(Date.now() + ret.body.expires_in),
+			}),
+		});
 	}
 
 	@action(PipelineType.OnSpotifyAddToPlaylist, ["playlistId"])
diff --git a/worker/src/services/youtube.ts b/worker/src/services/youtube.ts
index ade2b6c..9af0a1a 100644
--- a/worker/src/services/youtube.ts
+++ b/worker/src/services/youtube.ts
@@ -22,9 +22,19 @@ export class Youtube extends BaseService {
 			refresh_token: pipeline.userData["Google"].refreshToken,
 			access_token: pipeline.userData["Google"].accessToken,
 		});
-		// client.on("tokens", x => {
-		// 	
-		// });
+		client.on("tokens", x => {
+			fetch(`${process.env["WORKER_API_URL"]}/google/${pipeline.userId}?WORKER_API_KEY=${process.env["WORKER_API_KEY"]}`, {
+				method: "POST",
+				headers: {
+					"Content-Type": "application/json",
+				},
+				body: JSON.stringify({
+					accessToken: x.access_token,
+					refreshToken: x.refresh_token,
+					expires: x.expiry_date,
+				}),
+			});
+		});
 		this._youtube = new youtube_v3.Youtube({
 			auth: client,
 		});