@umsai/ums-code 0.3.0-v2 → 0.6.0-v1

package/dist/src/acp-integration/session/emitters/ToolCallEmitter.js

@@ -0,0 +1,248 @@
+/**
+ * @license
+ * Copyright 2025 Qwen
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseEmitter } from './BaseEmitter.js';
+import { PlanEmitter } from './PlanEmitter.js';
+import { TodoWriteTool, Kind, ExitPlanModeTool, } from '@umsai/ums-code-core';
+/**
+ * Unified tool call event emitter.
+ *
+ * Handles tool_call and tool_call_update for ALL flows:
+ * - Normal tool execution in runTool()
+ * - History replay in HistoryReplayer
+ * - SubAgent tool tracking in SubAgentTracker
+ *
+ * This ensures consistent behavior across all tool event sources,
+ * including special handling for tools like TodoWriteTool.
+ */
+export class ToolCallEmitter extends BaseEmitter {
+    planEmitter;
+    constructor(ctx) {
+        super(ctx);
+        this.planEmitter = new PlanEmitter(ctx);
+    }
+    /**
+     * Emits a tool call start event.
+     *
+     * @param params - Tool call start parameters
+     * @returns true if event was emitted, false if skipped (e.g., TodoWriteTool)
+     */
+    async emitStart(params) {
+        // Skip tool_call for TodoWriteTool - plan updates sent on result
+        if (this.isTodoWriteTool(params.toolName)) {
+            return false;
+        }
+        const { title, locations, kind } = this.resolveToolMetadata(params.toolName, params.args);
+        await this.sendUpdate({
+            sessionUpdate: 'tool_call',
+            toolCallId: params.callId,
+            status: params.status || 'pending',
+            title,
+            content: [],
+            locations,
+            kind,
+            rawInput: params.args ?? {},
+        });
+        return true;
+    }
+    /**
+     * Emits a tool call result event.
+     * Handles TodoWriteTool specially by routing to plan updates.
+     *
+     * @param params - Tool call result parameters
+     */
+    async emitResult(params) {
+        // Handle TodoWriteTool specially - send plan update instead
+        if (this.isTodoWriteTool(params.toolName)) {
+            const todos = this.planEmitter.extractTodos(params.resultDisplay, params.args);
+            // Match original behavior: send plan even if empty when args['todos'] exists
+            // This ensures the UI is updated even when all todos are removed
+            if (todos && todos.length > 0) {
+                await this.planEmitter.emitPlan(todos);
+            }
+            else if (params.args && Array.isArray(params.args['todos'])) {
+                // Send empty plan when args had todos but result has none
+                await this.planEmitter.emitPlan([]);
+            }
+            return; // Skip tool_call_update for TodoWriteTool
+        }
+        // Determine content for the update
+        let contentArray = [];
+        // Special case: diff result from edit tools (format from resultDisplay)
+        const diffContent = this.extractDiffContent(params.resultDisplay);
+        if (diffContent) {
+            contentArray = [diffContent];
+        }
+        else if (params.error) {
+            // Error case: show error message
+            contentArray = [
+                {
+                    type: 'content',
+                    content: { type: 'text', text: params.error.message },
+                },
+            ];
+        }
+        else {
+            // Normal case: transform message parts to ToolCallContent[]
+            contentArray = this.transformPartsToToolCallContent(params.message);
+        }
+        // Build the update
+        const update = {
+            sessionUpdate: 'tool_call_update',
+            toolCallId: params.callId,
+            status: params.success ? 'completed' : 'failed',
+            content: contentArray,
+        };
+        // Add rawOutput from resultDisplay
+        if (params.resultDisplay !== undefined) {
+            update['rawOutput'] = params.resultDisplay;
+        }
+        await this.sendUpdate(update);
+    }
+    /**
+     * Emits a tool call error event.
+     * Use this for explicit error handling when not using emitResult.
+     *
+     * @param callId - The tool call ID
+     * @param error - The error that occurred
+     */
+    async emitError(callId, error) {
+        await this.sendUpdate({
+            sessionUpdate: 'tool_call_update',
+            toolCallId: callId,
+            status: 'failed',
+            content: [
+                { type: 'content', content: { type: 'text', text: error.message } },
+            ],
+        });
+    }
+    // ==================== Public Utilities ====================
+    /**
+     * Checks if a tool name is the TodoWriteTool.
+     * Exposed for external use in components that need to check this.
+     */
+    isTodoWriteTool(toolName) {
+        return toolName === TodoWriteTool.Name;
+    }
+    /**
+     * Checks if a tool name is the ExitPlanModeTool.
+     */
+    isExitPlanModeTool(toolName) {
+        return toolName === ExitPlanModeTool.Name;
+    }
+    /**
+     * Resolves tool metadata from the registry.
+     * Falls back to defaults if tool not found or build fails.
+     *
+     * @param toolName - Name of the tool
+     * @param args - Tool call arguments (used to build invocation)
+     */
+    resolveToolMetadata(toolName, args) {
+        const toolRegistry = this.config.getToolRegistry();
+        const tool = toolRegistry.getTool(toolName);
+        let title = tool?.displayName ?? toolName;
+        let locations = [];
+        let kind = 'other';
+        if (tool && args) {
+            try {
+                const invocation = tool.build(args);
+                title = `${title}: ${invocation.getDescription()}`;
+                // Map locations to ensure line is null instead of undefined (for ACP consistency)
+                locations = invocation.toolLocations().map((loc) => ({
+                    path: loc.path,
+                    line: loc.line ?? null,
+                }));
+                // Pass tool name to handle special cases like exit_plan_mode -> switch_mode
+                kind = this.mapToolKind(tool.kind, toolName);
+            }
+            catch {
+                // Use defaults on build failure
+            }
+        }
+        return { title, locations, kind };
+    }
+    /**
+     * Maps core Tool Kind enum to ACP ToolKind string literals.
+     *
+     * @param kind - The core Kind enum value
+     * @param toolName - Optional tool name to handle special cases like exit_plan_mode
+     */
+    mapToolKind(kind, toolName) {
+        // Special case: exit_plan_mode uses 'switch_mode' kind per ACP spec
+        if (toolName && this.isExitPlanModeTool(toolName)) {
+            return 'switch_mode';
+        }
+        const kindMap = {
+            [Kind.Read]: 'read',
+            [Kind.Edit]: 'edit',
+            [Kind.Delete]: 'delete',
+            [Kind.Move]: 'move',
+            [Kind.Search]: 'search',
+            [Kind.Execute]: 'execute',
+            [Kind.Think]: 'think',
+            [Kind.Fetch]: 'fetch',
+            [Kind.Other]: 'other',
+        };
+        return kindMap[kind] ?? 'other';
+    }
+    // ==================== Private Helpers ====================
+    /**
+     * Extracts diff content from resultDisplay if it's a diff type (edit tool result).
+     * Returns null if not a diff.
+     */
+    extractDiffContent(resultDisplay) {
+        if (!resultDisplay || typeof resultDisplay !== 'object')
+            return null;
+        const obj = resultDisplay;
+        // Check if this is a diff display (edit tool result)
+        if ('fileName' in obj && 'newContent' in obj) {
+            return {
+                type: 'diff',
+                path: obj['fileName'],
+                oldText: obj['originalContent'] ?? '',
+                newText: obj['newContent'],
+            };
+        }
+        return null;
+    }
+    /**
+     * Transforms Part[] to ToolCallContent[].
+     * Extracts text from functionResponse parts and text parts.
+     */
+    transformPartsToToolCallContent(parts) {
+        const result = [];
+        for (const part of parts) {
+            // Handle text parts
+            if ('text' in part && part.text) {
+                result.push({
+                    type: 'content',
+                    content: { type: 'text', text: part.text },
+                });
+            }
+            // Handle functionResponse parts - stringify the response
+            if ('functionResponse' in part && part.functionResponse) {
+                try {
+                    const resp = part.functionResponse.response;
+                    const outputField = resp['output'];
+                    const errorField = resp['error'];
+                    const responseText = typeof outputField === 'string'
+                        ? outputField
+                        : typeof errorField === 'string'
+                            ? errorField
+                            : JSON.stringify(resp);
+                    result.push({
+                        type: 'content',
+                        content: { type: 'text', text: responseText },
+                    });
+                }
+                catch {
+                    // Ignore serialization errors
+                }
+            }
+        }
+        return result;
+    }
+}
+//# sourceMappingURL=ToolCallEmitter.js.map

package/dist/src/acp-integration/session/emitters/ToolCallEmitter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolCallEmitter.js","sourceRoot":"","sources":["../../../../../src/acp-integration/session/emitters/ToolCallEmitter.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAS/C,OAAO,EACL,aAAa,EACb,IAAI,EACJ,gBAAgB,GACjB,MAAM,sBAAsB,CAAC;AAE9B;;;;;;;;;;GAUG;AACH,MAAM,OAAO,eAAgB,SAAQ,WAAW;IAC7B,WAAW,CAAc;IAE1C,YAAY,GAAmB;QAC7B,KAAK,CAAC,GAAG,CAAC,CAAC;QACX,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1C,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CAAC,MAA2B;QACzC,iEAAiE;QACjE,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1C,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,mBAAmB,CACzD,MAAM,CAAC,QAAQ,EACf,MAAM,CAAC,IAAI,CACZ,CAAC;QAEF,MAAM,IAAI,CAAC,UAAU,CAAC;YACpB,aAAa,EAAE,WAAW;YAC1B,UAAU,EAAE,MAAM,CAAC,MAAM;YACzB,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,SAAS;YAClC,KAAK;YACL,OAAO,EAAE,EAAE;YACX,SAAS;YACT,IAAI;YACJ,QAAQ,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;SAC5B,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,UAAU,CAAC,MAA4B;QAC3C,4DAA4D;QAC5D,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CACzC,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,IAAI,CACZ,CAAC;YACF,6EAA6E;YAC7E,iEAAiE;YACjE,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9B,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YACzC,CAAC;iBAAM,IAAI,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;gBAC9D,0DAA0D;gBAC1D,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACtC,CAAC;YACD,OAAO,CAAC,0CAA0C;QACpD,CAAC;QAED,mCAAmC;QACnC,IAAI,YAAY,GAA0B,EAAE,CAAC;QAE7C,wEAAwE;QACxE,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAClE,IAAI,WAAW,EAAE,CAAC;YAChB,YAAY,GAAG,CAAC,WAAW,CAAC,CAAC;QAC/B,CAAC;aAAM,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACxB,iCAAiC;YACjC,YAAY,GAAG;gBACb;oBACE,IAAI,EAAE,SAAS;oBACf,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE;iBACtD;aACF,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,4DAA4D;YAC5D,YAAY,GAAG,IAAI,CAAC,+BAA+B,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACtE,CAAC;QAED,mBAAmB;QACnB,MAAM,MAAM,GAA0C;YACpD,aAAa,EAAE,kBAAkB;YACjC,UAAU,EAAE,MAAM,CAAC,MAAM;YACzB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;YAC/C,OAAO,EAAE,YAAY;SACtB,CAAC;QAEF,mCAAmC;QACnC,IAAI,MAAM,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;YACtC,MAAkC,CAAC,WAAW,CAAC,GAAG,MAAM,CAAC,aAAa,CAAC;QAC1E,CAAC;QAED,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,KAAY;QAC1C,MAAM,IAAI,CAAC,UAAU,CAAC;YACpB,aAAa,EAAE,kBAAkB;YACjC,UAAU,EAAE,MAAM;YAClB,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE;aACpE;SACF,CAAC,CAAC;IACL,CAAC;IAED,6DAA6D;IAE7D;;;OAGG;IACH,eAAe,CAAC,QAAgB;QAC9B,OAAO,QAAQ,KAAK,aAAa,CAAC,IAAI,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,QAAgB;QACjC,OAAO,QAAQ,KAAK,gBAAgB,CAAC,IAAI,CAAC;IAC5C,CAAC;IAED;;;;;;OAMG;IACH,mBAAmB,CACjB,QAAgB,EAChB,IAA8B;QAE9B,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;QACnD,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAE5C,IAAI,KAAK,GAAG,IAAI,EAAE,WAAW,IAAI,QAAQ,CAAC;QAC1C,IAAI,SAAS,GAA2B,EAAE,CAAC;QAC3C,IAAI,IAAI,GAAiB,OAAO,CAAC;QAEjC,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;YACjB,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACpC,KAAK,GAAG,GAAG,KAAK,KAAK,UAAU,CAAC,cAAc,EAAE,EAAE,CAAC;gBACnD,kFAAkF;gBAClF,SAAS,GAAG,UAAU,CAAC,aAAa,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBACnD,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,IAAI;iBACvB,CAAC,CAAC,CAAC;gBACJ,4EAA4E;gBAC5E,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YAC/C,CAAC;YAAC,MAAM,CAAC;gBACP,gCAAgC;YAClC,CAAC;QACH,CAAC;QAED,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;IACpC,CAAC;IAED;;;;;OAKG;IACH,WAAW,CAAC,IAAU,EAAE,QAAiB;QACvC,oEAAoE;QACpE,IAAI,QAAQ,IAAI,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClD,OAAO,aAAa,CAAC;QACvB,CAAC;QAED,MAAM,OAAO,GAA+B;YAC1C,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM;YACnB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM;YACnB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,QAAQ;YACvB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM;YACnB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,QAAQ;YACvB,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,SAAS;YACzB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO;YACrB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO;YACrB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO;SACtB,CAAC;QACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC;IAClC,CAAC;IAED,4DAA4D;IAE5D;;;OAGG;IACK,kBAAkB,CACxB,aAAsB;QAEtB,IAAI,CAAC,aAAa,IAAI,OAAO,aAAa,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAC;QAErE,MAAM,GAAG,GAAG,aAAwC,CAAC;QAErD,qDAAqD;QACrD,IAAI,UAAU,IAAI,GAAG,IAAI,YAAY,IAAI,GAAG,EAAE,CAAC;YAC7C,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,GAAG,CAAC,UAAU,CAAW;gBAC/B,OAAO,EAAG,GAAG,CAAC,iBAAiB,CAAY,IAAI,EAAE;gBACjD,OAAO,EAAE,GAAG,CAAC,YAAY,CAAW;aACrC,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;OAGG;IACK,+BAA+B,CACrC,KAAa;QAEb,MAAM,MAAM,GAA0B,EAAE,CAAC;QAEzC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,oBAAoB;YACpB,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBAChC,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,SAAS;oBACf,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE;iBAC3C,CAAC,CAAC;YACL,CAAC;YAED,yDAAyD;YACzD,IAAI,kBAAkB,IAAI,IAAI,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACxD,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAGlC,CAAC;oBACF,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;oBACnC,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;oBACjC,MAAM,YAAY,GAChB,OAAO,WAAW,KAAK,QAAQ;wBAC7B,CAAC,CAAC,WAAW;wBACb,CAAC,CAAC,OAAO,UAAU,KAAK,QAAQ;4BAC9B,CAAC,CAAC,UAAU;4BACZ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;oBAC7B,MAAM,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,SAAS;wBACf,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE;qBAC9C,CAAC,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC;oBACP,8BAA8B;gBAChC,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}

package/dist/src/acp-integration/session/emitters/ToolCallEmitter.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Qwen
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};