@tentou-tech/poly-websockets 1.0.0 → 1.0.1

package/dist/WSSubscriptionManager.d.ts

@@ -121,7 +121,7 @@ declare class WSSubscriptionManager {
     /**
      * Establish the WebSocket connection.
      */
-    private connect;
+    connect(): Promise<void>;
     /**
      * Sets up event handlers for the WebSocket connection.
      */

package/dist/WSSubscriptionManager.js

@@ -376,7 +376,8 @@ class WSSubscriptionManager {
         // (pendingUnsubscribeAssetIds is already cleared at this point)
         if (this.subscribedAssetIds.size === 0 &&
             this.pendingSubscribeAssetIds.size === 0 &&
-            this.pendingUnsubscribeAssetIds.size === 0) {
+            this.pendingUnsubscribeAssetIds.size === 0 &&
+            !this.enableCustomFeatures) {
             this.closeWebSocket();
         }
     }
@@ -408,8 +409,9 @@ class WSSubscriptionManager {
     async checkReconnection() {
         // If we have pending assets but no connection, reconnect
         const hasPendingAssets = this.pendingSubscribeAssetIds.size > 0;
+        const shouldReconnect = hasPendingAssets || this.enableCustomFeatures;
         const isDisconnected = !this.wsClient || this.wsClient.readyState !== ws_1.default.OPEN;
-        if (hasPendingAssets && isDisconnected && !this.connecting) {
+        if (shouldReconnect && isDisconnected && !this.connecting) {
             logger_1.logger.info({
                 message: 'Reconnection check - attempting to reconnect',
                 managerId: this.managerId,
@@ -432,7 +434,9 @@ class WSSubscriptionManager {
             return;
         }
         // No assets to subscribe to
-        if (this.pendingSubscribeAssetIds.size === 0 && this.subscribedAssetIds.size === 0) {
+        if (this.pendingSubscribeAssetIds.size === 0 &&
+            this.subscribedAssetIds.size === 0 &&
+            !this.enableCustomFeatures) {
             return;
         }
         this.connecting = true;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tentou-tech/poly-websockets",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "Plug-and-play Polymarket WebSocket price alerts",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/WSSubscriptionManager.ts

@@ -452,7 +452,8 @@ class WSSubscriptionManager {
         // (pendingUnsubscribeAssetIds is already cleared at this point)
         if (this.subscribedAssetIds.size === 0 && 
             this.pendingSubscribeAssetIds.size === 0 && 
-            this.pendingUnsubscribeAssetIds.size === 0) {
+            this.pendingUnsubscribeAssetIds.size === 0 &&
+            !this.enableCustomFeatures) {
             this.closeWebSocket();
         }
     }
@@ -489,9 +490,10 @@ class WSSubscriptionManager {
     private async checkReconnection() {
         // If we have pending assets but no connection, reconnect
         const hasPendingAssets = this.pendingSubscribeAssetIds.size > 0;
+        const shouldReconnect = hasPendingAssets || this.enableCustomFeatures;
         const isDisconnected = !this.wsClient || this.wsClient.readyState !== WebSocket.OPEN;
         
-        if (hasPendingAssets && isDisconnected && !this.connecting) {
+        if (shouldReconnect && isDisconnected && !this.connecting) {
             logger.info({
                 message: 'Reconnection check - attempting to reconnect',
                 managerId: this.managerId,
@@ -508,7 +510,7 @@ class WSSubscriptionManager {
     /**
      * Establish the WebSocket connection.
      */
-    private async connect(): Promise<void> {
+    public async connect(): Promise<void> {
         if (this.connecting) {
             return;
         }
@@ -517,7 +519,9 @@ class WSSubscriptionManager {
         }
 
         // No assets to subscribe to
-        if (this.pendingSubscribeAssetIds.size === 0 && this.subscribedAssetIds.size === 0) {
+        if (this.pendingSubscribeAssetIds.size === 0 && 
+            this.subscribedAssetIds.size === 0 && 
+            !this.enableCustomFeatures) {
             return;
         }