@kinotic-ai/core 1.3.0 → 1.3.1

package/dist/index.cjs

@@ -210,6 +210,8 @@ class StompConnectionManager {
       this.serverHeadersSubscription = null;
       const url = "ws" + (connectionInfo.useSSL ? "s" : "") + "://" + connectionInfo.host + (connectionInfo.port ? ":" + connectionInfo.port : "") + "/v1";
       this.rxStomp = new import_rx_stomp.RxStomp;
+      let preparedSocket = null;
+      const userWebSocketFactory = connectionInfo.webSocketFactory;
       const stompConfig = {
         brokerURL: url,
         connectHeaders: {
@@ -218,7 +220,7 @@ class StompConnectionManager {
         heartbeatIncoming: 120000,
         heartbeatOutgoing: 30000,
         reconnectDelay: this.INITIAL_RECONNECT_DELAY,
-        webSocketFactory: connectionInfo.webSocketFactory,
+        webSocketFactory: userWebSocketFactory ? () => preparedSocket : undefined,
         beforeConnect: async () => {
           if (connectionInfo?.maxConnectionAttempts) {
             this.connectionAttempts++;
@@ -229,12 +231,23 @@ class StompConnectionManager {
                 let message = this.lastWebsocketError?.message ? this.lastWebsocketError?.message : "UNKNOWN";
                 reject(`Max number of reconnection attempts reached. Last WS Error ${message}`);
               }
+              return;
             } else {
               await this.connectionJitterDelay();
             }
           } else {
             await this.connectionJitterDelay();
           }
+          if (userWebSocketFactory) {
+            try {
+              preparedSocket = await userWebSocketFactory();
+            } catch (e) {
+              await this.deactivate();
+              if (!this.initialConnectionSuccessful) {
+                reject(e);
+              }
+            }
+          }
         }
       };
       if (this.debugLogger.enabled) {
@@ -989,7 +1002,7 @@ var import_operators2 = require("rxjs/operators");
 // packages/core/package.json
 var package_default = {
   name: "@kinotic-ai/core",
-  version: "1.3.0",
+  version: "1.3.1",
   type: "module",
   files: [
     "dist"

package/dist/index.d.cts

@@ -17,11 +17,12 @@ interface IWebSocket {
 /**
 * Factory invoked on every (re)connect to produce the WebSocket the STOMP
 * client will use. Supply this in Node when you need to set headers on the
-* upgrade request (for example, an Authorization header). Browser callers
-* normally leave this unset and rely on the session cookie established by a
-* prior REST login.
+* upgrade request (for example, an Authorization header). It may be async —
+* for example, to refresh a short-lived access token before each connect.
+* Browser callers normally leave this unset and rely on the session cookie
+* established by a prior REST login.
 */
-type WebSocketFactory = () => IWebSocket;
+type WebSocketFactory = () => IWebSocket | Promise<IWebSocket>;
 declare class ServerInfo {
 	host: string;
 	port?: number | null;

package/dist/index.d.ts

@@ -17,11 +17,12 @@ interface IWebSocket {
 /**
 * Factory invoked on every (re)connect to produce the WebSocket the STOMP
 * client will use. Supply this in Node when you need to set headers on the
-* upgrade request (for example, an Authorization header). Browser callers
-* normally leave this unset and rely on the session cookie established by a
-* prior REST login.
+* upgrade request (for example, an Authorization header). It may be async —
+* for example, to refresh a short-lived access token before each connect.
+* Browser callers normally leave this unset and rely on the session cookie
+* established by a prior REST login.
 */
-type WebSocketFactory = () => IWebSocket;
+type WebSocketFactory = () => IWebSocket | Promise<IWebSocket>;
 declare class ServerInfo {
 	host: string;
 	port?: number | null;

package/dist/index.js

@@ -108,6 +108,8 @@ class StompConnectionManager {
       this.serverHeadersSubscription = null;
       const url = "ws" + (connectionInfo.useSSL ? "s" : "") + "://" + connectionInfo.host + (connectionInfo.port ? ":" + connectionInfo.port : "") + "/v1";
       this.rxStomp = new RxStomp;
+      let preparedSocket = null;
+      const userWebSocketFactory = connectionInfo.webSocketFactory;
       const stompConfig = {
         brokerURL: url,
         connectHeaders: {
@@ -116,7 +118,7 @@ class StompConnectionManager {
         heartbeatIncoming: 120000,
         heartbeatOutgoing: 30000,
         reconnectDelay: this.INITIAL_RECONNECT_DELAY,
-        webSocketFactory: connectionInfo.webSocketFactory,
+        webSocketFactory: userWebSocketFactory ? () => preparedSocket : undefined,
         beforeConnect: async () => {
           if (connectionInfo?.maxConnectionAttempts) {
             this.connectionAttempts++;
@@ -127,12 +129,23 @@ class StompConnectionManager {
                 let message = this.lastWebsocketError?.message ? this.lastWebsocketError?.message : "UNKNOWN";
                 reject(`Max number of reconnection attempts reached. Last WS Error ${message}`);
               }
+              return;
             } else {
               await this.connectionJitterDelay();
             }
           } else {
             await this.connectionJitterDelay();
           }
+          if (userWebSocketFactory) {
+            try {
+              preparedSocket = await userWebSocketFactory();
+            } catch (e) {
+              await this.deactivate();
+              if (!this.initialConnectionSuccessful) {
+                reject(e);
+              }
+            }
+          }
         }
       };
       if (this.debugLogger.enabled) {
@@ -890,7 +903,7 @@ import { first, map as map2 } from "rxjs/operators";
 // packages/core/package.json
 var package_default = {
   name: "@kinotic-ai/core",
-  version: "1.3.0",
+  version: "1.3.1",
   type: "module",
   files: [
     "dist"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kinotic-ai/core",
-  "version": "1.3.0",
+  "version": "1.3.1",
   "type": "module",
   "files": [
     "dist"