[Init] Initial commit - NetMesh terminal manager

infrastructure/ai/harness/agentStop.ts

@@ -0,0 +1,73 @@
+import { clearAllPendingApprovals } from '../shared/approvalGate';
+import { globalTraceStore } from './traceStore';
+import type { AgentBackend } from './types';
+
+export type StopAgentTurnReason = 'user' | 'slash';
+
+export interface AgentStopBridge {
+  aiCattyCancelExec?(chatSessionId: string): Promise<unknown>;
+  aiSdkAgentCancel?(requestId: string, chatSessionId?: string): Promise<{ ok: boolean; error?: string }>;
+  aiSetChatSessionCancelled?(chatSessionId: string, cancelled?: boolean): Promise<{ ok: boolean; error?: string }>;
+}
+
+export interface StopAgentTurnParams {
+  chatSessionId: string;
+  abortController?: AbortController | null;
+  bridge?: AgentStopBridge | null;
+  reason?: StopAgentTurnReason;
+  backend?: AgentBackend;
+}
+
+let stopEventCounter = 0;
+
+function nextStopEventId(): string {
+  stopEventCounter += 1;
+  return `turn-end-${Date.now()}-${stopEventCounter}`;
+}
+
+/**
+ * Unified stop entry for Catty, external SDK, and MCP tool surfaces.
+ */
+export async function stopAgentTurn({
+  chatSessionId,
+  abortController,
+  bridge,
+  reason = 'user',
+  backend = 'catty',
+}: StopAgentTurnParams): Promise<void> {
+  abortController?.abort();
+  clearAllPendingApprovals(chatSessionId);
+
+  const tasks: Array<Promise<unknown>> = [];
+  if (bridge?.aiCattyCancelExec) {
+    tasks.push(bridge.aiCattyCancelExec(chatSessionId).catch(() => {}));
+  }
+  if (bridge?.aiSdkAgentCancel) {
+    tasks.push(bridge.aiSdkAgentCancel('', chatSessionId).catch(() => {}));
+  }
+  if (bridge?.aiSetChatSessionCancelled) {
+    tasks.push(bridge.aiSetChatSessionCancelled(chatSessionId, true).catch(() => {}));
+  }
+  await Promise.all(tasks);
+
+  globalTraceStore.append({
+    id: nextStopEventId(),
+    type: 'turn_end',
+    sessionId: chatSessionId,
+    chatSessionId,
+    backend,
+    timestamp: Date.now(),
+    reason: 'aborted',
+    ...(reason === 'slash' ? { backendLabel: 'slash-stop' } : {}),
+  });
+}
+
+/**
+ * Clear the main-process cancelled flag when a new agent turn begins.
+ */
+export async function clearChatSessionCancelled(
+  chatSessionId: string,
+  bridge?: AgentStopBridge | null,
+): Promise<void> {
+  await bridge?.aiSetChatSessionCancelled?.(chatSessionId, false).catch(() => {});
+}