mstro-app 0.3.6 → 0.3.8

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,6 +107,11 @@ export interface WebSocketMessage {
     // Settings message types
     | 'getSettings'
     | 'updateSettings'
+    // Quality message types
+    | 'qualityDetectTools'
+    | 'qualityScan'
+    | 'qualityInstallTools'
+    | 'qualityCodeReview'
     // File upload message types (chunked remote uploads)
     | 'fileUploadStart'
     | 'fileUploadChunk'
@@ -219,6 +224,15 @@ export interface WebSocketResponse {
     // Settings response types
     | 'settings'
     | 'settingsUpdated'
+    // Quality response types
+    | 'qualityToolsDetected'
+    | 'qualityScanProgress'
+    | 'qualityScanResults'
+    | 'qualityInstallProgress'
+    | 'qualityInstallComplete'
+    | 'qualityCodeReview'
+    | 'qualityPostSession'
+    | 'qualityError'
     // File upload response types
     | 'fileUploadAck'
     | 'fileUploadReady'