@wu529778790/open-im 1.4.0 → 1.4.1

package/dist/wework/client.js

@@ -20,6 +20,8 @@ let connectionState = 'disconnected';
 let reconnectTimer = null;
 let heartbeatTimer = null;
 let reconnectAttempts = 0;
+let shouldReconnect = false;
+let isStopping = false;
 // Event handlers
 let messageHandler = null;
 let stateChangeHandler = null;
@@ -109,6 +111,8 @@ export async function initWeWork(cfg, eventHandler, onStateChange) {
     };
     messageHandler = eventHandler;
     stateChangeHandler = onStateChange ?? null;
+    isStopping = false;
+    shouldReconnect = false;
     log.info(`Initializing WeWork client (botId: ${config.botId})`);
     // 首次连接支持重试：单独启用企微时偶发 TLS 连接失败，加飞书后因初始化顺序有“预热”效果则稳定
     const maxAttempts = 3;
@@ -116,7 +120,7 @@ export async function initWeWork(cfg, eventHandler, onStateChange) {
     let lastErr = null;
     for (let attempt = 1; attempt <= maxAttempts; attempt++) {
         try {
-            await connectWebSocket(true); // true = 初始连接，close 时不 scheduleReconnect
+            await connectWebSocket();
             return;
         }
         catch (err) {
@@ -131,9 +135,8 @@ export async function initWeWork(cfg, eventHandler, onStateChange) {
 }
 /**
  * Connect to WeWork WebSocket server
- * @param isInitialConnect - 初始连接时为 true，close 时不 scheduleReconnect（由 initWeWork 重试）
  */
-async function connectWebSocket(isInitialConnect = false) {
+async function connectWebSocket() {
     if (connectionState === 'connecting') {
         log.warn('WebSocket connection already in progress');
         return;
@@ -165,6 +168,7 @@ async function connectWebSocket(isInitialConnect = false) {
                 // 发送认证订阅消息，并等待服务端确认（否则 aibot_send_msg 会报 846609 not subscribed）
                 try {
                     await sendSubscribeAndWaitAck(resolve, reject);
+                    shouldReconnect = true;
                     log.info('WeWork authentication successful');
                 }
                 catch (err) {
@@ -190,7 +194,7 @@ async function connectWebSocket(isInitialConnect = false) {
                 log.info('WeWork WebSocket closed');
                 stopHeartbeat();
                 updateState('disconnected');
-                if (!isInitialConnect) {
+                if (!isStopping && shouldReconnect) {
                     scheduleReconnect();
                 }
             });
@@ -355,16 +359,23 @@ function stopHeartbeat() {
  * Schedule reconnection attempt
  */
 function scheduleReconnect() {
+    if (isStopping || !shouldReconnect) {
+        return;
+    }
     if (reconnectAttempts >= MAX_RECONNECT_ATTEMPTS) {
         log.error('Max reconnect attempts reached');
         return;
     }
     const interval = 5000; // 5秒后重连
+    if (reconnectTimer) {
+        return;
+    }
     reconnectTimer = setTimeout(async () => {
+        reconnectTimer = null;
         reconnectAttempts++;
         log.info(`Reconnecting... Attempt ${reconnectAttempts}/${MAX_RECONNECT_ATTEMPTS}`);
         try {
-            await connectWebSocket(false); // 非初始连接，close 时继续 scheduleReconnect
+            await connectWebSocket();
         }
         catch (err) {
             log.error('Reconnection failed:', err);
@@ -375,6 +386,8 @@ function scheduleReconnect() {
  * Stop WeWork client
  */
 export function stopWeWork() {
+    isStopping = true;
+    shouldReconnect = false;
     stopHeartbeat();
     if (reconnectTimer) {
         clearTimeout(reconnectTimer);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wu529778790/open-im",
-  "version": "1.4.0",
+  "version": "1.4.1",
   "description": "Multi-platform IM bridge for AI CLI tools (Claude, Codex, Cursor)",
   "type": "module",
   "main": "dist/index.js",