import { describe, expect, test, vi } from 'vitest';
import type { RepomixConfigMerged } from '../../../src/config/configSchema.js';
import type { ProcessedFile } from '../../../src/core/file/fileTypes.js';
import { buildOutputGeneratorContext } from '../../../src/core/output/outputGenerate.js';

const createMockConfig = (overrides: Partial<RepomixConfigMerged> = {}): RepomixConfigMerged => ({
  cwd: '/repo',
  input: { maxFileSize: 1024 * 1024 },
  output: {
    filePath: 'repomix-output.json',
    style: 'json',
    filePathStyle: 'target-relative',
    parsableStyle: false,
    headerText: undefined,
    instructionFilePath: undefined,
    fileSummary: true,
    directoryStructure: true,
    files: true,
    removeComments: false,
    removeEmptyLines: false,
    compress: false,
    topFilesLength: 5,
    showLineNumbers: false,
    truncateBase64: false,
    copyToClipboard: false,
    includeEmptyDirectories: false,
    includeFullDirectoryStructure: true,
    tokenCountTree: false,
    git: {
      sortByChanges: false,
      sortByChangesMaxCommits: 10,
      includeDiffs: false,
      includeLogs: false,
      includeLogsCount: 5,
    },
  },
  include: ['src/**/*.ts'],
  ignore: {
    useGitignore: true,
    useDotIgnore: true,
    useDefaultPatterns: true,
    customPatterns: [],
  },
  security: {
    enableSecurityCheck: true,
  },
  tokenCount: {
    encoding: 'cl100k_base',
  },
  ...overrides,
});

describe('includeFullDirectoryStructure flag', () => {
  test('renders full repository tree including root-level files when flag is enabled and include patterns present', async () => {
    const config = createMockConfig();
    const processedFiles: ProcessedFile[] = [{ path: 'src/a/index.ts', content: 'export const a = 1;\n' }];
    const allFilePaths = processedFiles.map((f) => f.path);

    const deps = {
      // Return a directory set that includes paths outside of the included files
      listDirectories: async () => ['src', 'src/a', 'src/b', 'docs', 'docs/guide'],
      // Files across the repo (subject to ignores)
      listFiles: async () => ['README.md', 'LICENSE.md', 'src/a/index.ts', 'src/b/other.ts', 'docs/guide/intro.md'],
      // Not used in full-tree branch, but included for completeness
      searchFiles: async () => ({ filePaths: allFilePaths, emptyDirPaths: [] }),
    };

    const ctx = await buildOutputGeneratorContext(
      ['/repo'],
      config,
      allFilePaths,
      processedFiles,
      undefined,
      undefined,
      undefined,
      undefined,
      deps,
    );

    // Expect the tree to include root-level files beyond those derived from included files
    expect(ctx.treeString).toContain('README.md');
    expect(ctx.treeString).toContain('LICENSE.md');
    // Expect directories beyond those derived from files
    expect(ctx.treeString).toContain('docs');
    expect(ctx.treeString).toContain('src');
    // Should still include the file name within the tree
    expect(ctx.treeString).toContain('index.ts');
  });

  test('does not prefix single-root full-tree directories with the root basename', async () => {
    const config = createMockConfig();
    const processedFiles: ProcessedFile[] = [{ path: 'src/a/index.ts', content: 'export const a = 1;\n' }];
    const allFilePaths = processedFiles.map((f) => f.path);

    const deps = {
      listDirectories: async () => ['src', 'src/a', 'src/b', 'docs', 'docs/guide'],
      listFiles: async () => ['src/a/index.ts', 'src/b/other.ts', 'docs/guide/intro.md'],
      searchFiles: async () => ({ filePaths: allFilePaths, emptyDirPaths: [] }),
    };

    // pack() passes a one-entry filePathsByRoot for a single target-relative root,
    // whose rootLabel falls back to the target basename. That label must NOT prefix
    // the full-tree directories: the included file paths are left unprefixed for a
    // single root, so prefixing here would desync them and add a spurious `repo/` branch.
    const filePathsByRoot = [{ rootLabel: 'repo', files: allFilePaths }];

    const ctx = await buildOutputGeneratorContext(
      ['/repo'],
      config,
      allFilePaths,
      processedFiles,
      undefined,
      undefined,
      filePathsByRoot,
      undefined,
      deps,
    );

    // Directories from the full tree should appear unprefixed alongside the included file
    expect(ctx.treeString).toContain('src/');
    expect(ctx.treeString).toContain('docs/');
    expect(ctx.treeString).toContain('index.ts');
    // No spurious root-basename branch
    expect(ctx.treeString).not.toContain('repo/');
  });

  test('does not render full tree when include is empty even if flag is enabled', async () => {
    const config = createMockConfig({ include: [] });
    const processedFiles: ProcessedFile[] = [{ path: 'src/a/index.ts', content: 'export const a = 1;\n' }];
    const allFilePaths = processedFiles.map((f) => f.path);

    const deps = {
      // Would return extra directories if called, but should NOT be used in this case
      listDirectories: async () => ['src', 'src/a', 'docs', 'docs/guide'],
      listFiles: async () => ['README.md', 'LICENSE.md'],
      searchFiles: async () => ({ filePaths: allFilePaths, emptyDirPaths: [] }),
    };

    const ctx = await buildOutputGeneratorContext(
      ['/repo'],
      config,
      allFilePaths,
      processedFiles,
      undefined,
      undefined,
      undefined,
      undefined,
      deps,
    );

    // Should not include directories/files outside of file-derived tree ('docs' and root files should not appear)
    expect(ctx.treeString).not.toContain('docs');
    expect(ctx.treeString).not.toContain('README.md');
    expect(ctx.treeString).not.toContain('LICENSE.md');
    // Should include the file-derived structure
    expect(ctx.treeString).toContain('src');
    expect(ctx.treeString).toContain('index.ts');
  });
});

describe('includeEmptyDirectories with pre-computed emptyDirPaths', () => {
  const createEmptyDirConfig = (overrides: Partial<RepomixConfigMerged> = {}): RepomixConfigMerged => ({
    cwd: '/repo',
    input: { maxFileSize: 1024 * 1024 },
    output: {
      filePath: 'repomix-output.json',
      style: 'json',
      filePathStyle: 'target-relative',
      parsableStyle: false,
      headerText: undefined,
      instructionFilePath: undefined,
      fileSummary: true,
      directoryStructure: true,
      files: true,
      removeComments: false,
      removeEmptyLines: false,
      compress: false,
      topFilesLength: 5,
      showLineNumbers: false,
      truncateBase64: false,
      copyToClipboard: false,
      includeEmptyDirectories: true,
      includeFullDirectoryStructure: false,
      tokenCountTree: false,
      git: {
        sortByChanges: false,
        sortByChangesMaxCommits: 10,
        includeDiffs: false,
        includeLogs: false,
        includeLogsCount: 5,
      },
    },
    include: [],
    ignore: {
      useGitignore: true,
      useDotIgnore: true,
      useDefaultPatterns: true,
      customPatterns: [],
    },
    security: { enableSecurityCheck: true },
    tokenCount: { encoding: 'cl100k_base' },
    ...overrides,
  });

  test('uses pre-computed emptyDirPaths and skips searchFiles call', async () => {
    const config = createEmptyDirConfig();
    const processedFiles: ProcessedFile[] = [{ path: 'src/index.ts', content: 'export const a = 1;\n' }];
    const allFilePaths = processedFiles.map((f) => f.path);
    const preComputedEmptyDirs = ['empty-dir'];

    const deps = {
      listDirectories: vi.fn(),
      listFiles: vi.fn(),
      searchFiles: vi.fn().mockResolvedValue({ filePaths: allFilePaths, emptyDirPaths: ['should-not-use'] }),
    };

    const ctx = await buildOutputGeneratorContext(
      ['/repo'],
      config,
      allFilePaths,
      processedFiles,
      undefined,
      undefined,
      undefined,
      preComputedEmptyDirs,
      deps,
    );

    // searchFiles should NOT be called when emptyDirPaths is provided
    expect(deps.searchFiles).not.toHaveBeenCalled();
    // The pre-computed empty dir should appear in the tree
    expect(ctx.treeString).toContain('empty-dir');
  });

  test('falls back to searchFiles when emptyDirPaths is not provided', async () => {
    const config = createEmptyDirConfig();
    const processedFiles: ProcessedFile[] = [{ path: 'src/index.ts', content: 'export const a = 1;\n' }];
    const allFilePaths = processedFiles.map((f) => f.path);

    const deps = {
      listDirectories: vi.fn(),
      listFiles: vi.fn(),
      searchFiles: vi.fn().mockResolvedValue({ filePaths: allFilePaths, emptyDirPaths: ['fallback-empty-dir'] }),
    };

    const ctx = await buildOutputGeneratorContext(
      ['/repo'],
      config,
      allFilePaths,
      processedFiles,
      undefined,
      undefined,
      undefined,
      undefined,
      deps,
    );

    // searchFiles SHOULD be called as fallback
    expect(deps.searchFiles).toHaveBeenCalledWith('/repo', config);
    // The fallback empty dir should appear in the tree
    expect(ctx.treeString).toContain('fallback-empty-dir');
  });
});