Something went wrong. Try again.
source dump of claude code
Something went wrong. Try again.
4.2 kB · 142 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143import { LRUCache } from 'lru-cache'import { normalize } from 'path'
export type FileState = { content: string timestamp: number offset: number | undefined limit: number | undefined // True when this entry was populated by auto-injection (e.g. CLAUDE.md) and // the injected content did not match disk (stripped HTML comments, stripped // frontmatter, truncated MEMORY.md). The model has only seen a partial view; // Edit/Write must require an explicit Read first. `content` here holds the // RAW disk bytes (for getChangedFiles diffing), not what the model saw. isPartialView?: boolean}
// Default max entries for read file state cachesexport const READ_FILE_STATE_CACHE_SIZE = 100
// Default size limit for file state caches (25MB)// This prevents unbounded memory growth from large file contentsconst DEFAULT_MAX_CACHE_SIZE_BYTES = 25 * 1024 * 1024
/** * A file state cache that normalizes all path keys before access. * This ensures consistent cache hits regardless of whether callers pass * relative vs absolute paths with redundant segments (e.g. /foo/../bar) * or mixed path separators on Windows (/ vs \). */export class FileStateCache { private cache: LRUCache<string, FileState>
constructor(maxEntries: number, maxSizeBytes: number) { this.cache = new LRUCache<string, FileState>({ max: maxEntries, maxSize: maxSizeBytes, sizeCalculation: value => Math.max(1, Buffer.byteLength(value.content)), }) }
get(key: string): FileState | undefined { return this.cache.get(normalize(key)) }
set(key: string, value: FileState): this { this.cache.set(normalize(key), value) return this }
has(key: string): boolean { return this.cache.has(normalize(key)) }
delete(key: string): boolean { return this.cache.delete(normalize(key)) }
clear(): void { this.cache.clear() }
get size(): number { return this.cache.size }
get max(): number { return this.cache.max }
get maxSize(): number { return this.cache.maxSize }
get calculatedSize(): number { return this.cache.calculatedSize }
keys(): Generator<string> { return this.cache.keys() }
entries(): Generator<[string, FileState]> { return this.cache.entries() }
dump(): ReturnType<LRUCache<string, FileState>['dump']> { return this.cache.dump() }
load(entries: ReturnType<LRUCache<string, FileState>['dump']>): void { this.cache.load(entries) }}
/** * Factory function to create a size-limited FileStateCache. * Uses LRUCache's built-in size-based eviction to prevent memory bloat. * Note: Images are not cached (see FileReadTool) so size limit is mainly * for large text files, notebooks, and other editable content. */export function createFileStateCacheWithSizeLimit( maxEntries: number, maxSizeBytes: number = DEFAULT_MAX_CACHE_SIZE_BYTES,): FileStateCache { return new FileStateCache(maxEntries, maxSizeBytes)}
// Helper function to convert cache to object (used by compact.ts)export function cacheToObject( cache: FileStateCache,): Record<string, FileState> { return Object.fromEntries(cache.entries())}
// Helper function to get all keys from cache (used by several components)export function cacheKeys(cache: FileStateCache): string[] { return Array.from(cache.keys())}
// Helper function to clone a FileStateCache// Preserves size limit configuration from the source cacheexport function cloneFileStateCache(cache: FileStateCache): FileStateCache { const cloned = createFileStateCacheWithSizeLimit(cache.max, cache.maxSize) cloned.load(cache.dump()) return cloned}
// Merge two file state caches, with more recent entries (by timestamp) overriding older onesexport function mergeFileStateCaches( first: FileStateCache, second: FileStateCache,): FileStateCache { const merged = cloneFileStateCache(first) for (const [filePath, fileState] of second.entries()) { const existing = merged.get(filePath) // Only override if the new entry is more recent if (!existing || fileState.timestamp > existing.timestamp) { merged.set(filePath, fileState) } } return merged}