@sumicom/quicksave 0.8.8 → 0.8.10

package/dist/handlers/messageHandler.js

@@ -9,8 +9,12 @@ import { CommitSummaryCliService, CommitSummaryCliError } from '../ai/commitSumm
 import { CommitSummaryStateStore } from '../ai/commitSummaryStore.js';
 import { SessionManager } from '../ai/sessionManager.js';
 import { ClaudeCodeProvider } from '../ai/claudeCodeProvider.js';
+import { ClaudeTerminalProvider } from '../ai/claudeTerminal/index.js';
+import { OpenCodeProvider } from '../ai/openCodeProvider.js';
 import { AttachmentStaging } from '../ai/attachmentStaging.js';
 import { loadAttachment, removeSessionAttachments } from '../ai/attachmentStore.js';
+import { transcribeAudio, listModels } from '../ai/voiceTranscription.js';
+import { probeAudioSupport } from '../ai/voiceStream.js';
 import { CodexAppServerProvider } from '../ai/codexAppServer/index.js';
 import { CodexLoginManager } from '../ai/codexLogin.js';
 import { getTerminalManager } from '../terminal/terminalManager.js';
@@ -87,7 +91,7 @@ function projectAppServerModel(m) {
 }
 export class MessageHandler {
     repos;
-    agentVersion = '0.8.8';
+    agentVersion = '0.8.10';
     defaultRepoPath;
     clientRepos = new Map(); // peerAddress -> repoPath
     repoLocks = new Map(); // repoPath -> peerAddress holding lock
@@ -124,7 +128,12 @@ export class MessageHandler {
         this.defaultRepoPath = repos.length > 0 ? repos[0].path : '';
         this.productionBuild = productionBuild;
         this.claudeService = options?.sessionManager
-            ?? new SessionManager([new ClaudeCodeProvider(), new CodexAppServerProvider()]);
+            ?? new SessionManager([
+                new ClaudeCodeProvider(),
+                new ClaudeTerminalProvider(),
+                new CodexAppServerProvider(),
+                new OpenCodeProvider(),
+            ]);
         this.codexCacheDir = options?.codexCacheDir ?? join(homedir(), '.codex');
         // Load explicit coding paths only (repos and coding paths are independent)
         if (codingPaths) {
@@ -234,7 +243,7 @@ export class MessageHandler {
         try {
             handle = await spawnAppServer({
                 clientInfo: { name: 'quicksave-agent', title: 'Quicksave Agent', version: '0.0.0' },
-                capabilities: { experimentalApi: false, optOutNotificationMethods: null },
+                capabilities: { experimentalApi: true, optOutNotificationMethods: null },
             });
             const res = await handle.rpc.request('model/list', { cursor: null, limit: null, includeHidden: false });
             return res.data
@@ -525,6 +534,8 @@ export class MessageHandler {
                 return this.handleAgentUpdate(message);
             case 'agent:restart':
                 return this.handleAgentRestart(message);
+            case 'agent:probe':
+                return this.handleAgentProbe();
             case 'systemd:status':
                 return this.handleSystemdStatus(message);
             case 'systemd:install':
@@ -544,6 +555,10 @@ export class MessageHandler {
                 return this.handleClaudeStart(message, peerAddress);
             case 'claude:resume':
                 return this.handleClaudeResume(message, peerAddress);
+            case 'claude:interrupt':
+                return this.handleClaudeInterrupt(message);
+            case 'claude:steer-queued':
+                return this.handleClaudeSteerQueued(message);
             case 'claude:cancel':
                 return this.handleClaudeCancel(message);
             case 'claude:close':
@@ -570,6 +585,8 @@ export class MessageHandler {
                 return this.handleListArchived(message);
             case 'session:mark-read':
                 return this.handleMarkRead(message);
+            case 'session:dismiss-pending-mission':
+                return this.handleDismissPendingMission(message);
             case 'project:list-summaries':
                 return this.handleListProjectSummaries(message);
             case 'project:list-repos':
@@ -598,14 +615,19 @@ export class MessageHandler {
                 return this.handleAttachmentCancel(message, peerAddress);
             case 'attachment:fetch':
                 return this.handleAttachmentFetch(message);
+            case 'voice:transcribe':
+                return this.handleVoiceTranscribe(message);
+            case 'voice:list-models':
+                return this.handleVoiceListModels(message);
             default:
                 return this.createErrorResponse(message.id, 'UNKNOWN_MESSAGE_TYPE', `Unknown message type: ${message.type}`);
         }
     }
-    handleHandshake(message, peerAddress) {
+    async handleHandshake(message, peerAddress) {
         // Fire-and-forget: check npm registry + OpenAI models (12h dedup each)
         this.checkLatestVersion().catch(() => { });
         this.fetchCodexModels().catch(() => { });
+        const availableProviders = await this.claudeService.probeProviders();
         const response = createMessage('handshake:ack', {
             success: true,
             agentVersion: this.agentVersion,
@@ -617,6 +639,8 @@ export class MessageHandler {
             devBuild: !this.productionBuild || undefined,
             codexModels: this.codexModelsCache?.models,
             platform: normalizePlatform(osPlatform()),
+            availableProviders: availableProviders.length > 0 ? availableProviders : undefined,
+            audio: await probeAudioSupport(),
         });
         response.id = message.id;
         return response;
@@ -1157,6 +1181,40 @@ export class MessageHandler {
             return response;
         }
     }
+    async handleVoiceTranscribe(message) {
+        const { audioBase64, mimeType, config } = message.payload;
+        try {
+            const audio = Buffer.from(audioBase64, 'base64');
+            const text = await transcribeAudio(audio, mimeType, config);
+            const response = createMessage('voice:transcribe:response', { text });
+            response.id = message.id;
+            return response;
+        }
+        catch (error) {
+            const response = createMessage('voice:transcribe:response', {
+                text: '',
+                error: error instanceof Error ? error.message : 'Transcription failed',
+            });
+            response.id = message.id;
+            return response;
+        }
+    }
+    async handleVoiceListModels(message) {
+        try {
+            const models = await listModels(message.payload.config);
+            const response = createMessage('voice:list-models:response', { models });
+            response.id = message.id;
+            return response;
+        }
+        catch (error) {
+            const response = createMessage('voice:list-models:response', {
+                models: [],
+                error: error instanceof Error ? error.message : 'Failed to list models',
+            });
+            response.id = message.id;
+            return response;
+        }
+    }
     handleGetApiKeyStatus(message) {
         const response = createMessage('ai:get-api-key-status:response', {
             configured: hasAnthropicApiKey(),
@@ -1548,6 +1606,10 @@ export class MessageHandler {
             return response;
         }
     }
+    async handleAgentProbe() {
+        const availableProviders = await this.claudeService.probeProviders();
+        return createMessage('agent:probe:response', { availableProviders });
+    }
     // ==========================================================================
     // systemd user-unit (Linux auto-start at login)
     //
@@ -1735,6 +1797,7 @@ export class MessageHandler {
                 model,
                 reasoningEffort,
                 contextWindow,
+                mcpCorrId: this.claudeService.getSessionMcpCorrId(sessionId),
             });
             this.onHistoryUpdated?.(cwd, registry.getEntry(cwd, sessionId), 'upsert');
             const response = createMessage('claude:start:response', { success: true, sessionId });
@@ -1749,7 +1812,7 @@ export class MessageHandler {
         }
     }
     async handleClaudeResume(message, peerAddress) {
-        const { sessionId: requestedId, prompt, cwd: payloadCwd, agent, attachmentIds } = message.payload;
+        const { sessionId: requestedId, prompt, cwd: payloadCwd, agent, attachmentIds, interruptCurrentTurn } = message.payload;
         const legacyProvider = message.payload.provider;
         const cwd = payloadCwd || this.getClientRepoPath(peerAddress);
         const resolvedAgent = agent ?? (legacyProvider === 'codex-mcp' ? 'codex' : legacyProvider ? 'claude-code' : undefined);
@@ -1775,6 +1838,7 @@ export class MessageHandler {
                 cwd,
                 agent: resolvedAgent,
                 attachments,
+                interruptCurrentTurn,
             });
             console.log(`[agent:resume] session resumed: ${actualSessionId}`);
             getEventStore().record({
@@ -1794,9 +1858,14 @@ export class MessageHandler {
                 agent: actualAgent,
                 lastAccessedAt: Date.now(),
                 gitBranch,
+                mcpCorrId: this.claudeService.getSessionMcpCorrId(actualSessionId) ?? existing?.mcpCorrId,
             });
             this.onHistoryUpdated?.(cwd, registry.getEntry(cwd, actualSessionId), 'upsert');
-            const response = createMessage('claude:resume:response', { success: true, sessionId: actualSessionId });
+            const response = createMessage('claude:resume:response', {
+                success: true,
+                sessionId: actualSessionId,
+                queueState: this.claudeService.buildSessionUpdatePayload(actualSessionId).queueState,
+            });
             response.id = message.id;
             return response;
         }
@@ -1817,6 +1886,20 @@ export class MessageHandler {
         response.id = message.id;
         return response;
     }
+    async handleClaudeInterrupt(message) {
+        const { sessionId } = message.payload;
+        const success = await this.claudeService.interruptSession(sessionId);
+        const response = createMessage('claude:interrupt:response', { success, error: success ? undefined : 'Session not found or already ended' });
+        response.id = message.id;
+        return response;
+    }
+    async handleClaudeSteerQueued(message) {
+        const { sessionId, interruptCurrentTurn } = message.payload;
+        const success = await this.claudeService.steerQueuedMessage(sessionId, { interruptCurrentTurn });
+        const response = createMessage('claude:steer-queued:response', { success, error: success ? undefined : 'No queued message or active turn to steer' });
+        response.id = message.id;
+        return response;
+    }
     handleClaudeClose(message) {
         const { sessionId } = message.payload;
         const success = this.claudeService.closeSession(sessionId);
@@ -1861,8 +1944,15 @@ export class MessageHandler {
         return response;
     }
     handleClaudeUserInputResponse(message) {
-        const resolved = this.claudeService.resolveUserInput(message.payload);
-        console.log(`[agent:user-input-response] requestId=${message.payload.requestId} action=${message.payload.action} resolved=${resolved}`);
+        const { payload } = message;
+        // On plan approval, update permission level synchronously before resolving
+        // so the CLI sees the correct mode for its first tool call after ExitPlanMode.
+        if (payload.action === 'allow' && payload.permissionMode) {
+            this.claudeService.setPermissionLevel(payload.sessionId, payload.permissionMode)
+                .catch(err => console.error('[plan-approve] setPermissionLevel failed:', err));
+        }
+        const resolved = this.claudeService.resolveUserInput(payload);
+        console.log(`[agent:user-input-response] requestId=${payload.requestId} action=${payload.action} resolved=${resolved}`);
         // No dedicated response type — just acknowledge
         const response = createMessage('claude:user-input-response', { success: resolved });
         response.id = message.id;
@@ -2024,6 +2114,27 @@ export class MessageHandler {
         response.id = message.id;
         return response;
     }
+    handleDismissPendingMission(message) {
+        const { sessionId, cwd, dismissedAt } = message.payload;
+        const stamp = typeof dismissedAt === 'number' ? dismissedAt : Date.now();
+        const existing = getSessionRegistry().getEntry(cwd, sessionId);
+        if (!existing?.pendingMission) {
+            const failed = createMessage('session:dismiss-pending-mission:response', { success: false, error: 'Pending mission not found' });
+            failed.id = message.id;
+            return failed;
+        }
+        const pendingMission = {
+            ...existing.pendingMission,
+            dismissedAt: Math.max(existing.pendingMission.dismissedAt ?? 0, stamp),
+        };
+        const updated = getSessionRegistry().updateEntry(cwd, sessionId, { pendingMission });
+        const entry = updated ?? { ...existing, pendingMission };
+        this.onHistoryUpdated?.(cwd, entry, 'upsert');
+        this.claudeService.emitSessionUpdate(sessionId);
+        const response = createMessage('session:dismiss-pending-mission:response', { success: true, pendingMission });
+        response.id = message.id;
+        return response;
+    }
     handleListProjectSummaries(message) {
         const allEntries = getSessionRegistry().getEntriesForProject();
         // Group by cwd