From ffcabaaad9726bb910f03fc695e0945aad031542 Mon Sep 17 00:00:00 2001
From: Shekar Siri <sshekarsiri@gmail.com>
Date: Wed, 28 Sep 2022 18:08:58 +0530
Subject: [PATCH] change(ui) - assist agentToken

---
 frontend/app/player/MessageDistributor/MessageDistributor.ts | 2 +-
 .../app/player/MessageDistributor/managers/AssistManager.ts  | 5 ++++-
 frontend/app/types/session/session.ts                        | 5 +++--
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/frontend/app/player/MessageDistributor/MessageDistributor.ts b/frontend/app/player/MessageDistributor/MessageDistributor.ts
index 936f7409f..04054c9c6 100644
--- a/frontend/app/player/MessageDistributor/MessageDistributor.ts
+++ b/frontend/app/player/MessageDistributor/MessageDistributor.ts
@@ -102,7 +102,7 @@ export default class MessageDistributor extends StatedScreen {
 
     if (live) {
       initListsDepr({})
-      this.assistManager.connect();
+      this.assistManager.connect(this.session.agentToken);
     } else {
       this.activirtManager = new ActivityManager(this.session.duration.milliseconds);
       /* == REFACTOR_ME == */
diff --git a/frontend/app/player/MessageDistributor/managers/AssistManager.ts b/frontend/app/player/MessageDistributor/managers/AssistManager.ts
index 2ba90311e..c7d2a52f3 100644
--- a/frontend/app/player/MessageDistributor/managers/AssistManager.ts
+++ b/frontend/app/player/MessageDistributor/managers/AssistManager.ts
@@ -115,7 +115,7 @@ export default class AssistManager {
   }
 
   private socket: Socket | null = null
-  connect() {
+  connect(agentToken: string) {
     const jmr = new JSONRawMessageReader()
     const reader = new MStreamReader(jmr)
     let waitingForMessages = true
@@ -130,6 +130,9 @@ export default class AssistManager {
       // @ts-ignore WTF, socket.io ???
       const socket: Socket = this.socket = io(urlObject.origin, {
         path: '/ws-assist/socket',
+        auth: {
+          token: agentToken
+        },
         query: {
           peerId: this.peerID,
           identity: "agent",
diff --git a/frontend/app/types/session/session.ts b/frontend/app/types/session/session.ts
index a4ed48fe6..dbe949eec 100644
--- a/frontend/app/types/session/session.ts
+++ b/frontend/app/types/session/session.ts
@@ -79,8 +79,9 @@ export default Record({
   isIOS: false,
   revId: '',
   userSessionsCount: 0,
+  agentToken: '',
 }, {
-  fromJS:({ 
+  fromJS:({
     startTs=0,
     timestamp = 0,
     backendErrors=0,
@@ -115,7 +116,7 @@ export default Record({
     const missedResources = resources.filter(({ success }) => !success);
     const logs = List(session.logs).map(Log);
 
-    const stackEventsList = List(stackEvents)   
+    const stackEventsList = List(stackEvents)
       .concat(List(session.userEvents))
       .sortBy(se => se.timestamp)
       .map(se => StackEvent({ ...se, time: se.timestamp - startedAt }));