shell-mirror 1.5.106 → 1.5.108

package/mac-agent/agent.js

@@ -980,16 +980,41 @@ function setupDataChannel(clientId) {
         // Handle session closure request from client
         logToFile(`[AGENT] Client ${clientId} closing session ${message.sessionId}`);
 
-        sessionManager.terminateSession(message.sessionId);
+        // Get remaining sessions BEFORE termination
+        const closingSessionId = message.sessionId;
+        sessionManager.terminateSession(closingSessionId);
+
+        const remainingSessions = sessionManager.getAllSessions();
 
         // Send confirmation with updated session list
         dataChannel.send(JSON.stringify({
           type: 'session-closed',
-          sessionId: message.sessionId,
-          availableSessions: sessionManager.getAllSessions()
+          sessionId: closingSessionId,
+          availableSessions: remainingSessions
         }));
 
-        logToFile(`[AGENT] ✅ Session closed: ${message.sessionId}`);
+        // If client requested auto-switch to next session, do it atomically
+        if (message.switchToSessionId && remainingSessions.find(s => s.id === message.switchToSessionId)) {
+          logToFile(`[AGENT] Auto-switching client to session ${message.switchToSessionId}`);
+          if (sessionManager.connectClientToSession(clientId, message.switchToSessionId)) {
+            const newSession = sessionManager.getSession(message.switchToSessionId);
+            dataChannel.send(JSON.stringify({
+              type: 'session-switched',
+              sessionId: message.switchToSessionId,
+              sessionName: newSession.name
+            }));
+            // Send buffered output
+            const bufferedOutput = newSession.buffer.getAll();
+            if (bufferedOutput) {
+              sendLargeMessage(dataChannel, { type: 'output', data: bufferedOutput }, '[AGENT]');
+            }
+          }
+        }
+
+        // Send immediate heartbeat to update dashboard
+        sendHeartbeat();
+
+        logToFile(`[AGENT] ✅ Session closed: ${closingSessionId}`);
       }
     } catch (err) {
       logToFile(`[AGENT] Error parsing data channel message: ${err.message}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "shell-mirror",
-  "version": "1.5.106",
+  "version": "1.5.108",
   "description": "Access your Mac shell from any device securely. Perfect for mobile coding with Claude Code CLI, Gemini CLI, and any shell tool.",
   "main": "server.js",
   "bin": {

package/public/app/terminal.html

@@ -576,6 +576,6 @@
         }
     </script>
 
-    <script src="/app/terminal.js?v=1.5.90"></script>
+    <script src="/app/terminal.js?v=1.5.92"></script>
   </body>
 </html>

package/public/app/terminal.js

@@ -1080,10 +1080,16 @@ function closeSession(sessionId, event) {
 function doCloseSession(sessionId) {
     console.log('[CLIENT] 🗑️ Closing session:', sessionId);
 
-    // Send close request to agent
+    // Remove from available sessions
+    const remainingSessions = availableSessions.filter(s => s.id !== sessionId);
+    const isClosingCurrentSession = currentSession && currentSession.id === sessionId;
+    const nextSession = isClosingCurrentSession && remainingSessions.length > 0 ? remainingSessions[0] : null;
+
+    // Send close request to agent (with auto-switch if closing active session)
     const closeMessage = {
         type: 'close_session',
-        sessionId: sessionId
+        sessionId: sessionId,
+        switchToSessionId: nextSession ? nextSession.id : null  // Atomic close + switch
     };
 
     if (dataChannel && dataChannel.readyState === 'open') {
@@ -1092,16 +1098,25 @@ function doCloseSession(sessionId) {
         ws.send(JSON.stringify(closeMessage));
     }
 
-    // Remove from available sessions
-    availableSessions = availableSessions.filter(s => s.id !== sessionId);
+    // Update local state
+    availableSessions = remainingSessions;
 
-    // If closing current session, switch to another or show message
-    if (currentSession && currentSession.id === sessionId) {
-        if (availableSessions.length > 0) {
-            switchToSession(availableSessions[0].id);
+    // If closing current session, update UI immediately
+    if (isClosingCurrentSession) {
+        // Clear terminal IMMEDIATELY to prevent garbage from closed session
+        term.clear();
+
+        if (nextSession) {
+            // Update currentSession IMMEDIATELY so renderTabs shows correct active tab
+            currentSession = {
+                id: nextSession.id,
+                name: nextSession.name || 'Terminal Session'
+            };
+            // Update URL
+            updateUrlWithSession(nextSession.id);
+            // Note: Agent will send session-switched with buffered output
         } else {
             currentSession = null;
-            term.clear();
             term.write('\r\n\x1b[33mSession closed. Click + to create a new session.\x1b[0m\r\n');
         }
     }
@@ -1249,14 +1264,28 @@ function handleSessionMessage(message) {
             term.focus();
             break;
         case 'session-ended':
-            term.write(`\r\n\x1b[31m❌ Session ended: ${message.reason}\x1b[0m\r\n`);
-            if (message.code) {
-                term.write(`Exit code: ${message.code}\r\n`);
+            // Only show if this is for the current session (ignore closed session messages)
+            if (!message.sessionId || (currentSession && message.sessionId === currentSession.id)) {
+                term.write(`\r\n\x1b[31m❌ Session ended: ${message.reason}\x1b[0m\r\n`);
+                if (message.code) {
+                    term.write(`Exit code: ${message.code}\r\n`);
+                }
             }
             break;
         case 'session-terminated':
-            term.write(`\r\n\x1b[31m❌ Session terminated\x1b[0m\r\n`);
-            term.write('🔄 Click Dashboard to start a new session\r\n');
+            // Only show if this is for the current session (ignore closed session messages)
+            if (!message.sessionId || (currentSession && message.sessionId === currentSession.id)) {
+                term.write(`\r\n\x1b[31m❌ Session terminated\x1b[0m\r\n`);
+                term.write('🔄 Click Dashboard to start a new session\r\n');
+            }
+            break;
+        case 'session-closed':
+            console.log('[CLIENT] ✅ Session closed confirmed:', message.sessionId);
+            // Update available sessions from server response
+            if (message.availableSessions) {
+                availableSessions = message.availableSessions;
+                renderTabs();
+            }
             break;
         case 'error':
             term.write(`\r\n\x1b[31m❌ Error: ${message.message}\x1b[0m\r\n`);