summaryrefslogtreecommitdiff
path: root/packages/core/src/services/gitService.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/core/src/services/gitService.ts')
-rw-r--r--packages/core/src/services/gitService.ts80
1 files changed, 31 insertions, 49 deletions
diff --git a/packages/core/src/services/gitService.ts b/packages/core/src/services/gitService.ts
index 8cd6b887..38c1c378 100644
--- a/packages/core/src/services/gitService.ts
+++ b/packages/core/src/services/gitService.ts
@@ -6,13 +6,13 @@
import * as fs from 'fs/promises';
import * as path from 'path';
+import * as os from 'os';
+import * as crypto from 'crypto';
import { isNodeError } from '../utils/errors.js';
import { isGitRepository } from '../utils/gitUtils.js';
import { exec } from 'node:child_process';
import { simpleGit, SimpleGit, CheckRepoActions } from 'simple-git';
-export const historyDirName = '.gemini_cli_history';
-
export class GitService {
private projectRoot: string;
@@ -20,6 +20,14 @@ export class GitService {
this.projectRoot = path.resolve(projectRoot);
}
+ private getHistoryDir(): string {
+ const hash = crypto
+ .createHash('sha256')
+ .update(this.projectRoot)
+ .digest('hex');
+ return path.join(os.homedir(), '.gemini', 'history', hash);
+ }
+
async initialize(): Promise<void> {
if (!isGitRepository(this.projectRoot)) {
throw new Error('GitService requires a Git repository');
@@ -28,7 +36,7 @@ export class GitService {
if (!gitAvailable) {
throw new Error('GitService requires Git to be installed');
}
- this.setupHiddenGitRepository();
+ this.setupShadowGitRepository();
}
verifyGitAvailability(): Promise<boolean> {
@@ -47,66 +55,40 @@ export class GitService {
* Creates a hidden git repository in the project root.
* The Git repository is used to support checkpointing.
*/
- async setupHiddenGitRepository() {
- const historyDir = path.join(this.projectRoot, historyDirName);
- const repoDir = path.join(historyDir, 'repository');
+ async setupShadowGitRepository() {
+ const repoDir = this.getHistoryDir();
await fs.mkdir(repoDir, { recursive: true });
- const repoInstance: SimpleGit = simpleGit(repoDir);
- const isRepoDefined = await repoInstance.checkIsRepo(
+ const isRepoDefined = await simpleGit(repoDir).checkIsRepo(
CheckRepoActions.IS_REPO_ROOT,
);
+
if (!isRepoDefined) {
- await repoInstance.init();
- try {
- await repoInstance.raw([
- 'worktree',
- 'add',
- this.projectRoot,
- '--force',
- ]);
- } catch (error) {
- console.log('Failed to add worktree:', error);
- }
+ await simpleGit(repoDir).init(false, {
+ '--initial-branch': 'main',
+ });
+
+ const repo = simpleGit(repoDir);
+ await repo.commit('Initial commit', { '--allow-empty': null });
}
- const visibileGitIgnorePath = path.join(this.projectRoot, '.gitignore');
- const hiddenGitIgnorePath = path.join(repoDir, '.gitignore');
+ const userGitIgnorePath = path.join(this.projectRoot, '.gitignore');
+ const shadowGitIgnorePath = path.join(repoDir, '.gitignore');
- let visibileGitIgnoreContent = ``;
+ let userGitIgnoreContent = '';
try {
- visibileGitIgnoreContent = await fs.readFile(
- visibileGitIgnorePath,
- 'utf-8',
- );
+ userGitIgnoreContent = await fs.readFile(userGitIgnorePath, 'utf-8');
} catch (error) {
if (isNodeError(error) && error.code !== 'ENOENT') {
throw error;
}
}
- await fs.writeFile(hiddenGitIgnorePath, visibileGitIgnoreContent);
-
- if (!visibileGitIgnoreContent.includes(historyDirName)) {
- const updatedContent = `${visibileGitIgnoreContent}\n# Gemini CLI history directory\n${historyDirName}\n`;
- await fs.writeFile(visibileGitIgnorePath, updatedContent);
- }
-
- const commit = await repoInstance.raw([
- 'rev-list',
- '--all',
- '--max-count=1',
- ]);
- if (!commit) {
- await repoInstance.add(hiddenGitIgnorePath);
-
- await repoInstance.commit('Initial commit');
- }
+ await fs.writeFile(shadowGitIgnorePath, userGitIgnoreContent);
}
- private get hiddenGitRepository(): SimpleGit {
- const historyDir = path.join(this.projectRoot, historyDirName);
- const repoDir = path.join(historyDir, 'repository');
+ private get shadowGitRepository(): SimpleGit {
+ const repoDir = this.getHistoryDir();
return simpleGit(this.projectRoot).env({
GIT_DIR: path.join(repoDir, '.git'),
GIT_WORK_TREE: this.projectRoot,
@@ -114,19 +96,19 @@ export class GitService {
}
async getCurrentCommitHash(): Promise<string> {
- const hash = await this.hiddenGitRepository.raw('rev-parse', 'HEAD');
+ const hash = await this.shadowGitRepository.raw('rev-parse', 'HEAD');
return hash.trim();
}
async createFileSnapshot(message: string): Promise<string> {
- const repo = this.hiddenGitRepository;
+ const repo = this.shadowGitRepository;
await repo.add('.');
const commitResult = await repo.commit(message);
return commitResult.commit;
}
async restoreProjectFromSnapshot(commitHash: string): Promise<void> {
- const repo = this.hiddenGitRepository;
+ const repo = this.shadowGitRepository;
await repo.raw(['restore', '--source', commitHash, '.']);
}
}