From 8a6dfa33a106f4eef6bcb658fa95599ccd1f6dda Mon Sep 17 00:00:00 2001
From: Andrew Heard <3010484+andrewheard@users.noreply.github.com>
Date: Wed, 26 Jan 2022 21:39:21 -0500
Subject: [PATCH] Support custom API endpoints in SDM plugin

Allows the user to customize the endpoints used for authorization and
accessing the API in the google-device-access plugin by specifying
hostnames in storage. These can be configured using the
`authorizationHostname` and `apiHostname` keys, respectively.
---
 plugins/google-device-access/src/main.ts | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/plugins/google-device-access/src/main.ts b/plugins/google-device-access/src/main.ts
index e68982122..5cd73376c 100644
--- a/plugins/google-device-access/src/main.ts
+++ b/plugins/google-device-access/src/main.ts
@@ -455,6 +455,8 @@ class GoogleSmartDeviceAccess extends ScryptedDeviceBase implements OauthClient,
     authorizationUri: string;
     client: ClientOAuth2;
 
+    apiHostname: string;
+
     startup: Promise<void>;
 
     updateClient() {
@@ -465,7 +467,8 @@ class GoogleSmartDeviceAccess extends ScryptedDeviceBase implements OauthClient,
             this.log.a('Enter a valid project ID. Setup instructions for Nest: https://www.home-assistant.io/integrations/nest/');
         }
 
-        this.authorizationUri = `https://nestservices.google.com/partnerconnections/${this.projectId}/auth`
+        const authorizationHostname = this.storage.getItem('authorizationHostname') || 'nestservices.google.com';
+        this.authorizationUri = `https://${authorizationHostname}/partnerconnections/${this.projectId}/auth`
         this.client = new ClientOAuth2({
             clientId: this.clientId,
             clientSecret: this.clientSecret,
@@ -475,6 +478,8 @@ class GoogleSmartDeviceAccess extends ScryptedDeviceBase implements OauthClient,
                 'https://www.googleapis.com/auth/sdm.service',
             ]
         });
+
+        this.apiHostname = this.storage.getItem('apiHostname') || 'smartdevicemanagement.googleapis.com';
     }
 
     refreshThrottled = throttle(async () => {
@@ -678,7 +683,7 @@ class GoogleSmartDeviceAccess extends ScryptedDeviceBase implements OauthClient,
 
     async authGet(path: string) {
         await this.loadToken();
-        return axios(`https://smartdevicemanagement.googleapis.com/v1/enterprises/${this.projectId}${path}`, {
+        return axios(`https://${this.apiHostname}/v1/enterprises/${this.projectId}${path}`, {
             // validateStatus() {
             //     return true;
             // },
@@ -690,7 +695,7 @@ class GoogleSmartDeviceAccess extends ScryptedDeviceBase implements OauthClient,
 
     async authPost(path: string, data: any) {
         await this.loadToken();
-        return axios.post(`https://smartdevicemanagement.googleapis.com/v1/enterprises/${this.projectId}${path}`, data, {
+        return axios.post(`https://${this.apiHostname}/v1/enterprises/${this.projectId}${path}`, data, {
             headers: {
                 Authorization: `Bearer ${this.token.accessToken}`
             }