From fcd7ef9a386a27292f177279edf4b4395a2d4828 Mon Sep 17 00:00:00 2001
From: kridd <kayleigh.ridd@outlook.com>
Date: Mon, 6 Mar 2023 13:56:40 +0000
Subject: [PATCH] Add option to send account id.

---
 src/components/Instance.js | 14 ++++++--
 src/lib/authWindow.js      | 70 ++++++++++++++++++++++----------------
 2 files changed, 52 insertions(+), 32 deletions(-)

diff --git a/src/components/Instance.js b/src/components/Instance.js
index bd4f6c1..084efc0 100644
--- a/src/components/Instance.js
+++ b/src/components/Instance.js
@@ -28,7 +28,8 @@ export class Instance extends React.PureComponent {
         instanceState: undefined,
         configWizardSrc: undefined,
         authExternalId: undefined,
-        authUrlParams: ''
+        authUrlParams: '',
+        accountId: ''
     };
 
     openWizard = (openInIframe, addCustomValidation = false) => {
@@ -78,7 +79,7 @@ export class Instance extends React.PureComponent {
     onCreateAuth = () => {
         getAuthCreateUrl(this.props.id, this.state.authExternalId)
             .then(({body}) => {
-                openAuthWindow(`${body.data.popupUrl}&${this.state.authUrlParams}`);
+                openAuthWindow(`${body.data.popupUrl}&${this.state.authUrlParams}`, this.state.accountId);
             })
     };
 
@@ -202,6 +203,15 @@ export class Instance extends React.PureComponent {
                                     shrink: true,
                                 }}
                             />
+                            <TextField
+                                style={styles.textFields}
+                                label="Account id (For use with enhanced session transfer security)"
+                                value={this.state.accountId}
+                                onChange={this.handleChange('accountId')}
+                                InputLabelProps={{
+                                    shrink: true,
+                                }}
+                            />
                             <Button
                                 style={styles.button}
                                 onClick={this.onCreateAuth}
diff --git a/src/lib/authWindow.js b/src/lib/authWindow.js
index 653fb75..0e6d80c 100644
--- a/src/lib/authWindow.js
+++ b/src/lib/authWindow.js
@@ -1,36 +1,46 @@
-export const openAuthWindow = (url) => {
-    // Must open window from user interaction code otherwise it is likely
-    // to be blocked by a popup blocker:
-    const authWindow = window.open(
-        undefined,
-        '_blank',
-        'width=500,height=500,scrollbars=no',
-    );
+export const openAuthWindow = (url, accountId) => {
+  // Must open window from user interaction code otherwise it is likely
+  // to be blocked by a popup blocker:
+  const authWindow = window.open(
+    undefined,
+    "_blank",
+    "width=500,height=500,scrollbars=no"
+  );
 
-    const onmessage = e => {
-        console.log('message', e.data.type, e.data);
+  const onmessage = (e) => {
 
-        if (e.data.type === 'tray.authPopup.error') {
-            // Handle popup error message
-            alert(`Error: ${e.data.error}`);
-            authWindow.close();
-        }
-        if (e.data.type === 'tray.authpopup.close' || e.data.type === 'tray.authpopup.finish') {
-            authWindow.close();
-        }
-    };
-    window.addEventListener('message', onmessage);
+    if(e.data.type === "tray.authpopup.accountIdRequest"){
+        // If the user has the enhancedSessionSecurity option set, they will send a postMessage
+        // requesting the account ID. Unless the client app responds with the appropriate ID
+        // the session will not be validated
+        e.source.postMessage({type: 'tray.authpopup.accountIdRequest', data: accountId},
+                           e.origin);
+    }
+
+    if (e.data.type === "tray.authPopup.error") {
+      // Handle popup error message
+      alert(`Error: ${e.data.error}`);
+      authWindow.close();
+    }
+    if (
+      e.data.type === "tray.authpopup.close" ||
+      e.data.type === "tray.authpopup.finish"
+    ) {
+    //   authWindow.close();
+    }
+  };
+  window.addEventListener("message", onmessage);
 
-    // Check if popup window has been closed
-    const CHECK_TIMEOUT = 1000;
-    const checkClosedWindow = () => {
-        if (authWindow.closed) {
-            window.removeEventListener('message', onmessage);
-        } else {
-            setTimeout(checkClosedWindow, CHECK_TIMEOUT);
-        }
+  // Check if popup window has been closed
+  const CHECK_TIMEOUT = 1000;
+  const checkClosedWindow = () => {
+    if (authWindow.closed) {
+      window.removeEventListener("message", onmessage);
+    } else {
+      setTimeout(checkClosedWindow, CHECK_TIMEOUT);
     }
+  };
 
-    checkClosedWindow();
-    authWindow.location = url;
+  checkClosedWindow();
+  authWindow.location = url;
 };