fss-link 1.2.0 → 1.2.1

package/dist/src/services/FileCommandLoader.js

@@ -0,0 +1,210 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { promises as fs } from 'fs';
+import path from 'path';
+import toml from '@iarna/toml';
+import { glob } from 'glob';
+import { z } from 'zod';
+import { getProjectCommandsDir, getUserCommandsDir, } from 'fss-link-core';
+import { CommandKind, } from '../ui/commands/types.js';
+import { DefaultArgumentProcessor } from './prompt-processors/argumentProcessor.js';
+import { SHORTHAND_ARGS_PLACEHOLDER, SHELL_INJECTION_TRIGGER, } from './prompt-processors/types.js';
+import { ConfirmationRequiredError, ShellProcessor, } from './prompt-processors/shellProcessor.js';
+/**
+ * Defines the Zod schema for a command definition file. This serves as the
+ * single source of truth for both validation and type inference.
+ */
+const TomlCommandDefSchema = z.object({
+    prompt: z.string({
+        required_error: "The 'prompt' field is required.",
+        invalid_type_error: "The 'prompt' field must be a string.",
+    }),
+    description: z.string().optional(),
+});
+/**
+ * Discovers and loads custom slash commands from .toml files in both the
+ * user's global config directory and the current project's directory.
+ *
+ * This loader is responsible for:
+ * - Recursively scanning command directories.
+ * - Parsing and validating TOML files.
+ * - Adapting valid definitions into executable SlashCommand objects.
+ * - Handling file system errors and malformed files gracefully.
+ */
+export class FileCommandLoader {
+    config;
+    projectRoot;
+    constructor(config) {
+        this.config = config;
+        this.projectRoot = config?.getProjectRoot() || process.cwd();
+    }
+    /**
+     * Loads all commands from user, project, and extension directories.
+     * Returns commands in order: user → project → extensions (alphabetically).
+     *
+     * Order is important for conflict resolution in CommandService:
+     * - User/project commands (without extensionName) use "last wins" strategy
+     * - Extension commands (with extensionName) get renamed if conflicts exist
+     *
+     * @param signal An AbortSignal to cancel the loading process.
+     * @returns A promise that resolves to an array of all loaded SlashCommands.
+     */
+    async loadCommands(signal) {
+        const allCommands = [];
+        const globOptions = {
+            nodir: true,
+            dot: true,
+            signal,
+            follow: true,
+        };
+        // Load commands from each directory
+        const commandDirs = this.getCommandDirectories();
+        for (const dirInfo of commandDirs) {
+            try {
+                const files = await glob('**/*.toml', {
+                    ...globOptions,
+                    cwd: dirInfo.path,
+                });
+                const commandPromises = files.map((file) => this.parseAndAdaptFile(path.join(dirInfo.path, file), dirInfo.path, dirInfo.extensionName));
+                const commands = (await Promise.all(commandPromises)).filter((cmd) => cmd !== null);
+                // Add all commands without deduplication
+                allCommands.push(...commands);
+            }
+            catch (error) {
+                if (error.code !== 'ENOENT') {
+                    console.error(`[FileCommandLoader] Error loading commands from ${dirInfo.path}:`, error);
+                }
+            }
+        }
+        return allCommands;
+    }
+    /**
+     * Get all command directories in order for loading.
+     * User commands → Project commands → Extension commands
+     * This order ensures extension commands can detect all conflicts.
+     */
+    getCommandDirectories() {
+        const dirs = [];
+        // 1. User commands
+        dirs.push({ path: getUserCommandsDir() });
+        // 2. Project commands (override user commands)
+        dirs.push({ path: getProjectCommandsDir(this.projectRoot) });
+        // 3. Extension commands (processed last to detect all conflicts)
+        if (this.config) {
+            const activeExtensions = this.config
+                .getExtensions()
+                .filter((ext) => ext.isActive)
+                .sort((a, b) => a.name.localeCompare(b.name)); // Sort alphabetically for deterministic loading
+            const extensionCommandDirs = activeExtensions.map((ext) => ({
+                path: path.join(ext.path, 'commands'),
+                extensionName: ext.name,
+            }));
+            dirs.push(...extensionCommandDirs);
+        }
+        return dirs;
+    }
+    /**
+     * Parses a single .toml file and transforms it into a SlashCommand object.
+     * @param filePath The absolute path to the .toml file.
+     * @param baseDir The root command directory for name calculation.
+     * @param extensionName Optional extension name to prefix commands with.
+     * @returns A promise resolving to a SlashCommand, or null if the file is invalid.
+     */
+    async parseAndAdaptFile(filePath, baseDir, extensionName) {
+        let fileContent;
+        try {
+            fileContent = await fs.readFile(filePath, 'utf-8');
+        }
+        catch (error) {
+            console.error(`[FileCommandLoader] Failed to read file ${filePath}:`, error instanceof Error ? error.message : String(error));
+            return null;
+        }
+        let parsed;
+        try {
+            parsed = toml.parse(fileContent);
+        }
+        catch (error) {
+            console.error(`[FileCommandLoader] Failed to parse TOML file ${filePath}:`, error instanceof Error ? error.message : String(error));
+            return null;
+        }
+        const validationResult = TomlCommandDefSchema.safeParse(parsed);
+        if (!validationResult.success) {
+            console.error(`[FileCommandLoader] Skipping invalid command file: ${filePath}. Validation errors:`, validationResult.error.flatten());
+            return null;
+        }
+        const validDef = validationResult.data;
+        const relativePathWithExt = path.relative(baseDir, filePath);
+        const relativePath = relativePathWithExt.substring(0, relativePathWithExt.length - 5);
+        const baseCommandName = relativePath
+            .split(path.sep)
+            // Sanitize each path segment to prevent ambiguity. Since ':' is our
+            // namespace separator, we replace any literal colons in filenames
+            // with underscores to avoid naming conflicts.
+            .map((segment) => segment.replaceAll(':', '_'))
+            .join(':');
+        // Add extension name tag for extension commands
+        const defaultDescription = `Custom command from ${path.basename(filePath)}`;
+        let description = validDef.description || defaultDescription;
+        if (extensionName) {
+            description = `[${extensionName}] ${description}`;
+        }
+        const processors = [];
+        const usesArgs = validDef.prompt.includes(SHORTHAND_ARGS_PLACEHOLDER);
+        const usesShellInjection = validDef.prompt.includes(SHELL_INJECTION_TRIGGER);
+        // Interpolation (Shell Execution and Argument Injection)
+        // If the prompt uses either shell injection OR argument placeholders,
+        // we must use the ShellProcessor.
+        if (usesShellInjection || usesArgs) {
+            processors.push(new ShellProcessor(baseCommandName));
+        }
+        // Default Argument Handling
+        // If NO explicit argument injection ({{args}}) was used, we append the raw invocation.
+        if (!usesArgs) {
+            processors.push(new DefaultArgumentProcessor());
+        }
+        return {
+            name: baseCommandName,
+            description,
+            kind: CommandKind.FILE,
+            extensionName,
+            action: async (context, _args) => {
+                if (!context.invocation) {
+                    console.error(`[FileCommandLoader] Critical error: Command '${baseCommandName}' was executed without invocation context.`);
+                    return {
+                        type: 'submit_prompt',
+                        content: validDef.prompt, // Fallback to unprocessed prompt
+                    };
+                }
+                try {
+                    let processedPrompt = validDef.prompt;
+                    for (const processor of processors) {
+                        processedPrompt = await processor.process(processedPrompt, context);
+                    }
+                    return {
+                        type: 'submit_prompt',
+                        content: processedPrompt,
+                    };
+                }
+                catch (e) {
+                    // Check if it's our specific error type
+                    if (e instanceof ConfirmationRequiredError) {
+                        // Halt and request confirmation from the UI layer.
+                        return {
+                            type: 'confirm_shell_commands',
+                            commandsToConfirm: e.commandsToConfirm,
+                            originalInvocation: {
+                                raw: context.invocation.raw,
+                            },
+                        };
+                    }
+                    // Re-throw other errors to be handled by the global error handler.
+                    throw e;
+                }
+            },
+        };
+    }
+}
+//# sourceMappingURL=FileCommandLoader.js.map

package/dist/src/services/FileCommandLoader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FileCommandLoader.js","sourceRoot":"","sources":["../../../src/services/FileCommandLoader.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,IAAI,MAAM,aAAa,CAAC;AAC/B,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAEL,qBAAqB,EACrB,kBAAkB,GACnB,MAAM,eAAe,CAAC;AAEvB,OAAO,EAEL,WAAW,GAGZ,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,wBAAwB,EAAE,MAAM,0CAA0C,CAAC;AACpF,OAAO,EAEL,0BAA0B,EAC1B,uBAAuB,GACxB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACL,yBAAyB,EACzB,cAAc,GACf,MAAM,uCAAuC,CAAC;AAO/C;;;GAGG;AACH,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QACf,cAAc,EAAE,iCAAiC;QACjD,kBAAkB,EAAE,sCAAsC;KAC3D,CAAC;IACF,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACnC,CAAC,CAAC;AAEH;;;;;;;;;GASG;AACH,MAAM,OAAO,iBAAiB;IAGC;IAFZ,WAAW,CAAS;IAErC,YAA6B,MAAqB;QAArB,WAAM,GAAN,MAAM,CAAe;QAChD,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,cAAc,EAAE,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/D,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,YAAY,CAAC,MAAmB;QACpC,MAAM,WAAW,GAAmB,EAAE,CAAC;QACvC,MAAM,WAAW,GAAG;YAClB,KAAK,EAAE,IAAI;YACX,GAAG,EAAE,IAAI;YACT,MAAM;YACN,MAAM,EAAE,IAAI;SACb,CAAC;QAEF,oCAAoC;QACpC,MAAM,WAAW,GAAG,IAAI,CAAC,qBAAqB,EAAE,CAAC;QACjD,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE;oBACpC,GAAG,WAAW;oBACd,GAAG,EAAE,OAAO,CAAC,IAAI;iBAClB,CAAC,CAAC;gBAEH,MAAM,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACzC,IAAI,CAAC,iBAAiB,CACpB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,EAC7B,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,aAAa,CACtB,CACF,CAAC;gBAEF,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,MAAM,CAC1D,CAAC,GAAG,EAAuB,EAAE,CAAC,GAAG,KAAK,IAAI,CAC3C,CAAC;gBAEF,yCAAyC;gBACzC,WAAW,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;YAChC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACvD,OAAO,CAAC,KAAK,CACX,mDAAmD,OAAO,CAAC,IAAI,GAAG,EAClE,KAAK,CACN,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;;;OAIG;IACK,qBAAqB;QAC3B,MAAM,IAAI,GAAuB,EAAE,CAAC;QAEpC,mBAAmB;QACnB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,EAAE,CAAC,CAAC;QAE1C,+CAA+C;QAC/C,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAE7D,iEAAiE;QACjE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM;iBACjC,aAAa,EAAE;iBACf,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC;iBAC7B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,gDAAgD;YAEjG,MAAM,oBAAoB,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC1D,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC;gBACrC,aAAa,EAAE,GAAG,CAAC,IAAI;aACxB,CAAC,CAAC,CAAC;YAEJ,IAAI,CAAC,IAAI,CAAC,GAAG,oBAAoB,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;OAMG;IACK,KAAK,CAAC,iBAAiB,CAC7B,QAAgB,EAChB,OAAe,EACf,aAAsB;QAEtB,IAAI,WAAmB,CAAC;QACxB,IAAI,CAAC;YACH,WAAW,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACrD,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,OAAO,CAAC,KAAK,CACX,2CAA2C,QAAQ,GAAG,EACtD,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACvD,CAAC;YACF,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,OAAO,CAAC,KAAK,CACX,iDAAiD,QAAQ,GAAG,EAC5D,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACvD,CAAC;YACF,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAEhE,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,KAAK,CACX,sDAAsD,QAAQ,sBAAsB,EACpF,gBAAgB,CAAC,KAAK,CAAC,OAAO,EAAE,CACjC,CAAC;YACF,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,QAAQ,GAAG,gBAAgB,CAAC,IAAI,CAAC;QAEvC,MAAM,mBAAmB,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC7D,MAAM,YAAY,GAAG,mBAAmB,CAAC,SAAS,CAChD,CAAC,EACD,mBAAmB,CAAC,MAAM,GAAG,CAAC,CAC/B,CAAC;QACF,MAAM,eAAe,GAAG,YAAY;aACjC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC;YAChB,oEAAoE;YACpE,kEAAkE;YAClE,8CAA8C;aAC7C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;aAC9C,IAAI,CAAC,GAAG,CAAC,CAAC;QAEb,gDAAgD;QAChD,MAAM,kBAAkB,GAAG,uBAAuB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5E,IAAI,WAAW,GAAG,QAAQ,CAAC,WAAW,IAAI,kBAAkB,CAAC;QAC7D,IAAI,aAAa,EAAE,CAAC;YAClB,WAAW,GAAG,IAAI,aAAa,KAAK,WAAW,EAAE,CAAC;QACpD,CAAC;QAED,MAAM,UAAU,GAAuB,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;QACtE,MAAM,kBAAkB,GAAG,QAAQ,CAAC,MAAM,CAAC,QAAQ,CACjD,uBAAuB,CACxB,CAAC;QAEF,yDAAyD;QACzD,sEAAsE;QACtE,kCAAkC;QAClC,IAAI,kBAAkB,IAAI,QAAQ,EAAE,CAAC;YACnC,UAAU,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,eAAe,CAAC,CAAC,CAAC;QACvD,CAAC;QAED,4BAA4B;QAC5B,uFAAuF;QACvF,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,UAAU,CAAC,IAAI,CAAC,IAAI,wBAAwB,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,OAAO;YACL,IAAI,EAAE,eAAe;YACrB,WAAW;YACX,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,aAAa;YACb,MAAM,EAAE,KAAK,EACX,OAAuB,EACvB,KAAa,EACsB,EAAE;gBACrC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;oBACxB,OAAO,CAAC,KAAK,CACX,gDAAgD,eAAe,4CAA4C,CAC5G,CAAC;oBACF,OAAO;wBACL,IAAI,EAAE,eAAe;wBACrB,OAAO,EAAE,QAAQ,CAAC,MAAM,EAAE,iCAAiC;qBAC5D,CAAC;gBACJ,CAAC;gBAED,IAAI,CAAC;oBACH,IAAI,eAAe,GAAG,QAAQ,CAAC,MAAM,CAAC;oBACtC,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;wBACnC,eAAe,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;oBACtE,CAAC;oBAED,OAAO;wBACL,IAAI,EAAE,eAAe;wBACrB,OAAO,EAAE,eAAe;qBACzB,CAAC;gBACJ,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,wCAAwC;oBACxC,IAAI,CAAC,YAAY,yBAAyB,EAAE,CAAC;wBAC3C,mDAAmD;wBACnD,OAAO;4BACL,IAAI,EAAE,wBAAwB;4BAC9B,iBAAiB,EAAE,CAAC,CAAC,iBAAiB;4BACtC,kBAAkB,EAAE;gCAClB,GAAG,EAAE,OAAO,CAAC,UAAU,CAAC,GAAG;6BAC5B;yBACF,CAAC;oBACJ,CAAC;oBACD,mEAAmE;oBACnE,MAAM,CAAC,CAAC;gBACV,CAAC;YACH,CAAC;SACF,CAAC;IACJ,CAAC;CACF"}

package/dist/src/services/McpPromptLoader.d.ts

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Config } from 'fss-link-core';
+import { SlashCommand } from '../ui/commands/types.js';
+import { ICommandLoader } from './types.js';
+/**
+ * Discovers and loads executable slash commands from prompts exposed by
+ * Model-Context-Protocol (MCP) servers.
+ */
+export declare class McpPromptLoader implements ICommandLoader {
+    private readonly config;
+    constructor(config: Config | null);
+    /**
+     * Loads all available prompts from all configured MCP servers and adapts
+     * them into executable SlashCommand objects.
+     *
+     * @param _signal An AbortSignal (unused for this synchronous loader).
+     * @returns A promise that resolves to an array of loaded SlashCommands.
+     */
+    loadCommands(_signal: AbortSignal): Promise<SlashCommand[]>;
+    private parseArgs;
+}

package/dist/src/services/McpPromptLoader.js

@@ -0,0 +1,192 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { getErrorMessage, getMCPServerPrompts, } from 'fss-link-core';
+import { CommandKind, } from '../ui/commands/types.js';
+/**
+ * Discovers and loads executable slash commands from prompts exposed by
+ * Model-Context-Protocol (MCP) servers.
+ */
+export class McpPromptLoader {
+    config;
+    constructor(config) {
+        this.config = config;
+    }
+    /**
+     * Loads all available prompts from all configured MCP servers and adapts
+     * them into executable SlashCommand objects.
+     *
+     * @param _signal An AbortSignal (unused for this synchronous loader).
+     * @returns A promise that resolves to an array of loaded SlashCommands.
+     */
+    loadCommands(_signal) {
+        const promptCommands = [];
+        if (!this.config) {
+            return Promise.resolve([]);
+        }
+        const mcpServers = this.config.getMcpServers() || {};
+        for (const serverName in mcpServers) {
+            const prompts = getMCPServerPrompts(this.config, serverName) || [];
+            for (const prompt of prompts) {
+                const commandName = `${prompt.name}`;
+                const newPromptCommand = {
+                    name: commandName,
+                    description: prompt.description || `Invoke prompt ${prompt.name}`,
+                    kind: CommandKind.MCP_PROMPT,
+                    subCommands: [
+                        {
+                            name: 'help',
+                            description: 'Show help for this prompt',
+                            kind: CommandKind.MCP_PROMPT,
+                            action: async () => {
+                                if (!prompt.arguments || prompt.arguments.length === 0) {
+                                    return {
+                                        type: 'message',
+                                        messageType: 'info',
+                                        content: `Prompt "${prompt.name}" has no arguments.`,
+                                    };
+                                }
+                                let helpMessage = `Arguments for "${prompt.name}":\n\n`;
+                                if (prompt.arguments && prompt.arguments.length > 0) {
+                                    helpMessage += `You can provide arguments by name (e.g., --argName="value") or by position.\n\n`;
+                                    helpMessage += `e.g., ${prompt.name} ${prompt.arguments?.map((_) => `"foo"`)} is equivalent to ${prompt.name} ${prompt.arguments?.map((arg) => `--${arg.name}="foo"`)}\n\n`;
+                                }
+                                for (const arg of prompt.arguments) {
+                                    helpMessage += `  --${arg.name}\n`;
+                                    if (arg.description) {
+                                        helpMessage += `    ${arg.description}\n`;
+                                    }
+                                    helpMessage += `    (required: ${arg.required ? 'yes' : 'no'})\n\n`;
+                                }
+                                return {
+                                    type: 'message',
+                                    messageType: 'info',
+                                    content: helpMessage,
+                                };
+                            },
+                        },
+                    ],
+                    action: async (context, args) => {
+                        if (!this.config) {
+                            return {
+                                type: 'message',
+                                messageType: 'error',
+                                content: 'Config not loaded.',
+                            };
+                        }
+                        const promptInputs = this.parseArgs(args, prompt.arguments);
+                        if (promptInputs instanceof Error) {
+                            return {
+                                type: 'message',
+                                messageType: 'error',
+                                content: promptInputs.message,
+                            };
+                        }
+                        try {
+                            const mcpServers = this.config.getMcpServers() || {};
+                            const mcpServerConfig = mcpServers[serverName];
+                            if (!mcpServerConfig) {
+                                return {
+                                    type: 'message',
+                                    messageType: 'error',
+                                    content: `MCP server config not found for '${serverName}'.`,
+                                };
+                            }
+                            const result = await prompt.invoke(promptInputs);
+                            if (result['error']) {
+                                return {
+                                    type: 'message',
+                                    messageType: 'error',
+                                    content: `Error invoking prompt: ${result['error']}`,
+                                };
+                            }
+                            if (!result.messages?.[0]?.content?.['text']) {
+                                return {
+                                    type: 'message',
+                                    messageType: 'error',
+                                    content: 'Received an empty or invalid prompt response from the server.',
+                                };
+                            }
+                            return {
+                                type: 'submit_prompt',
+                                content: JSON.stringify(result.messages[0].content.text),
+                            };
+                        }
+                        catch (error) {
+                            return {
+                                type: 'message',
+                                messageType: 'error',
+                                content: `Error: ${getErrorMessage(error)}`,
+                            };
+                        }
+                    },
+                    completion: async (_, partialArg) => {
+                        if (!prompt || !prompt.arguments) {
+                            return [];
+                        }
+                        const suggestions = [];
+                        const usedArgNames = new Set((partialArg.match(/--([^=]+)/g) || []).map((s) => s.substring(2)));
+                        for (const arg of prompt.arguments) {
+                            if (!usedArgNames.has(arg.name)) {
+                                suggestions.push(`--${arg.name}=""`);
+                            }
+                        }
+                        return suggestions;
+                    },
+                };
+                promptCommands.push(newPromptCommand);
+            }
+        }
+        return Promise.resolve(promptCommands);
+    }
+    parseArgs(userArgs, promptArgs) {
+        const argValues = {};
+        const promptInputs = {};
+        // arg parsing: --key="value" or --key=value
+        const namedArgRegex = /--([^=]+)=(?:"((?:\\.|[^"\\])*)"|([^ ]*))/g;
+        let match;
+        const remainingArgs = [];
+        let lastIndex = 0;
+        while ((match = namedArgRegex.exec(userArgs)) !== null) {
+            const key = match[1];
+            const value = match[2] ?? match[3]; // Quoted or unquoted value
+            argValues[key] = value;
+            // Capture text between matches as potential positional args
+            if (match.index > lastIndex) {
+                remainingArgs.push(userArgs.substring(lastIndex, match.index).trim());
+            }
+            lastIndex = namedArgRegex.lastIndex;
+        }
+        // Capture any remaining text after the last named arg
+        if (lastIndex < userArgs.length) {
+            remainingArgs.push(userArgs.substring(lastIndex).trim());
+        }
+        const positionalArgs = remainingArgs.join(' ').split(/ +/);
+        if (!promptArgs) {
+            return promptInputs;
+        }
+        for (const arg of promptArgs) {
+            if (argValues[arg.name]) {
+                promptInputs[arg.name] = argValues[arg.name];
+            }
+        }
+        const unfilledArgs = promptArgs.filter((arg) => arg.required && !promptInputs[arg.name]);
+        const missingArgs = [];
+        for (let i = 0; i < unfilledArgs.length; i++) {
+            if (positionalArgs.length > i && positionalArgs[i]) {
+                promptInputs[unfilledArgs[i].name] = positionalArgs[i];
+            }
+            else {
+                missingArgs.push(unfilledArgs[i].name);
+            }
+        }
+        if (missingArgs.length > 0) {
+            const missingArgNames = missingArgs.map((name) => `--${name}`).join(', ');
+            return new Error(`Missing required argument(s): ${missingArgNames}`);
+        }
+        return promptInputs;
+    }
+}
+//# sourceMappingURL=McpPromptLoader.js.map

package/dist/src/services/McpPromptLoader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpPromptLoader.js","sourceRoot":"","sources":["../../../src/services/McpPromptLoader.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAEL,eAAe,EACf,mBAAmB,GACpB,MAAM,eAAe,CAAC;AACvB,OAAO,EAEL,WAAW,GAGZ,MAAM,yBAAyB,CAAC;AAIjC;;;GAGG;AACH,MAAM,OAAO,eAAe;IACG;IAA7B,YAA6B,MAAqB;QAArB,WAAM,GAAN,MAAM,CAAe;IAAG,CAAC;IAEtD;;;;;;OAMG;IACH,YAAY,CAAC,OAAoB;QAC/B,MAAM,cAAc,GAAmB,EAAE,CAAC;QAC1C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC;QACrD,KAAK,MAAM,UAAU,IAAI,UAAU,EAAE,CAAC;YACpC,MAAM,OAAO,GAAG,mBAAmB,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC;YACnE,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,MAAM,WAAW,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;gBACrC,MAAM,gBAAgB,GAAiB;oBACrC,IAAI,EAAE,WAAW;oBACjB,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,iBAAiB,MAAM,CAAC,IAAI,EAAE;oBACjE,IAAI,EAAE,WAAW,CAAC,UAAU;oBAC5B,WAAW,EAAE;wBACX;4BACE,IAAI,EAAE,MAAM;4BACZ,WAAW,EAAE,2BAA2B;4BACxC,IAAI,EAAE,WAAW,CAAC,UAAU;4BAC5B,MAAM,EAAE,KAAK,IAAuC,EAAE;gCACpD,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oCACvD,OAAO;wCACL,IAAI,EAAE,SAAS;wCACf,WAAW,EAAE,MAAM;wCACnB,OAAO,EAAE,WAAW,MAAM,CAAC,IAAI,qBAAqB;qCACrD,CAAC;gCACJ,CAAC;gCAED,IAAI,WAAW,GAAG,kBAAkB,MAAM,CAAC,IAAI,QAAQ,CAAC;gCACxD,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oCACpD,WAAW,IAAI,iFAAiF,CAAC;oCACjG,WAAW,IAAI,SAAS,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,qBAAqB,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC;gCAC9K,CAAC;gCACD,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oCACnC,WAAW,IAAI,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC;oCACnC,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC;wCACpB,WAAW,IAAI,OAAO,GAAG,CAAC,WAAW,IAAI,CAAC;oCAC5C,CAAC;oCACD,WAAW,IAAI,kBACb,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IACzB,OAAO,CAAC;gCACV,CAAC;gCACD,OAAO;oCACL,IAAI,EAAE,SAAS;oCACf,WAAW,EAAE,MAAM;oCACnB,OAAO,EAAE,WAAW;iCACrB,CAAC;4BACJ,CAAC;yBACF;qBACF;oBACD,MAAM,EAAE,KAAK,EACX,OAAuB,EACvB,IAAY,EACuB,EAAE;wBACrC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;4BACjB,OAAO;gCACL,IAAI,EAAE,SAAS;gCACf,WAAW,EAAE,OAAO;gCACpB,OAAO,EAAE,oBAAoB;6BAC9B,CAAC;wBACJ,CAAC;wBAED,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;wBAC5D,IAAI,YAAY,YAAY,KAAK,EAAE,CAAC;4BAClC,OAAO;gCACL,IAAI,EAAE,SAAS;gCACf,WAAW,EAAE,OAAO;gCACpB,OAAO,EAAE,YAAY,CAAC,OAAO;6BAC9B,CAAC;wBACJ,CAAC;wBAED,IAAI,CAAC;4BACH,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC;4BACrD,MAAM,eAAe,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;4BAC/C,IAAI,CAAC,eAAe,EAAE,CAAC;gCACrB,OAAO;oCACL,IAAI,EAAE,SAAS;oCACf,WAAW,EAAE,OAAO;oCACpB,OAAO,EAAE,oCAAoC,UAAU,IAAI;iCAC5D,CAAC;4BACJ,CAAC;4BACD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;4BAEjD,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gCACpB,OAAO;oCACL,IAAI,EAAE,SAAS;oCACf,WAAW,EAAE,OAAO;oCACpB,OAAO,EAAE,0BAA0B,MAAM,CAAC,OAAO,CAAC,EAAE;iCACrD,CAAC;4BACJ,CAAC;4BAED,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;gCAC7C,OAAO;oCACL,IAAI,EAAE,SAAS;oCACf,WAAW,EAAE,OAAO;oCACpB,OAAO,EACL,+DAA+D;iCAClE,CAAC;4BACJ,CAAC;4BAED,OAAO;gCACL,IAAI,EAAE,eAAe;gCACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;6BACzD,CAAC;wBACJ,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,OAAO;gCACL,IAAI,EAAE,SAAS;gCACf,WAAW,EAAE,OAAO;gCACpB,OAAO,EAAE,UAAU,eAAe,CAAC,KAAK,CAAC,EAAE;6BAC5C,CAAC;wBACJ,CAAC;oBACH,CAAC;oBACD,UAAU,EAAE,KAAK,EAAE,CAAiB,EAAE,UAAkB,EAAE,EAAE;wBAC1D,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;4BACjC,OAAO,EAAE,CAAC;wBACZ,CAAC;wBAED,MAAM,WAAW,GAAa,EAAE,CAAC;wBACjC,MAAM,YAAY,GAAG,IAAI,GAAG,CAC1B,CAAC,UAAU,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAClE,CAAC;wBAEF,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;4BACnC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gCAChC,WAAW,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,IAAI,KAAK,CAAC,CAAC;4BACvC,CAAC;wBACH,CAAC;wBAED,OAAO,WAAW,CAAC;oBACrB,CAAC;iBACF,CAAC;gBACF,cAAc,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IACzC,CAAC;IAEO,SAAS,CACf,QAAgB,EAChB,UAAwC;QAExC,MAAM,SAAS,GAA8B,EAAE,CAAC;QAChD,MAAM,YAAY,GAA4B,EAAE,CAAC;QAEjD,4CAA4C;QAC5C,MAAM,aAAa,GAAG,4CAA4C,CAAC;QACnE,IAAI,KAAK,CAAC;QACV,MAAM,aAAa,GAAa,EAAE,CAAC;QACnC,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,OAAO,CAAC,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;YACvD,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACrB,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,2BAA2B;YAC/D,SAAS,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YACvB,4DAA4D;YAC5D,IAAI,KAAK,CAAC,KAAK,GAAG,SAAS,EAAE,CAAC;gBAC5B,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YACxE,CAAC;YACD,SAAS,GAAG,aAAa,CAAC,SAAS,CAAC;QACtC,CAAC;QAED,sDAAsD;QACtD,IAAI,SAAS,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC;YAChC,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,cAAc,GAAG,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAE3D,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;YAC7B,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxB,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,UAAU,CAAC,MAAM,CACpC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CACjD,CAAC;QAEF,MAAM,WAAW,GAAa,EAAE,CAAC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7C,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnD,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;YACzD,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QAED,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,eAAe,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1E,OAAO,IAAI,KAAK,CAAC,iCAAiC,eAAe,EAAE,CAAC,CAAC;QACvE,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;CACF"}

package/dist/src/services/SearchEngineConfigProvider.d.ts

@@ -0,0 +1,33 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export interface SearchEngineConfig {
+    braveApiKey?: string;
+    tavilyApiKey?: string;
+}
+/**
+ * Configuration provider for search engine API keys
+ * Loads from database and sets environment variables
+ */
+export declare class SearchEngineConfigProvider {
+    private static instance;
+    static getInstance(): SearchEngineConfigProvider;
+    /**
+     * Load search engine configuration from database and set environment variables
+     */
+    loadConfiguration(): Promise<SearchEngineConfig>;
+    /**
+     * Save search engine configuration to database
+     */
+    saveConfiguration(config: SearchEngineConfig): Promise<void>;
+    /**
+     * Check if search engines are configured
+     */
+    hasConfiguration(): Promise<boolean>;
+    /**
+     * Get configuration status for display
+     */
+    getConfigurationStatus(): Promise<string>;
+}

package/dist/src/services/SearchEngineConfigProvider.js

@@ -0,0 +1,82 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { getFSSLinkDatabase } from '../config/database.js';
+/**
+ * Configuration provider for search engine API keys
+ * Loads from database and sets environment variables
+ */
+export class SearchEngineConfigProvider {
+    static instance = null;
+    static getInstance() {
+        if (!SearchEngineConfigProvider.instance) {
+            SearchEngineConfigProvider.instance = new SearchEngineConfigProvider();
+        }
+        return SearchEngineConfigProvider.instance;
+    }
+    /**
+     * Load search engine configuration from database and set environment variables
+     */
+    async loadConfiguration() {
+        const db = await getFSSLinkDatabase();
+        const braveApiKey = await db.getUserPreference('webscraper.brave_api_key');
+        const tavilyApiKey = await db.getUserPreference('webscraper.tavily_api_key');
+        // Set environment variables for immediate use
+        if (braveApiKey) {
+            process.env['BRAVE_SEARCH_API_KEY'] = braveApiKey;
+        }
+        if (tavilyApiKey) {
+            process.env['TAVILY_API_KEY'] = tavilyApiKey;
+        }
+        return {
+            braveApiKey: braveApiKey || undefined,
+            tavilyApiKey: tavilyApiKey || undefined,
+        };
+    }
+    /**
+     * Save search engine configuration to database
+     */
+    async saveConfiguration(config) {
+        const db = await getFSSLinkDatabase();
+        if (config.braveApiKey) {
+            await db.setUserPreference('webscraper.brave_api_key', config.braveApiKey);
+            await db.setUserPreference('webscraper.brave_enabled', 'true');
+            process.env['BRAVE_SEARCH_API_KEY'] = config.braveApiKey;
+        }
+        if (config.tavilyApiKey) {
+            await db.setUserPreference('webscraper.tavily_api_key', config.tavilyApiKey);
+            await db.setUserPreference('webscraper.tavily_enabled', 'true');
+            process.env['TAVILY_API_KEY'] = config.tavilyApiKey;
+        }
+    }
+    /**
+     * Check if search engines are configured
+     */
+    async hasConfiguration() {
+        const config = await this.loadConfiguration();
+        return !!(config.braveApiKey || config.tavilyApiKey);
+    }
+    /**
+     * Get configuration status for display
+     */
+    async getConfigurationStatus() {
+        const config = await this.loadConfiguration();
+        let status = '🔍 Search Engine Configuration:\n';
+        if (config.braveApiKey) {
+            status += `✅ Brave Search: Configured\n`;
+        }
+        else {
+            status += `❌ Brave Search: Not configured\n`;
+        }
+        if (config.tavilyApiKey) {
+            status += `✅ Tavily Search: Configured\n`;
+        }
+        else {
+            status += `❌ Tavily Search: Not configured\n`;
+        }
+        return status;
+    }
+}
+//# sourceMappingURL=SearchEngineConfigProvider.js.map

package/dist/src/services/SearchEngineConfigProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SearchEngineConfigProvider.js","sourceRoot":"","sources":["../../../src/services/SearchEngineConfigProvider.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAO3D;;;GAGG;AACH,MAAM,OAAO,0BAA0B;IAC7B,MAAM,CAAC,QAAQ,GAAsC,IAAI,CAAC;IAElE,MAAM,CAAC,WAAW;QAChB,IAAI,CAAC,0BAA0B,CAAC,QAAQ,EAAE,CAAC;YACzC,0BAA0B,CAAC,QAAQ,GAAG,IAAI,0BAA0B,EAAE,CAAC;QACzE,CAAC;QACD,OAAO,0BAA0B,CAAC,QAAQ,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB;QACrB,MAAM,EAAE,GAAG,MAAM,kBAAkB,EAAE,CAAC;QAEtC,MAAM,WAAW,GAAG,MAAM,EAAE,CAAC,iBAAiB,CAAC,0BAA0B,CAAC,CAAC;QAC3E,MAAM,YAAY,GAAG,MAAM,EAAE,CAAC,iBAAiB,CAAC,2BAA2B,CAAC,CAAC;QAE7E,8CAA8C;QAC9C,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,GAAG,WAAW,CAAC;QACpD,CAAC;QACD,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,GAAG,YAAY,CAAC;QAC/C,CAAC;QAED,OAAO;YACL,WAAW,EAAE,WAAW,IAAI,SAAS;YACrC,YAAY,EAAE,YAAY,IAAI,SAAS;SACxC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB,CAAC,MAA0B;QAChD,MAAM,EAAE,GAAG,MAAM,kBAAkB,EAAE,CAAC;QAEtC,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YACvB,MAAM,EAAE,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;YAC3E,MAAM,EAAE,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,MAAM,CAAC,CAAC;YAC/D,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,GAAG,MAAM,CAAC,WAAW,CAAC;QAC3D,CAAC;QAED,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACxB,MAAM,EAAE,CAAC,iBAAiB,CAAC,2BAA2B,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;YAC7E,MAAM,EAAE,CAAC,iBAAiB,CAAC,2BAA2B,EAAE,MAAM,CAAC,CAAC;YAChE,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC;QACtD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB;QACpB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC9C,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,YAAY,CAAC,CAAC;IACvD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,sBAAsB;QAC1B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE9C,IAAI,MAAM,GAAG,mCAAmC,CAAC;QAEjD,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YACvB,MAAM,IAAI,8BAA8B,CAAC;QAC3C,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,kCAAkC,CAAC;QAC/C,CAAC;QAED,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACxB,MAAM,IAAI,+BAA+B,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,mCAAmC,CAAC;QAChD,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC"}

package/dist/src/services/prompt-processors/argumentProcessor.d.ts

@@ -0,0 +1,16 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { IPromptProcessor } from './types.js';
+import { CommandContext } from '../../ui/commands/types.js';
+/**
+ * Appends the user's full command invocation to the prompt if arguments are
+ * provided, allowing the model to perform its own argument parsing.
+ *
+ * This processor is only used if the prompt does NOT contain {{args}}.
+ */
+export declare class DefaultArgumentProcessor implements IPromptProcessor {
+    process(prompt: string, context: CommandContext): Promise<string>;
+}

package/dist/src/services/prompt-processors/argumentProcessor.js

@@ -0,0 +1,20 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Appends the user's full command invocation to the prompt if arguments are
+ * provided, allowing the model to perform its own argument parsing.
+ *
+ * This processor is only used if the prompt does NOT contain {{args}}.
+ */
+export class DefaultArgumentProcessor {
+    async process(prompt, context) {
+        if (context.invocation.args) {
+            return `${prompt}\n\n${context.invocation.raw}`;
+        }
+        return prompt;
+    }
+}
+//# sourceMappingURL=argumentProcessor.js.map

package/dist/src/services/prompt-processors/argumentProcessor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"argumentProcessor.js","sourceRoot":"","sources":["../../../../src/services/prompt-processors/argumentProcessor.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH;;;;;GAKG;AACH,MAAM,OAAO,wBAAwB;IACnC,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAAuB;QACnD,IAAI,OAAO,CAAC,UAAW,CAAC,IAAI,EAAE,CAAC;YAC7B,OAAO,GAAG,MAAM,OAAO,OAAO,CAAC,UAAW,CAAC,GAAG,EAAE,CAAC;QACnD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}

package/dist/src/services/prompt-processors/shellProcessor.d.ts

@@ -0,0 +1,35 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { CommandContext } from '../../ui/commands/types.js';
+import { IPromptProcessor } from './types.js';
+export declare class ConfirmationRequiredError extends Error {
+    commandsToConfirm: string[];
+    constructor(message: string, commandsToConfirm: string[]);
+}
+/**
+ * Handles prompt interpolation, including shell command execution (`!{...}`)
+ * and context-aware argument injection (`{{args}}`).
+ *
+ * This processor ensures that:
+ * 1. `{{args}}` outside `!{...}` are replaced with raw input.
+ * 2. `{{args}}` inside `!{...}` are replaced with shell-escaped input.
+ * 3. Shell commands are executed securely after argument substitution.
+ * 4. Parsing correctly handles nested braces.
+ */
+export declare class ShellProcessor implements IPromptProcessor {
+    private readonly commandName;
+    constructor(commandName: string);
+    process(prompt: string, context: CommandContext): Promise<string>;
+    /**
+     * Iteratively parses the prompt string to extract shell injections (!{...}),
+     * correctly handling nested braces within the command.
+     *
+     * @param prompt The prompt string to parse.
+     * @returns An array of extracted ShellInjection objects.
+     * @throws Error if an unclosed injection (`!{`) is found.
+     */
+    private extractInjections;
+}