@limrun/api 0.16.1 → 0.17.0-rc.2

package/src/instance-client.ts

@@ -1,7 +1,7 @@
 import { WebSocket, Data } from 'ws';
 import { exec } from 'node:child_process';
 
-import { startTcpTunnel, isNonRetryableError } from './tunnel';
+import { startTcpTunnel } from './tunnel';
 import type { Tunnel } from './tunnel';
 
 /**
@@ -156,7 +156,6 @@ export async function createInstanceClient(options: InstanceClientOptions): Prom
   let reconnectAttempts = 0;
   let reconnectTimeout: NodeJS.Timeout | undefined;
   let intentionalDisconnect = false;
-  let lastError: string | undefined;
 
   const screenshotRequests: Map<
     string,
@@ -176,19 +175,19 @@ export async function createInstanceClient(options: InstanceClientOptions): Prom
 
   const stateChangeCallbacks: Set<ConnectionStateCallback> = new Set();
 
+  // Logger functions
   const logger = {
     debug: (...args: any[]) => {
-      if (logLevel === 'debug') console.log('[Endpoint]', ...args);
+      if (logLevel === 'debug') console.log(...args);
     },
     info: (...args: any[]) => {
-      if (logLevel === 'info' || logLevel === 'debug') console.log('[Endpoint]', ...args);
+      if (logLevel === 'info' || logLevel === 'debug') console.log(...args);
     },
     warn: (...args: any[]) => {
-      if (logLevel === 'warn' || logLevel === 'info' || logLevel === 'debug')
-        console.warn('[Endpoint]', ...args);
+      if (logLevel === 'warn' || logLevel === 'info' || logLevel === 'debug') console.warn(...args);
     },
     error: (...args: any[]) => {
-      if (logLevel !== 'none') console.error('[Endpoint]', ...args);
+      if (logLevel !== 'none') console.error(...args);
     },
   };
 
@@ -243,16 +242,8 @@ export async function createInstanceClient(options: InstanceClientOptions): Prom
         return;
       }
 
-      if (isNonRetryableError(lastError ?? '')) {
-        logger.debug('Skipping reconnection (non-retryable error)');
-        return;
-      }
-
       if (reconnectAttempts >= maxReconnectAttempts) {
-        logger.error(
-          `Max reconnection attempts (${maxReconnectAttempts}) reached. Giving up.`,
-          lastError ? `Last error: ${lastError}` : '',
-        );
+        logger.error(`Max reconnection attempts (${maxReconnectAttempts}) reached. Giving up.`);
         updateConnectionState('disconnected');
         return;
       }
@@ -359,9 +350,7 @@ export async function createInstanceClient(options: InstanceClientOptions): Prom
       });
 
       ws.on('error', (err: Error) => {
-        const errMessage = err.message;
-        lastError = errMessage;
-        logger.debug('WebSocket error:', errMessage);
+        logger.error('WebSocket error:', err.message);
         if (!hasResolved && (ws?.readyState === WebSocket.CONNECTING || ws?.readyState === WebSocket.OPEN)) {
           rejectConnection(err);
         }
@@ -373,10 +362,7 @@ export async function createInstanceClient(options: InstanceClientOptions): Prom
           pingInterval = undefined;
         }
 
-        const shouldReconnect =
-          !intentionalDisconnect &&
-          !isNonRetryableError(lastError ?? '') &&
-          connectionState !== 'disconnected';
+        const shouldReconnect = !intentionalDisconnect && connectionState !== 'disconnected';
         updateConnectionState('disconnected');
 
         logger.debug('Disconnected from server.');
@@ -385,19 +371,12 @@ export async function createInstanceClient(options: InstanceClientOptions): Prom
 
         if (shouldReconnect) {
           scheduleReconnect();
-        } else if (isNonRetryableError(lastError ?? '')) {
-          logger.error(`Closing connection due to non-retryable error: ${lastError}`);
-          cleanup();
-          updateConnectionState('disconnected');
-          failPendingRequests('Non-retryable error');
-          logger.debug('Non-retryable error. Closing connection.');
         }
       });
 
       ws.on('open', () => {
         logger.debug(`Connected to ${serverAddress}`);
         reconnectAttempts = 0;
-        lastError = undefined;
         updateConnectionState('connected');
 
         pingInterval = setInterval(() => {