@didim365/agent-cli-core 0.1.0

package/dist/src/services/gitService.test.js

@@ -0,0 +1,264 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
+import { GitService } from './gitService.js';
+import { Storage } from '../config/storage.js';
+import * as path from 'node:path';
+import * as fs from 'node:fs/promises';
+import * as os from 'node:os';
+import { getProjectHash, GEMINI_DIR, homedir as pathsHomedir, } from '../utils/paths.js';
+import { spawnAsync } from '../utils/shell-utils.js';
+vi.mock('../utils/shell-utils.js', () => ({
+    spawnAsync: vi.fn(),
+}));
+const hoistedMockEnv = vi.hoisted(() => vi.fn());
+const hoistedMockSimpleGit = vi.hoisted(() => vi.fn());
+const hoistedMockCheckIsRepo = vi.hoisted(() => vi.fn());
+const hoistedMockInit = vi.hoisted(() => vi.fn());
+const hoistedMockRaw = vi.hoisted(() => vi.fn());
+const hoistedMockAdd = vi.hoisted(() => vi.fn());
+const hoistedMockCommit = vi.hoisted(() => vi.fn());
+const hoistedMockStatus = vi.hoisted(() => vi.fn());
+vi.mock('simple-git', () => ({
+    simpleGit: hoistedMockSimpleGit.mockImplementation(() => ({
+        checkIsRepo: hoistedMockCheckIsRepo,
+        init: hoistedMockInit,
+        raw: hoistedMockRaw,
+        add: hoistedMockAdd,
+        commit: hoistedMockCommit,
+        status: hoistedMockStatus,
+        env: hoistedMockEnv,
+    })),
+    CheckRepoActions: { IS_REPO_ROOT: 'is-repo-root' },
+}));
+const hoistedIsGitRepositoryMock = vi.hoisted(() => vi.fn());
+vi.mock('../utils/gitUtils.js', () => ({
+    isGitRepository: hoistedIsGitRepositoryMock,
+}));
+const hoistedMockHomedir = vi.hoisted(() => vi.fn());
+vi.mock('node:os', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        homedir: hoistedMockHomedir,
+    };
+});
+vi.mock('../utils/paths.js', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        homedir: vi.fn(),
+    };
+});
+const hoistedMockDebugLogger = vi.hoisted(() => ({
+    debug: vi.fn(),
+    warn: vi.fn(),
+    error: vi.fn(),
+}));
+vi.mock('../utils/debugLogger.js', () => ({
+    debugLogger: hoistedMockDebugLogger,
+}));
+describe('GitService', () => {
+    let testRootDir;
+    let projectRoot;
+    let homedir;
+    let hash;
+    let storage;
+    beforeEach(async () => {
+        testRootDir = await fs.mkdtemp(path.join(os.tmpdir(), 'git-service-test-'));
+        projectRoot = path.join(testRootDir, 'project');
+        homedir = path.join(testRootDir, 'home');
+        await fs.mkdir(projectRoot, { recursive: true });
+        await fs.mkdir(homedir, { recursive: true });
+        hash = getProjectHash(projectRoot);
+        vi.clearAllMocks();
+        hoistedIsGitRepositoryMock.mockReturnValue(true);
+        spawnAsync.mockResolvedValue({
+            stdout: 'git version 2.0.0',
+            stderr: '',
+        });
+        hoistedMockHomedir.mockReturnValue(homedir);
+        pathsHomedir.mockReturnValue(homedir);
+        hoistedMockEnv.mockImplementation(() => ({
+            checkIsRepo: hoistedMockCheckIsRepo,
+            init: hoistedMockInit,
+            raw: hoistedMockRaw,
+            add: hoistedMockAdd,
+            commit: hoistedMockCommit,
+            status: hoistedMockStatus,
+        }));
+        hoistedMockSimpleGit.mockImplementation(() => ({
+            checkIsRepo: hoistedMockCheckIsRepo,
+            init: hoistedMockInit,
+            raw: hoistedMockRaw,
+            add: hoistedMockAdd,
+            commit: hoistedMockCommit,
+            status: hoistedMockStatus,
+            env: hoistedMockEnv,
+        }));
+        hoistedMockCheckIsRepo.mockResolvedValue(false);
+        hoistedMockInit.mockResolvedValue(undefined);
+        hoistedMockRaw.mockResolvedValue('');
+        hoistedMockAdd.mockResolvedValue(undefined);
+        hoistedMockCommit.mockResolvedValue({
+            commit: 'initial',
+        });
+        storage = new Storage(projectRoot);
+    });
+    afterEach(async () => {
+        vi.restoreAllMocks();
+        await fs.rm(testRootDir, { recursive: true, force: true });
+    });
+    describe('constructor', () => {
+        it('should successfully create an instance', () => {
+            expect(() => new GitService(projectRoot, storage)).not.toThrow();
+        });
+    });
+    describe('verifyGitAvailability', () => {
+        it('should resolve true if git --version command succeeds', async () => {
+            await expect(GitService.verifyGitAvailability()).resolves.toBe(true);
+            expect(spawnAsync).toHaveBeenCalledWith('git', ['--version']);
+        });
+        it('should resolve false if git --version command fails', async () => {
+            spawnAsync.mockRejectedValue(new Error('git not found'));
+            await expect(GitService.verifyGitAvailability()).resolves.toBe(false);
+        });
+    });
+    describe('initialize', () => {
+        it('should throw an error if Git is not available', async () => {
+            spawnAsync.mockRejectedValue(new Error('git not found'));
+            const service = new GitService(projectRoot, storage);
+            await expect(service.initialize()).rejects.toThrow('Checkpointing is enabled, but Git is not installed. Please install Git or disable checkpointing to continue.');
+        });
+        it('should call setupShadowGitRepository if Git is available', async () => {
+            const service = new GitService(projectRoot, storage);
+            const setupSpy = vi
+                .spyOn(service, 'setupShadowGitRepository')
+                .mockResolvedValue(undefined);
+            await service.initialize();
+            expect(setupSpy).toHaveBeenCalled();
+        });
+    });
+    describe('setupShadowGitRepository', () => {
+        let repoDir;
+        let gitConfigPath;
+        beforeEach(() => {
+            repoDir = path.join(homedir, GEMINI_DIR, 'history', hash);
+            gitConfigPath = path.join(repoDir, '.gitconfig');
+        });
+        it('should create history and repository directories', async () => {
+            const service = new GitService(projectRoot, storage);
+            await service.setupShadowGitRepository();
+            const stats = await fs.stat(repoDir);
+            expect(stats.isDirectory()).toBe(true);
+        });
+        it('should create a .gitconfig file with the correct content', async () => {
+            const service = new GitService(projectRoot, storage);
+            await service.setupShadowGitRepository();
+            const expectedConfigContent = '[user]\n  name = Gemini CLI\n  email = gemini-cli@google.com\n[commit]\n  gpgsign = false\n';
+            const actualConfigContent = await fs.readFile(gitConfigPath, 'utf-8');
+            expect(actualConfigContent).toBe(expectedConfigContent);
+        });
+        it('should initialize git repo in historyDir if not already initialized', async () => {
+            hoistedMockCheckIsRepo.mockResolvedValue(false);
+            const service = new GitService(projectRoot, storage);
+            await service.setupShadowGitRepository();
+            expect(hoistedMockSimpleGit).toHaveBeenCalledWith(repoDir);
+            expect(hoistedMockInit).toHaveBeenCalled();
+        });
+        it('should not initialize git repo if already initialized', async () => {
+            hoistedMockCheckIsRepo.mockResolvedValue(true);
+            const service = new GitService(projectRoot, storage);
+            await service.setupShadowGitRepository();
+            expect(hoistedMockInit).not.toHaveBeenCalled();
+        });
+        it('should copy .gitignore from projectRoot if it exists', async () => {
+            const gitignoreContent = 'node_modules/\n.env';
+            const visibleGitIgnorePath = path.join(projectRoot, '.gitignore');
+            await fs.writeFile(visibleGitIgnorePath, gitignoreContent);
+            const service = new GitService(projectRoot, storage);
+            await service.setupShadowGitRepository();
+            const hiddenGitIgnorePath = path.join(repoDir, '.gitignore');
+            const copiedContent = await fs.readFile(hiddenGitIgnorePath, 'utf-8');
+            expect(copiedContent).toBe(gitignoreContent);
+        });
+        it('should not create a .gitignore in shadow repo if project .gitignore does not exist', async () => {
+            const service = new GitService(projectRoot, storage);
+            await service.setupShadowGitRepository();
+            const hiddenGitIgnorePath = path.join(repoDir, '.gitignore');
+            // An empty string is written if the file doesn't exist.
+            const content = await fs.readFile(hiddenGitIgnorePath, 'utf-8');
+            expect(content).toBe('');
+        });
+        it('should throw an error if reading projectRoot .gitignore fails with other errors', async () => {
+            const visibleGitIgnorePath = path.join(projectRoot, '.gitignore');
+            // Create a directory instead of a file to cause a read error
+            await fs.mkdir(visibleGitIgnorePath);
+            const service = new GitService(projectRoot, storage);
+            // EISDIR is the expected error code on Unix-like systems
+            await expect(service.setupShadowGitRepository()).rejects.toThrow(/EISDIR: illegal operation on a directory, read|EBUSY: resource busy or locked, read/);
+        });
+        it('should make an initial commit if no commits exist in history repo', async () => {
+            hoistedMockCheckIsRepo.mockResolvedValue(false);
+            const service = new GitService(projectRoot, storage);
+            await service.setupShadowGitRepository();
+            expect(hoistedMockCommit).toHaveBeenCalledWith('Initial commit', {
+                '--allow-empty': null,
+            });
+        });
+        it('should not make an initial commit if commits already exist', async () => {
+            hoistedMockCheckIsRepo.mockResolvedValue(true);
+            const service = new GitService(projectRoot, storage);
+            await service.setupShadowGitRepository();
+            expect(hoistedMockCommit).not.toHaveBeenCalled();
+        });
+        it('should handle checkIsRepo failure gracefully and initialize repo', async () => {
+            // Simulate checkIsRepo failing (e.g., on certain Git versions like macOS 2.39.5)
+            hoistedMockCheckIsRepo.mockRejectedValue(new Error('git rev-parse --is-inside-work-tree failed'));
+            const service = new GitService(projectRoot, storage);
+            await service.setupShadowGitRepository();
+            // Should proceed to initialize the repo since checkIsRepo failed
+            expect(hoistedMockInit).toHaveBeenCalled();
+            // Should log the error using debugLogger
+            expect(hoistedMockDebugLogger.debug).toHaveBeenCalledWith(expect.stringContaining('checkIsRepo failed'));
+        });
+    });
+    describe('createFileSnapshot', () => {
+        it('should commit with --no-verify flag', async () => {
+            hoistedMockStatus.mockResolvedValue({ isClean: () => false });
+            const service = new GitService(projectRoot, storage);
+            await service.initialize();
+            await service.createFileSnapshot('test commit');
+            expect(hoistedMockCommit).toHaveBeenCalledWith('test commit', {
+                '--no-verify': null,
+            });
+        });
+        it('should create a new commit if there are staged changes', async () => {
+            hoistedMockStatus.mockResolvedValue({ isClean: () => false });
+            hoistedMockCommit.mockResolvedValue({ commit: 'new-commit-hash' });
+            const service = new GitService(projectRoot, storage);
+            const commitHash = await service.createFileSnapshot('test message');
+            expect(hoistedMockAdd).toHaveBeenCalledWith('.');
+            expect(hoistedMockStatus).toHaveBeenCalled();
+            expect(hoistedMockCommit).toHaveBeenCalledWith('test message', {
+                '--no-verify': null,
+            });
+            expect(commitHash).toBe('new-commit-hash');
+        });
+        it('should return the current HEAD commit hash if there are no staged changes', async () => {
+            hoistedMockStatus.mockResolvedValue({ isClean: () => true });
+            hoistedMockRaw.mockResolvedValue('current-head-hash');
+            const service = new GitService(projectRoot, storage);
+            const commitHash = await service.createFileSnapshot('test message');
+            expect(hoistedMockAdd).toHaveBeenCalledWith('.');
+            expect(hoistedMockStatus).toHaveBeenCalled();
+            expect(hoistedMockCommit).not.toHaveBeenCalled();
+            expect(hoistedMockRaw).toHaveBeenCalledWith('rev-parse', 'HEAD');
+            expect(commitHash).toBe('current-head-hash');
+        });
+    });
+});
+//# sourceMappingURL=gitService.test.js.map

package/dist/src/services/gitService.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gitService.test.js","sourceRoot":"","sources":["../../../src/services/gitService.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EACL,cAAc,EACd,UAAU,EACV,OAAO,IAAI,YAAY,GACxB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAErD,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;CACpB,CAAC,CAAC,CAAC;AAEJ,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACjD,MAAM,oBAAoB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACvD,MAAM,sBAAsB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACzD,MAAM,eAAe,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAClD,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACjD,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACjD,MAAM,iBAAiB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACpD,MAAM,iBAAiB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACpD,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3B,SAAS,EAAE,oBAAoB,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QACxD,WAAW,EAAE,sBAAsB;QACnC,IAAI,EAAE,eAAe;QACrB,GAAG,EAAE,cAAc;QACnB,GAAG,EAAE,cAAc;QACnB,MAAM,EAAE,iBAAiB;QACzB,MAAM,EAAE,iBAAiB;QACzB,GAAG,EAAE,cAAc;KACpB,CAAC,CAAC;IACH,gBAAgB,EAAE,EAAE,YAAY,EAAE,cAAc,EAAE;CACnD,CAAC,CAAC,CAAC;AAEJ,MAAM,0BAA0B,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAC7D,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,eAAe,EAAE,0BAA0B;CAC5C,CAAC,CAAC,CAAC;AAEJ,MAAM,kBAAkB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACrD,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC1C,MAAM,MAAM,GAAG,MAAM,cAAc,EAAa,CAAC;IACjD,OAAO;QACL,GAAG,MAAM;QACT,OAAO,EAAE,kBAAkB;KAC5B,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IACpD,MAAM,MAAM,GAAG,MAAM,cAAc,EAAsC,CAAC;IAC1E,OAAO;QACL,GAAG,MAAM;QACT,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;KACjB,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,MAAM,sBAAsB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IAC/C,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;IACd,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;IACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;CACf,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,WAAW,EAAE,sBAAsB;CACpC,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,IAAI,WAAmB,CAAC;IACxB,IAAI,WAAmB,CAAC;IACxB,IAAI,OAAe,CAAC;IACpB,IAAI,IAAY,CAAC;IACjB,IAAI,OAAgB,CAAC;IAErB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,WAAW,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QAC5E,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QAChD,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;QACzC,MAAM,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACjD,MAAM,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAE7C,IAAI,GAAG,cAAc,CAAC,WAAW,CAAC,CAAC;QAEnC,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,0BAA0B,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAChD,UAAmB,CAAC,iBAAiB,CAAC;YACrC,MAAM,EAAE,mBAAmB;YAC3B,MAAM,EAAE,EAAE;SACX,CAAC,CAAC;QAEH,kBAAkB,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAC3C,YAAqB,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAEhD,cAAc,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;YACvC,WAAW,EAAE,sBAAsB;YACnC,IAAI,EAAE,eAAe;YACrB,GAAG,EAAE,cAAc;YACnB,GAAG,EAAE,cAAc;YACnB,MAAM,EAAE,iBAAiB;YACzB,MAAM,EAAE,iBAAiB;SAC1B,CAAC,CAAC,CAAC;QACJ,oBAAoB,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;YAC7C,WAAW,EAAE,sBAAsB;YACnC,IAAI,EAAE,eAAe;YACrB,GAAG,EAAE,cAAc;YACnB,GAAG,EAAE,cAAc;YACnB,MAAM,EAAE,iBAAiB;YACzB,MAAM,EAAE,iBAAiB;YACzB,GAAG,EAAE,cAAc;SACpB,CAAC,CAAC,CAAC;QACJ,sBAAsB,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAChD,eAAe,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC7C,cAAc,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QACrC,cAAc,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC5C,iBAAiB,CAAC,iBAAiB,CAAC;YAClC,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC;QACH,OAAO,GAAG,IAAI,OAAO,CAAC,WAAW,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,EAAE,CAAC,eAAe,EAAE,CAAC;QACrB,MAAM,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,MAAM,CAAC,UAAU,CAAC,qBAAqB,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrE,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YAClE,UAAmB,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;YACnE,MAAM,MAAM,CAAC,UAAU,CAAC,qBAAqB,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC5D,UAAmB,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;YACnE,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAChD,8GAA8G,CAC/G,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,QAAQ,GAAG,EAAE;iBAChB,KAAK,CAAC,OAAO,EAAE,0BAA0B,CAAC;iBAC1C,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEhC,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;YAC3B,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,IAAI,OAAe,CAAC;QACpB,IAAI,aAAqB,CAAC;QAE1B,UAAU,CAAC,GAAG,EAAE;YACd,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;YAC1D,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC;YACzC,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC;YAEzC,MAAM,qBAAqB,GACzB,6FAA6F,CAAC;YAChG,MAAM,mBAAmB,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,sBAAsB,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAChD,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC;YACzC,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAC3D,MAAM,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,sBAAsB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC/C,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC;YACzC,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,gBAAgB,GAAG,qBAAqB,CAAC;YAC/C,MAAM,oBAAoB,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YAClE,MAAM,EAAE,CAAC,SAAS,CAAC,oBAAoB,EAAE,gBAAgB,CAAC,CAAC;YAE3D,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC;YAEzC,MAAM,mBAAmB,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAC7D,MAAM,aAAa,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;YAClG,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC;YAEzC,MAAM,mBAAmB,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAC7D,wDAAwD;YACxD,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;YAChE,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;YAC/F,MAAM,oBAAoB,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YAClE,6DAA6D;YAC7D,MAAM,EAAE,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;YAErC,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,yDAAyD;YACzD,MAAM,MAAM,CAAC,OAAO,CAAC,wBAAwB,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAC9D,qFAAqF,CACtF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,sBAAsB,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAChD,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC;YACzC,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,EAAE;gBAC/D,eAAe,EAAE,IAAI;aACtB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,sBAAsB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC/C,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC;YACzC,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,iFAAiF;YACjF,sBAAsB,CAAC,iBAAiB,CACtC,IAAI,KAAK,CAAC,4CAA4C,CAAC,CACxD,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC;YACzC,iEAAiE;YACjE,MAAM,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC3C,yCAAyC;YACzC,MAAM,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACvD,MAAM,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,CAC9C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,iBAAiB,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;YAC9D,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;YAC3B,MAAM,OAAO,CAAC,kBAAkB,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,aAAa,EAAE;gBAC5D,aAAa,EAAE,IAAI;aACpB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,iBAAiB,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;YAC9D,iBAAiB,CAAC,iBAAiB,CAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC;YACnE,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;YACpE,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;YACjD,MAAM,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC7C,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,cAAc,EAAE;gBAC7D,aAAa,EAAE,IAAI;aACpB,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,iBAAiB,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YAC7D,cAAc,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;YACtD,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;YACpE,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;YACjD,MAAM,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC7C,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACjD,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;YACjE,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/services/loopDetectionService.d.ts

@@ -0,0 +1,107 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { LlmEvent } from '../providers/events.js';
+import type { Config } from '../config/config.js';
+/**
+ * Service for detecting and preventing infinite loops in AI responses.
+ * Monitors tool call repetitions and content sentence repetitions.
+ */
+export declare class LoopDetectionService {
+    private readonly config;
+    private promptId;
+    private lastToolCallKey;
+    private toolCallRepetitionCount;
+    private streamContentHistory;
+    private contentStats;
+    private lastContentIndex;
+    private loopDetected;
+    private inCodeBlock;
+    private turnsInCurrentPrompt;
+    private llmCheckInterval;
+    private lastCheckTurn;
+    private disabledForSession;
+    constructor(config: Config);
+    /**
+     * Disables loop detection for the current session.
+     */
+    disableForSession(): void;
+    private getToolCallKey;
+    /**
+     * Processes a provider-independent LlmEvent and checks for loop conditions.
+     *
+     * @param event - The provider-independent LlmEvent to process
+     * @returns true if a loop is detected, false otherwise
+     */
+    addAndCheck(event: LlmEvent): boolean;
+    /**
+     * Signals the start of a new turn in the conversation.
+     *
+     * This method increments the turn counter and, if specific conditions are met,
+     * triggers an LLM-based check to detect potential conversation loops. The check
+     * is performed periodically based on the `llmCheckInterval`.
+     *
+     * @param signal - An AbortSignal to allow for cancellation of the asynchronous LLM check.
+     * @returns A promise that resolves to `true` if a loop is detected, and `false` otherwise.
+     */
+    turnStarted(signal: AbortSignal): Promise<boolean>;
+    private checkToolCallLoop;
+    /**
+     * Detects content loops by analyzing streaming text for repetitive patterns.
+     *
+     * The algorithm works by:
+     * 1. Appending new content to the streaming history
+     * 2. Truncating history if it exceeds the maximum length
+     * 3. Analyzing content chunks for repetitive patterns using hashing
+     * 4. Detecting loops when identical chunks appear frequently within a short distance
+     * 5. Disabling loop detection within code blocks to prevent false positives,
+     *    as repetitive code structures are common and not necessarily loops.
+     */
+    private checkContentLoop;
+    /**
+     * Truncates the content history to prevent unbounded memory growth.
+     * When truncating, adjusts all stored indices to maintain their relative positions.
+     */
+    private truncateAndUpdate;
+    /**
+     * Analyzes content in fixed-size chunks to detect repetitive patterns.
+     *
+     * Uses a sliding window approach:
+     * 1. Extract chunks of fixed size (CONTENT_CHUNK_SIZE)
+     * 2. Hash each chunk for efficient comparison
+     * 3. Track positions where identical chunks appear
+     * 4. Detect loops when chunks repeat frequently within a short distance
+     */
+    private analyzeContentChunksForLoop;
+    private hasMoreChunksToProcess;
+    /**
+     * Determines if a content chunk indicates a loop pattern.
+     *
+     * Loop detection logic:
+     * 1. Check if we've seen this hash before (new chunks are stored for future comparison)
+     * 2. Verify actual content matches to prevent hash collisions
+     * 3. Track all positions where this chunk appears
+     * 4. A loop is detected when the same chunk appears CONTENT_LOOP_THRESHOLD times
+     *    within a small average distance (≤ 5 * chunk size)
+     */
+    private isLoopDetectedForChunk;
+    /**
+     * Verifies that two chunks with the same hash actually contain identical content.
+     * This prevents false positives from hash collisions.
+     */
+    private isActualContentMatch;
+    private trimRecentHistory;
+    private checkForLoopWithLLM;
+    private queryLoopDetectionModel;
+    private handleConfirmedLoop;
+    private updateCheckInterval;
+    /**
+     * Resets all loop detection state.
+     */
+    reset(promptId: string): void;
+    private resetToolCallCount;
+    private resetContentTracking;
+    private resetLlmCheckTracking;
+}