@knowlearning/agents 0.9.90 → 0.9.91

package/agents/browser/root.js

@@ -4,18 +4,23 @@ import { getToken, login, logout } from './auth.js'
 import GenericAgent from '../generic/index.js'
 
 const DEVELOPMENT_HOST = 'localhost:32001'
-const REMOTE_HOST = 'api.knowlearning.systems'
+const REMOTE_HOST = localStorage.getItem('mode') === 'staging' ? 'api.staging.knowlearning.systems' : 'api.knowlearning.systems'
 
 function isLocal() { return localStorage.getItem('api') === 'local' }
 
+const API_HOST = isLocal() ? DEVELOPMENT_HOST : REMOTE_HOST
+
 export default options => {
   const { host, protocol } = window.location
 
   const Connection = function () {
-    const ws = new WebSocket(`${protocol === 'https:' ? 'wss' : 'ws'}://${isLocal() ? DEVELOPMENT_HOST : REMOTE_HOST}`)
+    const ws = new WebSocket(`${protocol === 'https:' ? 'wss' : 'ws'}://${API_HOST}`)
 
     this.send = message => ws.send(JSON.stringify(message))
-    this.close = () => ws.close()
+    this.close = info => {
+      this.send({ type: 'close', info })
+      ws.close()
+    }
 
     ws.onopen = () => this.onopen()
     ws.onmessage = ({ data }) => this.onmessage(data.length === 0 ? null : JSON.parse(data))
@@ -37,15 +42,12 @@ export default options => {
   })
 
   agent.local = () => {
-    if (isLocal()) return
-
     localStorage.setItem('api', 'local')
     location.reload()
   }
-  agent.remote = () => {
-    if (!isLocal()) return
-
-    localStorage.removeItem('api')
+  agent.remote = (mode='production') => {
+    localStorage.setItem('api', 'remote')
+    localStorage.setItem('mode', mode)
     location.reload()
   }
   agent.close = () => {

package/agents/generic/message-queue.js

@@ -236,7 +236,7 @@ export default function messageQueue({ token, Connection, watchers, states, appl
   function disconnect() {
     log('DISCONNECTED AGENT!!!!!!!!!!!!!!!')
     disconnected = true
-    connection.close()
+    connection.close({ keepalive: true })
   }
 
   function reconnect() {

package/package.json

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