@di-code/coding-agent 0.1.4 → 0.1.7

package/dist/modes/interactive.js

@@ -1,31 +1,38 @@
 import { statSync } from "node:fs";
-import { extname, resolve } from "node:path";
+import { homedir } from "node:os";
+import { extname, join, resolve } from "node:path";
 import { CombinedAutocompleteProvider, Editor, Key, matchesKey, SelectList, SettingsList, } from "@di-code/tui";
 import { cleanupStaleClipboardImages, clipboardImageDirectory, isClipboardImagePath, readClipboardImagePath, removeClipboardImage, } from "../core/clipboard-image.js";
 import { extractImageAttachments } from "../core/image-input.js";
+import { DEFAULT_LOCALE, translate } from "../i18n.js";
 import { showInteractiveProviderOnboarding, } from "../provider-onboarding.js";
-import { loadStartupConfiguration, removeGlobalProviderApiKey, resolveStartupRuntime } from "../startup.js";
+import { removeGlobalProviderApiKey, saveGlobalLocale, saveGlobalModelSelection, saveGlobalThinkingLevel, } from "../startup.js";
 import { AutocompleteMenu, InteractiveChat, InteractiveComposer, InteractiveFooter, InteractiveHeader, } from "./interactive-components.js";
 import { InteractiveLayout } from "./interactive-layout.js";
 import { InteractiveProjection } from "./interactive-state.js";
+import { TreeSelector } from "./tree-selector.js";
 export { InteractiveProjection } from "./interactive-state.js";
 const IMAGE_EXTENSIONS = new Set([".gif", ".jpeg", ".jpg", ".png", ".webp"]);
 const PASTE_IMAGE_KEY = process.platform === "win32" ? Key.alt("v") : Key.ctrl("v");
 const PASTE_IMAGE_SHORTCUT = process.platform === "win32" ? "Alt+V" : "Ctrl+V";
-const BUILTIN_SLASH_COMMANDS = [
-    { name: "help", description: "Show interactive commands" },
-    { name: "clear", description: "Clear visible messages" },
-    { name: "model", description: "Open the model selector" },
-    { name: "session", description: "Open the session selector" },
-    { name: "theme", description: "Open the theme selector" },
-    { name: "settings", description: "Open the settings selector" },
-    { name: "login", description: "Choose a Provider and save its global API key" },
-    { name: "logout", description: "Remove the current Provider global API key" },
-    { name: "compact", description: "Compact the persisted context now" },
-    { name: "usage", description: "Show token and cost usage" },
-    { name: "retry", description: "Retry the last failed prompt" },
-    { name: "steer", description: "Steer the running agent" },
-];
+function builtinSlashCommands(locale) {
+    const t = (key) => translate(locale, key);
+    return [
+        { name: "help", description: t("showInteractiveCommands") },
+        { name: "clear", description: t("clearVisibleMessages") },
+        { name: "model", description: t("openModelSelector") },
+        { name: "session", description: t("openSessionSelector") },
+        { name: "tree", description: t("openTreeSelector") },
+        { name: "theme", description: t("openThemeSelector") },
+        { name: "settings", description: t("openSettingsSelector") },
+        { name: "login", description: t("chooseProvider") },
+        { name: "logout", description: t("removeProviderKey") },
+        { name: "compact", description: t("compactContext") },
+        { name: "usage", description: t("showUsage") },
+        { name: "retry", description: t("retryPrompt") },
+        { name: "steer", description: t("steerAgent") },
+    ];
+}
 function asImageAttachmentReference(pasted, cwd) {
     if (pasted.includes("\n"))
         return pasted;
@@ -66,6 +73,7 @@ export class InteractiveMode {
     sessionChoices;
     extensionHost;
     providerOnboarding;
+    agentDir;
     readClipboardImagePath;
     clipboardDirectory;
     clipboardFiles = new Set();
@@ -85,6 +93,7 @@ export class InteractiveMode {
     spinnerTimer;
     shutdownEmitted = false;
     preserveClipboardFiles = false;
+    locale;
     constructor(options) {
         this.session = options.session;
         this.projection.configureFilePreview(this.session.allowedRoot, () => this.refresh());
@@ -93,8 +102,10 @@ export class InteractiveMode {
         this.sessionChoices = [...(options.sessions ?? [])];
         this.extensionHost = options.extensionHost;
         this.providerOnboarding = options.providerOnboarding;
+        this.agentDir = resolve(options.agentDir ?? join(homedir(), ".di-code"));
+        this.locale = options.locale ?? DEFAULT_LOCALE;
         this.readClipboardImagePath = options.readClipboardImagePath ?? readClipboardImagePath;
-        this.clipboardDirectory = clipboardImageDirectory(this.session.allowedRoot);
+        this.clipboardDirectory = clipboardImageDirectory(this.agentDir, this.session.allowedRoot);
         const autocomplete = {
             getSuggestions: (context, autocompleteOptions) => new CombinedAutocompleteProvider(this.listSlashCommands(), this.session.allowedRoot).getSuggestions(context, autocompleteOptions),
             applyCompletion: (context, item, prefix) => new CombinedAutocompleteProvider(this.listSlashCommands(), this.session.allowedRoot).applyCompletion(context, item, prefix),
@@ -102,9 +113,16 @@ export class InteractiveMode {
         this.editor = new Editor({
             maxHeight: 3,
             autocomplete,
+            submitAutocomplete: (context) => /^\/[^\s/]*$/.test(context.text.slice(0, context.cursor)),
         });
         this.editor.onSubmit = (text) => void this.submit(text);
-        this.editor.onEscape = () => this.activeAbort?.abort();
+        this.editor.onEscape = () => {
+            if (!this.activeAbort)
+                return;
+            this.projection.setStatus(translate(this.locale, "cancelled"));
+            this.activeAbort.abort();
+            this.refresh();
+        };
         this.editor.onCommand = (data) => this.handleCommand(data);
         this.editor.onInterrupt = () => this.exit();
         this.editor.onChange = () => {
@@ -120,6 +138,7 @@ export class InteractiveMode {
             ...this.projection.state,
             model: `${this.session.modelId}${this.session.thinkingLevel ? `(${this.session.thinkingLevel})` : ""}`,
             theme: this.theme,
+            locale: this.locale,
             pasteImageShortcut: PASTE_IMAGE_SHORTCUT,
         });
         this.root = new InteractiveLayout({
@@ -141,7 +160,7 @@ export class InteractiveMode {
         this.projection.replaceTranscript(this.session.transcript);
         this.projection.setUsage(this.session.usage);
         this.subscribeToSession();
-        void cleanupStaleClipboardImages(this.session.allowedRoot).catch((cause) => {
+        void cleanupStaleClipboardImages(this.agentDir, this.session.allowedRoot).catch((cause) => {
             this.projection.setError(cause instanceof Error ? cause.message : String(cause));
             this.refresh();
         });
@@ -190,6 +209,12 @@ export class InteractiveMode {
             try {
                 const level = this.session.cycleThinkingLevel();
                 this.projection.setStatus(level ? `thinking=${level}` : "Current model does not support thinking levels.");
+                if (level) {
+                    void saveGlobalThinkingLevel(this.agentDir, this.session.providerId, this.session.modelId, level).catch((cause) => {
+                        this.projection.setError(cause instanceof Error ? cause.message : String(cause));
+                        this.refresh();
+                    });
+                }
             }
             catch (cause) {
                 this.projection.setError(cause instanceof Error ? cause.message : String(cause));
@@ -227,7 +252,7 @@ export class InteractiveMode {
         try {
             const path = await this.readClipboardImagePath(this.clipboardDirectory);
             if (!path) {
-                this.projection.setStatus("Clipboard has no supported image.");
+                this.projection.setStatus(this.locale === "zh-CN" ? "剪贴板中没有受支持的图片。" : "Clipboard has no supported image.");
             }
             else {
                 this.editor.insertTextAtCursor(path);
@@ -241,9 +266,10 @@ export class InteractiveMode {
         this.refresh();
     }
     openThemeSelector() {
+        const t = (key) => translate(this.locale, key);
         const list = new SelectList([
-            { value: "dark", label: "Dark", description: "Dark terminal theme" },
-            { value: "light", label: "Light", description: "Light terminal theme" },
+            { value: "dark", label: t("dark"), description: t("darkTheme") },
+            { value: "light", label: t("light"), description: t("lightTheme") },
         ]);
         list.onSelect = (item) => {
             this.theme = item.value === "light" ? "light" : "dark";
@@ -264,6 +290,10 @@ export class InteractiveMode {
             try {
                 this.session.setModel(item.value);
                 this.projection.setStatus(`model=${item.value}`);
+                void saveGlobalModelSelection(this.agentDir, this.session.providerId, this.session.modelId).catch((cause) => {
+                    this.projection.setError(cause instanceof Error ? cause.message : String(cause));
+                    this.refresh();
+                });
                 this.closeOverlay();
                 this.refresh();
             }
@@ -275,11 +305,12 @@ export class InteractiveMode {
         this.showOverlay(list);
     }
     openSessionSelector() {
+        const t = (key) => translate(this.locale, key);
         const list = new SelectList([
             {
                 value: "__current__",
-                label: "Current session",
-                description: this.session.sessionFile ?? "In-memory session",
+                label: t("currentSession"),
+                description: this.session.sessionFile ?? t("inMemorySession"),
             },
             ...this.sessionChoices.map((choice) => ({
                 value: choice.id,
@@ -303,28 +334,124 @@ export class InteractiveMode {
         list.onCancel = () => this.closeOverlay();
         this.showOverlay(list);
     }
+    openTreeSelector() {
+        if (this.promptInFlight || this.compactionInFlight || this.sessionSwitching) {
+            this.projection.setError(this.locale === "zh-CN"
+                ? "提示词运行时不能浏览会话树。"
+                : "Cannot browse the session tree while a prompt is running.");
+            this.refresh();
+            return;
+        }
+        if (this.session.sessionTree.length === 0) {
+            this.projection.setStatus(this.locale === "zh-CN" ? "当前会话还没有历史节点。" : "The current session has no history nodes.");
+            this.refresh();
+            return;
+        }
+        const selector = new TreeSelector({
+            nodes: this.session.sessionTree,
+            leafId: this.session.sessionLeafId,
+            locale: this.locale,
+            onContinue: (entry) => {
+                this.closeOverlay();
+                void this.navigateTreeEntry(entry.id);
+            },
+            onEdit: (entry) => {
+                if (entry.type !== "message" || entry.message.role !== "user") {
+                    this.projection.setError(this.locale === "zh-CN" ? "只能编辑历史用户消息。" : "Only historical user messages can be edited.");
+                    this.refresh();
+                    return;
+                }
+                this.closeOverlay();
+                void this.navigateTreeEntry(entry.id);
+            },
+            onSummarize: (entry) => {
+                this.closeOverlay();
+                void this.summarizeTreeBranch(entry.id);
+            },
+            onCancel: () => this.closeOverlay(),
+        });
+        this.showOverlay(selector, true, "80%");
+    }
+    async navigateTreeEntry(entryId) {
+        try {
+            const result = await this.session.navigateTree(entryId);
+            if (!this.started)
+                return false;
+            this.projection.replaceTranscript(this.session.transcript);
+            this.editor.setValue(result.editorText ?? "");
+            this.projection.setStatus(result.imagesOmitted
+                ? this.locale === "zh-CN"
+                    ? "已恢复文本；请重新附加图片。"
+                    : "Text restored; reattach images before sending."
+                : `tree=${result.selectedEntryId}`);
+            return true;
+        }
+        catch (cause) {
+            this.projection.setError(cause instanceof Error ? cause.message : String(cause));
+            return false;
+        }
+        finally {
+            this.refresh();
+        }
+    }
+    async summarizeTreeBranch(entryId) {
+        if (this.compactionInFlight || this.promptInFlight)
+            return;
+        this.compactionInFlight = true;
+        this.projection.setStatus(this.locale === "zh-CN" ? "正在为分支生成摘要。" : "Summarizing selected branch.");
+        this.refresh();
+        try {
+            if (!(await this.navigateTreeEntry(entryId)))
+                return;
+            await this.session.compact();
+            this.editor.setValue("");
+            this.projection.replaceTranscript(this.session.transcript);
+            this.projection.setStatus(this.locale === "zh-CN" ? "摘要分支已创建。" : "Summary branch created.");
+        }
+        catch (cause) {
+            this.projection.setError(cause instanceof Error ? cause.message : String(cause));
+        }
+        finally {
+            this.compactionInFlight = false;
+            this.refresh();
+        }
+    }
     openSettingsSelector() {
+        const t = (key) => translate(this.locale, key);
         const list = new SettingsList([
             {
                 id: "compaction",
-                label: "Context compaction",
-                currentValue: this.session.compactionEnabled ? "on" : "off",
-                values: ["on", "off"],
-                description: "Summarize older persisted context before the model limit is reached.",
+                label: t("contextCompaction"),
+                currentValue: this.session.compactionEnabled ? t("on") : t("off"),
+                values: [t("on"), t("off")],
             },
-        ]);
+            {
+                id: "locale",
+                label: t("language"),
+                currentValue: this.locale,
+                values: ["en", "zh-CN"],
+            },
+        ], { title: t("settings") });
         list.onChange = (id, value) => {
+            if (id === "locale") {
+                this.locale = value === "zh-CN" ? "zh-CN" : "en";
+                void saveGlobalLocale(this.agentDir, this.locale).catch((cause) => {
+                    this.projection.setError(cause instanceof Error ? cause.message : String(cause));
+                    this.refresh();
+                });
+                this.closeOverlay();
+                this.openSettingsSelector();
+                return;
+            }
             if (id !== "compaction")
                 return;
             try {
-                const enabled = this.session.setCompactionEnabled(value === "on");
-                list.updateValue(id, enabled ? "on" : "off");
-                this.projection.setStatus(enabled === (value === "on")
-                    ? `compaction=${enabled ? "on" : "off"}`
-                    : "compaction unavailable without a persisted session");
+                const enabled = this.session.setCompactionEnabled(value === t("on"));
+                list.updateValue(id, enabled ? t("on") : t("off"));
+                this.projection.setStatus(enabled === (value === t("on")) ? `compaction=${enabled ? t("on") : t("off")}` : t("compactionUnavailable"));
             }
             catch (cause) {
-                list.updateValue(id, this.session.compactionEnabled ? "on" : "off");
+                list.updateValue(id, this.session.compactionEnabled ? t("on") : t("off"));
                 this.projection.setError(cause instanceof Error ? cause.message : String(cause));
                 this.closeOverlay();
             }
@@ -333,15 +460,16 @@ export class InteractiveMode {
         list.onCancel = () => this.closeOverlay();
         this.showOverlay(list);
     }
-    showOverlay(component) {
+    showOverlay(component, preserveLastLine = false, maxHeight = "60%") {
         this.editor.cancelAutocomplete();
         this.closeAutocompleteOverlay();
         this.closeOverlay();
         this.overlay = this.tui.showOverlay(component, {
             width: "70%",
-            maxHeight: "60%",
+            maxHeight,
             anchor: "center",
             margin: 1,
+            preserveLastLine,
         });
     }
     closeOverlay() {
@@ -361,7 +489,7 @@ export class InteractiveMode {
         const editorBounds = this.root.getEditorBounds(this.tui.columns);
         this.autocompleteOverlay = this.tui.showOverlay(menu, {
             width: "55%",
-            maxHeight: 8,
+            maxHeight: 9,
             anchor: "bottom-left",
             placement: {
                 anchorRow: editorBounds.end - 1,
@@ -369,6 +497,7 @@ export class InteractiveMode {
                 preferred: "below",
             },
             margin: 1,
+            preserveLastLine: true,
             nonCapturing: true,
         });
         this.refresh();
@@ -399,6 +528,11 @@ export class InteractiveMode {
         this.unsubscribeSession = this.session.subscribeSession((event) => {
             if (!this.started)
                 return;
+            if (event.type === "tree_navigated") {
+                this.projection.replaceTranscript(this.session.transcript);
+                this.refresh();
+                return;
+            }
             if (event.type === "queue_update") {
                 this.steeringPrompts = [...event.steering];
                 this.refreshQueue();
@@ -411,7 +545,7 @@ export class InteractiveMode {
     }
     async switchSession(choice) {
         if (this.promptInFlight || this.sessionSwitching) {
-            this.projection.setError("Cannot switch sessions while a prompt is running.");
+            this.projection.setError(this.locale === "zh-CN" ? "提示词运行时不能切换会话。" : "Cannot switch sessions while a prompt is running.");
             this.refresh();
             return;
         }
@@ -425,8 +559,8 @@ export class InteractiveMode {
             this.unsubscribeSession?.();
             this.session = next;
             this.projection.configureFilePreview(next.allowedRoot, () => this.refresh());
-            this.clipboardDirectory = clipboardImageDirectory(next.allowedRoot);
-            void cleanupStaleClipboardImages(next.allowedRoot);
+            this.clipboardDirectory = clipboardImageDirectory(this.agentDir, next.allowedRoot);
+            void cleanupStaleClipboardImages(this.agentDir, next.allowedRoot);
             this.queuedPrompts = [];
             this.steeringPrompts = [];
             this.refreshQueue();
@@ -458,12 +592,16 @@ export class InteractiveMode {
             return;
         }
         if (this.sessionSwitching) {
-            this.projection.setError("A session is opening; wait before submitting a prompt.");
+            this.projection.setError(this.locale === "zh-CN"
+                ? "会话正在打开，请等待后再提交提示词。"
+                : "A session is opening; wait before submitting a prompt.");
             this.refresh();
             return;
         }
         if (this.compactionInFlight) {
-            this.projection.setError("A compaction is running; wait before submitting a prompt.");
+            this.projection.setError(this.locale === "zh-CN"
+                ? "正在压缩上下文，请等待后再提交提示词。"
+                : "A compaction is running; wait before submitting a prompt.");
             this.refresh();
             return;
         }
@@ -509,22 +647,28 @@ export class InteractiveMode {
     async submitSteering(text) {
         const prompt = text.trim();
         if (prompt.length === 0) {
-            this.projection.setError("Steering content must not be empty.");
+            this.projection.setError(this.locale === "zh-CN" ? "引导内容不能为空。" : "Steering content must not be empty.");
             this.refresh();
             return;
         }
         if (this.sessionSwitching) {
-            this.projection.setError("A session is opening; wait before steering the agent.");
+            this.projection.setError(this.locale === "zh-CN"
+                ? "会话正在打开，请等待后再引导 Agent。"
+                : "A session is opening; wait before steering the agent.");
             this.refresh();
             return;
         }
         if (this.compactionInFlight) {
-            this.projection.setError("A compaction is running; wait before steering the agent.");
+            this.projection.setError(this.locale === "zh-CN"
+                ? "正在压缩上下文，请等待后再引导 Agent。"
+                : "A compaction is running; wait before steering the agent.");
             this.refresh();
             return;
         }
         if (!this.promptInFlight) {
-            this.projection.setError("Steering is only available while a prompt is running.");
+            this.projection.setError(this.locale === "zh-CN"
+                ? "只能在提示词运行时引导 Agent。"
+                : "Steering is only available while a prompt is running.");
             this.refresh();
             return;
         }
@@ -570,7 +714,7 @@ export class InteractiveMode {
                 break;
             case "clear":
                 this.projection.clearVisibleMessages();
-                this.projection.setStatus("visible messages cleared");
+                this.projection.setStatus(translate(this.locale, "visibleMessagesCleared"));
                 break;
             case "model":
                 this.openModelSelector();
@@ -578,6 +722,9 @@ export class InteractiveMode {
             case "session":
                 this.openSessionSelector();
                 return;
+            case "tree":
+                this.openTreeSelector();
+                return;
             case "theme":
                 this.openThemeSelector();
                 return;
@@ -600,7 +747,7 @@ export class InteractiveMode {
                 if (this.lastFailedPrompt && !this.promptInFlight)
                     void this.submit(this.lastFailedPrompt, true);
                 else
-                    this.projection.setStatus("nothing to retry");
+                    this.projection.setStatus(translate(this.locale, "nothingToRetry"));
                 break;
             default: {
                 if (this.extensionHost?.listCommands().some((entry) => entry.name === command)) {
@@ -628,7 +775,11 @@ export class InteractiveMode {
         }
         this.editor.cancelAutocomplete();
         this.closeAutocompleteOverlay();
-        void showInteractiveProviderOnboarding({ ...this.providerOnboarding, tui: this.tui })
+        void showInteractiveProviderOnboarding({
+            ...this.providerOnboarding,
+            configuration: { ...this.providerOnboarding.configuration, locale: this.locale },
+            tui: this.tui,
+        })
             .then((runtime) => {
             if (!this.started)
                 return;
@@ -660,7 +811,6 @@ export class InteractiveMode {
             return;
         }
         const providerId = this.session.providerId;
-        const modelId = this.session.modelId;
         try {
             const removed = await removeGlobalProviderApiKey(this.providerOnboarding.agentDir, providerId);
             if (!removed) {
@@ -668,14 +818,7 @@ export class InteractiveMode {
                 this.refresh();
                 return;
             }
-            const configuration = await loadStartupConfiguration(this.session.allowedRoot, this.providerOnboarding.configuration.environment, this.providerOnboarding.agentDir);
-            const runtime = resolveStartupRuntime({
-                ...configuration.environment,
-                DI_CODE_PROVIDER: providerId,
-                DI_CODE_MODEL: modelId,
-            }, configuration.providers, configuration.defaults);
-            this.session.setRuntime(runtime.provider, runtime.model);
-            this.projection.setStatus(`global API key removed for provider=${providerId}; environment variables may still apply`);
+            this.projection.setStatus(`global API key removed for provider=${providerId}; it remains active until this session ends`);
         }
         catch (cause) {
             this.projection.setError(cause instanceof Error ? cause.message : String(cause));
@@ -687,7 +830,7 @@ export class InteractiveMode {
             name: `skill:${skill.name}`,
             description: skill.description,
         }));
-        return [...BUILTIN_SLASH_COMMANDS, ...(this.extensionHost?.listCommands() ?? []), ...skills];
+        return [...builtinSlashCommands(this.locale), ...(this.extensionHost?.listCommands() ?? []), ...skills];
     }
     isLoadedSkillCommand(input) {
         const name = /^\/skill:([a-z0-9]+(?:-[a-z0-9]+)*)(?:\s|$)/.exec(input)?.[1];
@@ -695,20 +838,23 @@ export class InteractiveMode {
     }
     formatUsage() {
         const usage = this.session.usage;
+        if (this.locale === "zh-CN") {
+            return `用量：请求=${usage.requestCount} 输入=${usage.inputTokens} 输出=${usage.outputTokens} 总计=${usage.totalTokens} 费用=$${usage.cost.total.toFixed(4)} 上下文=${usage.estimatedContextTokens}/${usage.contextWindow}`;
+        }
         return `usage: requests=${usage.requestCount} input=${usage.inputTokens} output=${usage.outputTokens} total=${usage.totalTokens} cost=$${usage.cost.total.toFixed(4)} context=${usage.estimatedContextTokens}/${usage.contextWindow}`;
     }
     async runManualCompaction() {
         if (this.promptInFlight || this.compactionInFlight) {
-            this.projection.setError("Cannot compact while a prompt is running.");
+            this.projection.setError(this.locale === "zh-CN" ? "提示词运行时不能压缩上下文。" : "Cannot compact while a prompt is running.");
             this.refresh();
             return;
         }
         this.compactionInFlight = true;
-        this.projection.setStatus("starting manual compaction");
+        this.projection.setStatus(translate(this.locale, "manualCompactionStarting"));
         this.refresh();
         try {
             await this.session.compact();
-            this.projection.setStatus("manual compaction complete");
+            this.projection.setStatus(translate(this.locale, "manualCompactionComplete"));
         }
         catch (cause) {
             this.projection.setError(cause instanceof Error ? cause.message : String(cause));