@fleetagent/pi-coding-agent 0.0.10 → 0.0.12

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" || contextBackendInfo?.type === "remote" ? "☁" : "🖥";
                 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,20 +1033,20 @@ 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));
+                const skillCompactList = formatCompactList(skills.map((skill) => `${getSourceBackendIcon(skill.sourceInfo)} ${skill.name}`));
                 addLoadedSection("Skills", skillCompactList, skillList);
             }
             const rules = rulesResult.rules;
             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));
+                const ruleCompactList = formatCompactList(rules.map((rule) => `${getSourceBackendIcon(rule.sourceInfo)} ${rule.name}`));
                 addLoadedSection("Rules", ruleCompactList, ruleList);
             }
             const templates = this.session.promptTemplates;
@@ -1050,14 +1056,14 @@ 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}`));
+                const promptCompactList = formatCompactList(templates.map((template) => `${getSourceBackendIcon(template.sourceInfo)} /${template.name}`));
                 addLoadedSection("Prompts", promptCompactList, templateList);
             }
             if (extensions.length > 0) {
@@ -1277,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,
@@ -1330,10 +1342,12 @@ export class InteractiveMode {
         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}`);
+        if (info.type === "remote") {
+            return info.configured
+                ? theme.fg("accent", `tools: remote ${info.url} ${info.cwd}`)
+                : theme.fg("warning", `tools: remote not configured ${info.cwd}`);
         }
-        return theme.fg("warning", `tools: ssh not configured ${info.cwd}`);
+        return theme.fg("accent", `tools: ssh ${info.remote}:${info.cwd}`);
     }
     updateToolBackendStatus() {
         this.setExtensionStatus("toolBackend", this.formatToolBackendStatus(this.session.getToolBackendInfo()));
@@ -2051,9 +2065,9 @@ export class InteractiveMode {
                 this.editor.setText("");
                 return;
             }
-            if (text === "/ssh-sandbox" || text.startsWith("/ssh-sandbox ")) {
+            if (text === "/remote" || text.startsWith("/remote ")) {
                 this.editor.setText("");
-                await this.handleSshSandboxCommand(text);
+                await this.handleRemoteCommand(text);
                 return;
             }
             if (text === "/changelog") {
@@ -4383,40 +4397,59 @@ 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();
+    async reloadResourcesAfterBackendChange() {
+        await this.session.reload();
+        this.setupAutocompleteProvider();
+        this.showLoadedResources({ force: true, showDiagnosticsWhenQuiet: true });
+    }
+    async handleRemoteCommand(text) {
+        const args = text.replace(/^\/remote\s*/, "").trim();
         if (!args || args === "status") {
             this.showStatus(this.formatToolBackendStatus(this.session.getToolBackendInfo()));
             return;
         }
         if (args === "clear") {
             try {
-                this.session.clearSshSandbox();
+                this.session.clearRemoteSandbox();
+                await this.reloadResourcesAfterBackendChange();
                 this.updateToolBackendStatus();
-                this.showStatus("SSH sandbox cleared");
+                this.showStatus("Remote backend cleared");
             }
             catch (error) {
                 this.showError(error instanceof Error ? error.message : String(error));
             }
             return;
         }
-        const [targetArg, cwdArg] = args.split(/\s+/, 2);
+        const [kind, targetArg, cwdArg] = args.split(/\s+/, 3);
+        if (kind !== "ssh" && kind !== "daemon") {
+            this.showWarning("Usage: /remote <ssh|daemon> <url> [path]");
+            return;
+        }
         if (!targetArg) {
-            this.showWarning("Usage: /ssh-sandbox <user@host[:/path]> [path]");
+            this.showWarning(kind === "ssh" ? "Usage: /remote ssh <user@host[:/path]> [path]" : "Usage: /remote daemon <ws://url>");
             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 });
+            const info = kind === "ssh"
+                ? await this.session.configureRemoteSandbox({
+                    type: "ssh",
+                    ...this.parseSshRemoteTarget(targetArg, cwdArg),
+                })
+                : await this.session.configureRemoteSandbox({ type: "daemon", url: targetArg });
+            await this.reloadResourcesAfterBackendChange();
             this.updateToolBackendStatus();
             this.showStatus(this.formatToolBackendStatus(info));
         }
         catch (error) {
-            this.showError(`Failed to configure SSH sandbox: ${error instanceof Error ? error.message : String(error)}`);
+            this.showError(`Failed to configure remote backend: ${error instanceof Error ? error.message : String(error)}`);
         }
     }
+    parseSshRemoteTarget(targetArg, cwdArg) {
+        const separatorIndex = targetArg.indexOf(":");
+        const remote = separatorIndex === -1 ? targetArg : targetArg.slice(0, separatorIndex);
+        const cwd = cwdArg ?? (separatorIndex === -1 ? undefined : targetArg.slice(separatorIndex + 1));
+        return { remote, cwd };
+    }
     handleSessionCommand() {
         const stats = this.session.getSessionStats();
         const sessionName = this.activeSession.getSessionName();