summaryrefslogtreecommitdiff
path: root/packages/core/src/services
diff options
context:
space:
mode:
Diffstat (limited to 'packages/core/src/services')
-rw-r--r--packages/core/src/services/fileDiscoveryService.test.ts142
-rw-r--r--packages/core/src/services/fileDiscoveryService.ts89
2 files changed, 70 insertions, 161 deletions
diff --git a/packages/core/src/services/fileDiscoveryService.test.ts b/packages/core/src/services/fileDiscoveryService.test.ts
index 368f4d23..d7530cd6 100644
--- a/packages/core/src/services/fileDiscoveryService.test.ts
+++ b/packages/core/src/services/fileDiscoveryService.test.ts
@@ -8,15 +8,13 @@ import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest';
import type { Mocked } from 'vitest';
import { FileDiscoveryService } from './fileDiscoveryService.js';
import { GitIgnoreParser } from '../utils/gitIgnoreParser.js';
+import * as gitUtils from '../utils/gitUtils.js';
// Mock the GitIgnoreParser
vi.mock('../utils/gitIgnoreParser.js');
// Mock gitUtils module
-vi.mock('../utils/gitUtils.js', () => ({
- isGitRepository: vi.fn(() => true),
- findGitRoot: vi.fn(() => '/test/project'),
-}));
+vi.mock('../utils/gitUtils.js');
describe('FileDiscoveryService', () => {
let service: FileDiscoveryService;
@@ -24,16 +22,16 @@ describe('FileDiscoveryService', () => {
const mockProjectRoot = '/test/project';
beforeEach(() => {
- service = new FileDiscoveryService(mockProjectRoot);
-
mockGitIgnoreParser = {
initialize: vi.fn(),
isIgnored: vi.fn(),
- getIgnoredPatterns: vi.fn(() => ['.git/**', 'node_modules/**']),
- parseGitIgnoreContent: vi.fn(),
+ loadPatterns: vi.fn(),
+ loadGitRepoPatterns: vi.fn(),
} as unknown as Mocked<GitIgnoreParser>;
vi.mocked(GitIgnoreParser).mockImplementation(() => mockGitIgnoreParser);
+ vi.mocked(gitUtils.isGitRepository).mockReturnValue(true);
+ vi.mocked(gitUtils.findGitRoot).mockReturnValue('/test/project');
vi.clearAllMocks();
});
@@ -42,36 +40,30 @@ describe('FileDiscoveryService', () => {
});
describe('initialization', () => {
- it('should initialize git ignore parser by default', async () => {
- await service.initialize();
-
+ it('should initialize git ignore parser by default', () => {
+ service = new FileDiscoveryService(mockProjectRoot);
expect(GitIgnoreParser).toHaveBeenCalledWith(mockProjectRoot);
- expect(mockGitIgnoreParser.initialize).toHaveBeenCalled();
- });
-
- it('should not initialize git ignore parser when respectGitIgnore is false', async () => {
- await service.initialize({ respectGitIgnore: false });
-
- expect(GitIgnoreParser).not.toHaveBeenCalled();
- expect(mockGitIgnoreParser.initialize).not.toHaveBeenCalled();
+ expect(GitIgnoreParser).toHaveBeenCalledTimes(2);
+ expect(mockGitIgnoreParser.loadGitRepoPatterns).toHaveBeenCalled();
+ expect(mockGitIgnoreParser.loadPatterns).toHaveBeenCalled();
});
- it('should handle initialization errors gracefully', async () => {
- mockGitIgnoreParser.initialize.mockRejectedValue(
- new Error('Init failed'),
- );
+ it('should not initialize git ignore parser when not a git repo', () => {
+ vi.mocked(gitUtils.isGitRepository).mockReturnValue(false);
+ service = new FileDiscoveryService(mockProjectRoot);
- await expect(service.initialize()).rejects.toThrow('Init failed');
+ expect(GitIgnoreParser).toHaveBeenCalledOnce();
+ expect(mockGitIgnoreParser.loadGitRepoPatterns).not.toHaveBeenCalled();
});
});
describe('filterFiles', () => {
- beforeEach(async () => {
+ beforeEach(() => {
mockGitIgnoreParser.isIgnored.mockImplementation(
(path: string) =>
path.includes('node_modules') || path.includes('.git'),
);
- await service.initialize();
+ service = new FileDiscoveryService(mockProjectRoot);
});
it('should filter out git-ignored files by default', () => {
@@ -106,102 +98,22 @@ describe('FileDiscoveryService', () => {
});
});
- describe('shouldIgnoreFile', () => {
- beforeEach(async () => {
+ describe('shouldGitIgnoreFile', () => {
+ beforeEach(() => {
mockGitIgnoreParser.isIgnored.mockImplementation((path: string) =>
path.includes('node_modules'),
);
- await service.initialize();
+ service = new FileDiscoveryService(mockProjectRoot);
});
it('should return true for git-ignored files', () => {
- expect(service.shouldIgnoreFile('node_modules/package/index.js')).toBe(
+ expect(service.shouldGitIgnoreFile('node_modules/package/index.js')).toBe(
true,
);
});
it('should return false for non-ignored files', () => {
- expect(service.shouldIgnoreFile('src/index.ts')).toBe(false);
- });
-
- it('should return false when respectGitIgnore is false', () => {
- expect(
- service.shouldIgnoreFile('node_modules/package/index.js', {
- respectGitIgnore: false,
- }),
- ).toBe(false);
- });
-
- it('should return false when git ignore parser is not initialized', async () => {
- const uninitializedService = new FileDiscoveryService(mockProjectRoot);
- expect(
- uninitializedService.shouldIgnoreFile('node_modules/package/index.js'),
- ).toBe(false);
- });
- });
-
- describe('isGitRepository', () => {
- it('should return true when isGitRepo is explicitly set to true in options', () => {
- const result = service.isGitRepository({ isGitRepo: true });
- expect(result).toBe(true);
- });
-
- it('should return false when isGitRepo is explicitly set to false in options', () => {
- const result = service.isGitRepository({ isGitRepo: false });
- expect(result).toBe(false);
- });
-
- it('should use git utility function when isGitRepo is not specified', () => {
- const result = service.isGitRepository();
- expect(result).toBe(true); // mocked to return true
- });
-
- it('should use git utility function when options are undefined', () => {
- const result = service.isGitRepository(undefined);
- expect(result).toBe(true); // mocked to return true
- });
- });
-
- describe('initialization with isGitRepo config', () => {
- it('should initialize git ignore parser when isGitRepo is true in options', async () => {
- await service.initialize({ isGitRepo: true });
-
- expect(GitIgnoreParser).toHaveBeenCalledWith(mockProjectRoot);
- expect(mockGitIgnoreParser.initialize).toHaveBeenCalled();
- });
-
- it('should not initialize git ignore parser when isGitRepo is false in options', async () => {
- await service.initialize({ isGitRepo: false });
-
- expect(GitIgnoreParser).not.toHaveBeenCalled();
- expect(mockGitIgnoreParser.initialize).not.toHaveBeenCalled();
- });
-
- it('should initialize git ignore parser when isGitRepo is not specified but respectGitIgnore is true', async () => {
- await service.initialize({ respectGitIgnore: true });
-
- expect(GitIgnoreParser).toHaveBeenCalledWith(mockProjectRoot);
- expect(mockGitIgnoreParser.initialize).toHaveBeenCalled();
- });
- });
-
- describe('shouldIgnoreFile with isGitRepo config', () => {
- it('should respect isGitRepo option when checking if file should be ignored', async () => {
- mockGitIgnoreParser.isIgnored.mockImplementation((path: string) =>
- path.includes('node_modules'),
- );
- await service.initialize({ isGitRepo: true });
-
- expect(
- service.shouldIgnoreFile('node_modules/package/index.js', {
- isGitRepo: true,
- }),
- ).toBe(true);
- expect(
- service.shouldIgnoreFile('node_modules/package/index.js', {
- isGitRepo: false,
- }),
- ).toBe(false);
+ expect(service.shouldGitIgnoreFile('src/index.ts')).toBe(false);
});
});
@@ -211,13 +123,7 @@ describe('FileDiscoveryService', () => {
expect(relativeService).toBeInstanceOf(FileDiscoveryService);
});
- it('should handle undefined options', async () => {
- await service.initialize(undefined);
- expect(GitIgnoreParser).toHaveBeenCalled();
- });
-
- it('should handle filterFiles with undefined options', async () => {
- await service.initialize();
+ it('should handle filterFiles with undefined options', () => {
const files = ['src/index.ts'];
const filtered = service.filterFiles(files, undefined);
expect(filtered).toEqual(files);
diff --git a/packages/core/src/services/fileDiscoveryService.ts b/packages/core/src/services/fileDiscoveryService.ts
index 7531f90a..984f3f53 100644
--- a/packages/core/src/services/fileDiscoveryService.ts
+++ b/packages/core/src/services/fileDiscoveryService.ts
@@ -7,41 +7,37 @@
import { GitIgnoreParser, GitIgnoreFilter } from '../utils/gitIgnoreParser.js';
import { isGitRepository } from '../utils/gitUtils.js';
import * as path from 'path';
-import { glob, type GlobOptions } from 'glob';
-export interface FileDiscoveryOptions {
+const GEMINI_IGNORE_FILE_NAME = '.geminiignore';
+
+export interface FilterFilesOptions {
respectGitIgnore?: boolean;
- includeBuildArtifacts?: boolean;
- isGitRepo?: boolean;
+ respectGeminiIgnore?: boolean;
}
export class FileDiscoveryService {
private gitIgnoreFilter: GitIgnoreFilter | null = null;
+ private geminiIgnoreFilter: GitIgnoreFilter | null = null;
private projectRoot: string;
constructor(projectRoot: string) {
this.projectRoot = path.resolve(projectRoot);
- }
-
- async initialize(options: FileDiscoveryOptions = {}): Promise<void> {
- const isGitRepo = options.isGitRepo ?? isGitRepository(this.projectRoot);
-
- if (options.respectGitIgnore !== false && isGitRepo) {
+ if (isGitRepository(this.projectRoot)) {
const parser = new GitIgnoreParser(this.projectRoot);
- await parser.initialize();
+ try {
+ parser.loadGitRepoPatterns();
+ } catch (_error) {
+ // ignore file not found
+ }
this.gitIgnoreFilter = parser;
}
- }
-
- async glob(
- pattern: string | string[],
- options: GlobOptions = {},
- ): Promise<string[]> {
- const files = (await glob(pattern, {
- ...options,
- nocase: true,
- })) as string[];
- return this.filterFiles(files);
+ const gParser = new GitIgnoreParser(this.projectRoot);
+ try {
+ gParser.loadPatterns(GEMINI_IGNORE_FILE_NAME);
+ } catch (_error) {
+ // ignore file not found
+ }
+ this.geminiIgnoreFilter = gParser;
}
/**
@@ -49,42 +45,49 @@ export class FileDiscoveryService {
*/
filterFiles(
filePaths: string[],
- options: FileDiscoveryOptions = {},
+ options: FilterFilesOptions = {
+ respectGitIgnore: true,
+ respectGeminiIgnore: true,
+ },
): string[] {
return filePaths.filter((filePath) => {
- // Always respect git ignore unless explicitly disabled
- if (options.respectGitIgnore !== false && this.gitIgnoreFilter) {
- if (this.gitIgnoreFilter.isIgnored(filePath)) {
- return false;
- }
+ if (options.respectGitIgnore && this.shouldGitIgnoreFile(filePath)) {
+ return false;
+ }
+ if (
+ options.respectGeminiIgnore &&
+ this.shouldGeminiIgnoreFile(filePath)
+ ) {
+ return false;
}
-
return true;
});
}
/**
- * Checks if a single file should be ignored
+ * Checks if a single file should be git-ignored
*/
- shouldIgnoreFile(
- filePath: string,
- options: FileDiscoveryOptions = {},
- ): boolean {
- const isGitRepo = options.isGitRepo ?? isGitRepository(this.projectRoot);
- if (
- options.respectGitIgnore !== false &&
- isGitRepo &&
- this.gitIgnoreFilter
- ) {
+ shouldGitIgnoreFile(filePath: string): boolean {
+ if (this.gitIgnoreFilter) {
return this.gitIgnoreFilter.isIgnored(filePath);
}
return false;
}
/**
- * Returns whether the project is a git repository
+ * Checks if a single file should be gemini-ignored
+ */
+ shouldGeminiIgnoreFile(filePath: string): boolean {
+ if (this.geminiIgnoreFilter) {
+ return this.geminiIgnoreFilter.isIgnored(filePath);
+ }
+ return false;
+ }
+
+ /**
+ * Returns loaded patterns from .geminiignore
*/
- isGitRepository(options: FileDiscoveryOptions = {}): boolean {
- return options.isGitRepo ?? isGitRepository(this.projectRoot);
+ getGeminiIgnorePatterns(): string[] {
+ return this.geminiIgnoreFilter?.getPatterns() ?? [];
}
}