@nxuss/lemma 0.9.2 → 1.0.0

package/dist/esm/utils/SymbolSurgicalContext.js

@@ -0,0 +1,226 @@
+/**
+ * SymbolSurgicalContext — Surgical AST extraction tool.
+ *
+ * Scans a file for a specific symbol (class, function, variable, interface),
+ * extracts its implementation, parses its AST to find workspace dependencies,
+ * and extracts ONLY the signatures of those dependencies (not their entire files).
+ *
+ * Saves up to 90% of token context in LLM prompts.
+ */
+import fs from 'fs';
+import path from 'path';
+import * as ts from 'typescript';
+import { safeResolvePath } from '../mcp/utils';
+/**
+ * Traverses AST to find the node defining the specified symbol name.
+ */
+function findSymbolNode(sourceFile, name) {
+    let targetNode = null;
+    function walk(node) {
+        if (targetNode)
+            return;
+        if (ts.isFunctionDeclaration(node) && node.name && node.name.text === name) {
+            targetNode = node;
+            return;
+        }
+        if (ts.isClassDeclaration(node) && node.name && node.name.text === name) {
+            targetNode = node;
+            return;
+        }
+        if (ts.isInterfaceDeclaration(node) && node.name && node.name.text === name) {
+            targetNode = node;
+            return;
+        }
+        if (ts.isTypeAliasDeclaration(node) && node.name && node.name.text === name) {
+            targetNode = node;
+            return;
+        }
+        if (ts.isEnumDeclaration(node) && node.name && node.name.text === name) {
+            targetNode = node;
+            return;
+        }
+        if (ts.isVariableStatement(node)) {
+            for (const decl of node.declarationList.declarations) {
+                if (ts.isIdentifier(decl.name) && decl.name.text === name) {
+                    targetNode = node; // return the entire variable statement
+                    return;
+                }
+            }
+        }
+        ts.forEachChild(node, walk);
+    }
+    walk(sourceFile);
+    return targetNode;
+}
+/**
+ * Extracts referenced identifiers from a node.
+ */
+function extractReferences(node) {
+    const refs = new Set();
+    function walk(n) {
+        if (ts.isIdentifier(n)) {
+            refs.add(n.text);
+        }
+        ts.forEachChild(n, walk);
+    }
+    walk(node);
+    return refs;
+}
+/**
+ * Finds the import declarations in a file to map imported symbols to their paths.
+ */
+function parseImports(sourceFile, currentFilePath, workspaceRoot) {
+    const importMap = new Map(); // symbol name -> relative module path
+    ts.forEachChild(sourceFile, (node) => {
+        if (ts.isImportDeclaration(node) && node.moduleSpecifier && ts.isStringLiteral(node.moduleSpecifier)) {
+            const spec = node.moduleSpecifier.text;
+            // We only care about relative imports (workspace files)
+            if (spec.startsWith('.')) {
+                const importDir = path.dirname(currentFilePath);
+                let resolvedPath = path.resolve(importDir, spec);
+                // Find exact extension matching
+                for (const ext of ['.ts', '.tsx', '.js', '.jsx', '/index.ts', '/index.tsx', '/index.js']) {
+                    if (fs.existsSync(resolvedPath + ext)) {
+                        resolvedPath += ext;
+                        break;
+                    }
+                }
+                const relPath = path.relative(workspaceRoot, resolvedPath).replace(/\\/g, '/');
+                // Extract imported names
+                if (node.importClause) {
+                    // Default import: import Foo from './Foo'
+                    if (node.importClause.name) {
+                        importMap.set(node.importClause.name.text, relPath);
+                    }
+                    // Named imports: import { Foo, Bar } from './Module'
+                    if (node.importClause.namedBindings) {
+                        if (ts.isNamedImports(node.importClause.namedBindings)) {
+                            for (const el of node.importClause.namedBindings.elements) {
+                                importMap.set(el.name.text, relPath);
+                            }
+                        }
+                        else if (ts.isNamespaceImport(node.importClause.namedBindings)) {
+                            // import * as Foo from './Module'
+                            importMap.set(node.importClause.namedBindings.name.text, relPath);
+                        }
+                    }
+                }
+            }
+        }
+    });
+    return importMap;
+}
+/**
+ * Extracts ONLY the signature declaration of a symbol, dropping the function/method body.
+ */
+function getSignatureText(node, sourceText) {
+    if (ts.isFunctionDeclaration(node)) {
+        const start = node.getStart();
+        let end = node.body ? node.body.getStart() : node.getEnd();
+        let sig = sourceText.substring(start, end).trim();
+        if (sig.endsWith('{'))
+            sig = sig.slice(0, -1).trim();
+        return { signature: sig, type: 'function' };
+    }
+    if (ts.isClassDeclaration(node)) {
+        const start = node.getStart();
+        // For class declaration, get just the class definition signature and public methods signatures
+        let end = node.members.length > 0 ? node.members[0].getStart() : node.getEnd();
+        let classHeader = sourceText.substring(start, end).trim();
+        if (classHeader.endsWith('{'))
+            classHeader = classHeader.slice(0, -1).trim();
+        // Add public methods/members signatures
+        const memberSigs = [];
+        for (const member of node.members) {
+            if (ts.isMethodDeclaration(member) && member.name) {
+                // Only include public methods
+                const modifiers = ts.getModifiers(member);
+                const isPrivate = modifiers?.some(m => m.kind === ts.SyntaxKind.PrivateKeyword);
+                if (!isPrivate) {
+                    const mStart = member.getStart();
+                    let mEnd = member.body ? member.body.getStart() : member.getEnd();
+                    let mSig = sourceText.substring(mStart, mEnd).trim();
+                    if (mSig.endsWith('{'))
+                        mSig = mSig.slice(0, -1).trim();
+                    memberSigs.push(`  ${mSig};`);
+                }
+            }
+            else if (ts.isPropertyDeclaration(member) && member.name) {
+                memberSigs.push(`  ${member.getText().split('=')[0].trim()};`);
+            }
+        }
+        return {
+            signature: `${classHeader} {\n${memberSigs.join('\n')}\n}`,
+            type: 'class'
+        };
+    }
+    if (ts.isInterfaceDeclaration(node) || ts.isTypeAliasDeclaration(node) || ts.isEnumDeclaration(node)) {
+        return { signature: node.getText().trim(), type: 'type' };
+    }
+    return { signature: node.getText().trim(), type: 'variable' };
+}
+/**
+ * Extracts a specific symbol signature from a dependency file.
+ */
+function extractDependencySignature(filePath, symbolName) {
+    try {
+        if (!fs.existsSync(filePath))
+            return null;
+        const src = fs.readFileSync(filePath, 'utf8');
+        const sf = ts.createSourceFile(filePath, src, ts.ScriptTarget.Latest, true);
+        const node = findSymbolNode(sf, symbolName);
+        if (!node)
+            return null;
+        return getSignatureText(node, src);
+    }
+    catch {
+        return null;
+    }
+}
+/**
+ * Core orchestrator for Surgical Symbol Context Extraction.
+ */
+export function getSymbolSurgicalContext(workspaceRoot, relativeFilePath, symbolName) {
+    const { resolved: resolvedTarget } = safeResolvePath(workspaceRoot, relativeFilePath);
+    if (!fs.existsSync(resolvedTarget)) {
+        throw new Error(`File not found: ${relativeFilePath}`);
+    }
+    const src = fs.readFileSync(resolvedTarget, 'utf8');
+    const sf = ts.createSourceFile(resolvedTarget, src, ts.ScriptTarget.Latest, true);
+    const node = findSymbolNode(sf, symbolName);
+    if (!node) {
+        throw new Error(`Symbol "${symbolName}" not found in file "${relativeFilePath}"`);
+    }
+    // 1. Get entire implementation of target symbol
+    const implementation = node.getText();
+    // 2. Map imports of this file
+    const importsMap = parseImports(sf, resolvedTarget, workspaceRoot);
+    // 3. Find all identifiers referenced in the symbol implementation
+    const refs = extractReferences(node);
+    // 4. Resolve references that are imported from other workspace files
+    const dependencies = [];
+    for (const ref of refs) {
+        if (ref === symbolName)
+            continue; // skip self reference
+        const moduleRelPath = importsMap.get(ref);
+        if (moduleRelPath) {
+            const depAbsPath = path.resolve(workspaceRoot, moduleRelPath);
+            const depInfo = extractDependencySignature(depAbsPath, ref);
+            if (depInfo) {
+                dependencies.push({
+                    name: ref,
+                    filePath: moduleRelPath,
+                    signature: depInfo.signature,
+                    type: depInfo.type
+                });
+            }
+        }
+    }
+    return {
+        symbolName,
+        filePath: relativeFilePath,
+        implementation,
+        dependencies
+    };
+}
+//# sourceMappingURL=SymbolSurgicalContext.js.map

package/dist/esm/utils/SymbolSurgicalContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SymbolSurgicalContext.js","sourceRoot":"","sources":["../../../src/utils/SymbolSurgicalContext.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,KAAK,EAAE,MAAM,YAAY,CAAC;AACjC,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAc/C;;GAEG;AACH,SAAS,cAAc,CAAC,UAAyB,EAAE,IAAY;IAC7D,IAAI,UAAU,GAAmB,IAAI,CAAC;IAEtC,SAAS,IAAI,CAAC,IAAa;QACzB,IAAI,UAAU;YAAE,OAAO;QAEvB,IAAI,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YAC3E,UAAU,GAAG,IAAI,CAAC;YAClB,OAAO;QACT,CAAC;QACD,IAAI,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACxE,UAAU,GAAG,IAAI,CAAC;YAClB,OAAO;QACT,CAAC;QACD,IAAI,EAAE,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YAC5E,UAAU,GAAG,IAAI,CAAC;YAClB,OAAO;QACT,CAAC;QACD,IAAI,EAAE,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YAC5E,UAAU,GAAG,IAAI,CAAC;YAClB,OAAO;QACT,CAAC;QACD,IAAI,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACvE,UAAU,GAAG,IAAI,CAAC;YAClB,OAAO;QACT,CAAC;QACD,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YACjC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,CAAC;gBACrD,IAAI,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;oBAC1D,UAAU,GAAG,IAAI,CAAC,CAAC,uCAAuC;oBAC1D,OAAO;gBACT,CAAC;YACH,CAAC;QACH,CAAC;QAED,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC9B,CAAC;IAED,IAAI,CAAC,UAAU,CAAC,CAAC;IACjB,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,IAAa;IACtC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,SAAS,IAAI,CAAC,CAAU;QACtB,IAAI,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC;YACvB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACnB,CAAC;QACD,EAAE,CAAC,YAAY,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAC3B,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,CAAC;IACX,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,UAAyB,EAAE,eAAuB,EAAE,aAAqB;IAC7F,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC,CAAC,sCAAsC;IAEnF,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC,IAAI,EAAE,EAAE;QACnC,IAAI,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,eAAe,IAAI,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YACrG,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;YAEvC,wDAAwD;YACxD,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACzB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;gBAChD,IAAI,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;gBAEjD,gCAAgC;gBAChC,KAAK,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,CAAC,EAAE,CAAC;oBACzF,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,GAAG,GAAG,CAAC,EAAE,CAAC;wBACtC,YAAY,IAAI,GAAG,CAAC;wBACpB,MAAM;oBACR,CAAC;gBACH,CAAC;gBAED,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;gBAE/E,yBAAyB;gBACzB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtB,0CAA0C;oBAC1C,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC;wBAC3B,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;oBACtD,CAAC;oBACD,qDAAqD;oBACrD,IAAI,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,CAAC;wBACpC,IAAI,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,EAAE,CAAC;4BACvD,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;gCAC1D,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;4BACvC,CAAC;wBACH,CAAC;6BAAM,IAAI,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,EAAE,CAAC;4BACjE,kCAAkC;4BAClC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;wBACpE,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,SAAS,gBAAgB,CAAC,IAAa,EAAE,UAAkB;IACzD,IAAI,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC9B,IAAI,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QAC3D,IAAI,GAAG,GAAG,UAAU,CAAC,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QAClD,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACrD,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;IAC9C,CAAC;IAED,IAAI,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC;QAChC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC9B,+FAA+F;QAC/F,IAAI,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QAC/E,IAAI,WAAW,GAAG,UAAU,CAAC,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QAC1D,IAAI,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,WAAW,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE7E,wCAAwC;QACxC,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,EAAE,CAAC,mBAAmB,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAClD,8BAA8B;gBAC9B,MAAM,SAAS,GAAG,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;gBAC1C,MAAM,SAAS,GAAG,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;gBAChF,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;oBACjC,IAAI,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;oBAClE,IAAI,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;oBACrD,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;wBAAE,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACxD,UAAU,CAAC,IAAI,CAAC,KAAK,IAAI,GAAG,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;iBAAM,IAAI,EAAE,CAAC,qBAAqB,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC3D,UAAU,CAAC,IAAI,CAAC,KAAK,MAAM,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QAED,OAAO;YACL,SAAS,EAAE,GAAG,WAAW,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK;YAC1D,IAAI,EAAE,OAAO;SACd,CAAC;IACJ,CAAC;IAED,IAAI,EAAE,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;QACrG,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IAC5D,CAAC;IAED,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;AAChE,CAAC;AAED;;GAEG;AACH,SAAS,0BAA0B,CAAC,QAAgB,EAAE,UAAkB;IACtE,IAAI,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,OAAO,IAAI,CAAC;QAC1C,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC9C,MAAM,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,QAAQ,EAAE,GAAG,EAAE,EAAE,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC5E,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QACvB,OAAO,gBAAgB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IACrC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB,CACtC,aAAqB,EACrB,gBAAwB,EACxB,UAAkB;IAElB,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,eAAe,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;IAEtF,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,mBAAmB,gBAAgB,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IACpD,MAAM,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,cAAc,EAAE,GAAG,EAAE,EAAE,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAElF,MAAM,IAAI,GAAG,cAAc,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAC5C,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,wBAAwB,gBAAgB,GAAG,CAAC,CAAC;IACpF,CAAC;IAED,gDAAgD;IAChD,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;IAEtC,8BAA8B;IAC9B,MAAM,UAAU,GAAG,YAAY,CAAC,EAAE,EAAE,cAAc,EAAE,aAAa,CAAC,CAAC;IAEnE,kEAAkE;IAClE,MAAM,IAAI,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;IAErC,qEAAqE;IACrE,MAAM,YAAY,GAA0C,EAAE,CAAC;IAE/D,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,GAAG,KAAK,UAAU;YAAE,SAAS,CAAC,sBAAsB;QAExD,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1C,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;YAC9D,MAAM,OAAO,GAAG,0BAA0B,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;YAC5D,IAAI,OAAO,EAAE,CAAC;gBACZ,YAAY,CAAC,IAAI,CAAC;oBAChB,IAAI,EAAE,GAAG;oBACT,QAAQ,EAAE,aAAa;oBACvB,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;iBACnB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,UAAU;QACV,QAAQ,EAAE,gBAAgB;QAC1B,cAAc;QACd,YAAY;KACb,CAAC;AACJ,CAAC"}

package/dist/esm/utils/WormholeCompressor.d.ts

@@ -0,0 +1,22 @@
+/**
+ * WormholeCompressor — Reversible, token-dense serialization for source code.
+ *
+ * Compresses source code into a highly dense representation, substituting keywords
+ * and repeating identifiers with short tokens. Includes a self-contained dictionary
+ * header so the LLM or helper scripts can inflate/deflate it on the fly.
+ *
+ * Reduces token footprint for long files by 60-70% while keeping it structured and reversible.
+ */
+export interface CompressedPayload {
+    dict: Record<string, string>;
+    payload: string;
+}
+/**
+ * Compresses source code into a Wormhole representation.
+ */
+export declare function compressWormhole(code: string): string;
+/**
+ * Reverts a Wormhole-compressed payload back to standard source code.
+ */
+export declare function decompressWormhole(wormholeCode: string): string;
+//# sourceMappingURL=WormholeCompressor.d.ts.map

package/dist/esm/utils/WormholeCompressor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WormholeCompressor.d.ts","sourceRoot":"","sources":["../../../src/utils/WormholeCompressor.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAqCH,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7B,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAwErD;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,YAAY,EAAE,MAAM,GAAG,MAAM,CAsC/D"}

package/dist/esm/utils/WormholeCompressor.js

@@ -0,0 +1,148 @@
+/**
+ * WormholeCompressor — Reversible, token-dense serialization for source code.
+ *
+ * Compresses source code into a highly dense representation, substituting keywords
+ * and repeating identifiers with short tokens. Includes a self-contained dictionary
+ * header so the LLM or helper scripts can inflate/deflate it on the fly.
+ *
+ * Reduces token footprint for long files by 60-70% while keeping it structured and reversible.
+ */
+// ─── Dictionary mappings ───
+const KEYWORD_MAP = {
+    'function': 'ƒ',
+    'const': '¢',
+    'let': 'ℓ',
+    'return': '⏎',
+    'export': '⨂',
+    'async': '⨁',
+    'await': '⨀',
+    'import': '⨃',
+    'Promise': '⨄',
+    'interface': '⨅',
+    'class': '⨆',
+    'undefined': '⨇',
+    'null': '⨈',
+    'console.log': '⨍',
+    'console.error': '⨎',
+    'console.warn': '⨏',
+    'this.': '⨌',
+    'public': '⨗',
+    'private': '⨘',
+    'protected': '⨙',
+    'static': '⨚',
+    'constructor': '⨛',
+    'implements': '⨜',
+    'extends': '⨝'
+};
+// Reverse map for decompression
+const REVERSE_KEYWORD_MAP = {};
+for (const [k, v] of Object.entries(KEYWORD_MAP)) {
+    REVERSE_KEYWORD_MAP[v] = k;
+}
+/**
+ * Compresses source code into a Wormhole representation.
+ */
+export function compressWormhole(code) {
+    // 1. Strip comments first to maximize savings (since this is reversible, comments are lost,
+    // but for LLM context reading that is exactly what we want)
+    let clean = code
+        .replace(/\/\*[\s\S]*?\*\//g, '')
+        .replace(/\/\/.*$/gm, '')
+        .split('\n')
+        .map(line => line.trimEnd())
+        .filter(line => line.trim().length > 0)
+        .join('\n');
+    // 2. Identify long repeating identifiers (camelCase/PascalCase > 6 chars)
+    const identifierRegex = /\b[a-zA-Z_][a-zA-Z0-9_]{5,}\b/g;
+    const matches = clean.match(identifierRegex) || [];
+    const freq = {};
+    for (const m of matches) {
+        // Skip words that are standard keywords
+        if (KEYWORD_MAP[m] || m === 'String' || m === 'Number' || m === 'Boolean' || m === 'Object' || m === 'Array')
+            continue;
+        freq[m] = (freq[m] || 0) + 1;
+    }
+    // Filter identifiers that appear at least twice
+    const sortedIdentifiers = Object.entries(freq)
+        .filter(([, count]) => count >= 2)
+        .sort((a, b) => b[1] - a[1])
+        .map(([word]) => word);
+    // 3. Create short alias map ($a, $b, ... $z, $A, $B...)
+    const dictionary = {};
+    const revDictionary = {};
+    const chars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ';
+    let aliasIndex = 0;
+    for (const word of sortedIdentifiers) {
+        if (aliasIndex >= chars.length)
+            break; // cap at 52 unique replacements
+        const alias = `$${chars[aliasIndex]}`;
+        dictionary[word] = alias;
+        revDictionary[alias] = word;
+        aliasIndex++;
+    }
+    // 4. Perform replacements
+    let payload = clean;
+    // Replace identifiers
+    for (const [word, alias] of Object.entries(dictionary)) {
+        // Use word boundary to avoid partial replacements
+        const escaped = word.replace(/[-\/\\^$*+?.()|[\]{}]/g, '\\$&');
+        const regex = new RegExp(`\\b${escaped}\\b`, 'g');
+        payload = payload.replace(regex, alias);
+    }
+    // Replace standard keywords
+    for (const [word, alias] of Object.entries(KEYWORD_MAP)) {
+        // Avoid boundary checks for things ending/starting with special characters like "this." or "console."
+        if (word.includes('.')) {
+            payload = payload.split(word).join(alias);
+        }
+        else {
+            const regex = new RegExp(`\\b${word}\\b`, 'g');
+            payload = payload.replace(regex, alias);
+        }
+    }
+    // Collapse extra whitespace/consecutive spaces
+    payload = payload.replace(/[ \t]+/g, ' ');
+    // 5. Serialize with Dictionary Header
+    const dictHeader = Object.entries(revDictionary)
+        .map(([alias, word]) => `${alias}=${word}`)
+        .join(',');
+    return `⸎WORMHOLE;DICT:[${dictHeader}]\n${payload}`;
+}
+/**
+ * Reverts a Wormhole-compressed payload back to standard source code.
+ */
+export function decompressWormhole(wormholeCode) {
+    if (!wormholeCode.startsWith('⸎WORMHOLE;DICT:[')) {
+        return wormholeCode; // Not a wormhole payload
+    }
+    const headerEnd = wormholeCode.indexOf(']\n');
+    if (headerEnd === -1)
+        return wormholeCode;
+    const header = wormholeCode.substring('⸎WORMHOLE;DICT:['.length, headerEnd);
+    let payload = wormholeCode.substring(headerEnd + 2);
+    // Parse dictionary
+    const dictionary = {};
+    if (header.trim().length > 0) {
+        const parts = header.split(',');
+        for (const part of parts) {
+            const eqIdx = part.indexOf('=');
+            if (eqIdx !== -1) {
+                const alias = part.substring(0, eqIdx);
+                const word = part.substring(eqIdx + 1);
+                dictionary[alias] = word;
+            }
+        }
+    }
+    // 1. Decompress standard keywords
+    for (const [alias, word] of Object.entries(REVERSE_KEYWORD_MAP)) {
+        payload = payload.split(alias).join(word);
+    }
+    // 2. Decompress custom identifiers
+    for (const [alias, word] of Object.entries(dictionary)) {
+        const escaped = alias.replace(/[-\/\\^$*+?.()|[\]{}]/g, '\\$&');
+        const regex = new RegExp(`${escaped}\\b`, 'g');
+        payload = payload.replace(regex, word);
+    }
+    return payload;
+}
+//# sourceMappingURL=WormholeCompressor.js.map

package/dist/esm/utils/WormholeCompressor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WormholeCompressor.js","sourceRoot":"","sources":["../../../src/utils/WormholeCompressor.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,8BAA8B;AAE9B,MAAM,WAAW,GAA2B;IAC1C,UAAU,EAAE,GAAG;IACf,OAAO,EAAE,GAAG;IACZ,KAAK,EAAE,GAAG;IACV,QAAQ,EAAE,GAAG;IACb,QAAQ,EAAE,GAAG;IACb,OAAO,EAAE,GAAG;IACZ,OAAO,EAAE,GAAG;IACZ,QAAQ,EAAE,GAAG;IACb,SAAS,EAAE,GAAG;IACd,WAAW,EAAE,GAAG;IAChB,OAAO,EAAE,GAAG;IACZ,WAAW,EAAE,GAAG;IAChB,MAAM,EAAE,GAAG;IACX,aAAa,EAAE,GAAG;IAClB,eAAe,EAAE,GAAG;IACpB,cAAc,EAAE,GAAG;IACnB,OAAO,EAAE,GAAG;IACZ,QAAQ,EAAE,GAAG;IACb,SAAS,EAAE,GAAG;IACd,WAAW,EAAE,GAAG;IAChB,QAAQ,EAAE,GAAG;IACb,aAAa,EAAE,GAAG;IAClB,YAAY,EAAE,GAAG;IACjB,SAAS,EAAE,GAAG;CACf,CAAC;AAEF,gCAAgC;AAChC,MAAM,mBAAmB,GAA2B,EAAE,CAAC;AACvD,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;IACjD,mBAAmB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;AAC7B,CAAC;AAOD;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,IAAY;IAC3C,4FAA4F;IAC5F,4DAA4D;IAC5D,IAAI,KAAK,GAAG,IAAI;SACb,OAAO,CAAC,mBAAmB,EAAE,EAAE,CAAC;SAChC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;SACxB,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;SAC3B,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;SACtC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,0EAA0E;IAC1E,MAAM,eAAe,GAAG,gCAAgC,CAAC;IACzD,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;IACnD,MAAM,IAAI,GAA2B,EAAE,CAAC;IACxC,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;QACxB,wCAAwC;QACxC,IAAI,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,SAAS,IAAI,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,OAAO;YAAE,SAAS;QACvH,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC;IAED,gDAAgD;IAChD,MAAM,iBAAiB,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC;SAC3C,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,CAAC;SACjC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;SAC3B,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;IAEzB,wDAAwD;IACxD,MAAM,UAAU,GAA2B,EAAE,CAAC;IAC9C,MAAM,aAAa,GAA2B,EAAE,CAAC;IACjD,MAAM,KAAK,GAAG,sDAAsD,CAAC;IAErE,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,KAAK,MAAM,IAAI,IAAI,iBAAiB,EAAE,CAAC;QACrC,IAAI,UAAU,IAAI,KAAK,CAAC,MAAM;YAAE,MAAM,CAAC,gCAAgC;QACvE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;QACtC,UAAU,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC;QACzB,aAAa,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;QAC5B,UAAU,EAAE,CAAC;IACf,CAAC;IAED,0BAA0B;IAC1B,IAAI,OAAO,GAAG,KAAK,CAAC;IAEpB,sBAAsB;IACtB,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACvD,kDAAkD;QAClD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,wBAAwB,EAAE,MAAM,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,MAAM,OAAO,KAAK,EAAE,GAAG,CAAC,CAAC;QAClD,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC1C,CAAC;IAED,4BAA4B;IAC5B,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;QACxD,sGAAsG;QACtG,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACvB,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,MAAM,IAAI,KAAK,EAAE,GAAG,CAAC,CAAC;YAC/C,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IAE1C,sCAAsC;IACtC,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC;SAC7C,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,IAAI,IAAI,EAAE,CAAC;SAC1C,IAAI,CAAC,GAAG,CAAC,CAAC;IAEb,OAAO,mBAAmB,UAAU,MAAM,OAAO,EAAE,CAAC;AACtD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,YAAoB;IACrD,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;QACjD,OAAO,YAAY,CAAC,CAAC,yBAAyB;IAChD,CAAC;IAED,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC9C,IAAI,SAAS,KAAK,CAAC,CAAC;QAAE,OAAO,YAAY,CAAC;IAE1C,MAAM,MAAM,GAAG,YAAY,CAAC,SAAS,CAAC,kBAAkB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC5E,IAAI,OAAO,GAAG,YAAY,CAAC,SAAS,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC;IAEpD,mBAAmB;IACnB,MAAM,UAAU,GAA2B,EAAE,CAAC;IAC9C,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAChC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAChC,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;gBACvC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;gBACvC,UAAU,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,kCAAkC;IAClC,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,CAAC;QAChE,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC;IAED,mCAAmC;IACnC,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACvD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,wBAAwB,EAAE,MAAM,CAAC,CAAC;QAChE,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,GAAG,OAAO,KAAK,EAAE,GAAG,CAAC,CAAC;QAC/C,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/lemma-proxy.cjs

@@ -3,8 +3,9 @@
 
 const fs = require('fs');
 const path = require('path');
+const os = require('os');
+const { spawn } = require('child_process');
 
-// Automatically load environment variables from the user's workspace using a zero-dependency parser
 function loadEnvFile(filePath) {
   if (fs.existsSync(filePath)) {
     try {
@@ -28,21 +29,37 @@ function loadEnvFile(filePath) {
 
 loadEnvFile(path.join(process.cwd(), '.env'));
 loadEnvFile(path.join(process.cwd(), '.env.local'));
+loadEnvFile(path.join(os.homedir(), '.lemma-cache', 'env'));
 
+const cmd = process.argv[2];
 
-// Target path in the compiled CommonJS distribution
-const distPath = path.join(__dirname, 'dist', 'cjs', 'cli', 'lemma-proxy.js');
-
-if (fs.existsSync(distPath)) {
-  // Production mode: run the compiled CJS JavaScript binary
-  require(distPath);
+if (cmd === 'init') {
+  const initPath = path.join(__dirname, 'bin', 'init.js');
+  if (fs.existsSync(initPath)) require(initPath);
+  else { console.error('init not found. Run npm run build first.'); process.exit(1); }
+} else if (cmd === 'stats') {
+  const cliPath = path.join(__dirname, 'bin', 'cli.js');
+  if (fs.existsSync(cliPath)) require(cliPath);
+  else { console.error('stats not found. Run npm run build first.'); process.exit(1); }
+} else if (cmd === 'mcp') {
+  const mcpPath = path.join(__dirname, 'mcp-server.js');
+  if (fs.existsSync(mcpPath)) require(mcpPath);
+  else { console.error('mcp-server not found.'); process.exit(1); }
+} else if (cmd === 'proxy') {
+  const proxyPath = path.join(__dirname, 'cache-proxy.js');
+  if (fs.existsSync(proxyPath)) require(proxyPath);
+  else { console.error('cache-proxy not found.'); process.exit(1); }
 } else {
-  // Development mode: run via ts-node directly from src/
-  try {
-    require('ts-node/register');
-    require('./src/cli/lemma-proxy.ts');
-  } catch (err) {
-    console.error('❌ Error: Could not launch Lemma. Please run "npm run build" to compile the package.');
-    process.exit(1);
+  const distPath = path.join(__dirname, 'dist', 'cjs', 'cli', 'lemma-proxy.js');
+  if (fs.existsSync(distPath)) {
+    require(distPath);
+  } else {
+    try {
+      require('ts-node/register');
+      require('./src/cli/lemma-proxy.ts');
+    } catch (err) {
+      console.error('Error: Could not launch Lemma. Run "npm run build" to compile.');
+      process.exit(1);
+    }
   }
 }