fix: re-register environment instead of crashing (#1235)

Co-authored-by: Letta <noreply@letta.com>

src/cli/subcommands/listen.tsx

@@ -187,6 +187,22 @@ export async function runListenSubcommand(argv: string[]): Promise<number> {
       "../../websocket/listen-client"
     );
 
+    // Re-register helper for when the server closes with 1008 (environment not found)
+    const reregister = async (): Promise<{
+      connectionId: string;
+      wsUrl: string;
+    }> => {
+      sessionLog.log("Environment expired, re-registering...");
+      const result = await registerWithCloud({
+        serverUrl,
+        apiKey,
+        deviceId,
+        connectionName,
+      });
+      sessionLog.log(`Re-registered: connectionId=${result.connectionId}`);
+      return result;
+    };
+
     // WS event logger: always writes to file, console only in --debug
     const wsEventLogger = (
       direction: "send" | "recv",
@@ -204,9 +220,13 @@ export async function runListenSubcommand(argv: string[]): Promise<number> {
 
     if (debugMode) {
       // Debug mode: plain-text event logging, no Ink UI
+      const startDebugClient = async (
+        connId: string,
+        url: string,
+      ): Promise<void> => {
         await startListenerClient({
-        connectionId,
-        wsUrl,
+          connectionId: connId,
+          wsUrl: url,
           deviceId,
           connectionName,
           onWsEvent: wsEventLogger,
@@ -229,6 +249,23 @@ export async function runListenSubcommand(argv: string[]): Promise<number> {
               `[${formatTimestamp()}] Reconnecting (attempt ${attempt}, retry in ${Math.round(nextRetryIn / 1000)}s)`,
             );
           },
+          onNeedsReregister: async () => {
+            console.log(
+              `[${formatTimestamp()}] Environment expired, re-registering...`,
+            );
+            try {
+              const result = await reregister();
+              await startDebugClient(result.connectionId, result.wsUrl);
+            } catch (error) {
+              const msg =
+                error instanceof Error ? error.message : String(error);
+              sessionLog.log(`Re-registration failed: ${msg}`);
+              console.error(
+                `[${formatTimestamp()}] Re-registration failed: ${msg}`,
+              );
+              process.exit(1);
+            }
+          },
           onDisconnected: () => {
             sessionLog.log("Disconnected.");
             console.log(`[${formatTimestamp()}] Disconnected.`);
@@ -240,6 +277,8 @@ export async function runListenSubcommand(argv: string[]): Promise<number> {
             process.exit(1);
           },
         });
+      };
+      await startDebugClient(connectionId, wsUrl);
     } else {
       // Normal mode: interactive Ink UI
       console.clear();
@@ -264,9 +303,13 @@ export async function runListenSubcommand(argv: string[]): Promise<number> {
         />,
       );
 
+      const startNormalClient = async (
+        connId: string,
+        url: string,
+      ): Promise<void> => {
         await startListenerClient({
-        connectionId,
-        wsUrl,
+          connectionId: connId,
+          wsUrl: url,
           deviceId,
           connectionName,
           onWsEvent: wsEventLogger,
@@ -286,6 +329,20 @@ export async function runListenSubcommand(argv: string[]): Promise<number> {
             );
             updateRetryStatusCallback?.(attempt, nextRetryIn);
           },
+          onNeedsReregister: async () => {
+            sessionLog.log("Environment expired, re-registering...");
+            try {
+              const result = await reregister();
+              await startNormalClient(result.connectionId, result.wsUrl);
+            } catch (error) {
+              const msg =
+                error instanceof Error ? error.message : String(error);
+              sessionLog.log(`Re-registration failed: ${msg}`);
+              unmount();
+              console.error(`\n\u2717 Re-registration failed: ${msg}\n`);
+              process.exit(1);
+            }
+          },
           onDisconnected: () => {
             sessionLog.log("Disconnected.");
             unmount();
@@ -300,6 +357,8 @@ export async function runListenSubcommand(argv: string[]): Promise<number> {
             process.exit(1);
           },
         });
+      };
+      await startNormalClient(connectionId, wsUrl);
     }
 
     // Keep process alive