@fleetagent/pi-coding-agent 0.0.9 → 0.0.11

package/dist/modes/interactive/interactive-mode.js

@@ -11,6 +11,7 @@ import { CombinedAutocompleteProvider, Container, fuzzyFilter, getCapabilities,
 import { spawn, spawnSync } from "child_process";
 import { APP_NAME, APP_TITLE, getAgentDir, getAuthPath, getDebugLogPath, getDocsPath, getShareViewerUrl, PACKAGE_NAME, VERSION, } from "../../config.js";
 import { parseSkillBlock } from "../../core/agent-session.js";
+import { executeBashWithOperations } from "../../core/bash-executor.js";
 import { FooterDataProvider } from "../../core/footer-data-provider.js";
 import { KeybindingsManager } from "../../core/keybindings.js";
 import { createCompactionSummaryMessage } from "../../core/messages.js";
@@ -21,6 +22,7 @@ import { BUILT_IN_PROVIDER_DISPLAY_NAMES } from "../../core/provider-display-nam
 import { formatMissingSessionCwdPrompt, MissingSessionCwdError } from "../../core/session-cwd.js";
 import { LocalSessionManager } from "../../core/session-manager.js";
 import { BUILTIN_SLASH_COMMANDS } from "../../core/slash-commands.js";
+import { getSourceBackendIcon } from "../../core/source-info.js";
 import { isInstallTelemetryEnabled } from "../../core/telemetry.js";
 import { getChangelogPath, getNewEntries, parseChangelog } from "../../utils/changelog.js";
 import { copyToClipboard } from "../../utils/clipboard.js";
@@ -265,7 +267,8 @@ export class InteractiveMode {
         if (!sourceTag) {
             return description;
         }
-        return description ? `[${sourceTag}] ${description}` : `[${sourceTag}]`;
+        const backendIcon = getSourceBackendIcon(sourceInfo);
+        return description ? `[${backendIcon} ${sourceTag}] ${description}` : `[${backendIcon} ${sourceTag}]`;
     }
     getBuiltInCommandConflictDiagnostics(extensionRunner) {
         const builtinNames = new Set(BUILTIN_SLASH_COMMANDS.map((command) => command.name));
@@ -914,7 +917,8 @@ export class InteractiveMode {
         if (sourceInfo) {
             const shortPath = this.getShortPath(p, sourceInfo);
             const { label, scopeLabel } = this.getDisplaySourceInfo(sourceInfo);
-            const labelText = scopeLabel ? `${label} (${scopeLabel})` : label;
+            const backendIcon = getSourceBackendIcon(sourceInfo);
+            const labelText = scopeLabel ? `${backendIcon} ${label} (${scopeLabel})` : `${backendIcon} ${label}`;
             return `${labelText} ${shortPath}`;
         }
         return this.formatDisplayPath(p);
@@ -1017,8 +1021,10 @@ export class InteractiveMode {
             const contextFiles = this.session.resourceLoader.getAgentsFiles().agentsFiles;
             if (contextFiles.length > 0) {
                 this.chatContainer.addChild(new Spacer(1));
+                const contextBackendInfo = this.session.getToolBackendInfo?.();
+                const contextBackendIcon = contextBackendInfo?.type === "ssh" ? "☁" : "🖥";
                 const contextList = contextFiles
-                    .map((f) => theme.fg("dim", `  ${this.formatDisplayPath(f.path)}`))
+                    .map((f) => theme.fg("dim", `  ${f.path.startsWith(getAgentDir()) ? "🖥" : contextBackendIcon} ${this.formatDisplayPath(f.path)}`))
                     .join("\n");
                 const contextCompactList = formatCompactList(contextFiles.map((contextFile) => this.formatContextPath(contextFile.path)), { sort: false });
                 addLoadedSection("Context", contextCompactList, contextList);
@@ -1027,8 +1033,8 @@ export class InteractiveMode {
             if (skills.length > 0) {
                 const groups = this.buildScopeGroups(skills.map((skill) => ({ path: skill.filePath, sourceInfo: skill.sourceInfo })));
                 const skillList = this.formatScopeGroups(groups, {
-                    formatPath: (item) => this.formatDisplayPath(item.path),
-                    formatPackagePath: (item) => this.getShortPath(item.path, item.sourceInfo),
+                    formatPath: (item) => `${getSourceBackendIcon(item.sourceInfo)} ${this.formatDisplayPath(item.path)}`,
+                    formatPackagePath: (item) => `${getSourceBackendIcon(item.sourceInfo)} ${this.getShortPath(item.path, item.sourceInfo)}`,
                 });
                 const skillCompactList = formatCompactList(skills.map((skill) => skill.name));
                 addLoadedSection("Skills", skillCompactList, skillList);
@@ -1037,8 +1043,8 @@ export class InteractiveMode {
             if (rules.length > 0) {
                 const groups = this.buildScopeGroups(rules.map((rule) => ({ path: rule.filePath, sourceInfo: rule.sourceInfo })));
                 const ruleList = this.formatScopeGroups(groups, {
-                    formatPath: (item) => this.formatDisplayPath(item.path),
-                    formatPackagePath: (item) => this.getShortPath(item.path, item.sourceInfo),
+                    formatPath: (item) => `${getSourceBackendIcon(item.sourceInfo)} ${this.formatDisplayPath(item.path)}`,
+                    formatPackagePath: (item) => `${getSourceBackendIcon(item.sourceInfo)} ${this.getShortPath(item.path, item.sourceInfo)}`,
                 });
                 const ruleCompactList = formatCompactList(rules.map((rule) => rule.name));
                 addLoadedSection("Rules", ruleCompactList, ruleList);
@@ -1050,11 +1056,11 @@ export class InteractiveMode {
                 const templateList = this.formatScopeGroups(groups, {
                     formatPath: (item) => {
                         const template = templateByPath.get(item.path);
-                        return template ? `/${template.name}` : this.formatDisplayPath(item.path);
+                        return `${getSourceBackendIcon(item.sourceInfo)} ${template ? `/${template.name}` : this.formatDisplayPath(item.path)}`;
                     },
                     formatPackagePath: (item) => {
                         const template = templateByPath.get(item.path);
-                        return template ? `/${template.name}` : this.formatDisplayPath(item.path);
+                        return `${getSourceBackendIcon(item.sourceInfo)} ${template ? `/${template.name}` : this.formatDisplayPath(item.path)}`;
                     },
                 });
                 const promptCompactList = formatCompactList(templates.map((template) => `/${template.name}`));
@@ -1211,6 +1217,7 @@ export class InteractiveMode {
         });
         setRegisteredThemes(this.session.resourceLoader.getThemes().themes);
         this.setupAutocompleteProvider();
+        this.updateToolBackendStatus();
         const extensionRunner = this.session.extensionRunner;
         this.setupExtensionShortcuts(extensionRunner);
         this.showLoadedResources({ force: false, showDiagnosticsWhenQuiet: true });
@@ -1276,6 +1283,12 @@ export class InteractiveMode {
             ui: this.createExtensionUIContext(),
             hasUI: true,
             cwd: this.activeSession.getCwd(),
+            toolOperations: this.session.getToolOperations(),
+            getToolBackendInfo: () => this.session.getToolBackendInfo(),
+            execToolBackend: (command, options) => {
+                const operations = this.session.getToolOperations();
+                return executeBashWithOperations(command, options?.cwd ?? operations.cwd, operations, options);
+            },
             session: this.activeSession,
             modelRegistry: this.session.modelRegistry,
             model: this.session.model,
@@ -1325,6 +1338,18 @@ export class InteractiveMode {
         this.footerDataProvider.setExtensionStatus(key, text);
         this.ui.requestRender();
     }
+    formatToolBackendStatus(info) {
+        if (info.type === "local") {
+            return theme.fg("dim", `tools: local ${info.cwd}`);
+        }
+        if (info.configured) {
+            return theme.fg("accent", `tools: ssh ${info.remote}:${info.cwd}`);
+        }
+        return theme.fg("warning", `tools: ssh not configured ${info.cwd}`);
+    }
+    updateToolBackendStatus() {
+        this.setExtensionStatus("toolBackend", this.formatToolBackendStatus(this.session.getToolBackendInfo()));
+    }
     getWorkingLoaderMessage() {
         return this.workingMessage ?? this.defaultWorkingMessage;
     }
@@ -2038,6 +2063,11 @@ export class InteractiveMode {
                 this.editor.setText("");
                 return;
             }
+            if (text === "/ssh-sandbox" || text.startsWith("/ssh-sandbox ")) {
+                this.editor.setText("");
+                await this.handleSshSandboxCommand(text);
+                return;
+            }
             if (text === "/changelog") {
                 this.handleChangelogCommand();
                 this.editor.setText("");
@@ -4365,6 +4395,40 @@ export class InteractiveMode {
         this.chatContainer.addChild(new Text(theme.fg("dim", `Session name set: ${name}`), 1, 0));
         this.ui.requestRender();
     }
+    async handleSshSandboxCommand(text) {
+        const args = text.replace(/^\/ssh-sandbox\s*/, "").trim();
+        if (!args || args === "status") {
+            this.showStatus(this.formatToolBackendStatus(this.session.getToolBackendInfo()));
+            return;
+        }
+        if (args === "clear") {
+            try {
+                this.session.clearSshSandbox();
+                this.updateToolBackendStatus();
+                this.showStatus("SSH sandbox cleared");
+            }
+            catch (error) {
+                this.showError(error instanceof Error ? error.message : String(error));
+            }
+            return;
+        }
+        const [targetArg, cwdArg] = args.split(/\s+/, 2);
+        if (!targetArg) {
+            this.showWarning("Usage: /ssh-sandbox <user@host[:/path]> [path]");
+            return;
+        }
+        const separatorIndex = targetArg.indexOf(":");
+        const remote = separatorIndex === -1 ? targetArg : targetArg.slice(0, separatorIndex);
+        const cwd = cwdArg ?? (separatorIndex === -1 ? undefined : targetArg.slice(separatorIndex + 1));
+        try {
+            const info = await this.session.configureSshSandbox({ remote, cwd });
+            this.updateToolBackendStatus();
+            this.showStatus(this.formatToolBackendStatus(info));
+        }
+        catch (error) {
+            this.showError(`Failed to configure SSH sandbox: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
     handleSessionCommand() {
         const stats = this.session.getSessionStats();
         const sessionName = this.activeSession.getSessionName();