@vinaes/succ 1.3.19

package/dist/lib/chunker.js

@@ -0,0 +1,358 @@
+import { getConfig } from './config.js';
+/** Chunking stats — tracks how many files used AST vs regex chunking */
+let chunkingStats = { astFiles: 0, regexFiles: 0 };
+export function getChunkingStats() {
+    return { ...chunkingStats };
+}
+export function resetChunkingStats() {
+    chunkingStats = { astFiles: 0, regexFiles: 0 };
+}
+/**
+ * Count net brace depth change ({=+1, }=-1) ignoring braces inside strings and comments
+ */
+function countBracesOutsideStrings(line) {
+    let depth = 0;
+    let inString = null; // Track quote type: ', ", or `
+    let inLineComment = false;
+    let escaped = false; // Track escape state for proper \\ and \" handling
+    for (let i = 0; i < line.length; i++) {
+        const char = line[i];
+        const nextChar = i < line.length - 1 ? line[i + 1] : '';
+        // If previous char was escape, this char is escaped - skip it
+        if (escaped) {
+            escaped = false;
+            continue;
+        }
+        // Check for escape character inside strings
+        if (inString && char === '\\') {
+            escaped = true;
+            continue;
+        }
+        // Check for line comments (// or #)
+        if (!inString && !inLineComment) {
+            if (char === '/' && nextChar === '/') {
+                inLineComment = true;
+                continue;
+            }
+            if (char === '#') {
+                // Python-style comments
+                inLineComment = true;
+                continue;
+            }
+        }
+        if (inLineComment)
+            continue;
+        // Handle string boundaries
+        if (!inString && (char === '"' || char === "'" || char === '`')) {
+            inString = char;
+            continue;
+        }
+        if (inString && char === inString) {
+            inString = null;
+            continue;
+        }
+        // Count braces only outside strings
+        if (!inString) {
+            if (char === '{')
+                depth++;
+            if (char === '}')
+                depth--;
+        }
+    }
+    return depth;
+}
+/**
+ * Build an enriched text for embedding by prepending symbol metadata.
+ * The original content is stored as-is in the DB; this enriched version
+ * is only used for embedding generation to improve semantic search quality.
+ */
+export function enrichForEmbedding(chunk) {
+    if (!chunk.symbolName && !chunk.signature)
+        return chunk.content;
+    const type = chunk.symbolType || 'symbol';
+    const sig = chunk.signature ? `(${chunk.signature})` : '';
+    return `[${type}: ${chunk.symbolName}${sig}]\n${chunk.content}`;
+}
+/**
+ * Split text into overlapping chunks
+ */
+export function chunkText(text, _filePath) {
+    const config = getConfig();
+    const { chunk_size, chunk_overlap } = config;
+    const lines = text.split('\n');
+    const chunks = [];
+    let currentChunk = [];
+    let currentSize = 0;
+    let startLine = 0;
+    for (let i = 0; i < lines.length; i++) {
+        const line = lines[i];
+        const lineSize = line.length + 1; // +1 for newline
+        // If adding this line exceeds chunk size, save current chunk
+        if (currentSize + lineSize > chunk_size && currentChunk.length > 0) {
+            chunks.push({
+                content: currentChunk.join('\n'),
+                startLine: startLine + 1, // 1-indexed
+                endLine: i, // Previous line
+            });
+            // Keep overlap lines
+            const overlapLines = [];
+            let overlapSize = 0;
+            const previousChunkStartLine = startLine;
+            const previousChunkLineCount = currentChunk.length;
+            for (let j = currentChunk.length - 1; j >= 0 && overlapSize < chunk_overlap; j--) {
+                overlapLines.unshift(currentChunk[j]);
+                overlapSize += currentChunk[j].length + 1;
+            }
+            currentChunk = overlapLines;
+            currentSize = overlapSize;
+            // The overlap lines start from the end of the previous chunk
+            startLine = previousChunkStartLine + previousChunkLineCount - overlapLines.length;
+        }
+        currentChunk.push(line);
+        currentSize += lineSize;
+    }
+    // Save remaining chunk
+    if (currentChunk.length > 0) {
+        chunks.push({
+            content: currentChunk.join('\n'),
+            startLine: startLine + 1,
+            endLine: lines.length,
+        });
+    }
+    return chunks;
+}
+// Max chunk size in characters (for embedding models with context limits)
+// 2048 tokens ~= 4000 chars for BERT-based models (conservative: ~2 chars/token)
+// This ensures compatibility with models like nomic-embed-text (2048 ctx)
+const MAX_CHUNK_CHARS = 4000;
+/**
+ * Split a large chunk into smaller pieces while preserving some structure
+ */
+function splitLargeChunk(chunk) {
+    const { content, startLine } = chunk;
+    if (content.length <= MAX_CHUNK_CHARS) {
+        return [chunk];
+    }
+    const lines = content.split('\n');
+    const result = [];
+    let currentLines = [];
+    let currentSize = 0;
+    let currentStartLine = startLine;
+    for (let i = 0; i < lines.length; i++) {
+        const line = lines[i];
+        const lineSize = line.length + 1;
+        if (currentSize + lineSize > MAX_CHUNK_CHARS && currentLines.length > 0) {
+            result.push({
+                content: currentLines.join('\n'),
+                startLine: currentStartLine,
+                endLine: currentStartLine + currentLines.length - 1,
+            });
+            currentLines = [];
+            currentSize = 0;
+            currentStartLine = startLine + i;
+        }
+        currentLines.push(line);
+        currentSize += lineSize;
+    }
+    if (currentLines.length > 0) {
+        result.push({
+            content: currentLines.join('\n'),
+            startLine: currentStartLine,
+            endLine: currentStartLine + currentLines.length - 1,
+        });
+    }
+    return result;
+}
+/**
+ * Chunk code files by logical units (functions, classes, etc.)
+ * Falls back to line-based chunking if no structure detected
+ * Large chunks are automatically split to fit model context limits
+ */
+export function chunkCode(content, filePath) {
+    const ext = filePath.split('.').pop()?.toLowerCase() || '';
+    const lines = content.split('\n');
+    // Language-specific patterns for function/class detection
+    const patterns = {
+        // TypeScript/JavaScript
+        ts: [
+            /^(export\s+)?(async\s+)?function\s+\w+/,
+            /^(export\s+)?(const|let|var)\s+\w+\s*=\s*(async\s+)?\(/,
+            /^(export\s+)?(const|let|var)\s+\w+\s*=\s*(async\s+)?function/,
+            /^(export\s+)?class\s+\w+/,
+            /^(export\s+)?interface\s+\w+/,
+            /^(export\s+)?type\s+\w+/,
+            /^(export\s+)?enum\s+\w+/,
+        ],
+        js: [
+            /^(export\s+)?(async\s+)?function\s+\w+/,
+            /^(export\s+)?(const|let|var)\s+\w+\s*=\s*(async\s+)?\(/,
+            /^(export\s+)?class\s+\w+/,
+        ],
+        // Python
+        py: [/^(async\s+)?def\s+\w+/, /^class\s+\w+/],
+        // Go
+        go: [/^func\s+(\(\w+\s+\*?\w+\)\s+)?\w+/, /^type\s+\w+\s+(struct|interface)/],
+        // Rust
+        rs: [
+            /^(pub\s+)?(async\s+)?fn\s+\w+/,
+            /^(pub\s+)?struct\s+\w+/,
+            /^(pub\s+)?enum\s+\w+/,
+            /^(pub\s+)?impl\s+/,
+            /^(pub\s+)?trait\s+\w+/,
+        ],
+        // Java/Kotlin
+        java: [
+            /^(public|private|protected)?\s*(static)?\s*(final)?\s*(class|interface|enum)\s+\w+/,
+            /^(public|private|protected)?\s*(static)?\s*\w+\s+\w+\s*\(/,
+        ],
+        kt: [/^(fun|suspend\s+fun)\s+\w+/, /^(class|interface|object|data\s+class)\s+\w+/],
+    };
+    // Map extensions to pattern keys
+    const extMap = {
+        ts: 'ts',
+        tsx: 'ts',
+        js: 'js',
+        jsx: 'js',
+        mjs: 'js',
+        cjs: 'js',
+        py: 'py',
+        go: 'go',
+        rs: 'rs',
+        java: 'java',
+        kt: 'kt',
+        kts: 'kt',
+    };
+    const patternKey = extMap[ext];
+    const langPatterns = patternKey ? patterns[patternKey] : null;
+    // If no patterns for this language, use simple chunking
+    if (!langPatterns) {
+        return chunkText(content, filePath);
+    }
+    const chunks = [];
+    let currentChunk = [];
+    let chunkStartLine = 0;
+    let braceDepth = 0;
+    let inDefinition = false;
+    for (let i = 0; i < lines.length; i++) {
+        const line = lines[i];
+        const trimmedLine = line.trim();
+        // Check if this line starts a new definition
+        const startsNewDef = langPatterns.some((p) => p.test(trimmedLine));
+        // For brace-based languages, track depth (ignoring braces in strings/comments)
+        if (['ts', 'js', 'go', 'rs', 'java', 'kt'].includes(patternKey || '')) {
+            braceDepth += countBracesOutsideStrings(line);
+        }
+        // For Python, track indentation
+        const pythonEndOfDef = patternKey === 'py' &&
+            inDefinition &&
+            trimmedLine !== '' &&
+            !line.startsWith(' ') &&
+            !line.startsWith('\t');
+        // Start new chunk if we hit a new definition
+        if (startsNewDef || pythonEndOfDef) {
+            // Save previous chunk if not empty
+            if (currentChunk.length > 0 && currentChunk.some((l) => l.trim())) {
+                chunks.push({
+                    content: currentChunk.join('\n'),
+                    startLine: chunkStartLine + 1,
+                    endLine: i,
+                });
+            }
+            currentChunk = [];
+            chunkStartLine = i;
+            inDefinition = true;
+        }
+        currentChunk.push(line);
+        // Check if definition ended (brace depth returned to 0)
+        if (inDefinition &&
+            braceDepth === 0 &&
+            currentChunk.length > 1 &&
+            ['ts', 'js', 'go', 'rs', 'java', 'kt'].includes(patternKey || '')) {
+            // Check if the closing brace is on this line
+            if (trimmedLine === '}' || trimmedLine.endsWith('}')) {
+                chunks.push({
+                    content: currentChunk.join('\n'),
+                    startLine: chunkStartLine + 1,
+                    endLine: i + 1,
+                });
+                currentChunk = [];
+                chunkStartLine = i + 1;
+                inDefinition = false;
+            }
+        }
+    }
+    // Save remaining content
+    if (currentChunk.length > 0 && currentChunk.some((l) => l.trim())) {
+        chunks.push({
+            content: currentChunk.join('\n'),
+            startLine: chunkStartLine + 1,
+            endLine: lines.length,
+        });
+    }
+    // If we ended up with too few chunks, fall back to line-based
+    if (chunks.length < 2 && lines.length > 100) {
+        return chunkText(content, filePath);
+    }
+    // Filter out empty chunks and chunks that are too small
+    const filtered = chunks.filter((c) => c.content.trim().length > 20);
+    // Split any chunks that exceed the max size
+    return filtered.flatMap(splitLargeChunk);
+}
+/**
+ * Async code chunker: tries tree-sitter first, falls back to regex chunker.
+ *
+ * Use this instead of chunkCode() when async is acceptable (indexer, MCP tools).
+ * Tree-sitter provides precise AST boundaries and metadata (symbolName, symbolType, etc.)
+ * for 36+ languages. Falls back to regex-based chunkCode() if tree-sitter is unavailable.
+ */
+export async function chunkCodeAsync(content, filePath) {
+    try {
+        const { chunkCodeWithTreeSitter } = await import('./tree-sitter/chunker-ts.js');
+        const tsChunks = await chunkCodeWithTreeSitter(content, filePath);
+        if (tsChunks !== null) {
+            // Tree-sitter handled this file (even if 0 chunks for empty files)
+            chunkingStats.astFiles++;
+            if (tsChunks.length > 0)
+                return tsChunks;
+            // Empty result — fall through to regex for non-empty content
+            if (content.trim().length === 0)
+                return tsChunks;
+        }
+    }
+    catch {
+        // Tree-sitter unavailable or failed — fall through to regex
+    }
+    chunkingStats.regexFiles++;
+    return chunkCode(content, filePath);
+}
+/**
+ * Extract frontmatter from markdown
+ */
+export function extractFrontmatter(content) {
+    const match = content.match(/^---\n([\s\S]*?)\n---\n([\s\S]*)$/);
+    if (!match) {
+        return { frontmatter: {}, body: content };
+    }
+    try {
+        // Simple YAML parsing (key: value)
+        const frontmatter = {};
+        const lines = match[1].split('\n');
+        for (const line of lines) {
+            const colonIndex = line.indexOf(':');
+            if (colonIndex > 0) {
+                const key = line.slice(0, colonIndex).trim();
+                let value = line.slice(colonIndex + 1).trim();
+                // Remove quotes
+                if (value.startsWith('"') && value.endsWith('"')) {
+                    value = value.slice(1, -1);
+                }
+                frontmatter[key] = value;
+            }
+        }
+        return { frontmatter, body: match[2] };
+    }
+    catch {
+        return { frontmatter: {}, body: content };
+    }
+}
+//# sourceMappingURL=chunker.js.map

package/dist/lib/chunker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chunker.js","sourceRoot":"","sources":["../../src/lib/chunker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAGxC,wEAAwE;AACxE,IAAI,aAAa,GAAG,EAAE,QAAQ,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;AAEnD,MAAM,UAAU,gBAAgB;IAC9B,OAAO,EAAE,GAAG,aAAa,EAAE,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,aAAa,GAAG,EAAE,QAAQ,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;AACjD,CAAC;AAED;;GAEG;AACH,SAAS,yBAAyB,CAAC,IAAY;IAC7C,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,QAAQ,GAAkB,IAAI,CAAC,CAAC,+BAA+B;IACnE,IAAI,aAAa,GAAG,KAAK,CAAC;IAC1B,IAAI,OAAO,GAAG,KAAK,CAAC,CAAC,mDAAmD;IAExE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACrC,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACrB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAExD,8DAA8D;QAC9D,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,GAAG,KAAK,CAAC;YAChB,SAAS;QACX,CAAC;QAED,4CAA4C;QAC5C,IAAI,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAC9B,OAAO,GAAG,IAAI,CAAC;YACf,SAAS;QACX,CAAC;QAED,oCAAoC;QACpC,IAAI,CAAC,QAAQ,IAAI,CAAC,aAAa,EAAE,CAAC;YAChC,IAAI,IAAI,KAAK,GAAG,IAAI,QAAQ,KAAK,GAAG,EAAE,CAAC;gBACrC,aAAa,GAAG,IAAI,CAAC;gBACrB,SAAS;YACX,CAAC;YACD,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjB,wBAAwB;gBACxB,aAAa,GAAG,IAAI,CAAC;gBACrB,SAAS;YACX,CAAC;QACH,CAAC;QAED,IAAI,aAAa;YAAE,SAAS;QAE5B,2BAA2B;QAC3B,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YAChE,QAAQ,GAAG,IAAI,CAAC;YAChB,SAAS;QACX,CAAC;QAED,IAAI,QAAQ,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClC,QAAQ,GAAG,IAAI,CAAC;YAChB,SAAS;QACX,CAAC;QAED,oCAAoC;QACpC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,IAAI,IAAI,KAAK,GAAG;gBAAE,KAAK,EAAE,CAAC;YAC1B,IAAI,IAAI,KAAK,GAAG;gBAAE,KAAK,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAaD;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAY;IAC7C,IAAI,CAAC,KAAK,CAAC,UAAU,IAAI,CAAC,KAAK,CAAC,SAAS;QAAE,OAAO,KAAK,CAAC,OAAO,CAAC;IAChE,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,IAAI,QAAQ,CAAC;IAC1C,MAAM,GAAG,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;IAC1D,OAAO,IAAI,IAAI,KAAK,KAAK,CAAC,UAAU,GAAG,GAAG,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;AAClE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,SAAS,CAAC,IAAY,EAAE,SAAiB;IACvD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,EAAE,UAAU,EAAE,aAAa,EAAE,GAAG,MAAM,CAAC;IAE7C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/B,MAAM,MAAM,GAAY,EAAE,CAAC;IAE3B,IAAI,YAAY,GAAa,EAAE,CAAC;IAChC,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,iBAAiB;QAEnD,6DAA6D;QAC7D,IAAI,WAAW,GAAG,QAAQ,GAAG,UAAU,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnE,MAAM,CAAC,IAAI,CAAC;gBACV,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;gBAChC,SAAS,EAAE,SAAS,GAAG,CAAC,EAAE,YAAY;gBACtC,OAAO,EAAE,CAAC,EAAE,gBAAgB;aAC7B,CAAC,CAAC;YAEH,qBAAqB;YACrB,MAAM,YAAY,GAAa,EAAE,CAAC;YAClC,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,MAAM,sBAAsB,GAAG,SAAS,CAAC;YACzC,MAAM,sBAAsB,GAAG,YAAY,CAAC,MAAM,CAAC;YACnD,KAAK,IAAI,CAAC,GAAG,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,WAAW,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;gBACjF,YAAY,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;gBACtC,WAAW,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;YAC5C,CAAC;YAED,YAAY,GAAG,YAAY,CAAC;YAC5B,WAAW,GAAG,WAAW,CAAC;YAC1B,6DAA6D;YAC7D,SAAS,GAAG,sBAAsB,GAAG,sBAAsB,GAAG,YAAY,CAAC,MAAM,CAAC;QACpF,CAAC;QAED,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxB,WAAW,IAAI,QAAQ,CAAC;IAC1B,CAAC;IAED,uBAAuB;IACvB,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC;YACV,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;YAChC,SAAS,EAAE,SAAS,GAAG,CAAC;YACxB,OAAO,EAAE,KAAK,CAAC,MAAM;SACtB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,0EAA0E;AAC1E,iFAAiF;AACjF,0EAA0E;AAC1E,MAAM,eAAe,GAAG,IAAI,CAAC;AAE7B;;GAEG;AACH,SAAS,eAAe,CAAC,KAAY;IACnC,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,KAAK,CAAC;IACrC,IAAI,OAAO,CAAC,MAAM,IAAI,eAAe,EAAE,CAAC;QACtC,OAAO,CAAC,KAAK,CAAC,CAAC;IACjB,CAAC;IAED,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,MAAM,GAAY,EAAE,CAAC;IAC3B,IAAI,YAAY,GAAa,EAAE,CAAC;IAChC,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,gBAAgB,GAAG,SAAS,CAAC;IAEjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;QAEjC,IAAI,WAAW,GAAG,QAAQ,GAAG,eAAe,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxE,MAAM,CAAC,IAAI,CAAC;gBACV,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;gBAChC,SAAS,EAAE,gBAAgB;gBAC3B,OAAO,EAAE,gBAAgB,GAAG,YAAY,CAAC,MAAM,GAAG,CAAC;aACpD,CAAC,CAAC;YACH,YAAY,GAAG,EAAE,CAAC;YAClB,WAAW,GAAG,CAAC,CAAC;YAChB,gBAAgB,GAAG,SAAS,GAAG,CAAC,CAAC;QACnC,CAAC;QAED,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxB,WAAW,IAAI,QAAQ,CAAC;IAC1B,CAAC;IAED,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC;YACV,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;YAChC,SAAS,EAAE,gBAAgB;YAC3B,OAAO,EAAE,gBAAgB,GAAG,YAAY,CAAC,MAAM,GAAG,CAAC;SACpD,CAAC,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,SAAS,CAAC,OAAe,EAAE,QAAgB;IACzD,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;IAC3D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,0DAA0D;IAC1D,MAAM,QAAQ,GAA6B;QACzC,wBAAwB;QACxB,EAAE,EAAE;YACF,wCAAwC;YACxC,wDAAwD;YACxD,8DAA8D;YAC9D,0BAA0B;YAC1B,8BAA8B;YAC9B,yBAAyB;YACzB,yBAAyB;SAC1B;QACD,EAAE,EAAE;YACF,wCAAwC;YACxC,wDAAwD;YACxD,0BAA0B;SAC3B;QACD,SAAS;QACT,EAAE,EAAE,CAAC,uBAAuB,EAAE,cAAc,CAAC;QAC7C,KAAK;QACL,EAAE,EAAE,CAAC,mCAAmC,EAAE,kCAAkC,CAAC;QAC7E,OAAO;QACP,EAAE,EAAE;YACF,+BAA+B;YAC/B,wBAAwB;YACxB,sBAAsB;YACtB,mBAAmB;YACnB,uBAAuB;SACxB;QACD,cAAc;QACd,IAAI,EAAE;YACJ,oFAAoF;YACpF,2DAA2D;SAC5D;QACD,EAAE,EAAE,CAAC,4BAA4B,EAAE,8CAA8C,CAAC;KACnF,CAAC;IAEF,iCAAiC;IACjC,MAAM,MAAM,GAA2B;QACrC,EAAE,EAAE,IAAI;QACR,GAAG,EAAE,IAAI;QACT,EAAE,EAAE,IAAI;QACR,GAAG,EAAE,IAAI;QACT,GAAG,EAAE,IAAI;QACT,GAAG,EAAE,IAAI;QACT,EAAE,EAAE,IAAI;QACR,EAAE,EAAE,IAAI;QACR,EAAE,EAAE,IAAI;QACR,IAAI,EAAE,MAAM;QACZ,EAAE,EAAE,IAAI;QACR,GAAG,EAAE,IAAI;KACV,CAAC;IAEF,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAC/B,MAAM,YAAY,GAAG,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAE9D,wDAAwD;IACxD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACtC,CAAC;IAED,MAAM,MAAM,GAAY,EAAE,CAAC;IAC3B,IAAI,YAAY,GAAa,EAAE,CAAC;IAChC,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,YAAY,GAAG,KAAK,CAAC;IAEzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAEhC,6CAA6C;QAC7C,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;QAEnE,+EAA+E;QAC/E,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,QAAQ,CAAC,UAAU,IAAI,EAAE,CAAC,EAAE,CAAC;YACtE,UAAU,IAAI,yBAAyB,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC;QAED,gCAAgC;QAChC,MAAM,cAAc,GAClB,UAAU,KAAK,IAAI;YACnB,YAAY;YACZ,WAAW,KAAK,EAAE;YAClB,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;YACrB,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAEzB,6CAA6C;QAC7C,IAAI,YAAY,IAAI,cAAc,EAAE,CAAC;YACnC,mCAAmC;YACnC,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;gBAClE,MAAM,CAAC,IAAI,CAAC;oBACV,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;oBAChC,SAAS,EAAE,cAAc,GAAG,CAAC;oBAC7B,OAAO,EAAE,CAAC;iBACX,CAAC,CAAC;YACL,CAAC;YACD,YAAY,GAAG,EAAE,CAAC;YAClB,cAAc,GAAG,CAAC,CAAC;YACnB,YAAY,GAAG,IAAI,CAAC;QACtB,CAAC;QAED,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAExB,wDAAwD;QACxD,IACE,YAAY;YACZ,UAAU,KAAK,CAAC;YAChB,YAAY,CAAC,MAAM,GAAG,CAAC;YACvB,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,QAAQ,CAAC,UAAU,IAAI,EAAE,CAAC,EACjE,CAAC;YACD,6CAA6C;YAC7C,IAAI,WAAW,KAAK,GAAG,IAAI,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrD,MAAM,CAAC,IAAI,CAAC;oBACV,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;oBAChC,SAAS,EAAE,cAAc,GAAG,CAAC;oBAC7B,OAAO,EAAE,CAAC,GAAG,CAAC;iBACf,CAAC,CAAC;gBACH,YAAY,GAAG,EAAE,CAAC;gBAClB,cAAc,GAAG,CAAC,GAAG,CAAC,CAAC;gBACvB,YAAY,GAAG,KAAK,CAAC;YACvB,CAAC;QACH,CAAC;IACH,CAAC;IAED,yBAAyB;IACzB,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;QAClE,MAAM,CAAC,IAAI,CAAC;YACV,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;YAChC,SAAS,EAAE,cAAc,GAAG,CAAC;YAC7B,OAAO,EAAE,KAAK,CAAC,MAAM;SACtB,CAAC,CAAC;IACL,CAAC;IAED,8DAA8D;IAC9D,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;QAC5C,OAAO,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACtC,CAAC;IAED,wDAAwD;IACxD,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;IAEpE,4CAA4C;IAC5C,OAAO,QAAQ,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;AAC3C,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,OAAe,EAAE,QAAgB;IACpE,IAAI,CAAC;QACH,MAAM,EAAE,uBAAuB,EAAE,GAAG,MAAM,MAAM,CAAC,6BAA6B,CAAC,CAAC;QAChF,MAAM,QAAQ,GAAG,MAAM,uBAAuB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAClE,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;YACtB,mEAAmE;YACnE,aAAa,CAAC,QAAQ,EAAE,CAAC;YACzB,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC;gBAAE,OAAO,QAAQ,CAAC;YACzC,6DAA6D;YAC7D,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,QAAQ,CAAC;QACnD,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,4DAA4D;IAC9D,CAAC;IAED,aAAa,CAAC,UAAU,EAAE,CAAC;IAC3B,OAAO,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;AACtC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,OAAe;IAIhD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;IAEjE,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;IAC5C,CAAC;IAED,IAAI,CAAC;QACH,mCAAmC;QACnC,MAAM,WAAW,GAA4B,EAAE,CAAC;QAChD,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEnC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACrC,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;gBACnB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC7C,IAAI,KAAK,GAAY,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAEvD,gBAAgB;gBAChB,IAAK,KAAgB,CAAC,UAAU,CAAC,GAAG,CAAC,IAAK,KAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;oBACzE,KAAK,GAAI,KAAgB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBACzC,CAAC;gBAED,WAAW,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IACzC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;IAC5C,CAAC;AACH,CAAC"}

package/dist/lib/claude-ws-transport.d.ts

@@ -0,0 +1,76 @@
+/**
+ * Claude WebSocket Transport
+ *
+ * Persistent WebSocket connection to Claude Code CLI via the --sdk-url flag.
+ * NDJSON protocol over WebSocket — eliminates process-per-call overhead.
+ *
+ * Protocol based on reverse-engineering by The-Vibe-Company/companion:
+ * https://github.com/The-Vibe-Company/companion
+ * See their WEBSOCKET_PROTOCOL_REVERSED.md for full protocol spec.
+ */
+import type { ChatMessage } from './llm.js';
+export interface WSSendOptions {
+    model?: string;
+    timeout?: number;
+}
+export declare class ClaudeWSTransport {
+    private wss;
+    private ws;
+    private cliProcess;
+    private sessionId;
+    private port;
+    private pending;
+    private queue;
+    private processing;
+    private idleTimer;
+    /** Default idle timeout (5 min). Override via llm.ws_idle_timeout (seconds). */
+    private static readonly DEFAULT_IDLE_TIMEOUT_MS;
+    private readyResolve;
+    private readyReject;
+    private readyPromise;
+    private static instance;
+    private static initPromise;
+    private constructor();
+    /**
+     * Get or create the singleton instance.
+     * First call starts WS server + spawns CLI + waits for system/init.
+     */
+    static getInstance(): Promise<ClaudeWSTransport>;
+    /**
+     * Fire-and-forget warmup — call early so first real call is instant.
+     */
+    static warmup(): void;
+    /**
+     * Shutdown: kill CLI process, close WS server, reset singleton.
+     */
+    static shutdown(): Promise<void>;
+    /**
+     * Send a single prompt and get the response text.
+     */
+    send(prompt: string, opts?: WSSendOptions): Promise<string>;
+    /**
+     * Send multi-turn chat messages (native multi-turn via WS).
+     * Sends messages sequentially — each user message waits for its result.
+     */
+    sendChat(messages: ChatMessage[], opts?: WSSendOptions): Promise<string>;
+    private start;
+    private startServer;
+    private launchCLI;
+    private stop;
+    private initRequestId;
+    /**
+     * Send initialize control_request to CLI.
+     * Required by the SDK protocol before sending any user messages.
+     */
+    private sendInitialize;
+    private handleMessage;
+    private autoApproveTool;
+    private getIdleTimeoutMs;
+    private resetIdleTimer;
+    private clearIdleTimer;
+    private enqueue;
+    private processNext;
+    private sendPrompt;
+    private rejectPending;
+}
+//# sourceMappingURL=claude-ws-transport.d.ts.map

package/dist/lib/claude-ws-transport.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"claude-ws-transport.d.ts","sourceRoot":"","sources":["../../src/lib/claude-ws-transport.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAQH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAgB5C,MAAM,WAAW,aAAa;IAC5B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AA8DD,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,GAAG,CAAgC;IAC3C,OAAO,CAAC,EAAE,CAAuB;IACjC,OAAO,CAAC,UAAU,CAA6B;IAC/C,OAAO,CAAC,SAAS,CAAM;IACvB,OAAO,CAAC,IAAI,CAAK;IACjB,OAAO,CAAC,OAAO,CAA+B;IAC9C,OAAO,CAAC,KAAK,CAAyB;IACtC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,SAAS,CAA8C;IAE/D,gFAAgF;IAChF,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAAiB;IAEhE,OAAO,CAAC,YAAY,CAA6B;IACjD,OAAO,CAAC,WAAW,CAAuC;IAC1D,OAAO,CAAC,YAAY,CAAgB;IAGpC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAkC;IACzD,OAAO,CAAC,MAAM,CAAC,WAAW,CAA2C;IAErE,OAAO;IAOP;;;OAGG;WACU,WAAW,IAAI,OAAO,CAAC,iBAAiB,CAAC;IAoBtD;;OAEG;IACH,MAAM,CAAC,MAAM,IAAI,IAAI;IAMrB;;OAEG;WACU,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAYtC;;OAEG;IACG,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC;IAKjE;;;OAGG;IACG,QAAQ,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,IAAI,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC;YA6BhE,KAAK;IA2BnB,OAAO,CAAC,WAAW;IA8DnB,OAAO,CAAC,SAAS;YAkEH,IAAI;IA4ClB,OAAO,CAAC,aAAa,CAAM;IAE3B;;;OAGG;IACH,OAAO,CAAC,cAAc;IActB,OAAO,CAAC,aAAa;IA0ErB,OAAO,CAAC,eAAe;IAsBvB,OAAO,CAAC,gBAAgB;IAaxB,OAAO,CAAC,cAAc;IAStB,OAAO,CAAC,cAAc;IAWtB,OAAO,CAAC,OAAO;IAgBf,OAAO,CAAC,WAAW;IAUnB,OAAO,CAAC,UAAU;IAwClB,OAAO,CAAC,aAAa;CAWtB"}