/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */
import { vi, describe, it, expect, beforeEach, afterEach, } from 'vitest';
const mockShellExecutionService = vi.hoisted(() => vi.fn());
vi.mock('../services/shellExecutionService.js', () => ({
    ShellExecutionService: { execute: mockShellExecutionService },
}));
vi.mock('fs');
vi.mock('os');
vi.mock('crypto');
vi.mock('../utils/summarizer.js');
import { isCommandAllowed } from '../utils/shell-utils.js';
import { ShellTool } from './shell.js';
import * as fs from 'fs';
import * as os from 'os';
import * as path from 'path';
import * as crypto from 'crypto';
import * as summarizer from '../utils/summarizer.js';
import { ToolConfirmationOutcome } from './tools.js';
import { OUTPUT_UPDATE_INTERVAL_MS } from './shell.js';
import { createMockWorkspaceContext } from '../test-utils/mockWorkspaceContext.js';
describe('ShellTool', () => {
    let shellTool;
    let mockConfig;
    let mockShellOutputCallback;
    let resolveExecutionPromise;
    beforeEach(() => {
        vi.clearAllMocks();
        mockConfig = {
            getCoreTools: vi.fn().mockReturnValue([]),
            getExcludeTools: vi.fn().mockReturnValue([]),
            getDebugMode: vi.fn().mockReturnValue(false),
            getTargetDir: vi.fn().mockReturnValue('/test/dir'),
            getSummarizeToolOutputConfig: vi.fn().mockReturnValue(undefined),
            getWorkspaceContext: () => createMockWorkspaceContext('.'),
            getGeminiClient: vi.fn(),
            getGitCoAuthor: vi.fn().mockReturnValue({
                enabled: true,
                name: 'Qwen-Coder',
                email: 'qwen-coder@alibabacloud.com',
            }),
        };
        shellTool = new ShellTool(mockConfig);
        vi.mocked(os.platform).mockReturnValue('linux');
        vi.mocked(os.tmpdir).mockReturnValue('/tmp');
        vi.mocked(crypto.randomBytes).mockReturnValue(Buffer.from('abcdef', 'hex'));
        // Capture the output callback to simulate streaming events from the service
        mockShellExecutionService.mockImplementation((_cmd, _cwd, callback) => {
            mockShellOutputCallback = callback;
            return {
                pid: 12345,
                result: new Promise((resolve) => {
                    resolveExecutionPromise = resolve;
                }),
            };
        });
    });
    describe('isCommandAllowed', () => {
        it('should allow a command if no restrictions are provided', () => {
            mockConfig.getCoreTools.mockReturnValue(undefined);
            mockConfig.getExcludeTools.mockReturnValue(undefined);
            expect(isCommandAllowed('ls -l', mockConfig).allowed).toBe(true);
        });
        it('should block a command with command substitution using $()', () => {
            expect(isCommandAllowed('echo $(rm -rf /)', mockConfig).allowed).toBe(false);
        });
    });
    describe('validateToolParams', () => {
        it('should return null for a valid command', () => {
            expect(shellTool.validateToolParams({ command: 'ls -l' })).toBeNull();
        });
        it('should return an error for an empty command', () => {
            expect(shellTool.validateToolParams({ command: ' ' })).toBe('Command cannot be empty.');
        });
        it('should return an error for a non-existent directory', () => {
            vi.mocked(fs.existsSync).mockReturnValue(false);
            expect(shellTool.validateToolParams({ command: 'ls', directory: 'rel/path' })).toBe("Directory 'rel/path' is not a registered workspace directory.");
        });
    });
    describe('execute', () => {
        const mockAbortSignal = new AbortController().signal;
        const resolveShellExecution = (result = {}) => {
            const fullResult = {
                rawOutput: Buffer.from(result.output || ''),
                output: 'Success',
                stdout: 'Success',
                stderr: '',
                exitCode: 0,
                signal: null,
                error: null,
                aborted: false,
                pid: 12345,
                ...result,
            };
            resolveExecutionPromise(fullResult);
        };
        it('should wrap command on linux and parse pgrep output', async () => {
            const promise = shellTool.execute({ command: 'my-command &' }, mockAbortSignal);
            resolveShellExecution({ pid: 54321 });
            vi.mocked(fs.existsSync).mockReturnValue(true);
            vi.mocked(fs.readFileSync).mockReturnValue('54321\n54322\n'); // Service PID and background PID
            const result = await promise;
            const tmpFile = path.join(os.tmpdir(), 'shell_pgrep_abcdef.tmp');
            const wrappedCommand = `{ my-command & }; __code=$?; pgrep -g 0 >${tmpFile} 2>&1; exit $__code;`;
            expect(mockShellExecutionService).toHaveBeenCalledWith(wrappedCommand, expect.any(String), expect.any(Function), mockAbortSignal);
            expect(result.llmContent).toContain('Background PIDs: 54322');
            expect(vi.mocked(fs.unlinkSync)).toHaveBeenCalledWith(tmpFile);
        });
        it('should not wrap command on windows', async () => {
            vi.mocked(os.platform).mockReturnValue('win32');
            const promise = shellTool.execute({ command: 'dir' }, mockAbortSignal);
            resolveExecutionPromise({
                rawOutput: Buffer.from(''),
                output: '',
                stdout: '',
                stderr: '',
                exitCode: 0,
                signal: null,
                error: null,
                aborted: false,
                pid: 12345,
            });
            await promise;
            expect(mockShellExecutionService).toHaveBeenCalledWith('dir', expect.any(String), expect.any(Function), mockAbortSignal);
        });
        it('should format error messages correctly', async () => {
            const error = new Error('wrapped command failed');
            const promise = shellTool.execute({ command: 'user-command' }, mockAbortSignal);
            resolveShellExecution({
                error,
                exitCode: 1,
                output: 'err',
                stderr: 'err',
                rawOutput: Buffer.from('err'),
                stdout: '',
                signal: null,
                aborted: false,
                pid: 12345,
            });
            const result = await promise;
            // The final llmContent should contain the user's command, not the wrapper
            expect(result.llmContent).toContain('Error: wrapped command failed');
            expect(result.llmContent).not.toContain('pgrep');
        });
        it('should summarize output when configured', async () => {
            mockConfig.getSummarizeToolOutputConfig.mockReturnValue({
                [shellTool.name]: { tokenBudget: 1000 },
            });
            vi.mocked(summarizer.summarizeToolOutput).mockResolvedValue('summarized output');
            const promise = shellTool.execute({ command: 'ls' }, mockAbortSignal);
            resolveExecutionPromise({
                output: 'long output',
                rawOutput: Buffer.from('long output'),
                stdout: 'long output',
                stderr: '',
                exitCode: 0,
                signal: null,
                error: null,
                aborted: false,
                pid: 12345,
            });
            const result = await promise;
            expect(summarizer.summarizeToolOutput).toHaveBeenCalledWith(expect.any(String), mockConfig.getGeminiClient(), mockAbortSignal, 1000);
            expect(result.llmContent).toBe('summarized output');
            expect(result.returnDisplay).toBe('long output');
        });
        it('should clean up the temp file on synchronous execution error', async () => {
            const error = new Error('sync spawn error');
            mockShellExecutionService.mockImplementation(() => {
                throw error;
            });
            vi.mocked(fs.existsSync).mockReturnValue(true); // Pretend the file exists
            await expect(shellTool.execute({ command: 'a-command' }, mockAbortSignal)).rejects.toThrow(error);
            const tmpFile = path.join(os.tmpdir(), 'shell_pgrep_abcdef.tmp');
            expect(vi.mocked(fs.unlinkSync)).toHaveBeenCalledWith(tmpFile);
        });
        describe('Streaming to `updateOutput`', () => {
            let updateOutputMock;
            beforeEach(() => {
                vi.useFakeTimers({ toFake: ['Date'] });
                updateOutputMock = vi.fn();
            });
            afterEach(() => {
                vi.useRealTimers();
            });
            it('should throttle text output updates', async () => {
                const promise = shellTool.execute({ command: 'stream' }, mockAbortSignal, updateOutputMock);
                // First chunk, should be throttled.
                mockShellOutputCallback({
                    type: 'data',
                    stream: 'stdout',
                    chunk: 'hello ',
                });
                expect(updateOutputMock).not.toHaveBeenCalled();
                // Advance time past the throttle interval.
                await vi.advanceTimersByTimeAsync(OUTPUT_UPDATE_INTERVAL_MS + 1);
                // Send a second chunk. THIS event triggers the update with the CUMULATIVE content.
                mockShellOutputCallback({
                    type: 'data',
                    stream: 'stderr',
                    chunk: 'world',
                });
                // It should have been called once now with the combined output.
                expect(updateOutputMock).toHaveBeenCalledOnce();
                expect(updateOutputMock).toHaveBeenCalledWith('hello \nworld');
                resolveExecutionPromise({
                    rawOutput: Buffer.from(''),
                    output: '',
                    stdout: '',
                    stderr: '',
                    exitCode: 0,
                    signal: null,
                    error: null,
                    aborted: false,
                    pid: 12345,
                });
                await promise;
            });
            it('should immediately show binary detection message and throttle progress', async () => {
                const promise = shellTool.execute({ command: 'cat img' }, mockAbortSignal, updateOutputMock);
                mockShellOutputCallback({ type: 'binary_detected' });
                expect(updateOutputMock).toHaveBeenCalledOnce();
                expect(updateOutputMock).toHaveBeenCalledWith('[Binary output detected. Halting stream...]');
                mockShellOutputCallback({
                    type: 'binary_progress',
                    bytesReceived: 1024,
                });
                expect(updateOutputMock).toHaveBeenCalledOnce();
                // Advance time past the throttle interval.
                await vi.advanceTimersByTimeAsync(OUTPUT_UPDATE_INTERVAL_MS + 1);
                // Send a SECOND progress event. This one will trigger the flush.
                mockShellOutputCallback({
                    type: 'binary_progress',
                    bytesReceived: 2048,
                });
                // Now it should be called a second time with the latest progress.
                expect(updateOutputMock).toHaveBeenCalledTimes(2);
                expect(updateOutputMock).toHaveBeenLastCalledWith('[Receiving binary output... 2.0 KB received]');
                resolveExecutionPromise({
                    rawOutput: Buffer.from(''),
                    output: '',
                    stdout: '',
                    stderr: '',
                    exitCode: 0,
                    signal: null,
                    error: null,
                    aborted: false,
                    pid: 12345,
                });
                await promise;
            });
        });
    });
    describe('shouldConfirmExecute', () => {
        it('should request confirmation for a new command and whitelist it on "Always"', async () => {
            const params = { command: 'npm install' };
            const confirmation = await shellTool.shouldConfirmExecute(params, new AbortController().signal);
            expect(confirmation).not.toBe(false);
            expect(confirmation && confirmation.type).toBe('exec');
            // eslint-disable-next-line @typescript-eslint/no-explicit-any
            await confirmation.onConfirm(ToolConfirmationOutcome.ProceedAlways);
            // Should now be whitelisted
            const secondConfirmation = await shellTool.shouldConfirmExecute({ command: 'npm test' }, new AbortController().signal);
            expect(secondConfirmation).toBe(false);
        });
        it('should skip confirmation if validation fails', async () => {
            const confirmation = await shellTool.shouldConfirmExecute({ command: '' }, new AbortController().signal);
            expect(confirmation).toBe(false);
        });
    });
    describe('addCoAuthorToGitCommit', () => {
        it('should add co-author to git commit with double quotes', () => {
            const command = 'git commit -m "Initial commit"';
            // Use public test method
            const result = shellTool.addCoAuthorToGitCommit(command);
            expect(result).toBe(`git commit -m "Initial commit

Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>"`);
        });
        it('should add co-author to git commit with single quotes', () => {
            const command = "git commit -m 'Fix bug'";
            const result = shellTool.addCoAuthorToGitCommit(command);
            expect(result).toBe(`git commit -m 'Fix bug

Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>'`);
        });
        it('should handle git commit with additional flags', () => {
            const command = 'git commit -a -m "Add feature"';
            const result = shellTool.addCoAuthorToGitCommit(command);
            expect(result).toBe(`git commit -a -m "Add feature

Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>"`);
        });
        it('should not modify non-git commands', () => {
            const command = 'npm install';
            const result = shellTool.addCoAuthorToGitCommit(command);
            expect(result).toBe('npm install');
        });
        it('should not modify git commands without -m flag', () => {
            const command = 'git commit';
            const result = shellTool.addCoAuthorToGitCommit(command);
            expect(result).toBe('git commit');
        });
        it('should handle git commit with escaped quotes in message', () => {
            const command = 'git commit -m "Fix \\"quoted\\" text"';
            const result = shellTool.addCoAuthorToGitCommit(command);
            expect(result).toBe(`git commit -m "Fix \\"quoted\\" text

Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>"`);
        });
        it('should not add co-author when disabled in config', () => {
            // Mock config with disabled co-author
            mockConfig.getGitCoAuthor.mockReturnValue({
                enabled: false,
                name: 'Qwen-Coder',
                email: 'qwen-coder@alibabacloud.com',
            });
            const command = 'git commit -m "Initial commit"';
            const result = shellTool.addCoAuthorToGitCommit(command);
            expect(result).toBe('git commit -m "Initial commit"');
        });
        it('should use custom name and email from config', () => {
            // Mock config with custom co-author details
            mockConfig.getGitCoAuthor.mockReturnValue({
                enabled: true,
                name: 'Custom Bot',
                email: 'custom@example.com',
            });
            const command = 'git commit -m "Test commit"';
            const result = shellTool.addCoAuthorToGitCommit(command);
            expect(result).toBe(`git commit -m "Test commit

Co-authored-by: Custom Bot <custom@example.com>"`);
        });
    });
});
describe('validateToolParams', () => {
    it('should return null for valid directory', () => {
        const config = {
            getCoreTools: () => undefined,
            getExcludeTools: () => undefined,
            getTargetDir: () => '/root',
            getWorkspaceContext: () => createMockWorkspaceContext('/root', ['/users/test']),
        };
        const shellTool = new ShellTool(config);
        const result = shellTool.validateToolParams({
            command: 'ls',
            directory: 'test',
        });
        expect(result).toBeNull();
    });
    it('should return error for directory outside workspace', () => {
        const config = {
            getCoreTools: () => undefined,
            getExcludeTools: () => undefined,
            getTargetDir: () => '/root',
            getWorkspaceContext: () => createMockWorkspaceContext('/root', ['/users/test']),
        };
        const shellTool = new ShellTool(config);
        const result = shellTool.validateToolParams({
            command: 'ls',
            directory: 'test2',
        });
        expect(result).toContain('is not a registered workspace directory');
    });
});
describe('validateToolParams', () => {
    it('should return null for valid directory', () => {
        const config = {
            getCoreTools: () => undefined,
            getExcludeTools: () => undefined,
            getTargetDir: () => '/root',
            getWorkspaceContext: () => createMockWorkspaceContext('/root', ['/users/test']),
        };
        const shellTool = new ShellTool(config);
        const result = shellTool.validateToolParams({
            command: 'ls',
            directory: 'test',
        });
        expect(result).toBeNull();
    });
    it('should return error for directory outside workspace', () => {
        const config = {
            getCoreTools: () => undefined,
            getExcludeTools: () => undefined,
            getTargetDir: () => '/root',
            getWorkspaceContext: () => createMockWorkspaceContext('/root', ['/users/test']),
        };
        const shellTool = new ShellTool(config);
        const result = shellTool.validateToolParams({
            command: 'ls',
            directory: 'test2',
        });
        expect(result).toContain('is not a registered workspace directory');
    });
});
//# sourceMappingURL=shell.test.js.map