/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */
import { debugLogger, flatMapTextParts, readPathFromWorkspace, } from 'sixth-cli-core';
import { MessageType } from '../../ui/types.js';
import { AT_FILE_INJECTION_TRIGGER, } from './types.js';
import { extractInjections } from './injectionParser.js';
export class AtFileProcessor {
    commandName;
    constructor(commandName) {
        this.commandName = commandName;
    }
    async process(input, context) {
        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 = [];
            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}`;
                    // `context.invocation` should always be present at this point.
                    debugLogger.error(`Error while loading custom command (${context.invocation.name}) ${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;
        });
    }
}
//# sourceMappingURL=atFileProcessor.js.map