Repomix

import * as fs from 'node:fs/promises'; import path from 'node:path'; import { beforeEach, describe, expect, test, vi } from 'vitest'; import type { DefaultActionRunnerResult } from '../../../src/cli/actions/defaultAction.js'; import { copyOutputToCurrentDirectory, runRemoteAction } from '../../../src/cli/actions/remoteAction.js'; import { createMockConfig } from '../../testing/testUtils.js'; vi.mock('node:fs/promises', async (importOriginal) => { const actual = await importOriginal<typeof import('node:fs/promises')>(); return { ...actual, copyFile: vi.fn(), mkdir: vi.fn(), }; }); vi.mock('../../../src/shared/logger'); vi.mock('../../../src/cli/cliSpinner'); describe('remoteAction functions', () => { beforeEach(() => { vi.resetAllMocks(); }); describe('runRemoteAction', () => { test('should clone the repository when not a GitHub repo', async () => { const execGitShallowCloneMock = vi.fn(async (_url: string, directory: string) => { await fs.writeFile(path.join(directory, 'README.md'), 'Hello, world!'); }); vi.mocked(fs.copyFile).mockResolvedValue(undefined); await runRemoteAction( 'https://gitlab.com/owner/repo.git', {}, { isGitInstalled: async () => Promise.resolve(true), execGitShallowClone: execGitShallowCloneMock, getRemoteRefs: async () => Promise.resolve(['main']), runDefaultAction: async () => { return { packResult: { totalFiles: 1, totalCharacters: 1, totalTokens: 1, fileCharCounts: {}, fileTokenCounts: {}, suspiciousFilesResults: [], suspiciousGitDiffResults: [], suspiciousGitLogResults: [], processedFiles: [], safeFilePaths: [], gitDiffTokenCount: 0, gitLogTokenCount: 0, skippedFiles: [], }, config: createMockConfig(), } satisfies DefaultActionRunnerResult; }, downloadGitHubArchive: vi.fn().mockRejectedValue(new Error('Archive download not implemented in test')), isGitHubRepository: vi.fn().mockReturnValue(false), parseGitHubRepoInfo: vi.fn().mockReturnValue(null), isArchiveDownloadSupported: vi.fn().mockReturnValue(false), }, ); expect(execGitShallowCloneMock).toHaveBeenCalledTimes(1); }); test('should download GitHub archive successfully without git installed', async () => { const downloadGitHubArchiveMock = vi.fn().mockResolvedValue(undefined); const execGitShallowCloneMock = vi.fn(); const isGitInstalledMock = vi.fn().mockResolvedValue(false); // Git is NOT installed vi.mocked(fs.copyFile).mockResolvedValue(undefined); await runRemoteAction( 'yamadashy/repomix', {}, { isGitInstalled: isGitInstalledMock, execGitShallowClone: execGitShallowCloneMock, getRemoteRefs: async () => Promise.resolve(['main']), runDefaultAction: async () => { return { packResult: { totalFiles: 1, totalCharacters: 1, totalTokens: 1, fileCharCounts: {}, fileTokenCounts: {}, suspiciousFilesResults: [], suspiciousGitDiffResults: [], suspiciousGitLogResults: [], processedFiles: [], safeFilePaths: [], gitDiffTokenCount: 0, gitLogTokenCount: 0, skippedFiles: [], }, config: createMockConfig(), } satisfies DefaultActionRunnerResult; }, downloadGitHubArchive: downloadGitHubArchiveMock, isGitHubRepository: vi.fn().mockReturnValue(true), parseGitHubRepoInfo: vi.fn().mockReturnValue({ owner: 'yamadashy', repo: 'repomix' }), isArchiveDownloadSupported: vi.fn().mockReturnValue(true), }, ); expect(downloadGitHubArchiveMock).toHaveBeenCalledTimes(1); expect(execGitShallowCloneMock).not.toHaveBeenCalled(); expect(isGitInstalledMock).not.toHaveBeenCalled(); // Git check should not be called when archive succeeds }); test('should fallback to git clone when archive download fails', async () => { const downloadGitHubArchiveMock = vi.fn().mockRejectedValue(new Error('Archive download failed')); const execGitShallowCloneMock = vi.fn(async (_url: string, directory: string) => { await fs.writeFile(path.join(directory, 'README.md'), 'Hello, world!'); }); vi.mocked(fs.copyFile).mockResolvedValue(undefined); await runRemoteAction( 'yamadashy/repomix', {}, { isGitInstalled: async () => Promise.resolve(true), execGitShallowClone: execGitShallowCloneMock, getRemoteRefs: async () => Promise.resolve(['main']), runDefaultAction: async () => { return { packResult: { totalFiles: 1, totalCharacters: 1, totalTokens: 1, fileCharCounts: {}, fileTokenCounts: {}, suspiciousFilesResults: [], suspiciousGitDiffResults: [], suspiciousGitLogResults: [], processedFiles: [], safeFilePaths: [], gitDiffTokenCount: 0, gitLogTokenCount: 0, skippedFiles: [], }, config: createMockConfig(), } satisfies DefaultActionRunnerResult; }, downloadGitHubArchive: downloadGitHubArchiveMock, isGitHubRepository: vi.fn().mockReturnValue(true), parseGitHubRepoInfo: vi.fn().mockReturnValue({ owner: 'yamadashy', repo: 'repomix' }), isArchiveDownloadSupported: vi.fn().mockReturnValue(true), }, ); expect(downloadGitHubArchiveMock).toHaveBeenCalledTimes(1); expect(execGitShallowCloneMock).toHaveBeenCalledTimes(1); }); test('should fail when archive download fails and git is not installed', async () => { const downloadGitHubArchiveMock = vi.fn().mockRejectedValue(new Error('Archive download failed')); const execGitShallowCloneMock = vi.fn(); const isGitInstalledMock = vi.fn().mockResolvedValue(false); // Git is NOT installed vi.mocked(fs.copyFile).mockResolvedValue(undefined); await expect( runRemoteAction( 'yamadashy/repomix', {}, { isGitInstalled: isGitInstalledMock, execGitShallowClone: execGitShallowCloneMock, getRemoteRefs: async () => Promise.resolve(['main']), runDefaultAction: async () => { return { packResult: { totalFiles: 1, totalCharacters: 1, totalTokens: 1, fileCharCounts: {}, fileTokenCounts: {}, suspiciousFilesResults: [], suspiciousGitDiffResults: [], suspiciousGitLogResults: [], processedFiles: [], safeFilePaths: [], gitDiffTokenCount: 0, gitLogTokenCount: 0, skippedFiles: [], }, config: createMockConfig(), } satisfies DefaultActionRunnerResult; }, downloadGitHubArchive: downloadGitHubArchiveMock, isGitHubRepository: vi.fn().mockReturnValue(true), parseGitHubRepoInfo: vi.fn().mockReturnValue({ owner: 'yamadashy', repo: 'repomix' }), isArchiveDownloadSupported: vi.fn().mockReturnValue(true), }, ), ).rejects.toThrow('Git is not installed or not in the system PATH.'); expect(downloadGitHubArchiveMock).toHaveBeenCalledTimes(1); expect(isGitInstalledMock).toHaveBeenCalledTimes(1); // Git check should be called when fallback to git clone expect(execGitShallowCloneMock).not.toHaveBeenCalled(); }); }); describe('copyOutputToCurrentDirectory', () => { test('should copy output file when source and target are different', async () => { const sourceDir = '/source/dir'; const targetDir = '/target/dir'; const fileName = 'output.txt'; vi.mocked(fs.copyFile).mockResolvedValue(); await copyOutputToCurrentDirectory(sourceDir, targetDir, fileName); expect(fs.copyFile).toHaveBeenCalledWith(path.resolve(sourceDir, fileName), path.resolve(targetDir, fileName)); }); test('should skip copy when source and target are the same', async () => { const sourceDir = '/tmp/dir'; const targetDir = '/tmp/dir'; const fileName = 'output.txt'; vi.mocked(fs.copyFile).mockResolvedValue(); await copyOutputToCurrentDirectory(sourceDir, targetDir, fileName); // Should not call copyFile when source and target are the same expect(fs.copyFile).not.toHaveBeenCalled(); }); test('should skip copy when absolute path resolves to same location', async () => { const sourceDir = '/tmp/repomix-123'; const targetDir = process.cwd(); const absolutePath = '/tmp/my_private_dir/output.xml'; vi.mocked(fs.copyFile).mockResolvedValue(); await copyOutputToCurrentDirectory(sourceDir, targetDir, absolutePath); // When absolute path is used, both source and target resolve to the same path // path.resolve('/tmp/repomix-123', '/tmp/my_private_dir/output.xml') -> '/tmp/my_private_dir/output.xml' // path.resolve(process.cwd(), '/tmp/my_private_dir/output.xml') -> '/tmp/my_private_dir/output.xml' expect(fs.copyFile).not.toHaveBeenCalled(); }); test('should throw error when copy fails', async () => { const sourceDir = '/source/dir'; const targetDir = '/target/dir'; const fileName = 'output.txt'; vi.mocked(fs.copyFile).mockRejectedValue(new Error('Permission denied')); await expect(copyOutputToCurrentDirectory(sourceDir, targetDir, fileName)).rejects.toThrow( 'Failed to copy output file', ); }); }); });