/**
 * @license
 * Copyright 2026 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */
import fs from 'node:fs';
import path from 'node:path';
import process from 'node:process';
import { GEMINI_DIR } from '../utils/paths.js';
import { ApprovalMode } from '../policy/types.js';
import * as snippets from './snippets.js';
import * as legacySnippets from './snippets.legacy.js';
import { resolvePathFromEnv, applySubstitutions, isSectionEnabled, } from './utils.js';
import { CodebaseInvestigatorAgent } from '../agents/codebase-investigator.js';
import { isGitRepository } from '../utils/gitUtils.js';
import { WRITE_TODOS_TOOL_NAME, READ_FILE_TOOL_NAME, ENTER_PLAN_MODE_TOOL_NAME, GLOB_TOOL_NAME, GREP_TOOL_NAME, } from '../tools/tool-names.js';
import { resolveModel, supportsModernFeatures } from '../config/models.js';
import { DiscoveredMCPTool } from '../tools/mcp-tool.js';
import { getAllGeminiMdFilenames } from '../tools/memoryTool.js';
/**
 * Orchestrates prompt generation by gathering context and building options.
 */
export class PromptProvider {
    /**
     * Generates the core system prompt.
     */
    getCoreSystemPrompt(context, userMemory, interactiveOverride) {
        const systemMdResolution = resolvePathFromEnv(process.env['GEMINI_SYSTEM_MD']);
        const interactiveMode = interactiveOverride ?? context.config.isInteractive();
        const approvalMode = context.config.getApprovalMode?.() ?? ApprovalMode.DEFAULT;
        const isPlanMode = approvalMode === ApprovalMode.PLAN;
        const isYoloMode = approvalMode === ApprovalMode.YOLO;
        const skills = context.config.getSkillManager().getSkills();
        const toolNames = context.toolRegistry.getAllToolNames();
        const enabledToolNames = new Set(toolNames);
        const approvedPlanPath = context.config.getApprovedPlanPath();
        const desiredModel = resolveModel(context.config.getActiveModel(), context.config.getGemini31LaunchedSync?.() ?? false);
        const isModernModel = supportsModernFeatures(desiredModel);
        const activeSnippets = isModernModel ? snippets : legacySnippets;
        const contextFilenames = getAllGeminiMdFilenames();
        // --- Context Gathering ---
        let planModeToolsList = '';
        if (isPlanMode) {
            const allTools = context.toolRegistry.getAllTools();
            planModeToolsList = allTools
                .map((t) => {
                if (t instanceof DiscoveredMCPTool) {
                    return `  <tool>\`${t.name}\` (${t.serverName})</tool>`;
                }
                return `  <tool>\`${t.name}\`</tool>`;
            })
                .join('\n');
        }
        let basePrompt;
        // --- Template File Override ---
        if (systemMdResolution.value && !systemMdResolution.isDisabled) {
            let systemMdPath = path.resolve(path.join(GEMINI_DIR, 'system.md'));
            if (!systemMdResolution.isSwitch) {
                systemMdPath = systemMdResolution.value;
            }
            if (!fs.existsSync(systemMdPath)) {
                throw new Error(`missing system prompt file '${systemMdPath}'`);
            }
            basePrompt = fs.readFileSync(systemMdPath, 'utf8');
            const skillsPrompt = activeSnippets.renderAgentSkills(skills.map((s) => ({
                name: s.name,
                description: s.description,
                location: s.location,
            })));
            basePrompt = applySubstitutions(basePrompt, context.config, skillsPrompt, isModernModel);
        }
        else {
            // --- Standard Composition ---
            const hasHierarchicalMemory = typeof userMemory === 'object' &&
                userMemory !== null &&
                (!!userMemory.global?.trim() ||
                    !!userMemory.extension?.trim() ||
                    !!userMemory.project?.trim());
            const options = {
                preamble: this.withSection('preamble', () => ({
                    interactive: interactiveMode,
                })),
                coreMandates: this.withSection('coreMandates', () => ({
                    interactive: interactiveMode,
                    hasSkills: skills.length > 0,
                    hasHierarchicalMemory,
                    contextFilenames,
                    topicUpdateNarration: context.config.isTopicUpdateNarrationEnabled(),
                })),
                subAgents: this.withSection('agentContexts', () => context.config
                    .getAgentRegistry()
                    .getAllDefinitions()
                    .map((d) => ({
                    name: d.name,
                    description: d.description,
                }))),
                agentSkills: this.withSection('agentSkills', () => skills.map((s) => ({
                    name: s.name,
                    description: s.description,
                    location: s.location,
                })), skills.length > 0),
                hookContext: isSectionEnabled('hookContext') || undefined,
                primaryWorkflows: this.withSection('primaryWorkflows', () => ({
                    interactive: interactiveMode,
                    enableCodebaseInvestigator: enabledToolNames.has(CodebaseInvestigatorAgent.name),
                    enableWriteTodosTool: enabledToolNames.has(WRITE_TODOS_TOOL_NAME),
                    enableEnterPlanModeTool: enabledToolNames.has(ENTER_PLAN_MODE_TOOL_NAME),
                    enableGrep: enabledToolNames.has(GREP_TOOL_NAME),
                    enableGlob: enabledToolNames.has(GLOB_TOOL_NAME),
                    approvedPlan: approvedPlanPath
                        ? { path: approvedPlanPath }
                        : undefined,
                    taskTracker: context.config.isTrackerEnabled(),
                    topicUpdateNarration: context.config.isTopicUpdateNarrationEnabled(),
                }), !isPlanMode),
                planningWorkflow: this.withSection('planningWorkflow', () => ({
                    planModeToolsList,
                    plansDir: context.config.storage.getPlansDir(),
                    approvedPlanPath: context.config.getApprovedPlanPath(),
                    taskTracker: context.config.isTrackerEnabled(),
                }), isPlanMode),
                taskTracker: context.config.isTrackerEnabled(),
                operationalGuidelines: this.withSection('operationalGuidelines', () => ({
                    interactive: interactiveMode,
                    enableShellEfficiency: context.config.getEnableShellOutputEfficiency(),
                    interactiveShellEnabled: context.config.isInteractiveShellEnabled(),
                    topicUpdateNarration: context.config.isTopicUpdateNarrationEnabled(),
                })),
                sandbox: this.withSection('sandbox', () => getSandboxMode()),
                interactiveYoloMode: this.withSection('interactiveYoloMode', () => true, isYoloMode && interactiveMode),
                gitRepo: this.withSection('git', () => ({ interactive: interactiveMode }), isGitRepository(process.cwd()) ? true : false),
                finalReminder: isModernModel
                    ? undefined
                    : this.withSection('finalReminder', () => ({
                        readFileToolName: READ_FILE_TOOL_NAME,
                    })),
            };
            // eslint-disable-next-line @typescript-eslint/no-unsafe-type-assertion
            const getCoreSystemPrompt = activeSnippets.getCoreSystemPrompt;
            basePrompt = getCoreSystemPrompt(options);
        }
        // --- Finalization (Shell) ---
        const finalPrompt = activeSnippets.renderFinalShell(basePrompt, userMemory, contextFilenames);
        // Sanitize erratic newlines from composition
        const sanitizedPrompt = finalPrompt.replace(/\n{3,}/g, '\n\n');
        // Write back to file if requested
        this.maybeWriteSystemMd(sanitizedPrompt, systemMdResolution, path.resolve(path.join(GEMINI_DIR, 'system.md')));
        return sanitizedPrompt;
    }
    getCompressionPrompt(context) {
        const desiredModel = resolveModel(context.config.getActiveModel(), context.config.getGemini31LaunchedSync?.() ?? false);
        const isModernModel = supportsModernFeatures(desiredModel);
        const activeSnippets = isModernModel ? snippets : legacySnippets;
        return activeSnippets.getCompressionPrompt(context.config.getApprovedPlanPath());
    }
    withSection(key, factory, guard = true) {
        return guard && isSectionEnabled(key) ? factory() : undefined;
    }
    maybeWriteSystemMd(basePrompt, resolution, defaultPath) {
        const writeSystemMdResolution = resolvePathFromEnv(process.env['GEMINI_WRITE_SYSTEM_MD']);
        if (writeSystemMdResolution.value && !writeSystemMdResolution.isDisabled) {
            const writePath = writeSystemMdResolution.isSwitch
                ? defaultPath
                : writeSystemMdResolution.value;
            fs.mkdirSync(path.dirname(writePath), { recursive: true });
            fs.writeFileSync(writePath, basePrompt);
        }
    }
}
// --- Internal Context Helpers ---
function getSandboxMode() {
    if (process.env['SANDBOX'] === 'sandbox-exec')
        return 'macos-seatbelt';
    if (process.env['SANDBOX'])
        return 'generic';
    return 'outside';
}
//# sourceMappingURL=promptProvider.js.map