mirror of
https://github.com/siteboon/claudecodeui.git
synced 2026-05-01 18:28:38 +00:00
refactor: move search to module
This commit is contained in:
Haileyesus
@@ -2,6 +2,7 @@ import express, { type Request, type Response } from 'express';
|
||||
|
||||
import { providerAuthService } from '@/modules/providers/services/provider-auth.service.js';
|
||||
import { providerMcpService } from '@/modules/providers/services/mcp.service.js';
|
||||
import { sessionConversationsSearchService } from '@/modules/providers/services/session-conversations-search.service.js';
|
||||
import { sessionsService } from '@/modules/providers/services/sessions.service.js';
|
||||
import type { LLMProvider, McpScope, McpTransport, UpsertProviderMcpServerInput } from '@/shared/types.js';
|
||||
import { AppError, asyncHandler, createApiSuccessResponse } from '@/shared/utils.js';
|
||||
@@ -141,19 +142,19 @@ const parseMcpUpsertPayload = (payload: unknown): UpsertProviderMcpServerInput =
|
||||
args: Array.isArray(body.args) ? body.args.filter((entry): entry is string => typeof entry === 'string') : undefined,
|
||||
env: typeof body.env === 'object' && body.env !== null
|
||||
? Object.fromEntries(
|
||||
Object.entries(body.env as Record<string, unknown>).filter(
|
||||
(entry): entry is [string, string] => typeof entry[1] === 'string',
|
||||
),
|
||||
)
|
||||
Object.entries(body.env as Record<string, unknown>).filter(
|
||||
(entry): entry is [string, string] => typeof entry[1] === 'string',
|
||||
),
|
||||
)
|
||||
: undefined,
|
||||
cwd: readOptionalQueryString(body.cwd),
|
||||
url: readOptionalQueryString(body.url),
|
||||
headers: typeof body.headers === 'object' && body.headers !== null
|
||||
? Object.fromEntries(
|
||||
Object.entries(body.headers as Record<string, unknown>).filter(
|
||||
(entry): entry is [string, string] => typeof entry[1] === 'string',
|
||||
),
|
||||
)
|
||||
Object.entries(body.headers as Record<string, unknown>).filter(
|
||||
(entry): entry is [string, string] => typeof entry[1] === 'string',
|
||||
),
|
||||
)
|
||||
: undefined,
|
||||
envVars: Array.isArray(body.envVars)
|
||||
? body.envVars.filter((entry): entry is string => typeof entry === 'string')
|
||||
@@ -161,10 +162,10 @@ const parseMcpUpsertPayload = (payload: unknown): UpsertProviderMcpServerInput =
|
||||
bearerTokenEnvVar: readOptionalQueryString(body.bearerTokenEnvVar),
|
||||
envHttpHeaders: typeof body.envHttpHeaders === 'object' && body.envHttpHeaders !== null
|
||||
? Object.fromEntries(
|
||||
Object.entries(body.envHttpHeaders as Record<string, unknown>).filter(
|
||||
(entry): entry is [string, string] => typeof entry[1] === 'string',
|
||||
),
|
||||
)
|
||||
Object.entries(body.envHttpHeaders as Record<string, unknown>).filter(
|
||||
(entry): entry is [string, string] => typeof entry[1] === 'string',
|
||||
),
|
||||
)
|
||||
: undefined,
|
||||
};
|
||||
};
|
||||
@@ -208,6 +209,35 @@ const parseSessionRenameSummary = (payload: unknown): string => {
|
||||
return summary;
|
||||
};
|
||||
|
||||
const parseSessionSearchQuery = (value: unknown): string => {
|
||||
const query = readOptionalQueryString(value) ?? '';
|
||||
if (query.length < 2) {
|
||||
throw new AppError('Query must be at least 2 characters', {
|
||||
code: 'INVALID_SEARCH_QUERY',
|
||||
statusCode: 400,
|
||||
});
|
||||
}
|
||||
|
||||
return query;
|
||||
};
|
||||
|
||||
const parseSessionSearchLimit = (value: unknown): number => {
|
||||
const raw = readOptionalQueryString(value);
|
||||
if (!raw) {
|
||||
return 50;
|
||||
}
|
||||
|
||||
const parsed = Number.parseInt(raw, 10);
|
||||
if (Number.isNaN(parsed)) {
|
||||
throw new AppError('limit must be a valid integer.', {
|
||||
code: 'INVALID_QUERY_PARAMETER',
|
||||
statusCode: 400,
|
||||
});
|
||||
}
|
||||
|
||||
return Math.max(1, Math.min(parsed, 100));
|
||||
};
|
||||
|
||||
router.get(
|
||||
'/:provider/auth/status',
|
||||
asyncHandler(async (req: Request, res: Response) => {
|
||||
@@ -217,6 +247,7 @@ router.get(
|
||||
}),
|
||||
);
|
||||
|
||||
// ----------------- MCP routes -----------------
|
||||
router.get(
|
||||
'/:provider/mcp/servers',
|
||||
asyncHandler(async (req: Request, res: Response) => {
|
||||
@@ -279,6 +310,7 @@ router.post(
|
||||
}),
|
||||
);
|
||||
|
||||
// ----------------- Session routes -----------------
|
||||
router.delete(
|
||||
'/sessions/:sessionId',
|
||||
asyncHandler(async (req: Request, res: Response) => {
|
||||
@@ -331,4 +363,56 @@ router.get(
|
||||
}),
|
||||
);
|
||||
|
||||
router.get('/search/sessions', asyncHandler(async (req: Request, res: Response) => {
|
||||
const query = parseSessionSearchQuery(req.query.q);
|
||||
const limit = parseSessionSearchLimit(req.query.limit);
|
||||
|
||||
res.writeHead(200, {
|
||||
'Content-Type': 'text/event-stream',
|
||||
'Cache-Control': 'no-cache',
|
||||
Connection: 'keep-alive',
|
||||
'X-Accel-Buffering': 'no',
|
||||
});
|
||||
|
||||
let closed = false;
|
||||
const abortController = new AbortController();
|
||||
req.on('close', () => {
|
||||
closed = true;
|
||||
abortController.abort();
|
||||
});
|
||||
|
||||
try {
|
||||
await sessionConversationsSearchService.search({
|
||||
query,
|
||||
limit,
|
||||
signal: abortController.signal,
|
||||
onProgress: ({ projectResult, totalMatches, scannedProjects, totalProjects }) => {
|
||||
if (closed) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (projectResult) {
|
||||
res.write(`event: result\ndata: ${JSON.stringify({ projectResult, totalMatches, scannedProjects, totalProjects })}\n\n`);
|
||||
return;
|
||||
}
|
||||
|
||||
res.write(`event: progress\ndata: ${JSON.stringify({ totalMatches, scannedProjects, totalProjects })}\n\n`);
|
||||
},
|
||||
});
|
||||
|
||||
if (!closed) {
|
||||
res.write('event: done\ndata: {}\n\n');
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('Error searching conversations:', error);
|
||||
if (!closed) {
|
||||
res.write(`event: error\ndata: ${JSON.stringify({ error: 'Search failed' })}\n\n`);
|
||||
}
|
||||
} finally {
|
||||
if (!closed) {
|
||||
res.end();
|
||||
}
|
||||
}
|
||||
}));
|
||||
|
||||
export default router;
|
||||
|
||||
@@ -0,0 +1,918 @@
|
||||
import fsSync, { promises as fs } from 'node:fs';
|
||||
import os from 'node:os';
|
||||
import path from 'node:path';
|
||||
import readline from 'node:readline';
|
||||
|
||||
import { projectsDb } from '@/modules/database/index.js';
|
||||
import { generateDisplayName } from '@/modules/projects/index.js';
|
||||
import sessionManager from '@/sessionManager.js';
|
||||
|
||||
type AnyRecord = Record<string, any>;
|
||||
|
||||
type SearchSnippetHighlight = {
|
||||
start: number;
|
||||
end: number;
|
||||
};
|
||||
|
||||
type SessionConversationMatch = {
|
||||
role: string;
|
||||
snippet: string;
|
||||
highlights: SearchSnippetHighlight[];
|
||||
timestamp: string | null;
|
||||
provider: 'claude' | 'codex' | 'gemini';
|
||||
messageUuid?: string | null;
|
||||
};
|
||||
|
||||
type SessionConversationResult = {
|
||||
sessionId: string;
|
||||
provider: 'claude' | 'codex' | 'gemini';
|
||||
sessionSummary: string;
|
||||
matches: SessionConversationMatch[];
|
||||
};
|
||||
|
||||
type ProjectConversationResult = {
|
||||
projectId: string | null;
|
||||
projectName: string;
|
||||
projectDisplayName: string;
|
||||
sessions: SessionConversationResult[];
|
||||
};
|
||||
|
||||
export type SessionConversationSearchProgressUpdate = {
|
||||
projectResult: ProjectConversationResult | null;
|
||||
totalMatches: number;
|
||||
scannedProjects: number;
|
||||
totalProjects: number;
|
||||
};
|
||||
|
||||
type SearchSessionConversationsInput = {
|
||||
query: string;
|
||||
limit: number;
|
||||
signal?: AbortSignal;
|
||||
onProgress?: (update: SessionConversationSearchProgressUpdate) => void;
|
||||
};
|
||||
|
||||
const projectDirectoryCache = new Map<string, string>();
|
||||
|
||||
async function loadProjectConfig(): Promise<Record<string, AnyRecord>> {
|
||||
const configPath = path.join(os.homedir(), '.claude', 'project-config.json');
|
||||
try {
|
||||
const configData = await fs.readFile(configPath, 'utf8');
|
||||
return JSON.parse(configData) as Record<string, AnyRecord>;
|
||||
} catch {
|
||||
return {};
|
||||
}
|
||||
}
|
||||
|
||||
async function extractProjectDirectory(projectName: string): Promise<string> {
|
||||
if (projectDirectoryCache.has(projectName)) {
|
||||
return projectDirectoryCache.get(projectName) as string;
|
||||
}
|
||||
|
||||
const config = await loadProjectConfig();
|
||||
if (config[projectName]?.originalPath) {
|
||||
const originalPath = String(config[projectName].originalPath);
|
||||
projectDirectoryCache.set(projectName, originalPath);
|
||||
return originalPath;
|
||||
}
|
||||
|
||||
const projectDir = path.join(os.homedir(), '.claude', 'projects', projectName);
|
||||
const cwdCounts = new Map<string, number>();
|
||||
let latestTimestamp = 0;
|
||||
let latestCwd: string | null = null;
|
||||
let extractedPath: string;
|
||||
|
||||
try {
|
||||
await fs.access(projectDir);
|
||||
|
||||
const files = await fs.readdir(projectDir);
|
||||
const jsonlFiles = files.filter((file) => file.endsWith('.jsonl'));
|
||||
|
||||
if (jsonlFiles.length === 0) {
|
||||
extractedPath = projectName.replace(/-/g, '/');
|
||||
} else {
|
||||
for (const file of jsonlFiles) {
|
||||
const jsonlFile = path.join(projectDir, file);
|
||||
const fileStream = fsSync.createReadStream(jsonlFile);
|
||||
const rl = readline.createInterface({
|
||||
input: fileStream,
|
||||
crlfDelay: Infinity,
|
||||
});
|
||||
|
||||
for await (const line of rl) {
|
||||
if (!line.trim()) {
|
||||
continue;
|
||||
}
|
||||
|
||||
try {
|
||||
const entry = JSON.parse(line) as AnyRecord;
|
||||
if (!entry.cwd) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const cwd = String(entry.cwd);
|
||||
cwdCounts.set(cwd, (cwdCounts.get(cwd) || 0) + 1);
|
||||
|
||||
const timestamp = new Date(entry.timestamp || 0).getTime();
|
||||
if (timestamp > latestTimestamp) {
|
||||
latestTimestamp = timestamp;
|
||||
latestCwd = cwd;
|
||||
}
|
||||
} catch {
|
||||
// Skip malformed lines.
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (cwdCounts.size === 0) {
|
||||
extractedPath = projectName.replace(/-/g, '/');
|
||||
} else if (cwdCounts.size === 1) {
|
||||
extractedPath = Array.from(cwdCounts.keys())[0] as string;
|
||||
} else {
|
||||
const latestCount = latestCwd ? (cwdCounts.get(latestCwd) || 0) : 0;
|
||||
const maxCount = Math.max(...cwdCounts.values());
|
||||
|
||||
if (latestCount >= maxCount * 0.25 && latestCwd) {
|
||||
extractedPath = latestCwd;
|
||||
} else {
|
||||
let mostFrequentPath = '';
|
||||
for (const [cwd, count] of cwdCounts.entries()) {
|
||||
if (count === maxCount) {
|
||||
mostFrequentPath = cwd;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
extractedPath = mostFrequentPath || latestCwd || projectName.replace(/-/g, '/');
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
projectDirectoryCache.set(projectName, extractedPath);
|
||||
return extractedPath;
|
||||
} catch (error) {
|
||||
const code = (error as NodeJS.ErrnoException).code;
|
||||
if (code === 'ENOENT') {
|
||||
extractedPath = projectName.replace(/-/g, '/');
|
||||
} else {
|
||||
console.error(`Error extracting project directory for ${projectName}:`, error);
|
||||
extractedPath = projectName.replace(/-/g, '/');
|
||||
}
|
||||
|
||||
projectDirectoryCache.set(projectName, extractedPath);
|
||||
return extractedPath;
|
||||
}
|
||||
}
|
||||
|
||||
function normalizeComparablePath(inputPath: string): string {
|
||||
if (!inputPath || typeof inputPath !== 'string') {
|
||||
return '';
|
||||
}
|
||||
|
||||
const withoutLongPathPrefix = inputPath.startsWith('\\\\?\\')
|
||||
? inputPath.slice(4)
|
||||
: inputPath;
|
||||
const normalized = path.normalize(withoutLongPathPrefix.trim());
|
||||
if (!normalized) {
|
||||
return '';
|
||||
}
|
||||
|
||||
const resolved = path.resolve(normalized);
|
||||
return process.platform === 'win32' ? resolved.toLowerCase() : resolved;
|
||||
}
|
||||
|
||||
async function findCodexJsonlFiles(dir: string): Promise<string[]> {
|
||||
const files: string[] = [];
|
||||
|
||||
try {
|
||||
const entries = await fs.readdir(dir, { withFileTypes: true });
|
||||
for (const entry of entries) {
|
||||
const fullPath = path.join(dir, entry.name);
|
||||
if (entry.isDirectory()) {
|
||||
files.push(...await findCodexJsonlFiles(fullPath));
|
||||
} else if (entry.name.endsWith('.jsonl')) {
|
||||
files.push(fullPath);
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
// Skip directories we can't read.
|
||||
}
|
||||
|
||||
return files;
|
||||
}
|
||||
|
||||
async function searchCodexSessionsForProject(
|
||||
projectPath: string,
|
||||
projectResult: ProjectConversationResult,
|
||||
allWordsMatch: (textLower: string) => boolean,
|
||||
buildSnippet: (text: string, textLower: string) => { snippet: string; highlights: SearchSnippetHighlight[] },
|
||||
limit: number,
|
||||
getTotalMatches: () => number,
|
||||
addMatches: (count: number) => void,
|
||||
isAborted: () => boolean,
|
||||
): Promise<void> {
|
||||
const normalizedProjectPath = normalizeComparablePath(projectPath);
|
||||
if (!normalizedProjectPath) {
|
||||
return;
|
||||
}
|
||||
|
||||
const codexSessionsDir = path.join(os.homedir(), '.codex', 'sessions');
|
||||
try {
|
||||
await fs.access(codexSessionsDir);
|
||||
} catch {
|
||||
return;
|
||||
}
|
||||
|
||||
const jsonlFiles = await findCodexJsonlFiles(codexSessionsDir);
|
||||
|
||||
for (const filePath of jsonlFiles) {
|
||||
if (getTotalMatches() >= limit || isAborted()) {
|
||||
break;
|
||||
}
|
||||
|
||||
try {
|
||||
const fileStream = fsSync.createReadStream(filePath);
|
||||
const rl = readline.createInterface({ input: fileStream, crlfDelay: Infinity });
|
||||
|
||||
let sessionMeta: AnyRecord | null = null;
|
||||
for await (const line of rl) {
|
||||
if (!line.trim()) {
|
||||
continue;
|
||||
}
|
||||
|
||||
try {
|
||||
const entry = JSON.parse(line) as AnyRecord;
|
||||
if (entry.type === 'session_meta' && entry.payload) {
|
||||
sessionMeta = entry.payload as AnyRecord;
|
||||
break;
|
||||
}
|
||||
} catch {
|
||||
// Skip malformed lines.
|
||||
}
|
||||
}
|
||||
|
||||
if (!sessionMeta) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const sessionProjectPath = normalizeComparablePath(String(sessionMeta.cwd || ''));
|
||||
if (sessionProjectPath !== normalizedProjectPath) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const fileStream2 = fsSync.createReadStream(filePath);
|
||||
const rl2 = readline.createInterface({ input: fileStream2, crlfDelay: Infinity });
|
||||
let latestUserMessageText: string | null = null;
|
||||
const matches: SessionConversationMatch[] = [];
|
||||
|
||||
for await (const line of rl2) {
|
||||
if (getTotalMatches() >= limit || isAborted()) {
|
||||
break;
|
||||
}
|
||||
if (!line.trim()) {
|
||||
continue;
|
||||
}
|
||||
|
||||
let entry: AnyRecord;
|
||||
try {
|
||||
entry = JSON.parse(line) as AnyRecord;
|
||||
} catch {
|
||||
continue;
|
||||
}
|
||||
|
||||
let text: string | null = null;
|
||||
let role: string | null = null;
|
||||
|
||||
if (entry.type === 'event_msg' && entry.payload?.type === 'user_message' && entry.payload.message) {
|
||||
text = String(entry.payload.message);
|
||||
role = 'user';
|
||||
latestUserMessageText = text;
|
||||
} else if (entry.type === 'response_item' && entry.payload?.type === 'message') {
|
||||
const contentParts = Array.isArray(entry.payload.content) ? entry.payload.content : [];
|
||||
if (entry.payload.role === 'user') {
|
||||
text = contentParts
|
||||
.filter((part: AnyRecord) => part.type === 'input_text' && part.text)
|
||||
.map((part: AnyRecord) => String(part.text))
|
||||
.join(' ');
|
||||
role = 'user';
|
||||
if (text) {
|
||||
latestUserMessageText = text;
|
||||
}
|
||||
} else if (entry.payload.role === 'assistant') {
|
||||
text = contentParts
|
||||
.filter((part: AnyRecord) => part.type === 'output_text' && part.text)
|
||||
.map((part: AnyRecord) => String(part.text))
|
||||
.join(' ');
|
||||
role = 'assistant';
|
||||
}
|
||||
}
|
||||
|
||||
if (!text || !role) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const textLower = text.toLowerCase();
|
||||
if (!allWordsMatch(textLower)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (matches.length < 2) {
|
||||
const { snippet, highlights } = buildSnippet(text, textLower);
|
||||
matches.push({
|
||||
role,
|
||||
snippet,
|
||||
highlights,
|
||||
timestamp: entry.timestamp ? String(entry.timestamp) : null,
|
||||
provider: 'codex',
|
||||
});
|
||||
addMatches(1);
|
||||
}
|
||||
}
|
||||
|
||||
if (matches.length > 0) {
|
||||
projectResult.sessions.push({
|
||||
sessionId: String(sessionMeta.id || ''),
|
||||
provider: 'codex',
|
||||
sessionSummary: latestUserMessageText
|
||||
? (latestUserMessageText.length > 50 ? `${latestUserMessageText.substring(0, 50)}...` : latestUserMessageText)
|
||||
: 'Codex Session',
|
||||
matches,
|
||||
});
|
||||
}
|
||||
} catch {
|
||||
// Skip unreadable or malformed files.
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
async function searchGeminiSessionsForProject(
|
||||
projectPath: string,
|
||||
projectResult: ProjectConversationResult,
|
||||
allWordsMatch: (textLower: string) => boolean,
|
||||
buildSnippet: (text: string, textLower: string) => { snippet: string; highlights: SearchSnippetHighlight[] },
|
||||
limit: number,
|
||||
getTotalMatches: () => number,
|
||||
addMatches: (count: number) => void,
|
||||
): Promise<void> {
|
||||
for (const [sessionId, session] of sessionManager.sessions as Map<string, AnyRecord>) {
|
||||
if (getTotalMatches() >= limit) {
|
||||
break;
|
||||
}
|
||||
if (session.projectPath !== projectPath) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const matches: SessionConversationMatch[] = [];
|
||||
const sourceMessages = Array.isArray(session.messages) ? session.messages : [];
|
||||
|
||||
for (const msg of sourceMessages) {
|
||||
if (getTotalMatches() >= limit) {
|
||||
break;
|
||||
}
|
||||
if (msg.role !== 'user' && msg.role !== 'assistant') {
|
||||
continue;
|
||||
}
|
||||
|
||||
const text = typeof msg.content === 'string'
|
||||
? msg.content
|
||||
: Array.isArray(msg.content)
|
||||
? msg.content.filter((part: AnyRecord) => part.type === 'text').map((part: AnyRecord) => String(part.text)).join(' ')
|
||||
: '';
|
||||
if (!text) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const textLower = text.toLowerCase();
|
||||
if (!allWordsMatch(textLower)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (matches.length < 2) {
|
||||
const { snippet, highlights } = buildSnippet(text, textLower);
|
||||
matches.push({
|
||||
role: String(msg.role),
|
||||
snippet,
|
||||
highlights,
|
||||
timestamp: msg.timestamp ? new Date(msg.timestamp).toISOString() : null,
|
||||
provider: 'gemini',
|
||||
});
|
||||
addMatches(1);
|
||||
}
|
||||
}
|
||||
|
||||
if (matches.length > 0) {
|
||||
const firstUserMessage = sourceMessages.find((msg: AnyRecord) => msg.role === 'user');
|
||||
const summary = firstUserMessage?.content
|
||||
? (typeof firstUserMessage.content === 'string'
|
||||
? (firstUserMessage.content.length > 50 ? `${firstUserMessage.content.substring(0, 50)}...` : firstUserMessage.content)
|
||||
: 'Gemini Session')
|
||||
: 'Gemini Session';
|
||||
|
||||
projectResult.sessions.push({
|
||||
sessionId,
|
||||
provider: 'gemini',
|
||||
sessionSummary: summary,
|
||||
matches,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
const normalizedProjectPath = normalizeComparablePath(projectPath);
|
||||
if (!normalizedProjectPath) {
|
||||
return;
|
||||
}
|
||||
|
||||
const geminiTmpDir = path.join(os.homedir(), '.gemini', 'tmp');
|
||||
try {
|
||||
await fs.access(geminiTmpDir);
|
||||
} catch {
|
||||
return;
|
||||
}
|
||||
|
||||
const trackedSessionIds = new Set<string>();
|
||||
for (const [sid] of sessionManager.sessions as Map<string, AnyRecord>) {
|
||||
trackedSessionIds.add(String(sid));
|
||||
}
|
||||
|
||||
let projectDirs: string[];
|
||||
try {
|
||||
projectDirs = await fs.readdir(geminiTmpDir);
|
||||
} catch {
|
||||
return;
|
||||
}
|
||||
|
||||
for (const projectDir of projectDirs) {
|
||||
if (getTotalMatches() >= limit) {
|
||||
break;
|
||||
}
|
||||
|
||||
const projectRootFile = path.join(geminiTmpDir, projectDir, '.project_root');
|
||||
let projectRoot = '';
|
||||
try {
|
||||
projectRoot = (await fs.readFile(projectRootFile, 'utf8')).trim();
|
||||
} catch {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (normalizeComparablePath(projectRoot) !== normalizedProjectPath) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const chatsDir = path.join(geminiTmpDir, projectDir, 'chats');
|
||||
let chatFiles: string[];
|
||||
try {
|
||||
chatFiles = await fs.readdir(chatsDir);
|
||||
} catch {
|
||||
continue;
|
||||
}
|
||||
|
||||
for (const chatFile of chatFiles) {
|
||||
if (getTotalMatches() >= limit) {
|
||||
break;
|
||||
}
|
||||
if (!chatFile.endsWith('.json')) {
|
||||
continue;
|
||||
}
|
||||
|
||||
try {
|
||||
const filePath = path.join(chatsDir, chatFile);
|
||||
const data = await fs.readFile(filePath, 'utf8');
|
||||
const session = JSON.parse(data) as AnyRecord;
|
||||
if (!session.messages || !Array.isArray(session.messages)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const cliSessionId = String(session.sessionId || chatFile.replace('.json', ''));
|
||||
if (trackedSessionIds.has(cliSessionId)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const matches: SessionConversationMatch[] = [];
|
||||
let firstUserText: string | null = null;
|
||||
|
||||
for (const msg of session.messages as AnyRecord[]) {
|
||||
if (getTotalMatches() >= limit) {
|
||||
break;
|
||||
}
|
||||
|
||||
const role = msg.type === 'user'
|
||||
? 'user'
|
||||
: (msg.type === 'gemini' || msg.type === 'assistant')
|
||||
? 'assistant'
|
||||
: null;
|
||||
if (!role) {
|
||||
continue;
|
||||
}
|
||||
|
||||
let text = '';
|
||||
if (typeof msg.content === 'string') {
|
||||
text = msg.content;
|
||||
} else if (Array.isArray(msg.content)) {
|
||||
text = msg.content
|
||||
.filter((part: AnyRecord) => part.text)
|
||||
.map((part: AnyRecord) => String(part.text))
|
||||
.join(' ');
|
||||
}
|
||||
|
||||
if (!text) {
|
||||
continue;
|
||||
}
|
||||
if (role === 'user' && !firstUserText) {
|
||||
firstUserText = text;
|
||||
}
|
||||
|
||||
const textLower = text.toLowerCase();
|
||||
if (!allWordsMatch(textLower)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (matches.length < 2) {
|
||||
const { snippet, highlights } = buildSnippet(text, textLower);
|
||||
matches.push({
|
||||
role,
|
||||
snippet,
|
||||
highlights,
|
||||
timestamp: msg.timestamp ? String(msg.timestamp) : null,
|
||||
provider: 'gemini',
|
||||
});
|
||||
addMatches(1);
|
||||
}
|
||||
}
|
||||
|
||||
if (matches.length > 0) {
|
||||
const summary = firstUserText
|
||||
? (firstUserText.length > 50 ? `${firstUserText.substring(0, 50)}...` : firstUserText)
|
||||
: 'Gemini CLI Session';
|
||||
|
||||
projectResult.sessions.push({
|
||||
sessionId: cliSessionId,
|
||||
provider: 'gemini',
|
||||
sessionSummary: summary,
|
||||
matches,
|
||||
});
|
||||
}
|
||||
} catch {
|
||||
// Skip unreadable or malformed files.
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
export async function searchConversations(
|
||||
query: string,
|
||||
limit = 50,
|
||||
onProjectResult: ((update: SessionConversationSearchProgressUpdate) => void) | null = null,
|
||||
signal: AbortSignal | null = null,
|
||||
): Promise<{ results: ProjectConversationResult[]; totalMatches: number; query: string }> {
|
||||
const safeQuery = typeof query === 'string' ? query.trim() : '';
|
||||
const safeLimit = Math.max(1, Math.min(Number.isFinite(limit) ? limit : 50, 200));
|
||||
const claudeDir = path.join(os.homedir(), '.claude', 'projects');
|
||||
const config = await loadProjectConfig();
|
||||
const results: ProjectConversationResult[] = [];
|
||||
let totalMatches = 0;
|
||||
const words = safeQuery.toLowerCase().split(/\s+/).filter((word) => word.length > 0);
|
||||
if (words.length === 0) {
|
||||
return { results: [], totalMatches: 0, query: safeQuery };
|
||||
}
|
||||
|
||||
const isAborted = () => signal?.aborted === true;
|
||||
|
||||
const isSystemMessage = (textContent: string): boolean => {
|
||||
return typeof textContent === 'string' && (
|
||||
textContent.startsWith('<command-name>') ||
|
||||
textContent.startsWith('<command-message>') ||
|
||||
textContent.startsWith('<command-args>') ||
|
||||
textContent.startsWith('<local-command-stdout>') ||
|
||||
textContent.startsWith('<system-reminder>') ||
|
||||
textContent.startsWith('Caveat:') ||
|
||||
textContent.startsWith('This session is being continued from a previous') ||
|
||||
textContent.startsWith('Invalid API key') ||
|
||||
textContent.includes('{"subtasks":') ||
|
||||
textContent.includes('CRITICAL: You MUST respond with ONLY a JSON') ||
|
||||
textContent === 'Warmup'
|
||||
);
|
||||
};
|
||||
|
||||
const extractText = (content: unknown): string => {
|
||||
if (typeof content === 'string') {
|
||||
return content;
|
||||
}
|
||||
if (Array.isArray(content)) {
|
||||
return content
|
||||
.filter((part: AnyRecord) => part.type === 'text' && part.text)
|
||||
.map((part: AnyRecord) => String(part.text))
|
||||
.join(' ');
|
||||
}
|
||||
return '';
|
||||
};
|
||||
|
||||
const escapeRegex = (value: string): string => value.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
|
||||
const wordPatterns = words.map((word) => new RegExp(`(?<!\\p{L})${escapeRegex(word)}(?!\\p{L})`, 'u'));
|
||||
const allWordsMatch = (textLower: string): boolean => wordPatterns.every((pattern) => pattern.test(textLower));
|
||||
|
||||
const buildSnippet = (
|
||||
text: string,
|
||||
textLower: string,
|
||||
snippetLen = 150,
|
||||
): { snippet: string; highlights: SearchSnippetHighlight[] } => {
|
||||
let firstIndex = -1;
|
||||
let firstWordLen = 0;
|
||||
for (const word of words) {
|
||||
const regex = new RegExp(`(?<!\\p{L})${escapeRegex(word)}(?!\\p{L})`, 'u');
|
||||
const match = regex.exec(textLower);
|
||||
if (match && (firstIndex === -1 || match.index < firstIndex)) {
|
||||
firstIndex = match.index;
|
||||
firstWordLen = word.length;
|
||||
}
|
||||
}
|
||||
|
||||
if (firstIndex === -1) {
|
||||
firstIndex = 0;
|
||||
}
|
||||
|
||||
const halfLen = Math.floor(snippetLen / 2);
|
||||
const start = Math.max(0, firstIndex - halfLen);
|
||||
const end = Math.min(text.length, firstIndex + halfLen + firstWordLen);
|
||||
const prefix = start > 0 ? '...' : '';
|
||||
const suffix = end < text.length ? '...' : '';
|
||||
const snippet = `${prefix}${text.slice(start, end).replace(/\n/g, ' ')}${suffix}`;
|
||||
|
||||
const snippetLower = snippet.toLowerCase();
|
||||
const highlights: SearchSnippetHighlight[] = [];
|
||||
for (const word of words) {
|
||||
const regex = new RegExp(`(?<!\\p{L})${escapeRegex(word)}(?!\\p{L})`, 'gu');
|
||||
let match: RegExpExecArray | null;
|
||||
match = regex.exec(snippetLower);
|
||||
while (match !== null) {
|
||||
highlights.push({ start: match.index, end: match.index + word.length });
|
||||
match = regex.exec(snippetLower);
|
||||
}
|
||||
}
|
||||
|
||||
highlights.sort((left, right) => left.start - right.start);
|
||||
const merged: SearchSnippetHighlight[] = [];
|
||||
for (const highlight of highlights) {
|
||||
const previous = merged[merged.length - 1];
|
||||
if (previous && highlight.start <= previous.end) {
|
||||
previous.end = Math.max(previous.end, highlight.end);
|
||||
} else {
|
||||
merged.push({ ...highlight });
|
||||
}
|
||||
}
|
||||
|
||||
return { snippet, highlights: merged };
|
||||
};
|
||||
|
||||
try {
|
||||
await fs.access(claudeDir);
|
||||
const entries = await fs.readdir(claudeDir, { withFileTypes: true });
|
||||
const projectDirs = entries.filter((entry) => entry.isDirectory());
|
||||
let scannedProjects = 0;
|
||||
const totalProjects = projectDirs.length;
|
||||
|
||||
for (const projectEntry of projectDirs) {
|
||||
if (totalMatches >= safeLimit || isAborted()) {
|
||||
break;
|
||||
}
|
||||
|
||||
const projectName = projectEntry.name;
|
||||
const projectDir = path.join(claudeDir, projectName);
|
||||
const projectDisplayName = config[projectName]?.displayName
|
||||
? String(config[projectName].displayName)
|
||||
: await generateDisplayName(projectName);
|
||||
|
||||
let files: string[];
|
||||
try {
|
||||
files = await fs.readdir(projectDir);
|
||||
} catch {
|
||||
continue;
|
||||
}
|
||||
|
||||
const jsonlFiles = files.filter(
|
||||
(file) => file.endsWith('.jsonl') && !file.startsWith('agent-'),
|
||||
);
|
||||
|
||||
let searchProjectId: string | null = null;
|
||||
try {
|
||||
const resolvedPath = await extractProjectDirectory(projectName);
|
||||
const dbRow = projectsDb.getProjectPath(resolvedPath);
|
||||
if (dbRow?.project_id) {
|
||||
searchProjectId = String(dbRow.project_id);
|
||||
}
|
||||
} catch {
|
||||
// Best-effort project id resolution.
|
||||
}
|
||||
|
||||
const projectResult: ProjectConversationResult = {
|
||||
projectId: searchProjectId,
|
||||
projectName,
|
||||
projectDisplayName,
|
||||
sessions: [],
|
||||
};
|
||||
|
||||
for (const file of jsonlFiles) {
|
||||
if (totalMatches >= safeLimit || isAborted()) {
|
||||
break;
|
||||
}
|
||||
|
||||
const filePath = path.join(projectDir, file);
|
||||
const sessionMatches = new Map<string, SessionConversationMatch[]>();
|
||||
const sessionSummaries = new Map<string, string>();
|
||||
const pendingSummaries = new Map<string, string>();
|
||||
const sessionLastMessages = new Map<string, { user?: string; assistant?: string }>();
|
||||
let currentSessionId: string | null = null;
|
||||
|
||||
try {
|
||||
const fileStream = fsSync.createReadStream(filePath);
|
||||
const rl = readline.createInterface({
|
||||
input: fileStream,
|
||||
crlfDelay: Infinity,
|
||||
});
|
||||
|
||||
for await (const line of rl) {
|
||||
if (totalMatches >= safeLimit || isAborted()) {
|
||||
break;
|
||||
}
|
||||
if (!line.trim()) {
|
||||
continue;
|
||||
}
|
||||
|
||||
let entry: AnyRecord;
|
||||
try {
|
||||
entry = JSON.parse(line) as AnyRecord;
|
||||
} catch {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (entry.sessionId) {
|
||||
currentSessionId = String(entry.sessionId);
|
||||
}
|
||||
|
||||
if (entry.type === 'summary' && entry.summary) {
|
||||
const summary = String(entry.summary);
|
||||
const sid = entry.sessionId
|
||||
? String(entry.sessionId)
|
||||
: currentSessionId;
|
||||
if (sid) {
|
||||
sessionSummaries.set(sid, summary);
|
||||
} else if (entry.leafUuid) {
|
||||
pendingSummaries.set(String(entry.leafUuid), summary);
|
||||
}
|
||||
}
|
||||
|
||||
if (entry.parentUuid && currentSessionId && !sessionSummaries.has(currentSessionId)) {
|
||||
const pendingSummary = pendingSummaries.get(String(entry.parentUuid));
|
||||
if (pendingSummary) {
|
||||
sessionSummaries.set(currentSessionId, pendingSummary);
|
||||
}
|
||||
}
|
||||
|
||||
if (entry.message?.content && currentSessionId && !entry.isApiErrorMessage) {
|
||||
const role = entry.message.role;
|
||||
if (role === 'user' || role === 'assistant') {
|
||||
const text = extractText(entry.message.content);
|
||||
if (text && !isSystemMessage(text)) {
|
||||
if (!sessionLastMessages.has(currentSessionId)) {
|
||||
sessionLastMessages.set(currentSessionId, {});
|
||||
}
|
||||
|
||||
const messages = sessionLastMessages.get(currentSessionId) as {
|
||||
user?: string;
|
||||
assistant?: string;
|
||||
};
|
||||
if (role === 'user') {
|
||||
messages.user = text;
|
||||
} else {
|
||||
messages.assistant = text;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!entry.message?.content) {
|
||||
continue;
|
||||
}
|
||||
if (entry.message.role !== 'user' && entry.message.role !== 'assistant') {
|
||||
continue;
|
||||
}
|
||||
if (entry.isApiErrorMessage) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const text = extractText(entry.message.content);
|
||||
if (!text || isSystemMessage(text)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const textLower = text.toLowerCase();
|
||||
if (!allWordsMatch(textLower)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const resolvedSessionId = entry.sessionId
|
||||
? String(entry.sessionId)
|
||||
: currentSessionId || file.replace('.jsonl', '');
|
||||
if (!sessionMatches.has(resolvedSessionId)) {
|
||||
sessionMatches.set(resolvedSessionId, []);
|
||||
}
|
||||
|
||||
const matches = sessionMatches.get(resolvedSessionId) as SessionConversationMatch[];
|
||||
if (matches.length < 2) {
|
||||
const { snippet, highlights } = buildSnippet(text, textLower);
|
||||
matches.push({
|
||||
role: String(entry.message.role),
|
||||
snippet,
|
||||
highlights,
|
||||
timestamp: entry.timestamp ? String(entry.timestamp) : null,
|
||||
provider: 'claude',
|
||||
messageUuid: entry.uuid ? String(entry.uuid) : null,
|
||||
});
|
||||
totalMatches += 1;
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
// Skip unreadable or malformed files.
|
||||
}
|
||||
|
||||
for (const [sessionId, matches] of sessionMatches.entries()) {
|
||||
const lastMessages = sessionLastMessages.get(sessionId);
|
||||
const fallback = lastMessages?.user || lastMessages?.assistant;
|
||||
projectResult.sessions.push({
|
||||
sessionId,
|
||||
provider: 'claude',
|
||||
sessionSummary: sessionSummaries.get(sessionId)
|
||||
|| (fallback ? (fallback.length > 50 ? `${fallback.substring(0, 50)}...` : fallback) : 'New Session'),
|
||||
matches,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
const actualProjectDir = await extractProjectDirectory(projectName);
|
||||
if (actualProjectDir && !isAborted() && totalMatches < safeLimit) {
|
||||
await searchCodexSessionsForProject(
|
||||
actualProjectDir,
|
||||
projectResult,
|
||||
allWordsMatch,
|
||||
buildSnippet,
|
||||
safeLimit,
|
||||
() => totalMatches,
|
||||
(count) => { totalMatches += count; },
|
||||
isAborted,
|
||||
);
|
||||
}
|
||||
} catch {
|
||||
// Skip codex search errors.
|
||||
}
|
||||
|
||||
try {
|
||||
const actualProjectDir = await extractProjectDirectory(projectName);
|
||||
if (actualProjectDir && !isAborted() && totalMatches < safeLimit) {
|
||||
await searchGeminiSessionsForProject(
|
||||
actualProjectDir,
|
||||
projectResult,
|
||||
allWordsMatch,
|
||||
buildSnippet,
|
||||
safeLimit,
|
||||
() => totalMatches,
|
||||
(count) => { totalMatches += count; },
|
||||
);
|
||||
}
|
||||
} catch {
|
||||
// Skip gemini search errors.
|
||||
}
|
||||
|
||||
scannedProjects += 1;
|
||||
if (projectResult.sessions.length > 0) {
|
||||
results.push(projectResult);
|
||||
onProjectResult?.({ projectResult, totalMatches, scannedProjects, totalProjects });
|
||||
} else if (onProjectResult && scannedProjects % 10 === 0) {
|
||||
onProjectResult({ projectResult: null, totalMatches, scannedProjects, totalProjects });
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
// ~/.claude/projects does not exist.
|
||||
}
|
||||
|
||||
return { results, totalMatches, query: safeQuery };
|
||||
}
|
||||
|
||||
/**
|
||||
* Application service for session-conversation search.
|
||||
*
|
||||
* Provider routes call this service so route handlers stay focused on
|
||||
* request parsing/response formatting, while search execution remains
|
||||
* centralized in one place.
|
||||
*/
|
||||
export const sessionConversationsSearchService = {
|
||||
/**
|
||||
* Streams progress updates while the search scans provider session logs.
|
||||
*/
|
||||
async search(input: SearchSessionConversationsInput): Promise<void> {
|
||||
await searchConversations(
|
||||
input.query,
|
||||
input.limit,
|
||||
input.onProgress ?? null,
|
||||
input.signal ?? null,
|
||||
);
|
||||
},
|
||||
};
|
||||
Reference in New Issue
Block a user