@hautechai/sdk 0.3.6 → 0.3.8

package/dist/sdk/index.d.ts

@@ -1218,7 +1218,7 @@ export declare const createSDK: (options: SDKOptions) => {
             metadata?: PoseMetadata;
         }) => Promise<void>;
     };
-    close: () => void;
+    close: () => Promise<void>;
 };
 export type SDK = ReturnType<typeof createSDK>;
 export interface CollectionMetadata {

package/dist/sdk/index.js

@@ -41,6 +41,7 @@ export const createSDK = (options) => {
             : null,
         // TODO: Refactor the API initialization
         operations: () => getAPI(OperationsApi, optionsWithTokenRefresher),
+        allowPollingFallback: options.allowPollingFallback ?? true,
     });
     operationsListener.subscribe();
     return {
@@ -56,8 +57,8 @@ export const createSDK = (options) => {
         storage: storage(optionsWithTokenRefresher),
         utils: utils(optionsWithTokenRefresher),
         poses: poses(optionsWithTokenRefresher),
-        close: () => {
-            operationsListener.close();
+        close: async () => {
+            await operationsListener.close();
         },
     };
 };

package/dist/sdk/listeners/index.d.ts

@@ -7,12 +7,14 @@ export declare class OperationsListener {
     } | null;
     ws: WebSocketClient | null;
     operations: () => Promise<OperationsApi>;
-    constructor({ ws, operations, }: {
+    allowPollingFallback: boolean;
+    constructor({ ws, operations, allowPollingFallback, }: {
         ws: {
             endpoint: string;
             token: () => string | Promise<string>;
         } | null;
         operations: () => Promise<OperationsApi>;
+        allowPollingFallback: boolean;
     });
     operationsStore: Record<string, OperationEntity>;
     getOperation(id: string): Promise<OperationEntity | null>;
@@ -21,5 +23,5 @@ export declare class OperationsListener {
     onOpen(): void;
     onMessage(msg: string): void;
     onClose(number: number, reason: string): void;
-    close(): void;
+    close(): Promise<void>;
 }

package/dist/sdk/listeners/index.js

@@ -2,10 +2,12 @@ import { w3cwebsocket as WebSocketClient } from 'websocket';
 import { HautechError } from '../errors';
 // This is pretty much a dirty solution until we need to rework this part.
 export class OperationsListener {
-    constructor({ ws, operations, }) {
+    constructor({ ws, operations, allowPollingFallback = false, }) {
         this.useWebsocket = null;
         this.ws = null;
         this.operationsStore = {};
+        if (!ws)
+            allowPollingFallback = true;
         if (ws) {
             this.useWebsocket = {
                 endpoint: ws?.endpoint,
@@ -13,10 +15,11 @@ export class OperationsListener {
             };
         }
         this.operations = operations;
+        this.allowPollingFallback = allowPollingFallback;
     }
     async getOperation(id) {
-        const isWsReady = this.ws?.readyState === WebSocket.OPEN;
-        if (!this.operationsStore[id] || !isWsReady) {
+        const fallbackToPolling = this.allowPollingFallback && !(this.ws?.readyState === WebSocketClient.OPEN);
+        if (!this.operationsStore[id] || fallbackToPolling) {
             const api = await this.operations();
             const operation = await api.operationsControllerGetOperationV1(id);
             if (operation.status == 200)
@@ -34,7 +37,8 @@ export class OperationsListener {
         if (!this.useWebsocket)
             return;
         try {
-            this.ws = new WebSocketClient(this.useWebsocket.endpoint, ['1', await this.useWebsocket.token()]);
+            const token = await this.useWebsocket.token();
+            this.ws = new WebSocketClient(this.useWebsocket.endpoint, ['1', token]);
             this.ws.onopen = () => {
                 this.onOpen();
             };
@@ -90,7 +94,8 @@ export class OperationsListener {
         this.operationsStore = {};
         this.ws = null;
     }
-    close() {
+    async close() {
         this.ws?.close();
+        await new Promise((resolve) => setTimeout(resolve, 1000));
     }
 }

package/dist/types.d.ts

@@ -2,6 +2,7 @@ export interface SDKOptions {
     authToken: () => string | Promise<string>;
     endpoint?: string;
     useWebsocket?: boolean;
+    allowPollingFallback?: boolean;
 }
 export type ListProps = {
     cursor?: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hautechai/sdk",
-  "version": "0.3.6",
+  "version": "0.3.8",
   "license": "MIT",
   "keywords": [],
   "repository": {