feat: update Gemini session processing to support new file structure and formats

This commit is contained in:
Haileyesus
2026-03-30 16:38:15 +03:00
parent 6b4c435cd3
commit 49de006313
2 changed files with 65 additions and 8 deletions

View File

@@ -7,15 +7,54 @@ import { SessionData } from '@/shared/types/session.js';
export async function processGeminiSessionFile(file: string): Promise<SessionData | null> { export async function processGeminiSessionFile(file: string): Promise<SessionData | null> {
try { try {
// Gemini uses standard JSON (not JSONL), so we read the whole file at once
const fileContent = await fsp.readFile(file, 'utf8'); const fileContent = await fsp.readFile(file, 'utf8');
const data = JSON.parse(fileContent); const data = JSON.parse(fileContent);
if (data?.id && data?.projectPath) {
// Check for new format: data.sessionId
// Fallback for old format: data.id and data.projectPath
if (data?.sessionId || (data?.id && data?.projectPath)) {
let sessionId = data.sessionId || data.id;
let workspacePath = data.projectPath || '';
let sessionName = 'New Gemini Chat';
// Extract workspacePath for new format
if (data?.sessionId && file.includes(`${path.sep}chats${path.sep}`)) {
const chatsDir = path.dirname(file);
const workspaceDir = path.dirname(chatsDir);
const projectRootFile = path.join(workspaceDir, '.project_root');
try {
const rootContent = await fsp.readFile(projectRootFile, 'utf8');
if (rootContent) {
workspacePath = rootContent.trim();
}
} catch (e) {
// Ignore if .project_root doesn't exist
}
}
// Extract sessionName
if (data.messages && Array.isArray(data.messages) && data.messages.length > 0) {
const firstMessage = data.messages[0];
if (firstMessage?.content && Array.isArray(firstMessage.content) && firstMessage.content.length > 0) {
sessionName = firstMessage.content[0]?.text?.trim() || sessionName;
} else if (firstMessage?.content && typeof firstMessage.content === 'string') {
sessionName = firstMessage.content.trim() || sessionName;
}
} else if (data.messages?.[0]?.content) {
// old format fallback
sessionName = data.messages[0].content;
}
// Clean up sessionName
if (sessionName) {
sessionName = sessionName.replace(/\n/g, ' ').trim().substring(0, 100);
}
return { return {
sessionId: data.id, sessionId,
workspacePath: data.projectPath, workspacePath,
sessionName: data.messages?.[0]?.content || 'New Gemini Chat' sessionName
}; };
} }
} catch (e) { } catch (e) {
@@ -25,10 +64,24 @@ export async function processGeminiSessionFile(file: string): Promise<SessionDat
} }
export async function processGeminiSessions() { export async function processGeminiSessions() {
const geminiPath = path.join(os.homedir(), '.gemini', 'sessions'); const geminiHome = path.join(os.homedir(), '.gemini');
const files = await findFilesRecursivelyCreatedAfterLastScan(geminiPath, '.json');
// Process old sessions directory
const oldGeminiPath = path.join(geminiHome, 'sessions');
const oldFiles = await findFilesRecursivelyCreatedAfterLastScan(oldGeminiPath, '.json');
// Process new tmp/chats directories
const tmpGeminiPath = path.join(geminiHome, 'tmp');
const tmpFiles = await findFilesRecursivelyCreatedAfterLastScan(tmpGeminiPath, '.json');
const files = [...oldFiles, ...tmpFiles];
for (const file of files) { for (const file of files) {
// For tmp files, only process those inside a 'chats' directory
if (file.startsWith(tmpGeminiPath) && !file.includes(`${path.sep}chats${path.sep}`)) {
continue;
}
const result = await processGeminiSessionFile(file); const result = await processGeminiSessionFile(file);
if (result) { if (result) {
let createdAt: string | undefined; let createdAt: string | undefined;

View File

@@ -31,6 +31,10 @@ const PROVIDER_WATCH_PATHS: { provider: LLMProvider; rootPath: string }[] = [
provider: "gemini", provider: "gemini",
rootPath: path.join(os.homedir(), ".gemini", "sessions"), rootPath: path.join(os.homedir(), ".gemini", "sessions"),
}, },
{
provider: "gemini",
rootPath: path.join(os.homedir(), ".gemini", "tmp"),
},
]; ];
const WATCHER_IGNORED_PATTERNS = [ const WATCHER_IGNORED_PATTERNS = [