From 4ef9d43528444436b39a212bc0c3fab53947df7e Mon Sep 17 00:00:00 2001
From: Lordmau5 <mail@lordmau5.com>
Date: Wed, 1 Jul 2020 05:10:51 +0200
Subject: [PATCH] Fix Auto Host Management

Since Twitch has removed their old v4 endpoints for auto host related stuff this needed to be switched over to GQL.
---
 .../twitch-twilight/modules/autohost_list.gql |  13 ++
 .../modules/autohost_list_mutate.gql          |  17 ++
 .../modules/autohost_settings.gql             |  11 ++
 .../modules/autohost_settings_mutate.gql      |  24 +++
 .../twitch-twilight/modules/host-options.vue  |  44 ++----
 .../twitch-twilight/modules/host_button.js    | 149 +++++++-----------
 6 files changed, 131 insertions(+), 127 deletions(-)
 create mode 100644 src/sites/twitch-twilight/modules/autohost_list.gql
 create mode 100644 src/sites/twitch-twilight/modules/autohost_list_mutate.gql
 create mode 100644 src/sites/twitch-twilight/modules/autohost_settings.gql
 create mode 100644 src/sites/twitch-twilight/modules/autohost_settings_mutate.gql

diff --git a/src/sites/twitch-twilight/modules/autohost_list.gql b/src/sites/twitch-twilight/modules/autohost_list.gql
new file mode 100644
index 00000000..1c523171
--- /dev/null
+++ b/src/sites/twitch-twilight/modules/autohost_list.gql
@@ -0,0 +1,13 @@
+query FFZ_AutoHostList($id: ID, $login: String) {
+	user(id: $id, login: $login) {
+        id,
+		autohostChannels {
+            nodes {
+                id
+                login
+                displayName
+                profileImageURL(width: 50)
+            }
+        }
+	}
+}
\ No newline at end of file
diff --git a/src/sites/twitch-twilight/modules/autohost_list_mutate.gql b/src/sites/twitch-twilight/modules/autohost_list_mutate.gql
new file mode 100644
index 00000000..fc7dc7d3
--- /dev/null
+++ b/src/sites/twitch-twilight/modules/autohost_list_mutate.gql
@@ -0,0 +1,17 @@
+mutation FFZ_AutoHostListUpdate($userID: ID!, $channelIDs: [ID!]!) {
+  setAutohostChannels(input: {
+    userID: $userID,
+    channelIDs: $channelIDs
+  }) {
+    user {
+      autohostChannels {
+        nodes {
+          id
+          login
+          displayName
+          profileImageURL(width: 50)
+        }
+      }
+    }
+  }
+}
\ No newline at end of file
diff --git a/src/sites/twitch-twilight/modules/autohost_settings.gql b/src/sites/twitch-twilight/modules/autohost_settings.gql
new file mode 100644
index 00000000..8840d28c
--- /dev/null
+++ b/src/sites/twitch-twilight/modules/autohost_settings.gql
@@ -0,0 +1,11 @@
+query FFZ_AutoHostSettings($id: ID, $login: String) {
+	user(id: $id, login: $login) {
+        id,
+		autohostSettings {
+            deprioritizeVodcast,
+            enabled,
+            strategy,
+            teamHost
+        }
+	}
+}
\ No newline at end of file
diff --git a/src/sites/twitch-twilight/modules/autohost_settings_mutate.gql b/src/sites/twitch-twilight/modules/autohost_settings_mutate.gql
new file mode 100644
index 00000000..9facf19e
--- /dev/null
+++ b/src/sites/twitch-twilight/modules/autohost_settings_mutate.gql
@@ -0,0 +1,24 @@
+mutation FFZ_AutoHostSettingsUpdate(
+  $userID: ID!,
+  $isEnabled: Boolean,
+  $strategy: AutohostSettingsStrategy,
+  $willAutohostTeam: Boolean,
+  $willPrioritizeAutohost: Boolean
+) {
+  updateAutohostSettings(input: {
+    userID: $userID,
+    isEnabled: $isEnabled,
+    strategy: $strategy,
+    willAutohostTeam: $willAutohostTeam,
+    willPrioritizeAutohost: $willPrioritizeAutohost,
+  }) {
+    user {
+      autohostSettings {
+        deprioritizeVodcast,
+        enabled,
+        strategy,
+        teamHost
+      }
+    }
+  }
+}
\ No newline at end of file
diff --git a/src/sites/twitch-twilight/modules/host-options.vue b/src/sites/twitch-twilight/modules/host-options.vue
index 1f8af8d3..d5d40865 100644
--- a/src/sites/twitch-twilight/modules/host-options.vue
+++ b/src/sites/twitch-twilight/modules/host-options.vue
@@ -22,18 +22,18 @@
 						>
 							<div
 								v-for="host in hosts"
-								:key="host._id"
-								:data-id="host._id"
+								:key="host.id"
+								:data-id="host.id"
 								class="tw-border-t ffz--host-user"
 							>
 								<div class="tw-interactable tw-interactable--inverted">
 									<div class="tw-align-items-center tw-flex tw-flex-row tw-flex-nowrap tw-mg-x-1">
 										<figure class="ffz-i-ellipsis-vert handle" />
 										<div class="ffz-channel-avatar">
-											<img :src="host.logo" :alt="host.display_name + '(' + host.name + ')'">
+											<img :src="host.profileImageURL" :alt="host.displayName + '(' + host.login + ')'">
 										</div>
 										<p class="tw-ellipsis tw-flex-grow-1 tw-mg-l-1 tw-font-size-5">
-											{{ host.name }}
+											{{ host.login }}
 										</p>
 										<div class="tw-flex-grow-1 tw-pd-x-2" />
 										<button class="tw-button-icon tw-mg-x-05 ffz--host-remove-user" @click="removeFromHosts">
@@ -91,14 +91,14 @@
 								<div class="ffz--widget ffz--checkbox">
 									<div class="tw-flex tw-align-items-center tw-checkbox">
 										<input
-											id="autoHostSettings:team_host"
-											:checked="autoHostSettings.team_host"
+											id="autoHostSettings:teamHost"
+											:checked="autoHostSettings.teamHost"
 											type="checkbox"
 											class="tw-checkbox__input"
-											data-setting="team_host"
+											data-setting="teamHost"
 											@change="updateCheckbox"
 										>
-										<label for="autoHostSettings:team_host" class="tw-checkbox__label">
+										<label for="autoHostSettings:teamHost" class="tw-checkbox__label">
 											<span class="tw-mg-l-1">
 												{{ t('metadata.host.setting.team-hosting.title', 'Team Hosting') }}
 											</span>
@@ -113,14 +113,14 @@
 								<div class="ffz--widget ffz--checkbox">
 									<div class="tw-flex tw-align-items-center tw-checkbox">
 										<input
-											id="autoHostSettings:vodcast_hosting"
-											:checked="!autoHostSettings.deprioritize_vodcast"
+											id="autoHostSettings:deprioritizeVodcast"
+											:checked="!autoHostSettings.deprioritizeVodcast"
 											type="checkbox"
 											class="tw-checkbox__input"
-											data-setting="deprioritize_vodcast"
+											data-setting="deprioritizeVodcast"
 											@change="updateCheckbox"
 										>
-										<label for="autoHostSettings:vodcast_hosting" class="tw-checkbox__label">
+										<label for="autoHostSettings:deprioritizeVodcast" class="tw-checkbox__label">
 											<span class="tw-mg-l-1">
 												{{ t('metadata.host.setting.vodcast-hosting.title', 'Vodcast Hosting') }}
 											</span>
@@ -131,26 +131,6 @@
 										<a href="https://blog.twitch.tv/vodcast-brings-the-twitch-community-experience-to-uploads-54098498715" target="_blank" rel="noopener">{{ t('metadata.host.setting.vodcast-hosting.link', 'Learn about Vodcasts') }}</a>
 									</section>
 								</div>
-								<div class="ffz--widget ffz--checkbox">
-									<div class="tw-flex tw-align-items-center tw-checkbox">
-										<input
-											id="autoHostSettings:recommended_host"
-											:checked="autoHostSettings.recommended_host"
-											type="checkbox"
-											class="tw-checkbox__input"
-											data-setting="recommended_host"
-											@change="updateCheckbox"
-										>
-										<label for="autoHostSettings:recommended_host" class="tw-checkbox__label">
-											<span class="tw-mg-l-1">
-												{{ t('metadata.host.setting.recommended-hosting.title', 'Auto Host Channels Similar To Yours') }}
-											</span>
-										</label>
-									</div>
-									<section class="tw-c-text-alt-2 ffz-checkbox-description">
-										{{ t('metadata.host.setting.recommended-hosting.description', 'Streamers on your primary team &amp; host list will always be hosted first') }}
-									</section>
-								</div>
 								<div class="ffz--widget ffz--checkbox">
 									<div class="tw-flex tw-align-items-center tw-checkbox">
 										<input
diff --git a/src/sites/twitch-twilight/modules/host_button.js b/src/sites/twitch-twilight/modules/host_button.js
index 9adb1292..982972cc 100644
--- a/src/sites/twitch-twilight/modules/host_button.js
+++ b/src/sites/twitch-twilight/modules/host_button.js
@@ -5,6 +5,7 @@
 // ============================================================================
 
 import Module from 'utilities/module';
+import {get} from 'utilities/object';
 import {createElement} from 'utilities/dom';
 
 const HOST_ERRORS = {
@@ -27,6 +28,7 @@ export default class HostButton extends Module {
 		this.inject('site');
 		this.inject('site.fine');
 		this.inject('site.chat');
+		this.inject('site.twitch_data');
 		this.inject('i18n');
 		this.inject('metadata');
 		this.inject('settings');
@@ -224,14 +226,20 @@ export default class HostButton extends Module {
 					if (this._auto_host_tip) this._auto_host_tip.update();
 				},
 				updateCheckbox: e => {
-					const t = e.target,
-						setting = t.dataset.setting;
-					let state = t.checked;
+					const t = e.target;
+					let setting = t.dataset.setting,
+						state = t.checked;
 
-					if ( setting === 'strategy' )
+					if ( setting === 'enabled' )
+						setting = 'isEnabled';
+					else if ( setting === 'teamHost' )
+						setting = 'willAutohostTeam';
+					else if ( setting === 'strategy' )
 						state = state ? 'random' : 'ordered';
-					else if ( setting === 'deprioritize_vodcast' )
+					else if ( setting === 'deprioritizeVodcast' ) {
+						setting = 'willPrioritizeAutohost';
 						state = ! state;
+					}
 
 					this.updateAutoHostSetting(setting, state);
 				}
@@ -246,26 +254,17 @@ export default class HostButton extends Module {
 		if ( ! user )
 			return;
 
-		let data;
-		try {
-			data = await fetch('https://api.twitch.tv/kraken/autohost/list', {
-				headers: {
-					'Accept': 'application/vnd.twitchtv.v4+json',
-					'Authorization': `OAuth ${user.authToken}`
-				}
-			}).then(r => {
-				if ( r.ok )
-					return r.json();
+		const result = await this.twitch_data.queryApollo(
+			await import(/* webpackChunkName: 'queries' */ './autohost_list.gql'),
+			{
+				id: user.id
+			},
+			{
+				fetchPolicy: 'network-only'
+			}
+		);
 
-				throw r.status;
-			});
-
-		} catch(err) {
-			this.log.error('Error loading auto host list.', err);
-			return;
-		}
-
-		return this.autoHosts = data.targets;
+		return this.autoHosts = get('data.user.autohostChannels.nodes', result);
 	}
 
 	async fetchAutoHostSettings() {
@@ -273,26 +272,17 @@ export default class HostButton extends Module {
 		if ( ! user )
 			return;
 
-		let data;
-		try {
-			data = await fetch('https://api.twitch.tv/kraken/autohost/settings', {
-				headers: {
-					'Accept': 'application/vnd.twitchtv.v4+json',
-					'Authorization': `OAuth ${user.authToken}`
-				}
-			}).then(r => {
-				if ( r.ok )
-					return r.json();
+		const result = await this.twitch_data.queryApollo(
+			await import(/* webpackChunkName: 'queries' */ './autohost_settings.gql'),
+			{
+				id: user.id
+			},
+			{
+				fetchPolicy: 'network-only'
+			}
+		);
 
-				throw r.status;
-			});
-
-		} catch(err) {
-			this.log.error('Error loading auto host settings.', err);
-			return;
-		}
-
-		return this.autoHostSettings = data.settings;
+		return this.autoHostSettings = get('data.user.autohostSettings', result);
 	}
 
 	queueHostUpdate() {
@@ -312,12 +302,12 @@ export default class HostButton extends Module {
 	}
 
 	currentRoomInHosts() {
-		return this.getAutoHostIDs(this.autoHosts).includes(parseInt(this._current_channel_id, 10));
+		return this.getAutoHostIDs(this.autoHosts).includes(this._current_channel_id);
 	}
 
 	addCurrentRoomToHosts() {
 		const newHosts = this.autoHosts.slice(0);
-		newHosts.push({ _id: parseInt(this._current_channel_id, 10)});
+		newHosts.push({ id: this._current_channel_id});
 
 		this.updateAutoHosts(newHosts);
 	}
@@ -327,7 +317,7 @@ export default class HostButton extends Module {
 
 		const newHosts = [];
 		for (let i = 0; i < this.autoHosts.length; i++) {
-			if (this.autoHosts[i]._id != id) newHosts.push(this.autoHosts[i]);
+			if (this.autoHosts[i].id != id) newHosts.push(this.autoHosts[i]);
 		}
 
 		this.updateAutoHosts(newHosts);
@@ -337,7 +327,7 @@ export default class HostButton extends Module {
 		const ids = [];
 		if (hosts) {
 			for (let i = 0; i < hosts.length; i++) {
-				ids.push(hosts[i]._id);
+				ids.push(hosts[i].id);
 			}
 		}
 		return ids;
@@ -348,32 +338,17 @@ export default class HostButton extends Module {
 		if ( ! user )
 			return;
 
-		let data;
-		try {
-			const form = new URLSearchParams();
-			const autoHosts = this.getAutoHostIDs(newHosts);
-			form.append('targets', autoHosts.join(','));
+		const autoHosts = this.getAutoHostIDs(newHosts);
 
-			data = await fetch('https://api.twitch.tv/kraken/autohost/list', {
-				headers: {
-					'Accept': 'application/vnd.twitchtv.v4+json',
-					'Authorization': `OAuth ${user.authToken}`
-				},
-				method: autoHosts.length ? 'PUT' : 'DELETE',
-				body: autoHosts.length ? form : undefined
-			}).then(r => {
-				if ( r.ok )
-					return r.json();
+		const result = await this.twitch_data.mutate({
+			mutation: await import(/* webpackChunkName: 'queries' */ './autohost_list_mutate.gql'),
+			variables: {
+				userID: user.id,
+				channelIDs: autoHosts
+			}
+		});
 
-				throw r.status;
-			});
-
-		} catch(err) {
-			this.log.error('Error updating auto host list.', err);
-			return;
-		}
-
-		this.autoHosts = data.targets;
+		this.autoHosts = get('data.setAutohostChannels.user.autohostChannels.nodes', result);
 		if (this.vueHostMenu) {
 			this.vueHostMenu.data.hosts = this.autoHosts;
 			this.vueHostMenu.data.addedToHosts = this.currentRoomInHosts();
@@ -385,31 +360,15 @@ export default class HostButton extends Module {
 		if ( ! user )
 			return;
 
-		let data;
-		try {
-			const form = new URLSearchParams();
-			form.append(setting, newValue);
+		const result = await this.twitch_data.mutate({
+			mutation: await import(/* webpackChunkName: 'queries' */ './autohost_settings_mutate.gql'),
+			variables: {
+				userID: user.id,
+				[setting]: newValue
+			}
+		});
 
-			data = await fetch('https://api.twitch.tv/kraken/autohost/settings', {
-				headers: {
-					'Accept': 'application/vnd.twitchtv.v4+json',
-					'Authorization': `OAuth ${user.authToken}`
-				},
-				method: 'PUT',
-				body: form
-			}).then(r => {
-				if ( r.ok )
-					return r.json();
-
-				throw r.status;
-			});
-
-		} catch(err) {
-			this.log.error('Error updating auto host setting.', err);
-			return;
-		}
-
-		this.autoHostSettings = data.settings;
+		this.autoHostSettings = get('data.updateAutohostSettings.user.autohostSettings', result);
 		if (this.vueHostMenu) {
 			this.vueHostMenu.data.autoHostSettings = this.autoHostSettings;
 		}