feat(commands): Enable @file processing in TOML commands (#6716)

packages/cli/src/services/prompt-processors/atFileProcessor.ts

@@ -0,0 +1,96 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import {
+  flatMapTextParts,
+  readPathFromWorkspace,
+} from '@google/gemini-cli-core';
+import type { CommandContext } from '../../ui/commands/types.js';
+import { MessageType } from '../../ui/types.js';
+import {
+  AT_FILE_INJECTION_TRIGGER,
+  type IPromptProcessor,
+  type PromptPipelineContent,
+} from './types.js';
+import { extractInjections } from './injectionParser.js';
+
+export class AtFileProcessor implements IPromptProcessor {
+  constructor(private readonly commandName?: string) {}
+
+  async process(
+    input: PromptPipelineContent,
+    context: CommandContext,
+  ): Promise<PromptPipelineContent> {
+    const config = context.services.config;
+    if (!config) {
+      return input;
+    }
+
+    return flatMapTextParts(input, async (text) => {
+      if (!text.includes(AT_FILE_INJECTION_TRIGGER)) {
+        return [{ text }];
+      }
+
+      const injections = extractInjections(
+        text,
+        AT_FILE_INJECTION_TRIGGER,
+        this.commandName,
+      );
+      if (injections.length === 0) {
+        return [{ text }];
+      }
+
+      const output: PromptPipelineContent = [];
+      let lastIndex = 0;
+
+      for (const injection of injections) {
+        const prefix = text.substring(lastIndex, injection.startIndex);
+        if (prefix) {
+          output.push({ text: prefix });
+        }
+
+        const pathStr = injection.content;
+        try {
+          const fileContentParts = await readPathFromWorkspace(pathStr, config);
+          if (fileContentParts.length === 0) {
+            const uiMessage = `File '@{${pathStr}}' was ignored by .gitignore or .geminiignore and was not included in the prompt.`;
+            context.ui.addItem(
+              { type: MessageType.INFO, text: uiMessage },
+              Date.now(),
+            );
+          }
+          output.push(...fileContentParts);
+        } catch (error) {
+          const message =
+            error instanceof Error ? error.message : String(error);
+          const uiMessage = `Failed to inject content for '@{${pathStr}}': ${message}`;
+
+          console.error(
+            `[AtFileProcessor] ${uiMessage}. Leaving placeholder in prompt.`,
+          );
+          context.ui.addItem(
+            { type: MessageType.ERROR, text: uiMessage },
+            Date.now(),
+          );
+
+          const placeholder = text.substring(
+            injection.startIndex,
+            injection.endIndex,
+          );
+          output.push({ text: placeholder });
+        }
+        lastIndex = injection.endIndex;
+      }
+
+      const suffix = text.substring(lastIndex);
+      if (suffix) {
+        output.push({ text: suffix });
+      }
+
+      return output;
+    });
+  }
+}