mirror of
https://github.com/AgentSeal/codeburn.git
synced 2026-05-19 16:13:56 +00:00
feat(cursor-agent): add provider for cursor-agent CLI sessions
Discovers transcripts at ~/.cursor/projects/*/agent-transcripts/*.txt and joins against ~/.cursor/ai-tracking/ai-code-tracking.db for model attribution. Token counts are estimated from transcript character length since the attribution DB does not carry them; the model label surfaces the estimation with an (est.) suffix on every row. Deduplication keys prefix cursor-agent: to stay disjoint from the existing cursor: prefix so the two providers do not cross-dedupe on shared conversationId namespaces. Tests cover: empty ~/.cursor/projects/, single transcript, multiple projects, missing ai-code-tracking.db, unrecognized transcript format skip, non-UUID filename fallback, and sqlite metadata join. Closes #55
This commit is contained in:
Matt Van Horn
parent
508edcd62b
commit
554036d2a7
4 changed files with 685 additions and 2 deletions
423
src/providers/cursor-agent.ts
Normal file
423
src/providers/cursor-agent.ts
Normal file
|
|
@ -0,0 +1,423 @@
|
|||
import { createHash } from 'crypto'
|
||||
import { existsSync } from 'fs'
|
||||
import { readdir, readFile, stat } from 'fs/promises'
|
||||
import { join, basename } from 'path'
|
||||
import { homedir } from 'os'
|
||||
|
||||
import { calculateCost } from '../models.js'
|
||||
import { openDatabase, type SqliteDatabase } from '../sqlite.js'
|
||||
import type {
|
||||
Provider,
|
||||
SessionSource,
|
||||
SessionParser,
|
||||
ParsedProviderCall,
|
||||
} from './types.js'
|
||||
|
||||
type ConversationSummary = {
|
||||
conversationId: string
|
||||
model: string | null
|
||||
title: string | null
|
||||
updatedAt: string | null
|
||||
}
|
||||
|
||||
type AssistantTurn = {
|
||||
body: string
|
||||
reasoning: string
|
||||
tools: string[]
|
||||
}
|
||||
|
||||
type ParsedTurn = {
|
||||
userMessage: string
|
||||
assistant: AssistantTurn
|
||||
}
|
||||
|
||||
const CURSOR_AGENT_DEFAULT_MODEL = 'claude-sonnet-4-5'
|
||||
const CHARS_PER_TOKEN = 4
|
||||
const MAX_USER_TEXT_LENGTH = 500
|
||||
const DIGITS_ONLY = /^\d+$/
|
||||
const UUID_LIKE = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i
|
||||
const USER_MARKER = /^\s*user:\s*/i
|
||||
const ASSISTANT_MARKER = /^\s*A:\s*/
|
||||
const THINKING_MARKER = /^\s*\[Thinking\]\s*/
|
||||
const TOOL_CALL_MARKER = /^\s*\[Tool call\]\s*(.+?)\s*$/i
|
||||
const TOOL_RESULT_MARKER = /^\s*\[Tool result\]\b/i
|
||||
const USER_QUERY_OPEN = '<user_query>'
|
||||
const USER_QUERY_CLOSE = '</user_query>'
|
||||
const CONVERSATION_SUMMARY_QUERY = `
|
||||
SELECT conversationId, model, title, updatedAt
|
||||
FROM conversation_summaries
|
||||
WHERE conversationId = ?
|
||||
`
|
||||
|
||||
const modelDisplayNames: Record<string, string> = {
|
||||
'claude-4.5-opus-high-thinking': 'Opus 4.5 (Thinking)',
|
||||
'claude-4-opus': 'Opus 4',
|
||||
'claude-4-sonnet-thinking': 'Sonnet 4 (Thinking)',
|
||||
'claude-4.5-sonnet-thinking': 'Sonnet 4.5 (Thinking)',
|
||||
'claude-4.6-sonnet': 'Sonnet 4.6',
|
||||
'composer-1': 'Composer 1',
|
||||
'grok-code-fast-1': 'Grok Code Fast',
|
||||
'gemini-3-pro': 'Gemini 3 Pro',
|
||||
'gpt-5.1-codex-high': 'GPT-5.1 Codex',
|
||||
'gpt-5': 'GPT-5',
|
||||
'gpt-4.1': 'GPT-4.1',
|
||||
default: 'Auto (Sonnet est.)',
|
||||
}
|
||||
|
||||
function getCursorAgentBaseDir(baseDirOverride?: string): string {
|
||||
if (baseDirOverride) return baseDirOverride
|
||||
// Windows paths unverified; tracked as Open Question 3 in issue #55.
|
||||
return join(homedir(), '.cursor')
|
||||
}
|
||||
|
||||
function getProjectsDir(baseDir: string): string {
|
||||
return join(baseDir, 'projects')
|
||||
}
|
||||
|
||||
function getAttributionDbPath(baseDir: string): string {
|
||||
return join(baseDir, 'ai-tracking', 'ai-code-tracking.db')
|
||||
}
|
||||
|
||||
function estimateTokens(charCount: number): number {
|
||||
if (charCount <= 0) return 0
|
||||
return Math.ceil(charCount / CHARS_PER_TOKEN)
|
||||
}
|
||||
|
||||
function parseToolName(raw: string): string {
|
||||
const clean = raw.trim()
|
||||
if (clean.length === 0) return 'unknown'
|
||||
return clean.toLowerCase().replace(/\s+/g, '-')
|
||||
}
|
||||
|
||||
function normalizeTimestamp(raw: string | number | null | undefined): string | null {
|
||||
if (raw === null || raw === undefined) return null
|
||||
if (typeof raw === 'string') {
|
||||
const trimmed = raw.trim()
|
||||
if (trimmed.length === 0) return null
|
||||
if (DIGITS_ONLY.test(trimmed)) {
|
||||
const num = Number(trimmed)
|
||||
if (!Number.isNaN(num)) {
|
||||
const ms = num < 1e12 ? num * 1000 : num
|
||||
return new Date(ms).toISOString()
|
||||
}
|
||||
}
|
||||
const parsed = new Date(trimmed)
|
||||
if (!Number.isNaN(parsed.getTime())) return parsed.toISOString()
|
||||
return null
|
||||
}
|
||||
|
||||
const ms = raw < 1e12 ? raw * 1000 : raw
|
||||
return new Date(ms).toISOString()
|
||||
}
|
||||
|
||||
function prettifyProjectId(raw: string): string {
|
||||
if (!raw) return raw
|
||||
|
||||
if (DIGITS_ONLY.test(raw)) {
|
||||
const num = Number(raw)
|
||||
if (!Number.isNaN(num) && raw.length >= 13) {
|
||||
const iso = new Date(num).toISOString()
|
||||
return `cursor-agent:${iso}`
|
||||
}
|
||||
}
|
||||
|
||||
const withoutPrefix = raw.replace(/^-Users-/, '')
|
||||
const parts = withoutPrefix.split('-').filter(Boolean)
|
||||
if (parts.length > 0) return parts[parts.length - 1]!
|
||||
|
||||
return raw
|
||||
}
|
||||
|
||||
function resolveModel(raw: string | null | undefined): string {
|
||||
if (!raw || raw === 'default') return CURSOR_AGENT_DEFAULT_MODEL
|
||||
return raw
|
||||
}
|
||||
|
||||
function toConversationId(transcriptPath: string): string {
|
||||
const filename = basename(transcriptPath, '.txt')
|
||||
if (filename.length === 36 && UUID_LIKE.test(filename)) return filename
|
||||
return createHash('sha1').update(transcriptPath).digest('hex').slice(0, 16)
|
||||
}
|
||||
|
||||
function extractUserQuery(userBlock: string): string {
|
||||
const chunks: string[] = []
|
||||
let cursor = 0
|
||||
|
||||
while (cursor < userBlock.length) {
|
||||
const openIndex = userBlock.indexOf(USER_QUERY_OPEN, cursor)
|
||||
if (openIndex === -1) break
|
||||
const start = openIndex + USER_QUERY_OPEN.length
|
||||
const closeIndex = userBlock.indexOf(USER_QUERY_CLOSE, start)
|
||||
if (closeIndex === -1) {
|
||||
chunks.push(userBlock.slice(start).trim())
|
||||
break
|
||||
}
|
||||
chunks.push(userBlock.slice(start, closeIndex).trim())
|
||||
cursor = closeIndex + USER_QUERY_CLOSE.length
|
||||
}
|
||||
|
||||
const combined = chunks.filter(Boolean).join(' ').replace(/\s+/g, ' ').trim()
|
||||
return combined.slice(0, MAX_USER_TEXT_LENGTH)
|
||||
}
|
||||
|
||||
function parseTranscript(raw: string): { turns: ParsedTurn[]; recognized: boolean } {
|
||||
const lines = raw.split(/\r?\n/)
|
||||
let recognized = false
|
||||
|
||||
const pendingUsers: string[] = []
|
||||
const turns: ParsedTurn[] = []
|
||||
|
||||
let active: 'none' | 'user' | 'assistant' = 'none'
|
||||
let userLines: string[] = []
|
||||
let assistantLines: string[] = []
|
||||
|
||||
const flushUser = () => {
|
||||
if (userLines.length === 0) return
|
||||
const userQuery = extractUserQuery(userLines.join('\n'))
|
||||
if (userQuery.length > 0) pendingUsers.push(userQuery)
|
||||
userLines = []
|
||||
}
|
||||
|
||||
const flushAssistant = () => {
|
||||
if (assistantLines.length === 0) return
|
||||
|
||||
let output = ''
|
||||
let reasoning = ''
|
||||
const toolsByTurn: Record<string, boolean> = Object.create(null)
|
||||
|
||||
for (const line of assistantLines) {
|
||||
if (TOOL_RESULT_MARKER.test(line)) continue
|
||||
|
||||
const thinkingMatch = line.match(THINKING_MARKER)
|
||||
if (thinkingMatch) {
|
||||
const body = line.replace(THINKING_MARKER, '').trim()
|
||||
if (body.length > 0) reasoning += `${body}\n`
|
||||
continue
|
||||
}
|
||||
|
||||
const toolMatch = line.match(TOOL_CALL_MARKER)
|
||||
if (toolMatch) {
|
||||
const parsedTool = parseToolName(toolMatch[1] ?? '')
|
||||
const toolKey = `cursor:${parsedTool}`
|
||||
toolsByTurn[toolKey] = true
|
||||
continue
|
||||
}
|
||||
|
||||
output += `${line}\n`
|
||||
}
|
||||
|
||||
if (pendingUsers.length > 0) {
|
||||
const userMessage = pendingUsers.shift()!
|
||||
const tools = Object.keys(toolsByTurn)
|
||||
turns.push({
|
||||
userMessage,
|
||||
assistant: {
|
||||
body: output.trim(),
|
||||
reasoning: reasoning.trim(),
|
||||
tools,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
assistantLines = []
|
||||
}
|
||||
|
||||
for (const line of lines) {
|
||||
if (USER_MARKER.test(line)) {
|
||||
recognized = true
|
||||
if (active === 'user') flushUser()
|
||||
if (active === 'assistant') flushAssistant()
|
||||
active = 'user'
|
||||
userLines = [line.replace(USER_MARKER, '')]
|
||||
continue
|
||||
}
|
||||
|
||||
if (ASSISTANT_MARKER.test(line)) {
|
||||
recognized = true
|
||||
if (active === 'user') flushUser()
|
||||
if (active === 'assistant') flushAssistant()
|
||||
active = 'assistant'
|
||||
assistantLines = [line.replace(ASSISTANT_MARKER, '')]
|
||||
continue
|
||||
}
|
||||
|
||||
if (active === 'user') {
|
||||
userLines.push(line)
|
||||
continue
|
||||
}
|
||||
|
||||
if (active === 'assistant') {
|
||||
assistantLines.push(line)
|
||||
}
|
||||
}
|
||||
|
||||
if (active === 'user') flushUser()
|
||||
if (active === 'assistant') flushAssistant()
|
||||
|
||||
return { turns, recognized }
|
||||
}
|
||||
|
||||
function createParser(
|
||||
source: SessionSource,
|
||||
seenKeys: Set<string>,
|
||||
dbPath: string,
|
||||
summariesByConversationId: Record<string, ConversationSummary | undefined>,
|
||||
): SessionParser {
|
||||
return {
|
||||
async *parse(): AsyncGenerator<ParsedProviderCall> {
|
||||
const conversationId = toConversationId(source.path)
|
||||
|
||||
let summary = summariesByConversationId[conversationId]
|
||||
let db: SqliteDatabase | null = null
|
||||
|
||||
try {
|
||||
if (!summary) {
|
||||
if (existsSync(dbPath)) {
|
||||
try {
|
||||
db = openDatabase(dbPath)
|
||||
const rows = db.query<{
|
||||
conversationId: string
|
||||
model: string | null
|
||||
title: string | null
|
||||
updatedAt: string | number | null
|
||||
}>(CONVERSATION_SUMMARY_QUERY, [conversationId])
|
||||
|
||||
if (rows.length > 0) {
|
||||
const row = rows[0]!
|
||||
summary = {
|
||||
conversationId: row.conversationId,
|
||||
model: row.model,
|
||||
title: row.title,
|
||||
updatedAt: normalizeTimestamp(row.updatedAt),
|
||||
}
|
||||
summariesByConversationId[conversationId] = summary
|
||||
}
|
||||
} catch {
|
||||
summary = undefined
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
const transcript = await readFile(source.path, 'utf-8')
|
||||
const parsed = parseTranscript(transcript)
|
||||
|
||||
if (!parsed.recognized) {
|
||||
process.stderr.write(`codeburn: skipped ${basename(source.path)}: unrecognized cursor-agent transcript format\n`)
|
||||
return
|
||||
}
|
||||
|
||||
let timestamp = summary?.updatedAt ?? null
|
||||
if (!timestamp) {
|
||||
const fileStat = await stat(source.path)
|
||||
timestamp = fileStat.mtime.toISOString()
|
||||
}
|
||||
|
||||
const model = resolveModel(summary?.model ?? null)
|
||||
|
||||
for (let turnIndex = 0; turnIndex < parsed.turns.length; turnIndex++) {
|
||||
const turn = parsed.turns[turnIndex]!
|
||||
const inputTokens = estimateTokens(turn.userMessage.length)
|
||||
const outputTokens = estimateTokens(turn.assistant.body.length)
|
||||
const reasoningTokens = estimateTokens(turn.assistant.reasoning.length)
|
||||
const deduplicationKey = `cursor-agent:${conversationId}:${turnIndex}`
|
||||
|
||||
if (seenKeys.has(deduplicationKey)) continue
|
||||
seenKeys.add(deduplicationKey)
|
||||
|
||||
const costUSD = calculateCost(
|
||||
model,
|
||||
inputTokens,
|
||||
outputTokens + reasoningTokens,
|
||||
0,
|
||||
0,
|
||||
0,
|
||||
)
|
||||
|
||||
yield {
|
||||
provider: 'cursor-agent',
|
||||
model,
|
||||
inputTokens,
|
||||
outputTokens,
|
||||
cacheCreationInputTokens: 0,
|
||||
cacheReadInputTokens: 0,
|
||||
cachedInputTokens: 0,
|
||||
reasoningTokens,
|
||||
webSearchRequests: 0,
|
||||
costUSD,
|
||||
tools: turn.assistant.tools,
|
||||
bashCommands: [],
|
||||
timestamp,
|
||||
speed: 'standard',
|
||||
deduplicationKey,
|
||||
userMessage: turn.userMessage,
|
||||
sessionId: conversationId,
|
||||
}
|
||||
}
|
||||
} finally {
|
||||
db?.close()
|
||||
}
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
export function createCursorAgentProvider(baseDirOverride?: string): Provider {
|
||||
const baseDir = getCursorAgentBaseDir(baseDirOverride)
|
||||
const projectsDir = getProjectsDir(baseDir)
|
||||
const dbPath = getAttributionDbPath(baseDir)
|
||||
const summariesByConversationId: Record<string, ConversationSummary | undefined> = Object.create(null)
|
||||
|
||||
return {
|
||||
name: 'cursor-agent',
|
||||
displayName: 'Cursor Agent',
|
||||
|
||||
modelDisplayName(model: string): string {
|
||||
const label = modelDisplayNames[model] ?? modelDisplayNames.default
|
||||
if (model === 'default') return label
|
||||
return label.endsWith('(est.)') ? label : `${label} (est.)`
|
||||
},
|
||||
|
||||
toolDisplayName(rawTool: string): string {
|
||||
return rawTool
|
||||
},
|
||||
|
||||
async discoverSessions(): Promise<SessionSource[]> {
|
||||
if (!existsSync(projectsDir)) return []
|
||||
|
||||
const projectEntries = await readdir(projectsDir, { withFileTypes: true })
|
||||
const sources: SessionSource[] = []
|
||||
|
||||
for (const entry of projectEntries) {
|
||||
if (!entry.isDirectory()) continue
|
||||
|
||||
const projectId = prettifyProjectId(entry.name)
|
||||
const transcriptDir = join(projectsDir, entry.name, 'agent-transcripts')
|
||||
if (!existsSync(transcriptDir)) continue
|
||||
|
||||
const transcriptEntries = await readdir(transcriptDir, { withFileTypes: true })
|
||||
for (const transcript of transcriptEntries) {
|
||||
if (!transcript.isFile()) continue
|
||||
if (!transcript.name.endsWith('.txt')) continue
|
||||
|
||||
const transcriptPath = join(transcriptDir, transcript.name)
|
||||
sources.push({
|
||||
path: transcriptPath,
|
||||
project: projectId,
|
||||
provider: 'cursor-agent',
|
||||
fingerprintPath: transcriptPath,
|
||||
cacheStrategy: 'full-reparse',
|
||||
progressLabel: `cursor-agent:${basename(transcript.name)}`,
|
||||
parserVersion: 'cursor-agent:v1',
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
return sources
|
||||
},
|
||||
|
||||
createSessionParser(source: SessionSource, seenKeys: Set<string>): SessionParser {
|
||||
return createParser(source, seenKeys, dbPath, summariesByConversationId)
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
export const cursor_agent = createCursorAgentProvider()
|
||||
|
|
@ -22,6 +22,9 @@ async function loadCursor(): Promise<Provider | null> {
|
|||
let opencodeProvider: Provider | null = null
|
||||
let opencodeLoadAttempted = false
|
||||
|
||||
let cursorAgentProvider: Provider | null = null
|
||||
let cursorAgentLoadAttempted = false
|
||||
|
||||
async function loadOpenCode(): Promise<Provider | null> {
|
||||
if (opencodeLoadAttempted) return opencodeProvider
|
||||
opencodeLoadAttempted = true
|
||||
|
|
@ -34,13 +37,26 @@ async function loadOpenCode(): Promise<Provider | null> {
|
|||
}
|
||||
}
|
||||
|
||||
async function loadCursorAgent(): Promise<Provider | null> {
|
||||
if (cursorAgentLoadAttempted) return cursorAgentProvider
|
||||
cursorAgentLoadAttempted = true
|
||||
try {
|
||||
const { cursor_agent } = await import('./cursor-agent.js')
|
||||
cursorAgentProvider = cursor_agent
|
||||
return cursor_agent
|
||||
} catch {
|
||||
return null
|
||||
}
|
||||
}
|
||||
|
||||
const coreProviders: Provider[] = [claude, codex, copilot, pi]
|
||||
|
||||
export async function getAllProviders(): Promise<Provider[]> {
|
||||
const [cursor, opencode] = await Promise.all([loadCursor(), loadOpenCode()])
|
||||
const [cursor, opencode, cursorAgent] = await Promise.all([loadCursor(), loadOpenCode(), loadCursorAgent()])
|
||||
const all = [...coreProviders]
|
||||
if (cursor) all.push(cursor)
|
||||
if (opencode) all.push(opencode)
|
||||
if (cursorAgent) all.push(cursorAgent)
|
||||
return all
|
||||
}
|
||||
|
||||
|
|
@ -68,5 +84,9 @@ export async function getProvider(name: string): Promise<Provider | undefined> {
|
|||
const oc = await loadOpenCode()
|
||||
return oc ?? undefined
|
||||
}
|
||||
if (name === 'cursor-agent') {
|
||||
const ca = await loadCursorAgent()
|
||||
return ca ?? undefined
|
||||
}
|
||||
return coreProviders.find(p => p.name === name)
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue