mstro-app 0.2.0 → 0.3.0

package/dist/server/services/websocket/settings-handlers.js

@@ -0,0 +1,125 @@
+// Copyright (c) 2025-present Mstro, Inc. All rights reserved.
+// Licensed under the MIT License. See LICENSE file for details.
+import { spawn } from 'node:child_process';
+import { existsSync, mkdirSync, unlinkSync, writeFileSync } from 'node:fs';
+import { join } from 'node:path';
+import { getSettings, setModel } from '../settings.js';
+export function handleGetSettings(ctx, ws) {
+    ctx.send(ws, { type: 'settings', data: getSettings() });
+}
+export function handleUpdateSettings(ctx, _ws, msg) {
+    if (msg.data?.model !== undefined) {
+        setModel(msg.data.model);
+    }
+    ctx.broadcastToAll({ type: 'settingsUpdated', data: getSettings() });
+}
+export async function generateNotificationSummary(ctx, ws, tabId, userPrompt, output, workingDir) {
+    try {
+        const tempDir = join(workingDir, '.mstro', 'tmp');
+        if (!existsSync(tempDir)) {
+            mkdirSync(tempDir, { recursive: true });
+        }
+        let truncatedOutput = output;
+        if (output.length > 4000) {
+            const firstPart = output.slice(0, 2000);
+            const lastPart = output.slice(-1500);
+            truncatedOutput = `${firstPart}\n\n... [output truncated] ...\n\n${lastPart}`;
+        }
+        const summaryPrompt = `You are generating a SHORT browser notification summary for a completed task.
+The user ran a task and wants a brief notification to remind them what happened.
+
+USER'S ORIGINAL PROMPT:
+"${userPrompt}"
+
+TASK OUTPUT (may be truncated):
+${truncatedOutput}
+
+Generate a notification summary following these rules:
+1. Maximum 100 characters (this is a browser notification)
+2. Focus on the OUTCOME, not the process
+3. Be specific about what was accomplished
+4. Use past tense (e.g., "Fixed bug in auth.ts", "Added 3 new tests")
+5. If there was an error, mention it briefly
+6. No emojis, no markdown, just plain text
+
+Respond with ONLY the summary text, nothing else.`;
+        const promptFile = join(tempDir, `notif-summary-${Date.now()}.txt`);
+        writeFileSync(promptFile, summaryPrompt);
+        const systemPrompt = 'You are a notification summary assistant. Respond with only the summary text, no preamble or explanation.';
+        const args = [
+            '--print',
+            '--model', 'haiku',
+            '--system-prompt', systemPrompt,
+            promptFile
+        ];
+        const claude = spawn('claude', args, {
+            cwd: workingDir,
+            stdio: ['ignore', 'pipe', 'pipe']
+        });
+        let stdout = '';
+        let stderr = '';
+        claude.stdout?.on('data', (data) => {
+            stdout += data.toString();
+        });
+        claude.stderr?.on('data', (data) => {
+            stderr += data.toString();
+        });
+        claude.on('close', (code) => {
+            try {
+                unlinkSync(promptFile);
+            }
+            catch {
+                // Ignore cleanup errors
+            }
+            let summary;
+            if (code === 0 && stdout.trim()) {
+                summary = stdout.trim().slice(0, 150);
+            }
+            else {
+                console.error('[WebSocketImproviseHandler] Claude error:', stderr || 'Unknown error');
+                summary = createFallbackSummary(userPrompt);
+            }
+            ctx.send(ws, {
+                type: 'notificationSummary',
+                tabId,
+                data: { summary }
+            });
+        });
+        claude.on('error', (err) => {
+            console.error('[WebSocketImproviseHandler] Failed to spawn Claude:', err);
+            const summary = createFallbackSummary(userPrompt);
+            ctx.send(ws, {
+                type: 'notificationSummary',
+                tabId,
+                data: { summary }
+            });
+        });
+        // Timeout after 10 seconds
+        setTimeout(() => {
+            claude.kill();
+            const summary = createFallbackSummary(userPrompt);
+            ctx.send(ws, {
+                type: 'notificationSummary',
+                tabId,
+                data: { summary }
+            });
+        }, 10000);
+    }
+    catch (error) {
+        console.error('[WebSocketImproviseHandler] Error generating summary:', error);
+        const summary = createFallbackSummary(userPrompt);
+        ctx.send(ws, {
+            type: 'notificationSummary',
+            tabId,
+            data: { summary }
+        });
+    }
+}
+function createFallbackSummary(userPrompt) {
+    const truncated = userPrompt.slice(0, 60);
+    if (userPrompt.length > 60) {
+        return `Completed: "${truncated}..."`;
+    }
+    return `Completed: "${truncated}"`;
+}
+//# sourceMappingURL=settings-handlers.js.map

package/dist/server/services/websocket/settings-handlers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"settings-handlers.js","sourceRoot":"","sources":["../../../../server/services/websocket/settings-handlers.ts"],"names":[],"mappings":"AAAA,8DAA8D;AAC9D,gEAAgE;AAEhE,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC3E,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAIvD,MAAM,UAAU,iBAAiB,CAAC,GAAmB,EAAE,EAAa;IAClE,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;AAC1D,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,GAAmB,EAAE,GAAc,EAAE,GAAqB;IAC7F,IAAI,GAAG,CAAC,IAAI,EAAE,KAAK,KAAK,SAAS,EAAE,CAAC;QAClC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IACD,GAAG,CAAC,cAAc,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;AACvE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAC/C,GAAmB,EACnB,EAAa,EACb,KAAa,EACb,UAAkB,EAClB,MAAc,EACd,UAAkB;IAElB,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;QAClD,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACzB,SAAS,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,CAAC;QAED,IAAI,eAAe,GAAG,MAAM,CAAC;QAC7B,IAAI,MAAM,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;YACxC,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC;YACrC,eAAe,GAAG,GAAG,SAAS,qCAAqC,QAAQ,EAAE,CAAC;QAChF,CAAC;QAED,MAAM,aAAa,GAAG;;;;GAIvB,UAAU;;;EAGX,eAAe;;;;;;;;;;kDAUiC,CAAC;QAE/C,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,iBAAiB,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACpE,aAAa,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;QAEzC,MAAM,YAAY,GAAG,2GAA2G,CAAC;QAEjI,MAAM,IAAI,GAAG;YACX,SAAS;YACT,SAAS,EAAE,OAAO;YAClB,iBAAiB,EAAE,YAAY;YAC/B,UAAU;SACX,CAAC;QAEF,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE;YACnC,GAAG,EAAE,UAAU;YACf,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;SAClC,CAAC,CAAC;QAEH,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;YACzC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;YACzC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAmB,EAAE,EAAE;YACzC,IAAI,CAAC;gBACH,UAAU,CAAC,UAAU,CAAC,CAAC;YACzB,CAAC;YAAC,MAAM,CAAC;gBACP,wBAAwB;YAC1B,CAAC;YAED,IAAI,OAAe,CAAC;YACpB,IAAI,IAAI,KAAK,CAAC,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBAChC,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YACxC,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,2CAA2C,EAAE,MAAM,IAAI,eAAe,CAAC,CAAC;gBACtF,OAAO,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;YAC9C,CAAC;YAED,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;gBACX,IAAI,EAAE,qBAAqB;gBAC3B,KAAK;gBACL,IAAI,EAAE,EAAE,OAAO,EAAE;aAClB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;YAChC,OAAO,CAAC,KAAK,CAAC,qDAAqD,EAAE,GAAG,CAAC,CAAC;YAC1E,MAAM,OAAO,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;YAClD,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;gBACX,IAAI,EAAE,qBAAqB;gBAC3B,KAAK;gBACL,IAAI,EAAE,EAAE,OAAO,EAAE;aAClB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,2BAA2B;QAC3B,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,OAAO,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;YAClD,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;gBACX,IAAI,EAAE,qBAAqB;gBAC3B,KAAK;gBACL,IAAI,EAAE,EAAE,OAAO,EAAE;aAClB,CAAC,CAAC;QACL,CAAC,EAAE,KAAK,CAAC,CAAC;IAEZ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,uDAAuD,EAAE,KAAK,CAAC,CAAC;QAC9E,MAAM,OAAO,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAClD,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;YACX,IAAI,EAAE,qBAAqB;YAC3B,KAAK;YACL,IAAI,EAAE,EAAE,OAAO,EAAE;SAClB,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAAC,UAAkB;IAC/C,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC1C,IAAI,UAAU,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;QAC3B,OAAO,eAAe,SAAS,MAAM,CAAC;IACxC,CAAC;IACD,OAAO,eAAe,SAAS,GAAG,CAAC;AACrC,CAAC"}

package/dist/server/services/websocket/tab-handlers.d.ts

@@ -0,0 +1,10 @@
+import type { HandlerContext } from './handler-context.js';
+import type { WebSocketMessage, WSContext } from './types.js';
+export declare function handleGetActiveTabs(ctx: HandlerContext, ws: WSContext, workingDir: string): void;
+export declare function handleSyncTabMeta(ctx: HandlerContext, _ws: WSContext, msg: WebSocketMessage, tabId: string, workingDir: string): void;
+export declare function handleSyncPromptText(ctx: HandlerContext, _ws: WSContext, msg: WebSocketMessage, tabId: string): void;
+export declare function handleRemoveTab(ctx: HandlerContext, _ws: WSContext, tabId: string, workingDir: string): void;
+export declare function handleMarkTabViewed(ctx: HandlerContext, _ws: WSContext, tabId: string, workingDir: string): void;
+export declare function handleCreateTab(ctx: HandlerContext, ws: WSContext, workingDir: string, tabName?: string, optimisticTabId?: string): Promise<void>;
+export declare function handleReorderTabs(ctx: HandlerContext, _ws: WSContext, workingDir: string, tabOrder?: string[]): void;
+//# sourceMappingURL=tab-handlers.d.ts.map

package/dist/server/services/websocket/tab-handlers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tab-handlers.d.ts","sourceRoot":"","sources":["../../../../server/services/websocket/tab-handlers.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAE3D,OAAO,KAAK,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAE9D,wBAAgB,mBAAmB,CAAC,GAAG,EAAE,cAAc,EAAE,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI,CAiChG;AAED,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,cAAc,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI,CASrI;AAED,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,cAAc,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,CAOpH;AAED,wBAAgB,eAAe,CAAC,GAAG,EAAE,cAAc,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI,CAQ5G;AAED,wBAAgB,mBAAmB,CAAC,GAAG,EAAE,cAAc,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI,CAQhH;AAED,wBAAsB,eAAe,CAAC,GAAG,EAAE,cAAc,EAAE,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAiDvJ;AAED,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,cAAc,EAAE,GAAG,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI,CAcpH"}

package/dist/server/services/websocket/tab-handlers.js

@@ -0,0 +1,131 @@
+// Copyright (c) 2025-present Mstro, Inc. All rights reserved.
+// Licensed under the MIT License. See LICENSE file for details.
+import { ImprovisationSessionManager } from '../../cli/improvisation-session-manager.js';
+import { getModel } from '../settings.js';
+import { buildOutputHistory, setupSessionListeners } from './session-handlers.js';
+export function handleGetActiveTabs(ctx, ws, workingDir) {
+    const registry = ctx.getRegistry(workingDir);
+    const allTabs = registry.getAllTabs();
+    const tabs = {};
+    for (const [tabId, regTab] of Object.entries(allTabs)) {
+        const session = ctx.sessions.get(regTab.sessionId);
+        if (session) {
+            tabs[tabId] = {
+                tabName: regTab.tabName,
+                createdAt: regTab.createdAt,
+                order: regTab.order,
+                hasUnviewedCompletion: regTab.hasUnviewedCompletion,
+                sessionInfo: session.getSessionInfo(),
+                isExecuting: session.isExecuting,
+                outputHistory: buildOutputHistory(session),
+                executionEvents: session.isExecuting ? session.getExecutionEventLog() : undefined,
+                ...(session.isExecuting && session.executionStartTimestamp ? { executionStartTimestamp: session.executionStartTimestamp } : {}),
+            };
+        }
+        else {
+            tabs[tabId] = {
+                tabName: regTab.tabName,
+                createdAt: regTab.createdAt,
+                order: regTab.order,
+                hasUnviewedCompletion: regTab.hasUnviewedCompletion,
+                sessionId: regTab.sessionId,
+                isExecuting: false,
+                outputHistory: [],
+            };
+        }
+    }
+    ctx.send(ws, { type: 'activeTabs', data: { tabs } });
+}
+export function handleSyncTabMeta(ctx, _ws, msg, tabId, workingDir) {
+    const registry = ctx.getRegistry(workingDir);
+    if (msg.data?.tabName) {
+        registry.updateTabName(tabId, msg.data.tabName);
+        ctx.broadcastToAll({
+            type: 'tabRenamed',
+            data: { tabId, tabName: msg.data.tabName }
+        });
+    }
+}
+export function handleSyncPromptText(ctx, _ws, msg, tabId) {
+    if (typeof msg.data?.text !== 'string')
+        return;
+    ctx.broadcastToAll({
+        type: 'promptTextSync',
+        tabId,
+        data: { tabId, text: msg.data.text }
+    });
+}
+export function handleRemoveTab(ctx, _ws, tabId, workingDir) {
+    const registry = ctx.getRegistry(workingDir);
+    registry.unregisterTab(tabId);
+    ctx.broadcastToAll({
+        type: 'tabRemoved',
+        data: { tabId }
+    });
+}
+export function handleMarkTabViewed(ctx, _ws, tabId, workingDir) {
+    const registry = ctx.getRegistry(workingDir);
+    registry.markTabViewed(tabId);
+    ctx.broadcastToAll({
+        type: 'tabViewed',
+        data: { tabId }
+    });
+}
+export async function handleCreateTab(ctx, ws, workingDir, tabName, optimisticTabId) {
+    const registry = ctx.getRegistry(workingDir);
+    const tabId = optimisticTabId || `tab-${Date.now()}-${Math.random().toString(36).slice(2, 6)}`;
+    const existingSession = registry.getTabSession(tabId);
+    if (existingSession) {
+        const regTab = registry.getTab(tabId);
+        ctx.broadcastToAll({
+            type: 'tabCreated',
+            data: {
+                tabId,
+                tabName: regTab?.tabName || 'Chat',
+                createdAt: regTab?.createdAt,
+                order: regTab?.order,
+                sessionInfo: ctx.sessions.get(existingSession)?.getSessionInfo(),
+            }
+        });
+        return;
+    }
+    const session = new ImprovisationSessionManager({ workingDir, model: getModel() });
+    setupSessionListeners(ctx, session, ws, tabId);
+    const sessionId = session.getSessionInfo().sessionId;
+    ctx.sessions.set(sessionId, session);
+    const tabMap = ctx.connections.get(ws);
+    if (tabMap)
+        tabMap.set(tabId, sessionId);
+    registry.registerTab(tabId, sessionId, tabName);
+    const registeredTab = registry.getTab(tabId);
+    ctx.broadcastToAll({
+        type: 'tabCreated',
+        data: {
+            tabId,
+            tabName: registeredTab?.tabName || 'Chat',
+            createdAt: registeredTab?.createdAt,
+            order: registeredTab?.order,
+            sessionInfo: session.getSessionInfo(),
+        }
+    });
+    ctx.send(ws, {
+        type: 'tabInitialized',
+        tabId,
+        data: session.getSessionInfo()
+    });
+}
+export function handleReorderTabs(ctx, _ws, workingDir, tabOrder) {
+    if (!Array.isArray(tabOrder))
+        return;
+    const registry = ctx.getRegistry(workingDir);
+    registry.reorderTabs(tabOrder);
+    const allTabs = registry.getAllTabs();
+    const orderMap = tabOrder
+        .filter((id) => allTabs[id])
+        .map((id) => ({ tabId: id, order: allTabs[id].order }));
+    ctx.broadcastToAll({
+        type: 'tabsReordered',
+        data: { tabOrder: orderMap }
+    });
+}
+//# sourceMappingURL=tab-handlers.js.map

package/dist/server/services/websocket/tab-handlers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tab-handlers.js","sourceRoot":"","sources":["../../../../server/services/websocket/tab-handlers.ts"],"names":[],"mappings":"AAAA,8DAA8D;AAC9D,gEAAgE;AAEhE,OAAO,EAAE,2BAA2B,EAAE,MAAM,4CAA4C,CAAC;AACzF,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,OAAO,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAGlF,MAAM,UAAU,mBAAmB,CAAC,GAAmB,EAAE,EAAa,EAAE,UAAkB;IACxF,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAC7C,MAAM,OAAO,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC;IAEtC,MAAM,IAAI,GAAwB,EAAE,CAAC;IACrC,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACtD,MAAM,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACnD,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,KAAK,CAAC,GAAG;gBACZ,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,qBAAqB,EAAE,MAAM,CAAC,qBAAqB;gBACnD,WAAW,EAAE,OAAO,CAAC,cAAc,EAAE;gBACrC,WAAW,EAAE,OAAO,CAAC,WAAW;gBAChC,aAAa,EAAE,kBAAkB,CAAC,OAAO,CAAC;gBAC1C,eAAe,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC,CAAC,CAAC,SAAS;gBACjF,GAAG,CAAC,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,uBAAuB,CAAC,CAAC,CAAC,EAAE,uBAAuB,EAAE,OAAO,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAChI,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,GAAG;gBACZ,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,qBAAqB,EAAE,MAAM,CAAC,qBAAqB;gBACnD,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,WAAW,EAAE,KAAK;gBAClB,aAAa,EAAE,EAAE;aAClB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;AACvD,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,GAAmB,EAAE,GAAc,EAAE,GAAqB,EAAE,KAAa,EAAE,UAAkB;IAC7H,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAC7C,IAAI,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;QACtB,QAAQ,CAAC,aAAa,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChD,GAAG,CAAC,cAAc,CAAC;YACjB,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE;SAC3C,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,GAAmB,EAAE,GAAc,EAAE,GAAqB,EAAE,KAAa;IAC5G,IAAI,OAAO,GAAG,CAAC,IAAI,EAAE,IAAI,KAAK,QAAQ;QAAE,OAAO;IAC/C,GAAG,CAAC,cAAc,CAAC;QACjB,IAAI,EAAE,gBAAgB;QACtB,KAAK;QACL,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE;KACrC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,GAAmB,EAAE,GAAc,EAAE,KAAa,EAAE,UAAkB;IACpG,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAC7C,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;IAE9B,GAAG,CAAC,cAAc,CAAC;QACjB,IAAI,EAAE,YAAY;QAClB,IAAI,EAAE,EAAE,KAAK,EAAE;KAChB,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,GAAmB,EAAE,GAAc,EAAE,KAAa,EAAE,UAAkB;IACxG,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAC7C,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;IAE9B,GAAG,CAAC,cAAc,CAAC;QACjB,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,EAAE,KAAK,EAAE;KAChB,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,GAAmB,EAAE,EAAa,EAAE,UAAkB,EAAE,OAAgB,EAAE,eAAwB;IACtI,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAE7C,MAAM,KAAK,GAAG,eAAe,IAAI,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IAE/F,MAAM,eAAe,GAAG,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;IACtD,IAAI,eAAe,EAAE,CAAC;QACpB,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACtC,GAAG,CAAC,cAAc,CAAC;YACjB,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE;gBACJ,KAAK;gBACL,OAAO,EAAE,MAAM,EAAE,OAAO,IAAI,MAAM;gBAClC,SAAS,EAAE,MAAM,EAAE,SAAS;gBAC5B,KAAK,EAAE,MAAM,EAAE,KAAK;gBACpB,WAAW,EAAE,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,cAAc,EAAE;aACjE;SACF,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,2BAA2B,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;IACnF,qBAAqB,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;IAE/C,MAAM,SAAS,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,SAAS,CAAC;IACrD,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAErC,MAAM,MAAM,GAAG,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACvC,IAAI,MAAM;QAAE,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAEzC,QAAQ,CAAC,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAChD,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAE7C,GAAG,CAAC,cAAc,CAAC;QACjB,IAAI,EAAE,YAAY;QAClB,IAAI,EAAE;YACJ,KAAK;YACL,OAAO,EAAE,aAAa,EAAE,OAAO,IAAI,MAAM;YACzC,SAAS,EAAE,aAAa,EAAE,SAAS;YACnC,KAAK,EAAE,aAAa,EAAE,KAAK;YAC3B,WAAW,EAAE,OAAO,CAAC,cAAc,EAAE;SACtC;KACF,CAAC,CAAC;IAEH,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;QACX,IAAI,EAAE,gBAAgB;QACtB,KAAK;QACL,IAAI,EAAE,OAAO,CAAC,cAAc,EAAE;KAC/B,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,GAAmB,EAAE,GAAc,EAAE,UAAkB,EAAE,QAAmB;IAC5G,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;QAAE,OAAO;IACrC,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAC7C,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IAE/B,MAAM,OAAO,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC;IACtC,MAAM,QAAQ,GAAG,QAAQ;SACtB,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;SAC3B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAE1D,GAAG,CAAC,cAAc,CAAC;QACjB,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE;KAC7B,CAAC,CAAC;AACL,CAAC"}

package/dist/server/services/websocket/terminal-handlers.d.ts

@@ -0,0 +1,9 @@
+import type { HandlerContext } from './handler-context.js';
+import type { WebSocketMessage, WSContext } from './types.js';
+export declare function handleTerminalMessage(ctx: HandlerContext, ws: WSContext, msg: WebSocketMessage, tabId: string, workingDir: string, permission?: 'control' | 'view'): void;
+/**
+ * Clean up terminal subscribers for a disconnected WS context.
+ * Called from handler.ts handleClose().
+ */
+export declare function cleanupTerminalSubscribers(ctx: HandlerContext, ws: WSContext): void;
+//# sourceMappingURL=terminal-handlers.d.ts.map

package/dist/server/services/websocket/terminal-handlers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"terminal-handlers.d.ts","sourceRoot":"","sources":["../../../../server/services/websocket/terminal-handlers.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,KAAK,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAE9D,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,cAAc,EAAE,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,SAAS,GAAG,MAAM,GAAG,IAAI,CAsBzK;AA8OD;;;GAGG;AACH,wBAAgB,0BAA0B,CAAC,GAAG,EAAE,cAAc,EAAE,EAAE,EAAE,SAAS,GAAG,IAAI,CAInF"}

package/dist/server/services/websocket/terminal-handlers.js

@@ -0,0 +1,220 @@
+// Copyright (c) 2025-present Mstro, Inc. All rights reserved.
+// Licensed under the MIT License. See LICENSE file for details.
+import { AnalyticsEvents, trackEvent } from '../analytics.js';
+import { getPTYManager } from '../terminal/pty-manager.js';
+export function handleTerminalMessage(ctx, ws, msg, tabId, workingDir, permission) {
+    const termId = msg.terminalId || tabId;
+    switch (msg.type) {
+        case 'terminalInit':
+            handleTerminalInit(ctx, ws, termId, workingDir, msg.data?.shell, msg.data?.cols, msg.data?.rows, permission);
+            break;
+        case 'terminalReconnect':
+            handleTerminalReconnect(ctx, ws, termId);
+            break;
+        case 'terminalList':
+            handleTerminalList(ctx, ws);
+            break;
+        case 'terminalInput':
+            handleTerminalInput(ctx, ws, termId, msg.data?.input);
+            break;
+        case 'terminalResize':
+            handleTerminalResize(ctx, termId, msg.data?.cols, msg.data?.rows);
+            break;
+        case 'terminalClose':
+            handleTerminalClose(ctx, ws, termId);
+            break;
+    }
+}
+function handleTerminalInit(ctx, ws, terminalId, workingDir, requestedShell, cols, rows, permission) {
+    const ptyManager = getPTYManager();
+    if (!ptyManager.isPtyAvailable()) {
+        ctx.send(ws, {
+            type: 'terminalError',
+            terminalId,
+            data: {
+                error: 'PTY_NOT_AVAILABLE',
+                instructions: ptyManager.getPtyInstallInstructions()
+            }
+        });
+        return;
+    }
+    addTerminalSubscriber(ctx, terminalId, ws);
+    setupTerminalBroadcastListeners(ctx, terminalId);
+    try {
+        const { shell, cwd, isReconnect } = ptyManager.create(terminalId, workingDir, cols || 80, rows || 24, requestedShell, { sandboxed: permission === 'control' || permission === 'view' });
+        if (!isReconnect) {
+            ctx.broadcastToOthers(ws, {
+                type: 'terminalCreated',
+                data: { terminalId, shell, cwd }
+            });
+        }
+        ctx.send(ws, {
+            type: 'terminalReady',
+            terminalId,
+            data: { shell, cwd, isReconnect }
+        });
+        trackEvent(AnalyticsEvents.TERMINAL_SESSION_CREATED, {
+            shell,
+            is_reconnect: isReconnect,
+        });
+    }
+    catch (error) {
+        console.error(`[WebSocketImproviseHandler] Failed to create terminal:`, error);
+        ctx.send(ws, {
+            type: 'terminalError',
+            terminalId,
+            data: { error: error.message || 'Failed to create terminal' }
+        });
+        removeTerminalSubscriber(ctx, terminalId, ws);
+    }
+}
+function handleTerminalReconnect(ctx, ws, terminalId) {
+    const ptyManager = getPTYManager();
+    const sessionInfo = ptyManager.getSessionInfo(terminalId);
+    if (!sessionInfo) {
+        ctx.send(ws, {
+            type: 'terminalError',
+            terminalId,
+            data: { error: 'Terminal session not found', sessionNotFound: true }
+        });
+        return;
+    }
+    addTerminalSubscriber(ctx, terminalId, ws);
+    setupTerminalBroadcastListeners(ctx, terminalId);
+    ctx.send(ws, {
+        type: 'terminalReady',
+        terminalId,
+        data: {
+            shell: sessionInfo.shell,
+            cwd: sessionInfo.cwd,
+            isReconnect: true
+        }
+    });
+    ptyManager.resize(terminalId, sessionInfo.cols, sessionInfo.rows);
+}
+function handleTerminalList(ctx, ws) {
+    const ptyManager = getPTYManager();
+    const terminalIds = ptyManager.getActiveTerminals();
+    const terminals = terminalIds.map(id => {
+        const info = ptyManager.getSessionInfo(id);
+        return info ? { id, ...info } : null;
+    }).filter(Boolean);
+    ctx.send(ws, {
+        type: 'terminalList',
+        data: { terminals }
+    });
+}
+function handleTerminalInput(ctx, ws, terminalId, input) {
+    if (!input)
+        return;
+    const ptyManager = getPTYManager();
+    const success = ptyManager.write(terminalId, input);
+    if (!success) {
+        ctx.send(ws, {
+            type: 'terminalError',
+            terminalId,
+            data: { error: 'Terminal not found or write failed' }
+        });
+    }
+}
+function handleTerminalResize(_ctx, terminalId, cols, rows) {
+    if (!cols || !rows)
+        return;
+    const ptyManager = getPTYManager();
+    ptyManager.resize(terminalId, cols, rows);
+}
+function handleTerminalClose(ctx, ws, terminalId) {
+    trackEvent(AnalyticsEvents.TERMINAL_SESSION_CLOSED);
+    const listenerCleanup = ctx.terminalListenerCleanups.get(terminalId);
+    if (listenerCleanup) {
+        listenerCleanup();
+        ctx.terminalListenerCleanups.delete(terminalId);
+    }
+    const ptyManager = getPTYManager();
+    ptyManager.close(terminalId);
+    ctx.terminalSubscribers.delete(terminalId);
+    ctx.broadcastToOthers(ws, {
+        type: 'terminalClosed',
+        data: { terminalId }
+    });
+}
+function addTerminalSubscriber(ctx, terminalId, ws) {
+    let subs = ctx.terminalSubscribers.get(terminalId);
+    if (!subs) {
+        subs = new Set();
+        ctx.terminalSubscribers.set(terminalId, subs);
+    }
+    subs.add(ws);
+}
+function removeTerminalSubscriber(ctx, terminalId, ws) {
+    const subs = ctx.terminalSubscribers.get(terminalId);
+    if (!subs)
+        return;
+    subs.delete(ws);
+    if (subs.size > 0)
+        return;
+    ctx.terminalSubscribers.delete(terminalId);
+    const cleanup = ctx.terminalListenerCleanups.get(terminalId);
+    if (cleanup) {
+        cleanup();
+        ctx.terminalListenerCleanups.delete(terminalId);
+    }
+}
+function setupTerminalBroadcastListeners(ctx, terminalId) {
+    if (ctx.terminalListenerCleanups.has(terminalId))
+        return;
+    const ptyManager = getPTYManager();
+    const onOutput = (tid, data) => {
+        if (tid === terminalId) {
+            const subs = ctx.terminalSubscribers.get(terminalId);
+            if (subs) {
+                for (const ws of subs) {
+                    ctx.send(ws, { type: 'terminalOutput', terminalId, data: { output: data } });
+                }
+            }
+        }
+    };
+    const onExit = (tid, exitCode) => {
+        if (tid === terminalId) {
+            const subs = ctx.terminalSubscribers.get(terminalId);
+            if (subs) {
+                for (const ws of subs) {
+                    ctx.send(ws, { type: 'terminalExit', terminalId, data: { exitCode } });
+                }
+            }
+            ptyManager.off('output', onOutput);
+            ptyManager.off('exit', onExit);
+            ptyManager.off('error', onError);
+            ctx.terminalListenerCleanups.delete(terminalId);
+            ctx.terminalSubscribers.delete(terminalId);
+        }
+    };
+    const onError = (tid, error) => {
+        if (tid === terminalId) {
+            const subs = ctx.terminalSubscribers.get(terminalId);
+            if (subs) {
+                for (const ws of subs) {
+                    ctx.send(ws, { type: 'terminalError', terminalId, data: { error } });
+                }
+            }
+        }
+    };
+    ptyManager.on('output', onOutput);
+    ptyManager.on('exit', onExit);
+    ptyManager.on('error', onError);
+    ctx.terminalListenerCleanups.set(terminalId, () => {
+        ptyManager.off('output', onOutput);
+        ptyManager.off('exit', onExit);
+        ptyManager.off('error', onError);
+    });
+}
+/**
+ * Clean up terminal subscribers for a disconnected WS context.
+ * Called from handler.ts handleClose().
+ */
+export function cleanupTerminalSubscribers(ctx, ws) {
+    for (const subs of ctx.terminalSubscribers.values()) {
+        subs.delete(ws);
+    }
+}
+//# sourceMappingURL=terminal-handlers.js.map

package/dist/server/services/websocket/terminal-handlers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"terminal-handlers.js","sourceRoot":"","sources":["../../../../server/services/websocket/terminal-handlers.ts"],"names":[],"mappings":"AAAA,8DAA8D;AAC9D,gEAAgE;AAEhE,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAI3D,MAAM,UAAU,qBAAqB,CAAC,GAAmB,EAAE,EAAa,EAAE,GAAqB,EAAE,KAAa,EAAE,UAAkB,EAAE,UAA+B;IACjK,MAAM,MAAM,GAAG,GAAG,CAAC,UAAU,IAAI,KAAK,CAAC;IACvC,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;QACjB,KAAK,cAAc;YACjB,kBAAkB,CAAC,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;YAC7G,MAAM;QACR,KAAK,mBAAmB;YACtB,uBAAuB,CAAC,GAAG,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;YACzC,MAAM;QACR,KAAK,cAAc;YACjB,kBAAkB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC5B,MAAM;QACR,KAAK,eAAe;YAClB,mBAAmB,CAAC,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACtD,MAAM;QACR,KAAK,gBAAgB;YACnB,oBAAoB,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAClE,MAAM;QACR,KAAK,eAAe;YAClB,mBAAmB,CAAC,GAAG,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;YACrC,MAAM;IACV,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CACzB,GAAmB,EACnB,EAAa,EACb,UAAkB,EAClB,UAAkB,EAClB,cAAuB,EACvB,IAAa,EACb,IAAa,EACb,UAA+B;IAE/B,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IAEnC,IAAI,CAAC,UAAU,CAAC,cAAc,EAAE,EAAE,CAAC;QACjC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;YACX,IAAI,EAAE,eAAe;YACrB,UAAU;YACV,IAAI,EAAE;gBACJ,KAAK,EAAE,mBAAmB;gBAC1B,YAAY,EAAE,UAAU,CAAC,yBAAyB,EAAE;aACrD;SACF,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,qBAAqB,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;IAC3C,+BAA+B,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;IAEjD,IAAI,CAAC;QACH,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,UAAU,CAAC,MAAM,CACnD,UAAU,EACV,UAAU,EACV,IAAI,IAAI,EAAE,EACV,IAAI,IAAI,EAAE,EACV,cAAc,EACd,EAAE,SAAS,EAAE,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,MAAM,EAAE,CACjE,CAAC;QAEF,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,GAAG,CAAC,iBAAiB,CAAC,EAAE,EAAE;gBACxB,IAAI,EAAE,iBAAiB;gBACvB,IAAI,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,EAAE;aACjC,CAAC,CAAC;QACL,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;YACX,IAAI,EAAE,eAAe;YACrB,UAAU;YACV,IAAI,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,WAAW,EAAE;SAClC,CAAC,CAAC;QACH,UAAU,CAAC,eAAe,CAAC,wBAAwB,EAAE;YACnD,KAAK;YACL,YAAY,EAAE,WAAW;SAC1B,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,KAAK,CAAC,wDAAwD,EAAE,KAAK,CAAC,CAAC;QAC/E,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;YACX,IAAI,EAAE,eAAe;YACrB,UAAU;YACV,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,2BAA2B,EAAE;SAC9D,CAAC,CAAC;QACH,wBAAwB,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;IAChD,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAAC,GAAmB,EAAE,EAAa,EAAE,UAAkB;IACrF,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IAEnC,MAAM,WAAW,GAAG,UAAU,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;IAC1D,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;YACX,IAAI,EAAE,eAAe;YACrB,UAAU;YACV,IAAI,EAAE,EAAE,KAAK,EAAE,4BAA4B,EAAE,eAAe,EAAE,IAAI,EAAE;SACrE,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,qBAAqB,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;IAC3C,+BAA+B,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;IAEjD,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;QACX,IAAI,EAAE,eAAe;QACrB,UAAU;QACV,IAAI,EAAE;YACJ,KAAK,EAAE,WAAW,CAAC,KAAK;YACxB,GAAG,EAAE,WAAW,CAAC,GAAG;YACpB,WAAW,EAAE,IAAI;SAClB;KACF,CAAC,CAAC;IAEH,UAAU,CAAC,MAAM,CAAC,UAAU,EAAE,WAAW,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;AACpE,CAAC;AAED,SAAS,kBAAkB,CAAC,GAAmB,EAAE,EAAa;IAC5D,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,MAAM,WAAW,GAAG,UAAU,CAAC,kBAAkB,EAAE,CAAC;IAEpD,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;QACrC,MAAM,IAAI,GAAG,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;IACvC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAEnB,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;QACX,IAAI,EAAE,cAAc;QACpB,IAAI,EAAE,EAAE,SAAS,EAAE;KACpB,CAAC,CAAC;AACL,CAAC;AAED,SAAS,mBAAmB,CAC1B,GAAmB,EACnB,EAAa,EACb,UAAkB,EAClB,KAAc;IAEd,IAAI,CAAC,KAAK;QAAE,OAAO;IAEnB,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,MAAM,OAAO,GAAG,UAAU,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IAEpD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;YACX,IAAI,EAAE,eAAe;YACrB,UAAU;YACV,IAAI,EAAE,EAAE,KAAK,EAAE,oCAAoC,EAAE;SACtD,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAC3B,IAAoB,EACpB,UAAkB,EAClB,IAAa,EACb,IAAa;IAEb,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI;QAAE,OAAO;IAE3B,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,UAAU,CAAC,MAAM,CAAC,UAAU,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAC5C,CAAC;AAED,SAAS,mBAAmB,CAAC,GAAmB,EAAE,EAAa,EAAE,UAAkB;IACjF,UAAU,CAAC,eAAe,CAAC,uBAAuB,CAAC,CAAC;IAEpD,MAAM,eAAe,GAAG,GAAG,CAAC,wBAAwB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACrE,IAAI,eAAe,EAAE,CAAC;QACpB,eAAe,EAAE,CAAC;QAClB,GAAG,CAAC,wBAAwB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAE7B,GAAG,CAAC,mBAAmB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAE3C,GAAG,CAAC,iBAAiB,CAAC,EAAE,EAAE;QACxB,IAAI,EAAE,gBAAgB;QACtB,IAAI,EAAE,EAAE,UAAU,EAAE;KACrB,CAAC,CAAC;AACL,CAAC;AAED,SAAS,qBAAqB,CAAC,GAAmB,EAAE,UAAkB,EAAE,EAAa;IACnF,IAAI,IAAI,GAAG,GAAG,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACnD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,IAAI,GAAG,IAAI,GAAG,EAAE,CAAC;QACjB,GAAG,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;IAChD,CAAC;IACD,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;AACf,CAAC;AAED,SAAS,wBAAwB,CAAC,GAAmB,EAAE,UAAkB,EAAE,EAAa;IACtF,MAAM,IAAI,GAAG,GAAG,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACrD,IAAI,CAAC,IAAI;QAAE,OAAO;IAClB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAChB,IAAI,IAAI,CAAC,IAAI,GAAG,CAAC;QAAE,OAAO;IAC1B,GAAG,CAAC,mBAAmB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAC3C,MAAM,OAAO,GAAG,GAAG,CAAC,wBAAwB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAC7D,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,EAAE,CAAC;QACV,GAAG,CAAC,wBAAwB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAClD,CAAC;AACH,CAAC;AAED,SAAS,+BAA+B,CAAC,GAAmB,EAAE,UAAkB;IAC9E,IAAI,GAAG,CAAC,wBAAwB,CAAC,GAAG,CAAC,UAAU,CAAC;QAAE,OAAO;IAEzD,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IAEnC,MAAM,QAAQ,GAAG,CAAC,GAAW,EAAE,IAAY,EAAE,EAAE;QAC7C,IAAI,GAAG,KAAK,UAAU,EAAE,CAAC;YACvB,MAAM,IAAI,GAAG,GAAG,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACrD,IAAI,IAAI,EAAE,CAAC;gBACT,KAAK,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC;oBACtB,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC/E,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,MAAM,GAAG,CAAC,GAAW,EAAE,QAAgB,EAAE,EAAE;QAC/C,IAAI,GAAG,KAAK,UAAU,EAAE,CAAC;YACvB,MAAM,IAAI,GAAG,GAAG,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACrD,IAAI,IAAI,EAAE,CAAC;gBACT,KAAK,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC;oBACtB,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;gBACzE,CAAC;YACH,CAAC;YACD,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACnC,UAAU,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC/B,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACjC,GAAG,CAAC,wBAAwB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAChD,GAAG,CAAC,mBAAmB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,CAAC,GAAW,EAAE,KAAa,EAAE,EAAE;QAC7C,IAAI,GAAG,KAAK,UAAU,EAAE,CAAC;YACvB,MAAM,IAAI,GAAG,GAAG,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACrD,IAAI,IAAI,EAAE,CAAC;gBACT,KAAK,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC;oBACtB,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBACvE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IAEF,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAClC,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9B,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAEhC,GAAG,CAAC,wBAAwB,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,EAAE;QAChD,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACnC,UAAU,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC/B,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,0BAA0B,CAAC,GAAmB,EAAE,EAAa;IAC3E,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,mBAAmB,CAAC,MAAM,EAAE,EAAE,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAClB,CAAC;AACH,CAAC"}

package/dist/server/services/websocket/types.d.ts

@@ -15,7 +15,7 @@ export interface WSContext {
     _ws?: unknown;
 }
 export interface WebSocketMessage {
-    type: 'execute' | 'cancel' | 'getHistory' | 'getSessions' | 'getSessionsCount' | 'deleteSession' | 'getSessionById' | 'clearHistory' | 'searchHistory' | 'new' | 'autocomplete' | 'readFile' | 'ping' | 'initTab' | 'resumeSession' | 'approve' | 'reject' | 'recordSelection' | 'requestNotificationSummary' | 'terminalInit' | 'terminalReconnect' | 'terminalList' | 'terminalInput' | 'terminalResize' | 'terminalClose' | 'listDirectory' | 'writeFile' | 'createFile' | 'createDirectory' | 'deleteFile' | 'renameFile' | 'notifyFileOpened' | 'gitStatus' | 'gitStage' | 'gitUnstage' | 'gitCommit' | 'gitCommitWithAI' | 'gitPush' | 'gitLog' | 'gitDiscoverRepos' | 'gitSetDirectory' | 'gitGetRemoteInfo' | 'gitCreatePR' | 'gitGeneratePRDescription' | 'getActiveTabs' | 'createTab' | 'reorderTabs' | 'syncTabMeta' | 'syncPromptText' | 'removeTab' | 'markTabViewed' | 'getSettings' | 'updateSettings';
+    type: 'execute' | 'cancel' | 'getHistory' | 'getSessions' | 'getSessionsCount' | 'deleteSession' | 'getSessionById' | 'clearHistory' | 'searchHistory' | 'new' | 'autocomplete' | 'readFile' | 'ping' | 'initTab' | 'resumeSession' | 'approve' | 'reject' | 'recordSelection' | 'requestNotificationSummary' | 'terminalInit' | 'terminalReconnect' | 'terminalList' | 'terminalInput' | 'terminalResize' | 'terminalClose' | 'listDirectory' | 'writeFile' | 'createFile' | 'createDirectory' | 'deleteFile' | 'renameFile' | 'notifyFileOpened' | 'searchFileContents' | 'cancelSearch' | 'findDefinition' | 'gitStatus' | 'gitStage' | 'gitUnstage' | 'gitCommit' | 'gitCommitWithAI' | 'gitPush' | 'gitPull' | 'gitLog' | 'gitDiscoverRepos' | 'gitSetDirectory' | 'gitGetRemoteInfo' | 'gitCreatePR' | 'gitGeneratePRDescription' | 'gitListBranches' | 'gitCheckout' | 'gitCreateBranch' | 'gitDeleteBranch' | 'gitDiff' | 'gitListTags' | 'gitCreateTag' | 'gitPushTag' | 'gitWorktreeList' | 'gitWorktreeCreate' | 'gitWorktreeRemove' | 'tabWorktreeSwitch' | 'gitWorktreePush' | 'gitWorktreeCreatePR' | 'gitMergePreview' | 'gitWorktreeMerge' | 'gitMergeAbort' | 'gitMergeComplete' | 'getActiveTabs' | 'createTab' | 'reorderTabs' | 'syncTabMeta' | 'syncPromptText' | 'removeTab' | 'markTabViewed' | 'getSettings' | 'updateSettings';
     tabId?: string;
     terminalId?: string;
     data?: any;
@@ -23,7 +23,7 @@ export interface WebSocketMessage {
     _permission?: 'control' | 'view';
 }
 export interface WebSocketResponse {
-    type: 'output' | 'thinking' | 'movementStart' | 'movementComplete' | 'movementError' | 'sessionUpdate' | 'history' | 'sessions' | 'sessionsCount' | 'sessionDeleted' | 'sessionData' | 'historyCleared' | 'searchResults' | 'newSession' | 'autocomplete' | 'fileContent' | 'error' | 'pong' | 'tabInitialized' | 'approvalRequired' | 'toolUse' | 'notificationSummary' | 'terminalOutput' | 'terminalReady' | 'terminalExit' | 'terminalError' | 'terminalList' | 'directoryListing' | 'fileWritten' | 'fileCreated' | 'directoryCreated' | 'fileDeleted' | 'fileRenamed' | 'fileOpened' | 'fileContentChanged' | 'terminalCreated' | 'terminalClosed' | 'gitStatus' | 'gitStaged' | 'gitUnstaged' | 'gitCommitted' | 'gitCommitMessage' | 'gitPushed' | 'gitLog' | 'gitError' | 'gitReposDiscovered' | 'gitDirectorySet' | 'gitRemoteInfo' | 'gitPRCreated' | 'gitPRDescription' | 'activeTabs' | 'tabCreated' | 'tabRemoved' | 'tabRenamed' | 'tabsReordered' | 'promptTextSync' | 'tabViewed' | 'tabStateChanged' | 'settings' | 'settingsUpdated';
+    type: 'output' | 'thinking' | 'movementStart' | 'movementComplete' | 'movementError' | 'sessionUpdate' | 'history' | 'sessions' | 'sessionsCount' | 'sessionDeleted' | 'sessionData' | 'historyCleared' | 'searchResults' | 'newSession' | 'autocomplete' | 'fileContent' | 'error' | 'pong' | 'tabInitialized' | 'approvalRequired' | 'toolUse' | 'streamingTokens' | 'notificationSummary' | 'terminalOutput' | 'terminalReady' | 'terminalExit' | 'terminalError' | 'terminalList' | 'directoryListing' | 'fileWritten' | 'fileCreated' | 'directoryCreated' | 'fileDeleted' | 'fileRenamed' | 'fileOpened' | 'fileContentChanged' | 'contentSearchResults' | 'contentSearchComplete' | 'contentSearchError' | 'definitionResult' | 'terminalCreated' | 'terminalClosed' | 'gitStatus' | 'gitStaged' | 'gitUnstaged' | 'gitCommitted' | 'gitCommitMessage' | 'gitPushed' | 'gitPulled' | 'gitLog' | 'gitError' | 'gitReposDiscovered' | 'gitDirectorySet' | 'gitRemoteInfo' | 'gitPRCreated' | 'gitPRDescription' | 'gitBranchList' | 'gitCheckedOut' | 'gitBranchCreated' | 'gitBranchDeleted' | 'gitDiffResult' | 'gitTagList' | 'gitTagCreated' | 'gitTagPushed' | 'gitWorktreeListResult' | 'gitWorktreeCreated' | 'gitWorktreeRemoved' | 'tabWorktreeSwitched' | 'gitWorktreePushed' | 'gitWorktreePRCreated' | 'gitMergePreviewResult' | 'gitWorktreeMergeResult' | 'gitMergeAborted' | 'gitMergeCompleted' | 'activeTabs' | 'tabCreated' | 'tabRemoved' | 'tabRenamed' | 'tabsReordered' | 'promptTextSync' | 'tabViewed' | 'tabStateChanged' | 'settings' | 'settingsUpdated';
     tabId?: string;
     terminalId?: string;
     data?: any;
@@ -215,4 +215,65 @@ export interface GitDirectorySetResponse {
     /** Whether the directory is valid (contains .git) */
     isValid: boolean;
 }
+export interface GitBranchEntry {
+    /** Branch name, e.g. "feat/auth" or "origin/main" */
+    name: string;
+    /** Short commit hash */
+    shortHash: string;
+    /** Whether this is a remote branch */
+    isRemote: boolean;
+    /** Whether this is the currently checked out branch */
+    isCurrent: boolean;
+    /** Tracking branch, e.g. "origin/feat/auth" */
+    upstream?: string;
+}
+export interface GitTagEntry {
+    /** Tag name */
+    name: string;
+    /** Short commit hash */
+    shortHash: string;
+    /** Creation date (ISO string) */
+    date: string;
+    /** Tag message (empty for lightweight tags) */
+    message: string;
+}
+export interface WorktreeInfo {
+    /** Absolute path to the worktree directory */
+    path: string;
+    /** Branch checked out in this worktree */
+    branch: string;
+    /** HEAD commit hash */
+    head: string;
+    /** Whether this is the main working tree */
+    isMain: boolean;
+    /** Whether this is a bare repository */
+    isBare: boolean;
+    /** Whether this worktree can be pruned */
+    prunable?: boolean;
+}
+export interface MergePreviewResult {
+    /** Whether the merge can be done cleanly */
+    clean: boolean;
+    /** List of conflicting file paths */
+    conflicts: string[];
+    /** Diff stat summary */
+    stat: string;
+    /** List of commits to be merged */
+    commits: {
+        hash: string;
+        message: string;
+    }[];
+    /** Number of commits ahead */
+    ahead: number;
+}
+export interface WorktreeMergeResult {
+    /** Whether the merge succeeded */
+    success: boolean;
+    /** Merge commit hash (if successful) */
+    mergeCommit?: string;
+    /** Error message (if failed) */
+    error?: string;
+    /** List of conflicting files (if conflicts) */
+    conflictFiles?: string[];
+}
 //# sourceMappingURL=types.d.ts.map