mstro-app 0.3.7 → 0.3.9

package/server/services/websocket/session-handlers.ts

@@ -6,9 +6,29 @@ import { join } from 'node:path';
 import { type FileAttachment, ImprovisationSessionManager } from '../../cli/improvisation-session-manager.js';
 import { getModel } from '../settings.js';
 import type { HandlerContext } from './handler-context.js';
+import { runQualityScan } from './quality-service.js';
 import type { SessionRegistry } from './session-registry.js';
 import type { WebSocketMessage, WSContext } from './types.js';
 
+const WRITE_TOOL_NAMES = new Set(['Edit', 'Write', 'MultiEdit', 'NotebookEdit']);
+
+function movementHadWrites(movement: Record<string, unknown>): boolean {
+  const history = movement.toolUseHistory as Array<{ toolName: string; result?: string; isError?: boolean }> | undefined;
+  if (!history) return false;
+  return history.some((t) => WRITE_TOOL_NAMES.has(t.toolName) && t.result !== undefined && !t.isError);
+}
+
+function triggerPostSessionQualityScan(ctx: HandlerContext, workingDir: string): void {
+  runQualityScan(workingDir).then((results) => {
+    ctx.broadcastToAll({
+      type: 'qualityPostSession',
+      data: { path: '.', results },
+    });
+  }).catch(() => {
+    // Quality scan failure should not affect session flow
+  });
+}
+
 /** Convert tool history entries into OutputLine-compatible lines */
 function convertToolHistoryToLines(tools: Array<{ toolName: string; toolInput?: Record<string, unknown>; result?: string; isError?: boolean }>, ts: number): Array<Record<string, unknown>> {
   const lines: Array<Record<string, unknown>> = [];
@@ -104,6 +124,12 @@ export function setupSessionListeners(ctx: HandlerContext, session: Improvisatio
         movementId: `${movement.sequenceNumber}`
       });
     }
+
+    // Post-session quality gate: if the session wrote/edited files, re-scan
+    if (movementHadWrites(movement)) {
+      const scanDir = ctx.gitDirectories.get(tabId) || session.getSessionInfo().workingDir;
+      triggerPostSessionQualityScan(ctx, scanDir);
+    }
   });
 
   session.on('onMovementError', (error: Error) => {

package/server/services/websocket/types.ts

@@ -107,11 +107,36 @@ export interface WebSocketMessage {
     // Settings message types
     | 'getSettings'
     | 'updateSettings'
+    // Quality message types
+    | 'qualityDetectTools'
+    | 'qualityScan'
+    | 'qualityInstallTools'
+    | 'qualityCodeReview'
+    | 'qualityFixIssues'
+    | 'qualityLoadState'
+    | 'qualitySaveDirectories'
     // File upload message types (chunked remote uploads)
     | 'fileUploadStart'
     | 'fileUploadChunk'
     | 'fileUploadComplete'
-    | 'fileUploadCancel';
+    | 'fileUploadCancel'
+    // Plan message types
+    | 'planInit'
+    | 'planGetState'
+    | 'planListIssues'
+    | 'planGetIssue'
+    | 'planGetSprint'
+    | 'planGetMilestone'
+    | 'planCreateIssue'
+    | 'planUpdateIssue'
+    | 'planDeleteIssue'
+    | 'planScaffold'
+    | 'planPrompt'
+    | 'planExecute'
+    | 'planExecuteEpic'
+    | 'planPause'
+    | 'planStop'
+    | 'planResume';
   tabId?: string;
   terminalId?: string;
   // biome-ignore lint/suspicious/noExplicitAny: message envelope carries heterogeneous payloads
@@ -219,10 +244,45 @@ export interface WebSocketResponse {
     // Settings response types
     | 'settings'
     | 'settingsUpdated'
+    // Quality response types
+    | 'qualityToolsDetected'
+    | 'qualityScanProgress'
+    | 'qualityScanResults'
+    | 'qualityInstallProgress'
+    | 'qualityInstallComplete'
+    | 'qualityCodeReview'
+    | 'qualityCodeReviewProgress'
+    | 'qualityPostSession'
+    | 'qualityFixProgress'
+    | 'qualityFixComplete'
+    | 'qualityError'
+    | 'qualityStateLoaded'
     // File upload response types
     | 'fileUploadAck'
     | 'fileUploadReady'
-    | 'fileUploadError';
+    | 'fileUploadError'
+    // Plan response types
+    | 'planState'
+    | 'planIssueList'
+    | 'planIssue'
+    | 'planSprint'
+    | 'planMilestone'
+    | 'planNotFound'
+    | 'planStateUpdated'
+    | 'planIssueUpdated'
+    | 'planIssueCreated'
+    | 'planIssueDeleted'
+    | 'planScaffolded'
+    | 'planPromptStreaming'
+    | 'planPromptProgress'
+    | 'planPromptResponse'
+    | 'planExecutionStarted'
+    | 'planExecutionProgress'
+    | 'planExecutionOutput'
+    | 'planExecutionMetrics'
+    | 'planExecutionComplete'
+    | 'planExecutionError'
+    | 'planError';
   tabId?: string;
   terminalId?: string;
   // biome-ignore lint/suspicious/noExplicitAny: message envelope carries heterogeneous payloads