@knowlearning/agents 0.3.17 → 0.3.18

package/agents/generic.js

@@ -36,6 +36,8 @@ export default function Agent({ host, token, WebSocket, protocol='ws', uuid, fet
   let failedConnections = 0
   let mode = 'normal'
   const environmentPromise = new Promise(r => resolveEnvironment = r)
+  let lastSentSI = -1
+  let lastHeartbeat
 
   const sessionData = new MutableProxy({}, patch => queueMessage({scope: 'sessions', patch}))
 
@@ -46,6 +48,9 @@ export default function Agent({ host, token, WebSocket, protocol='ws', uuid, fet
     patches: {}
   }
 
+  log('INITIALIZING AGENT CONNECTION')
+  initWS()
+
   function log() {
     if (mode === 'debug') console.log(...arguments)
   }
@@ -55,11 +60,6 @@ export default function Agent({ host, token, WebSocket, protocol='ws', uuid, fet
     if (watcherIndex > -1) watchers[key].splice(watcherIndex, 1)
     else console.warn('TRIED TO REMOVE WATCHER THAT DOES NOT EXIST')
   }
-
-  log('INITIALIZING AGENT CONNECTION')
-  initWS()
-
-  let lastSentSI = -1
   function flushMessageQueue() {
     //  TODO: probably want to make this loop async so we don't try and push more to
     //        a closed connection
@@ -80,23 +80,34 @@ export default function Agent({ host, token, WebSocket, protocol='ws', uuid, fet
     flushMessageQueue()
   }
 
-  let lastHeartbeat
   function checkHeartbeat() {
     clearTimeout(lastHeartbeat)
     lastHeartbeat = setTimeout(
       () => {
         log('CLOSING DUE TO HEARTBEAT TIMEOUT')
-        ws.close()
+        restartConnection()
       },
       HEARTBEAT_TIMEOUT
     )
   }
 
+  async function restartConnection() {
+    if (authed) log(`CLOSED DOMAIN ${domain} USER ${user} SESSION ${session} CONNECTION TO SERVER ${server}`)
+    authed = false
+    ws.onmessage = () => {} // needs to be a no-op since a closing ws can still get messages
+    if (!disconnected) {
+      await new Promise(r => setTimeout(r, Math.min(1000, failedConnections * 100)))
+      failedConnections += 1
+      initWS() // TODO: don't do this if we are purposefully unloading...
+    }
+  }
+
   function initWS() {
     ws = new WebSocket(`${protocol}://${host}`)
 
+    let opened = false
     ws.onopen = async () => {
-      checkHeartbeat()
+      opened = true
       log('AUTHORIZING NEWLY OPENED WS FOR SESSION:', session)
       failedConnections = 0
       ws.send(JSON.stringify({ token: await token(), session }))
@@ -169,19 +180,16 @@ export default function Agent({ host, token, WebSocket, protocol='ws', uuid, fet
     }
 
     ws.onerror = async error => {
-      log('WS CONNECTION ERROR', error.message)
+      log('WS CONNECTION ERROR', error.message, opened)
+      if (!opened) restartConnection() // onclose won't trigger if never opened
     }
 
-    ws.onclose = async () => {
-      if (authed) log(`CLOSED DOMAIN ${domain} USER ${user} SESSION ${session} CONNECTION TO SERVER ${server}`)
-      authed = false
-      ws.onmessage = () => {} // needs to be a no-op since a closing ws can still get messages
-      if (!disconnected) {
-        await new Promise(r => setTimeout(r, Math.min(1000, failedConnections * 100)))
-        failedConnections += 1
-        initWS() // TODO: don't do this if we are purposefully unloading...
-      }
+    ws.onclose = async error => {
+      log('WS CLOSURE', error.message, opened)
+      restartConnection()
     }
+
+    checkHeartbeat()
   }
 
   function environment() { return environmentPromise }
@@ -223,7 +231,7 @@ export default function Agent({ host, token, WebSocket, protocol='ws', uuid, fet
       //        POSSIBLE: wait for most recent interaction update response for key k that we sent...
       await lastInteractionUpdateForWatchedScope[k]
       //  TODO: probably something like await updateMessageReceivedForLastInteractionWeSentForKey[k]
-      resolveState(states[k])
+      resolveState(structuredClone(await states[k]))
     })
 
     promise.watch = fn => {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@knowlearning/agents",
-    "version": "0.3.17",
+    "version": "0.3.18",
     "description": "API for embedding applications in KnowLearning systems.",
     "main": "node.js",
     "browser": "browser.js",