vrr/codeburn
2
0
Fork 0
mirror of https://github.com/AgentSeal/codeburn.git synced 2026-05-19 07:43:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fix(context-budget): use bounded readSessionFile helper

Browse source 
Config JSON, CLAUDE.md scans, and session-discovery reads now pass
through the 128 MB-capped helper. JSON.parse remains wrapped in
try/catch to preserve the previous 'null on malformed JSON' contract.
MEDIUM-1 coverage for the context-budget module.
This commit is contained in:
Ninym 2026-04-17 08:09:25 +02:00
parent 716e080cb3
commit 1bdbac4927
1 changed files with 21 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

39
src/context-budget.ts
View file

@ -1,8 +1,10 @@
import { readdir, readFile } from 'fs/promises'
import { readdir } from 'fs/promises'
import { existsSync } from 'fs'
import { join } from 'path'
import { homedir } from 'os'
import { readSessionFile } from './fs-utils.js'
const CHARS_PER_TOKEN = 4
const SYSTEM_BASE_TOKENS = 10400
const TOOL_TOKENS_OVERHEAD = 400
@ -23,9 +25,9 @@ function estimateTokens(text: string): number {
async function readConfigFile(path: string): Promise<Record<string, unknown> | null> {
if (!existsSync(path)) return null
try {
return JSON.parse(await readFile(path, 'utf-8'))
} catch { return null }
const raw = await readSessionFile(path)
if (raw === null) return null
try { return JSON.parse(raw) } catch { return null }
}
async function countMcpTools(projectPath?: string): Promise<number> {
@ -91,10 +93,9 @@ async function scanMemoryFiles(projectPath?: string): Promise<Array<{ name: stri
for (const { path, name } of paths) {
if (!existsSync(path)) continue
try {
const content = await readFile(path, 'utf-8')
files.push({ name, tokens: estimateTokens(content) })
} catch { continue }
const content = await readSessionFile(path)
if (content === null) continue
files.push({ name, tokens: estimateTokens(content) })
}
return files
@ -130,17 +131,19 @@ export async function estimateBudgetsByProject(projectPaths: Map<string, string>
}
export async function discoverProjectCwd(sessionDir: string): Promise<string | null> {
let files: string[]
try {
const files = (await readdir(sessionDir)).filter(f => f.endsWith('.jsonl'))
if (files.length === 0) return null
const content = await readFile(join(sessionDir, files[0]), 'utf-8')
for (const line of content.split('\n')) {
if (!line.trim()) continue
try {
const entry = JSON.parse(line)
if (entry.cwd && typeof entry.cwd === 'string') return entry.cwd
} catch { continue }
}
files = (await readdir(sessionDir)).filter(f => f.endsWith('.jsonl'))
} catch { return null }
if (files.length === 0) return null
const content = await readSessionFile(join(sessionDir, files[0]))
if (content === null) return null
for (const line of content.split('\n')) {
if (!line.trim()) continue
try {
const entry = JSON.parse(line)
if (entry.cwd && typeof entry.cwd === 'string') return entry.cwd
} catch { continue }
}
return null
}