import { jsxs as _jsxs, jsx as _jsx, Fragment as _Fragment } from "react/jsx-runtime";
/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */
import { describe, it, expect, vi } from 'vitest';
import { render } from '../../test-utils/render.js';
import { Text } from 'ink';
import { Composer } from './Composer.js';
import { UIStateContext } from '../contexts/UIStateContext.js';
import { UIActionsContext, } from '../contexts/UIActionsContext.js';
import { ConfigContext } from '../contexts/ConfigContext.js';
import { SettingsContext } from '../contexts/SettingsContext.js';
// Mock VimModeContext hook
vi.mock('../contexts/VimModeContext.js', () => ({
    useVimMode: vi.fn(() => ({
        vimEnabled: false,
        vimMode: 'NORMAL',
    })),
}));
import { ApprovalMode } from '@google/gemini-cli-core';
import { StreamingState } from '../types.js';
import { mergeSettings } from '../../config/settings.js';
// Mock child components
vi.mock('./LoadingIndicator.js', () => ({
    LoadingIndicator: ({ thought }) => (_jsxs(Text, { children: ["LoadingIndicator", thought ? `: ${thought}` : ''] })),
}));
vi.mock('./ContextSummaryDisplay.js', () => ({
    ContextSummaryDisplay: () => _jsx(Text, { children: "ContextSummaryDisplay" }),
}));
vi.mock('./HookStatusDisplay.js', () => ({
    HookStatusDisplay: () => _jsx(Text, { children: "HookStatusDisplay" }),
}));
vi.mock('./ApprovalModeIndicator.js', () => ({
    ApprovalModeIndicator: () => _jsx(Text, { children: "ApprovalModeIndicator" }),
}));
vi.mock('./ShellModeIndicator.js', () => ({
    ShellModeIndicator: () => _jsx(Text, { children: "ShellModeIndicator" }),
}));
vi.mock('./DetailedMessagesDisplay.js', () => ({
    DetailedMessagesDisplay: () => _jsx(Text, { children: "DetailedMessagesDisplay" }),
}));
vi.mock('./InputPrompt.js', () => ({
    InputPrompt: () => _jsx(Text, { children: "InputPrompt" }),
    calculatePromptWidths: vi.fn(() => ({
        inputWidth: 80,
        suggestionsWidth: 40,
        containerWidth: 84,
    })),
}));
vi.mock('./Footer.js', () => ({
    Footer: () => _jsx(Text, { children: "Footer" }),
}));
vi.mock('./ShowMoreLines.js', () => ({
    ShowMoreLines: () => _jsx(Text, { children: "ShowMoreLines" }),
}));
vi.mock('./QueuedMessageDisplay.js', () => ({
    QueuedMessageDisplay: ({ messageQueue }) => {
        if (messageQueue.length === 0) {
            return null;
        }
        return (_jsx(_Fragment, { children: messageQueue.map((message, index) => (_jsx(Text, { children: message }, index))) }));
    },
}));
// Mock contexts
vi.mock('../contexts/OverflowContext.js', () => ({
    OverflowProvider: ({ children }) => children,
}));
// Create mock context providers
const createMockUIState = (overrides = {}) => ({
    streamingState: null,
    contextFileNames: [],
    showApprovalModeIndicator: ApprovalMode.DEFAULT,
    messageQueue: [],
    showErrorDetails: false,
    constrainHeight: false,
    isInputActive: true,
    buffer: { text: '' },
    inputWidth: 80,
    suggestionsWidth: 40,
    userMessages: [],
    slashCommands: [],
    commandContext: null,
    shellModeActive: false,
    isFocused: true,
    thought: '',
    currentLoadingPhrase: '',
    elapsedTime: 0,
    ctrlCPressedOnce: false,
    ctrlDPressedOnce: false,
    showEscapePrompt: false,
    ideContextState: null,
    geminiMdFileCount: 0,
    renderMarkdown: true,
    filteredConsoleMessages: [],
    history: [],
    sessionStats: {
        lastPromptTokenCount: 0,
        sessionTokenCount: 0,
        totalPrompts: 0,
    },
    branchName: 'main',
    debugMessage: '',
    corgiMode: false,
    errorCount: 0,
    nightly: false,
    isTrustedFolder: true,
    activeHooks: [],
    ...overrides,
});
const createMockUIActions = () => ({
    handleFinalSubmit: vi.fn(),
    handleClearScreen: vi.fn(),
    setShellModeActive: vi.fn(),
    onEscapePromptChange: vi.fn(),
    vimHandleInput: vi.fn(),
    // eslint-disable-next-line @typescript-eslint/no-explicit-any
});
const createMockConfig = (overrides = {}) => ({
    getModel: vi.fn(() => 'gemini-1.5-pro'),
    getTargetDir: vi.fn(() => '/test/dir'),
    getDebugMode: vi.fn(() => false),
    getAccessibility: vi.fn(() => ({})),
    getMcpServers: vi.fn(() => ({})),
    getToolRegistry: () => ({
        getTool: vi.fn(),
    }),
    getSkillManager: () => ({
        getSkills: () => [],
        getDisplayableSkills: () => [],
    }),
    getMcpClientManager: () => ({
        getMcpServers: () => ({}),
        getBlockedMcpServers: () => [],
    }),
    ...overrides,
});
const createMockSettings = (merged = {}) => {
    const defaultMergedSettings = mergeSettings({}, {}, {}, {}, true);
    return {
        merged: {
            ...defaultMergedSettings,
            ui: {
                ...defaultMergedSettings.ui,
                hideFooter: false,
                showMemoryUsage: false,
                ...merged,
            },
        },
    };
};
/* eslint-disable @typescript-eslint/no-explicit-any */
const renderComposer = (uiState, settings = createMockSettings(), config = createMockConfig(), uiActions = createMockUIActions()) => render(_jsx(ConfigContext.Provider, { value: config, children: _jsx(SettingsContext.Provider, { value: settings, children: _jsx(UIStateContext.Provider, { value: uiState, children: _jsx(UIActionsContext.Provider, { value: uiActions, children: _jsx(Composer, {}) }) }) }) }));
/* eslint-enable @typescript-eslint/no-explicit-any */
describe('Composer', () => {
    describe('Footer Display Settings', () => {
        it('renders Footer by default when hideFooter is false', () => {
            const uiState = createMockUIState();
            const settings = createMockSettings({ hideFooter: false });
            const { lastFrame } = renderComposer(uiState, settings);
            expect(lastFrame()).toContain('Footer');
        });
        it('does NOT render Footer when hideFooter is true', () => {
            const uiState = createMockUIState();
            const settings = createMockSettings({ hideFooter: true });
            const { lastFrame } = renderComposer(uiState, settings);
            // Check for content that only appears IN the Footer component itself
            expect(lastFrame()).not.toContain('[NORMAL]'); // Vim mode indicator
            expect(lastFrame()).not.toContain('(main'); // Branch name with parentheses
        });
        it('passes correct props to Footer including vim mode when enabled', async () => {
            const uiState = createMockUIState({
                branchName: 'feature-branch',
                corgiMode: true,
                errorCount: 2,
                sessionStats: {
                    sessionId: 'test-session',
                    sessionStartTime: new Date(),
                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
                    metrics: {},
                    lastPromptTokenCount: 150,
                    promptCount: 5,
                },
            });
            const config = createMockConfig({
                getModel: vi.fn(() => 'gemini-1.5-flash'),
                getTargetDir: vi.fn(() => '/project/path'),
                getDebugMode: vi.fn(() => true),
            });
            const settings = createMockSettings({
                hideFooter: false,
                showMemoryUsage: true,
            });
            // Mock vim mode for this test
            const { useVimMode } = await import('../contexts/VimModeContext.js');
            vi.mocked(useVimMode).mockReturnValueOnce({
                vimEnabled: true,
                vimMode: 'INSERT',
                // eslint-disable-next-line @typescript-eslint/no-explicit-any
            });
            const { lastFrame } = renderComposer(uiState, settings, config);
            expect(lastFrame()).toContain('Footer');
            // Footer should be rendered with all the state passed through
        });
    });
    describe('Loading Indicator', () => {
        it('renders LoadingIndicator with thought when streaming', () => {
            const uiState = createMockUIState({
                streamingState: StreamingState.Responding,
                thought: {
                    subject: 'Processing',
                    description: 'Processing your request...',
                },
                currentLoadingPhrase: 'Analyzing',
                elapsedTime: 1500,
            });
            const { lastFrame } = renderComposer(uiState);
            const output = lastFrame();
            expect(output).toContain('LoadingIndicator');
        });
        it('renders LoadingIndicator without thought when accessibility disables loading phrases', () => {
            const uiState = createMockUIState({
                streamingState: StreamingState.Responding,
                thought: { subject: 'Hidden', description: 'Should not show' },
            });
            const config = createMockConfig({
                getAccessibility: vi.fn(() => ({ enableLoadingPhrases: false })),
            });
            const { lastFrame } = renderComposer(uiState, undefined, config);
            const output = lastFrame();
            expect(output).toContain('LoadingIndicator');
            expect(output).not.toContain('Should not show');
        });
        it('suppresses thought when waiting for confirmation', () => {
            const uiState = createMockUIState({
                streamingState: StreamingState.WaitingForConfirmation,
                thought: {
                    subject: 'Confirmation',
                    description: 'Should not show during confirmation',
                },
            });
            const { lastFrame } = renderComposer(uiState);
            const output = lastFrame();
            expect(output).toContain('LoadingIndicator');
            expect(output).not.toContain('Should not show during confirmation');
        });
    });
    describe('Message Queue Display', () => {
        it('displays queued messages when present', () => {
            const uiState = createMockUIState({
                messageQueue: [
                    'First queued message',
                    'Second queued message',
                    'Third queued message',
                ],
            });
            const { lastFrame } = renderComposer(uiState);
            const output = lastFrame();
            expect(output).toContain('First queued message');
            expect(output).toContain('Second queued message');
            expect(output).toContain('Third queued message');
        });
        it('renders QueuedMessageDisplay with empty message queue', () => {
            const uiState = createMockUIState({
                messageQueue: [],
            });
            const { lastFrame } = renderComposer(uiState);
            // The component should render but return null for empty queue
            // This test verifies that the component receives the correct prop
            const output = lastFrame();
            expect(output).toContain('InputPrompt'); // Verify basic Composer rendering
        });
    });
    describe('Context and Status Display', () => {
        it('shows ContextSummaryDisplay in normal state', () => {
            const uiState = createMockUIState({
                ctrlCPressedOnce: false,
                ctrlDPressedOnce: false,
                showEscapePrompt: false,
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).toContain('ContextSummaryDisplay');
        });
        it('renders HookStatusDisplay instead of ContextSummaryDisplay with active hooks', () => {
            const uiState = createMockUIState({
                activeHooks: [{ name: 'test-hook', eventName: 'before-agent' }],
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).toContain('HookStatusDisplay');
            expect(lastFrame()).not.toContain('ContextSummaryDisplay');
        });
        it('shows Ctrl+C exit prompt when ctrlCPressedOnce is true', () => {
            const uiState = createMockUIState({
                ctrlCPressedOnce: true,
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).toContain('Press Ctrl+C again to exit');
        });
        it('shows Ctrl+D exit prompt when ctrlDPressedOnce is true', () => {
            const uiState = createMockUIState({
                ctrlDPressedOnce: true,
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).toContain('Press Ctrl+D again to exit');
        });
        it('shows escape prompt when showEscapePrompt is true', () => {
            const uiState = createMockUIState({
                showEscapePrompt: true,
                history: [{ id: 1, type: 'user', text: 'test' }],
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).toContain('Press Esc again to rewind');
        });
    });
    describe('Input and Indicators', () => {
        it('renders InputPrompt when input is active', () => {
            const uiState = createMockUIState({
                isInputActive: true,
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).toContain('InputPrompt');
        });
        it('does not render InputPrompt when input is inactive', () => {
            const uiState = createMockUIState({
                isInputActive: false,
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).not.toContain('InputPrompt');
        });
        it('shows ApprovalModeIndicator when approval mode is not default and shell mode is inactive', () => {
            const uiState = createMockUIState({
                showApprovalModeIndicator: ApprovalMode.YOLO,
                shellModeActive: false,
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).toContain('ApprovalModeIndicator');
        });
        it('shows ShellModeIndicator when shell mode is active', () => {
            const uiState = createMockUIState({
                shellModeActive: true,
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).toContain('ShellModeIndicator');
        });
        it('shows RawMarkdownIndicator when renderMarkdown is false', () => {
            const uiState = createMockUIState({
                renderMarkdown: false,
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).toContain('raw markdown mode');
        });
        it('does not show RawMarkdownIndicator when renderMarkdown is true', () => {
            const uiState = createMockUIState({
                renderMarkdown: true,
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).not.toContain('raw markdown mode');
        });
    });
    describe('Error Details Display', () => {
        it('shows DetailedMessagesDisplay when showErrorDetails is true', () => {
            const uiState = createMockUIState({
                showErrorDetails: true,
                filteredConsoleMessages: [
                    { level: 'error', message: 'Test error', timestamp: new Date() },
                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
                ],
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).toContain('DetailedMessagesDisplay');
            expect(lastFrame()).toContain('ShowMoreLines');
        });
        it('does not show error details when showErrorDetails is false', () => {
            const uiState = createMockUIState({
                showErrorDetails: false,
            });
            const { lastFrame } = renderComposer(uiState);
            expect(lastFrame()).not.toContain('DetailedMessagesDisplay');
        });
    });
});
//# sourceMappingURL=Composer.test.js.map