From 9c31d5b8a0843e355dfa1f1489462f0648bb9c27 Mon Sep 17 00:00:00 2001
From: KatieFrogs <23621460+KatieFrogs@users.noreply.github.com>
Date: Fri, 11 Mar 2022 17:34:00 +0300
Subject: [PATCH] Use Google 3P authorization

---
 public/src/js/customsongs.js |  3 ++
 public/src/js/gpicker.js     | 90 ++++++++++++++++--------------------
 2 files changed, 42 insertions(+), 51 deletions(-)

diff --git a/public/src/js/customsongs.js b/public/src/js/customsongs.js
index 7a5a155..e9f2d86 100644
--- a/public/src/js/customsongs.js
+++ b/public/src/js/customsongs.js
@@ -516,6 +516,9 @@ class CustomSongs{
 			pageEvents.remove(document, ["dragover", "dragleave", "drop"])
 			delete this.dropzone
 		}
+		if(gpicker){
+			gpicker.tokenResolve = null
+		}
 		delete this.browse
 		delete this.linkLocalFolder
 		delete this.linkGdriveFolder
diff --git a/public/src/js/gpicker.js b/public/src/js/gpicker.js
index c7c0590..5f6eaa0 100644
--- a/public/src/js/gpicker.js
+++ b/public/src/js/gpicker.js
@@ -9,9 +9,9 @@ class Gpicker{
 		this.scope = "https://www.googleapis.com/auth/drive.readonly"
 		this.folder = "application/vnd.google-apps.folder"
 		this.filesUrl = "https://www.googleapis.com/drive/v3/files/"
-		this.discoveryDocs = ["https://www.googleapis.com/discovery/v1/apis/drive/v3/rest"]
 		this.resolveQueue = []
 		this.queueActive = false
+		this.clientCallbackBind = this.clientCallback.bind(this)
 	}
 	browse(lockedCallback, errorCallback){
 		return this.loadApi()
@@ -124,9 +124,12 @@ class Gpicker{
 		if(window.gapi && gapi.client && gapi.client.drive){
 			return Promise.resolve()
 		}
-		return loader.loadScript("https://apis.google.com/js/api.js")
-		.then(() => new Promise((resolve, reject) =>
-			gapi.load("auth2:picker:client", {
+		var promises = [
+			loader.loadScript("https://apis.google.com/js/api.js"),
+			loader.loadScript("https://accounts.google.com/gsi/client")
+		]
+		return Promise.all(promises).then(() => new Promise((resolve, reject) =>
+			gapi.load("picker:client", {
 				callback: resolve,
 				onerror: reject
 			})
@@ -135,68 +138,53 @@ class Gpicker{
 			gapi.client.load("drive", "v3").then(resolve, reject)
 		))
 	}
-	getAuth(errorCallback=()=>{}){
-		if(!this.auth){
-			return new Promise((resolve, reject) => {
-				gapi.auth2.init({
-					apiKey: this.apiKey,
-					clientId: this.oauthClientId,
-					discoveryDocs: this.discoveryDocs,
-					fetch_basic_profile: false,
-					scope: this.scope
-				}).then(() => {
-					this.auth = gapi.auth2.getAuthInstance()
-					resolve(this.auth)
-				}, e => {
-					if(e.details){
-						var errorStr = strings.gpicker.authError.replace("%s", e.details)
-						if(/cookie/i.test(e.details)){
-							errorStr += "\n\n" + strings.gpicker.cookieError
-						}
-						errorCallback(errorStr)
-					}
-					reject(e)
-				})
-			})
+	getClient(errorCallback=()=>{}, force){
+		var obj = {
+			client_id: this.oauthClientId,
+			scope: this.scope,
+			callback: this.clientCallbackBind
+		}
+		if(force){
+			if(!this.clientForce){
+				obj.select_account = true
+				this.clientForce = google.accounts.oauth2.initTokenClient(obj)
+			}
+			return this.clientForce
 		}else{
-			return Promise.resolve(this.auth)
+			if(!this.client){
+				this.client = google.accounts.oauth2.initTokenClient(obj)
+			}
+			return this.client
+		}
+	}
+	clientCallback(tokenResponse){
+		this.tokenResponse = tokenResponse
+		this.oauthToken = tokenResponse.access_token
+		if(this.oauthToken && this.tokenResolve){
+			this.tokenResolve()
 		}
 	}
 	getToken(lockedCallback=()=>{}, errorCallback=()=>{}, force){
 		if(this.oauthToken && !force){
 			return Promise.resolve()
 		}
-		return this.getAuth(errorCallback).then(auth => {
-			if(!force && auth.isSignedIn.get() && this.checkScope()){
-				return Promise.resolve()
-			}else{
-				lockedCallback(false)
-				return new Promise((resolve, reject) =>
-					auth.signIn({
-						prompt: force ? "select_account" : "consent",
-						scope: this.scope
-					}).then(resolve, reject)
-				)
-			}
-		}).then(() => {
+		var client = this.getClient(errorCallback, force)
+		var promise = new Promise(resolve => {
+			this.tokenResolve = resolve
+		})
+		lockedCallback(false)
+		client.requestAccessToken()
+		return promise.then(() => {
+			this.tokenResolve = null
 			if(this.checkScope()){
 				lockedCallback(true)
 			}else{
 				return Promise.reject("cancel")
 			}
-		}, e => {
-			console.error(e)
-			Promise.reject("cancel")
 		})
 	}
 	checkScope(){
-		var user = this.auth.currentUser.get()
-		if(user.hasGrantedScopes(this.scope)){
-			this.oauthToken = user.getAuthResponse(true).access_token
-			return this.oauthToken
-		}else{
-			return false
-		}
+		return google.accounts.oauth2.hasGrantedAnyScope(this.tokenResponse, this.scope)
 	}
 	switchAccounts(lockedCallback, errorCallback){
 		return this.loadApi().then(() => this.getToken(lockedCallback, errorCallback, true))