@codeyam/codeyam-cli 0.0.0 → 0.1.0-staging.036391e

package/packages/ai/src/lib/validateTypeStructure.js

@@ -0,0 +1,72 @@
+export default function validateTypeStructure({ generatedProps, entityProps, typeStructures, }) {
+    // console.log("CODEYAM DEBUG validateTypeStructure IN", JSON.stringify({ generatedProps, entityProps, typeStructures }, null, 2));
+    if (!Array.isArray(entityProps))
+        entityProps = [entityProps];
+    const validProps = generatedProps;
+    if (typeof validProps === 'object' && !Array.isArray(validProps)) {
+        const validPropsObj = validProps;
+        for (const attribute in validPropsObj) {
+            if (validPropsObj[attribute]?.functionResponse)
+                continue;
+            let fileProp = entityProps.find((fileProp) => fileProp.name === attribute);
+            if (!fileProp && entityProps.length === 1) {
+                if (entityProps[0].name === 'destructured') {
+                    fileProp = typeStructures['destructured'].find((fileProp) => fileProp.name === attribute);
+                }
+                else {
+                    fileProp = entityProps.find((fileProp) => fileProp.type === validPropsObj[attribute]);
+                    if (fileProp) {
+                        validPropsObj[fileProp.name] = validPropsObj[attribute];
+                        delete validPropsObj[attribute];
+                    }
+                }
+            }
+            if (!fileProp?.type)
+                continue;
+            if (!typeStructures)
+                typeStructures = fileProp?.typeStructures;
+            const filePropIsArray = fileProp.type.indexOf('[]') > -1;
+            const filePropType = fileProp.type.replace('[]', '');
+            const typeFileProps = typeStructures?.[filePropType];
+            if (typeFileProps) {
+                let generatedPropsForValidation;
+                if (typeof validPropsObj[fileProp.name] === 'object') {
+                    generatedPropsForValidation = validPropsObj[fileProp.name];
+                }
+                else {
+                    if (filePropType === 'any') {
+                        generatedPropsForValidation = validPropsObj[attribute];
+                    }
+                    else {
+                        generatedPropsForValidation = filePropType;
+                    }
+                }
+                const validTypeStructure = validateTypeStructure({
+                    generatedProps: generatedPropsForValidation,
+                    entityProps: typeFileProps,
+                    typeStructures,
+                });
+                if (fileProp.type.indexOf('[]') > -1 &&
+                    !Array.isArray(validTypeStructure)) {
+                    validPropsObj[fileProp.name] = [validTypeStructure];
+                }
+                else {
+                    validPropsObj[fileProp.name] = validTypeStructure;
+                }
+            }
+            else {
+                if (typeof validPropsObj[attribute] === 'object')
+                    continue;
+                if (filePropIsArray) {
+                    validPropsObj[attribute] = [filePropType];
+                }
+                else {
+                    validPropsObj[attribute] = filePropType;
+                }
+            }
+        }
+    }
+    // console.log("CODEYAM DEBUG: validProps OUT", JSON.stringify(validProps, null, 2));
+    return validProps;
+}
+//# sourceMappingURL=validateTypeStructure.js.map

package/packages/ai/src/lib/validateTypeStructure.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validateTypeStructure.js","sourceRoot":"","sources":["../../../../../../packages/ai/src/lib/validateTypeStructure.ts"],"names":[],"mappings":"AAQA,MAAM,CAAC,OAAO,UAAU,qBAAqB,CAAC,EAC5C,cAAc,EACd,WAAW,EACX,cAAc,GACY;IAC1B,mIAAmI;IAEnI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC;QAAE,WAAW,GAAG,CAAC,WAAW,CAAC,CAAC;IAE7D,MAAM,UAAU,GAAG,cAAc,CAAC;IAElC,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACjE,MAAM,aAAa,GAAG,UAAgC,CAAC;QACvD,KAAK,MAAM,SAAS,IAAI,aAAa,EAAE,CAAC;YACtC,IAAK,aAAa,CAAC,SAAS,CAAS,EAAE,gBAAgB;gBAAE,SAAS;YAElE,IAAI,QAAQ,GAAG,WAAW,CAAC,IAAI,CAC7B,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,KAAK,SAAS,CAC1C,CAAC;YAEF,IAAI,CAAC,QAAQ,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1C,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;oBAC3C,QAAQ,GAAI,cAAc,CAAC,cAAc,CAAgB,CAAC,IAAI,CAC5D,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,KAAK,SAAS,CAC1C,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,QAAQ,GAAG,WAAW,CAAC,IAAI,CACzB,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,KAAK,aAAa,CAAC,SAAS,CAAC,CACzD,CAAC;oBACF,IAAI,QAAQ,EAAE,CAAC;wBACb,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;wBACxD,OAAO,aAAa,CAAC,SAAS,CAAC,CAAC;oBAClC,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,CAAC,QAAQ,EAAE,IAAI;gBAAE,SAAS;YAE9B,IAAI,CAAC,cAAc;gBAAE,cAAc,GAAG,QAAQ,EAAE,cAAc,CAAC;YAE/D,MAAM,eAAe,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YACzD,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACrD,MAAM,aAAa,GAAG,cAAc,EAAE,CAAC,YAAY,CAAC,CAAC;YAErD,IAAI,aAAa,EAAE,CAAC;gBAClB,IAAI,2BAA2B,CAAC;gBAChC,IAAI,OAAO,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,QAAQ,EAAE,CAAC;oBACrD,2BAA2B,GAAG,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAC7D,CAAC;qBAAM,CAAC;oBACN,IAAI,YAAY,KAAK,KAAK,EAAE,CAAC;wBAC3B,2BAA2B,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;oBACzD,CAAC;yBAAM,CAAC;wBACN,2BAA2B,GAAG,YAAY,CAAC;oBAC7C,CAAC;gBACH,CAAC;gBAED,MAAM,kBAAkB,GAAG,qBAAqB,CAAC;oBAC/C,cAAc,EAAE,2BAA2B;oBAC3C,WAAW,EAAE,aAAa;oBAC1B,cAAc;iBACf,CAAC,CAAC;gBAEH,IACE,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAChC,CAAC,KAAK,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAClC,CAAC;oBACD,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAQ,CAAC;gBAC7D,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,kBAAyB,CAAC;gBAC3D,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,OAAO,aAAa,CAAC,SAAS,CAAC,KAAK,QAAQ;oBAAE,SAAS;gBAE3D,IAAI,eAAe,EAAE,CAAC;oBACpB,aAAa,CAAC,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBAC5C,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,SAAS,CAAC,GAAG,YAAY,CAAC;gBAC1C,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,qFAAqF;IACrF,OAAO,UAAU,CAAC;AACpB,CAAC"}

package/packages/ai/src/lib/worker/SerializableDataStructure.js

@@ -0,0 +1,128 @@
+/**
+ * Serializable data structure that contains all the pre-computed results
+ * from ScopeDataStructure that downstream code needs.
+ *
+ * This structure is returned from the worker thread and can be used directly
+ * by downstream code without needing the full ScopeDataStructure class.
+ */
+import { fuzzyMatchFunctionName, findFuzzyMatchingKey, } from "./fuzzyMatchFunctionName.js";
+/**
+ * Helper functions to access SerializableDataStructure
+ * These provide the same interface as ScopeDataStructure methods
+ *
+ * PRODUCTION REQUIREMENTS (used by analyzeEntities.ts):
+ * - getFunctionSignature(dataStructure, { functionName?, fillInUnknowns? })
+ * - getReturnValue(dataStructure, { functionName?, fillInUnknowns? })
+ * - getUsageEquivalencies(dataStructure, functionName?)
+ * - getSourceEquivalencies(dataStructure, functionName?)
+ * - getEquivalentSignatureVariables(dataStructure)
+ * - Direct access to dataStructure.externalFunctionCalls (for callSignature info)
+ *
+ * All helper functions must handle fuzzy name matching (e.g., "useLoaderData" matches "useLoaderData<typeof loader>")
+ * to match the behavior of the original ScopeDataStructure methods.
+ */
+export function getExternalFunctionCallInfo(dataStructure, functionName) {
+    return dataStructure.externalFunctionCalls.find((efc) => fuzzyMatchFunctionName(functionName, efc.name));
+}
+export function getFunctionSignature(dataStructure, options = {}) {
+    const { functionName, fillInUnknowns = false } = options;
+    if (!functionName) {
+        // Root function
+        return fillInUnknowns
+            ? dataStructure.rootFunction.signatureWithUnknowns
+            : dataStructure.rootFunction.signature;
+    }
+    // Find function result with fuzzy matching
+    const matchingKey = findFuzzyMatchingKey(dataStructure.functionResults, functionName);
+    const functionResult = matchingKey
+        ? dataStructure.functionResults[matchingKey]
+        : undefined;
+    if (!functionResult) {
+        return undefined;
+    }
+    return fillInUnknowns
+        ? functionResult.signatureWithUnknowns
+        : functionResult.signature;
+}
+export function getReturnValue(dataStructure, options = {}) {
+    const { functionName, fillInUnknowns = false } = options;
+    if (!functionName) {
+        // Root function
+        return fillInUnknowns
+            ? dataStructure.rootFunction.returnValueWithUnknowns
+            : dataStructure.rootFunction.returnValue;
+    }
+    // Find function result with fuzzy matching
+    const matchingKey = findFuzzyMatchingKey(dataStructure.functionResults, functionName);
+    const functionResult = matchingKey
+        ? dataStructure.functionResults[matchingKey]
+        : undefined;
+    if (!functionResult) {
+        return undefined;
+    }
+    return fillInUnknowns
+        ? functionResult.returnValueWithUnknowns
+        : functionResult.returnValue;
+}
+export function getUsageEquivalencies(dataStructure, functionName) {
+    if (!functionName) {
+        return dataStructure.rootFunction.usageEquivalencies;
+    }
+    // Find function result with fuzzy matching
+    const matchingKey = findFuzzyMatchingKey(dataStructure.functionResults, functionName);
+    const functionResult = matchingKey
+        ? dataStructure.functionResults[matchingKey]
+        : undefined;
+    return functionResult?.usageEquivalencies ?? {};
+}
+export function getSourceEquivalencies(dataStructure, functionName) {
+    if (!functionName) {
+        return dataStructure.rootFunction.sourceEquivalencies;
+    }
+    // Find function result with fuzzy matching
+    const matchingKey = findFuzzyMatchingKey(dataStructure.functionResults, functionName);
+    const functionResult = matchingKey
+        ? dataStructure.functionResults[matchingKey]
+        : undefined;
+    return functionResult?.sourceEquivalencies ?? {};
+}
+export function getEquivalentSignatureVariables(dataStructure) {
+    return dataStructure.equivalentSignatureVariables;
+}
+/**
+ * Get enriched conditional usages with source tracing.
+ * Returns the conditional usages map, or empty object if not available.
+ */
+export function getConditionalUsages(dataStructure) {
+    return dataStructure.conditionalUsages ?? {};
+}
+/**
+ * Get conditional effects (setter calls inside conditionals).
+ * Returns the array of conditional effects, or empty array if not available.
+ */
+export function getConditionalEffects(dataStructure) {
+    return dataStructure.conditionalEffects ?? [];
+}
+/**
+ * Get compound conditionals (grouped conditions that must all be true).
+ * Returns the array of compound conditionals, or empty array if not available.
+ */
+export function getCompoundConditionals(dataStructure) {
+    return dataStructure.compoundConditionals ?? [];
+}
+/**
+ * Get child boundary gating conditions.
+ * Returns the mapping of child component names to their gating conditions,
+ * or empty object if not available.
+ */
+export function getChildBoundaryGatingConditions(dataStructure) {
+    return dataStructure.childBoundaryGatingConditions ?? {};
+}
+/**
+ * Get JSX rendering usages (arrays rendered via .map() and strings interpolated in JSX).
+ * Returns the array of JSX rendering usages, or empty array if not available.
+ */
+export function getJsxRenderingUsages(dataStructure) {
+    return dataStructure.jsxRenderingUsages ?? [];
+}
+//# sourceMappingURL=SerializableDataStructure.js.map

package/packages/ai/src/lib/worker/SerializableDataStructure.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SerializableDataStructure.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/worker/SerializableDataStructure.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EACL,sBAAsB,EACtB,oBAAoB,GACrB,MAAM,0BAA0B,CAAC;AA+JlC;;;;;;;;;;;;;;GAcG;AAEH,MAAM,UAAU,2BAA2B,CACzC,aAAwC,EACxC,YAAoB;IAEpB,OAAO,aAAa,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CACtD,sBAAsB,CAAC,YAAY,EAAE,GAAG,CAAC,IAAI,CAAC,CAC/C,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,aAAwC,EACxC,UAA+D,EAAE;IAEjE,MAAM,EAAE,YAAY,EAAE,cAAc,GAAG,KAAK,EAAE,GAAG,OAAO,CAAC;IAEzD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,gBAAgB;QAChB,OAAO,cAAc;YACnB,CAAC,CAAC,aAAa,CAAC,YAAY,CAAC,qBAAqB;YAClD,CAAC,CAAC,aAAa,CAAC,YAAY,CAAC,SAAS,CAAC;IAC3C,CAAC;IAED,2CAA2C;IAC3C,MAAM,WAAW,GAAG,oBAAoB,CACtC,aAAa,CAAC,eAAe,EAC7B,YAAY,CACb,CAAC;IACF,MAAM,cAAc,GAAG,WAAW;QAChC,CAAC,CAAC,aAAa,CAAC,eAAe,CAAC,WAAW,CAAC;QAC5C,CAAC,CAAC,SAAS,CAAC;IAEd,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,cAAc;QACnB,CAAC,CAAC,cAAc,CAAC,qBAAqB;QACtC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,aAAwC,EACxC,UAA+D,EAAE;IAEjE,MAAM,EAAE,YAAY,EAAE,cAAc,GAAG,KAAK,EAAE,GAAG,OAAO,CAAC;IAEzD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,gBAAgB;QAChB,OAAO,cAAc;YACnB,CAAC,CAAC,aAAa,CAAC,YAAY,CAAC,uBAAuB;YACpD,CAAC,CAAC,aAAa,CAAC,YAAY,CAAC,WAAW,CAAC;IAC7C,CAAC;IAED,2CAA2C;IAC3C,MAAM,WAAW,GAAG,oBAAoB,CACtC,aAAa,CAAC,eAAe,EAC7B,YAAY,CACb,CAAC;IACF,MAAM,cAAc,GAAG,WAAW;QAChC,CAAC,CAAC,aAAa,CAAC,eAAe,CAAC,WAAW,CAAC;QAC5C,CAAC,CAAC,SAAS,CAAC;IAEd,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,cAAc;QACnB,CAAC,CAAC,cAAc,CAAC,uBAAuB;QACxC,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,aAAwC,EACxC,YAAqB;IAErB,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,aAAa,CAAC,YAAY,CAAC,kBAAkB,CAAC;IACvD,CAAC;IAED,2CAA2C;IAC3C,MAAM,WAAW,GAAG,oBAAoB,CACtC,aAAa,CAAC,eAAe,EAC7B,YAAY,CACb,CAAC;IACF,MAAM,cAAc,GAAG,WAAW;QAChC,CAAC,CAAC,aAAa,CAAC,eAAe,CAAC,WAAW,CAAC;QAC5C,CAAC,CAAC,SAAS,CAAC;IAEd,OAAO,cAAc,EAAE,kBAAkB,IAAI,EAAE,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,aAAwC,EACxC,YAAqB;IAErB,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,aAAa,CAAC,YAAY,CAAC,mBAAmB,CAAC;IACxD,CAAC;IAED,2CAA2C;IAC3C,MAAM,WAAW,GAAG,oBAAoB,CACtC,aAAa,CAAC,eAAe,EAC7B,YAAY,CACb,CAAC;IACF,MAAM,cAAc,GAAG,WAAW;QAChC,CAAC,CAAC,aAAa,CAAC,eAAe,CAAC,WAAW,CAAC;QAC5C,CAAC,CAAC,SAAS,CAAC;IAEd,OAAO,cAAc,EAAE,mBAAmB,IAAI,EAAE,CAAC;AACnD,CAAC;AAED,MAAM,UAAU,+BAA+B,CAC7C,aAAwC;IAExC,OAAO,aAAa,CAAC,4BAA4B,CAAC;AACpD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAClC,aAAwC;IAExC,OAAO,aAAa,CAAC,iBAAiB,IAAI,EAAE,CAAC;AAC/C,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,qBAAqB,CACnC,aAAwC;IAExC,OAAO,aAAa,CAAC,kBAAkB,IAAI,EAAE,CAAC;AAChD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CACrC,aAAwC;IAExC,OAAO,aAAa,CAAC,oBAAoB,IAAI,EAAE,CAAC;AAClD,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,gCAAgC,CAC9C,aAAwC;IAExC,OAAO,aAAa,CAAC,6BAA6B,IAAI,EAAE,CAAC;AAC3D,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,qBAAqB,CACnC,aAAwC;IAExC,OAAO,aAAa,CAAC,kBAAkB,IAAI,EAAE,CAAC;AAChD,CAAC"}

package/packages/ai/src/lib/worker/analyzeScopeWorker.js

@@ -0,0 +1,158 @@
+/**
+ * Worker thread for analyzeScope
+ *
+ * This worker maintains a long-lived ProjectAnalyzer and cached FileAnalyzer
+ * to avoid expensive re-initialization on every call.
+ */
+import { FileAnalyzer, ProjectAnalyzer } from "../../../../../packages/analyze/index.js";
+import { analyzeScopeLocal } from "../analyzeScope.js";
+import { resetScopeDataStructureMetrics, } from "../dataStructure/ScopeDataStructure.js";
+import { setThreadContext } from "../../../../../packages/utils/index.js";
+import { threadId } from 'worker_threads';
+import v8 from 'v8';
+// Set thread context for logging
+setThreadContext(`<worker-${threadId}>`);
+// Long-lived state within the worker
+let projectAnalyzer = null;
+let projectAnalyzerKey = null;
+let fileAnalyzer = null;
+let fileAnalyzerPath = null;
+// Memory tracking
+let entitiesProcessed = 0;
+let lastMemoryLogAt = 0;
+const MEMORY_LOG_INTERVAL = 100; // Log every N entities
+const MEMORY_LOG_TIME_INTERVAL = 60000; // Or every 60 seconds
+/**
+ * Get memory stats for logging
+ */
+function getWorkerMemoryStats() {
+    const mem = process.memoryUsage();
+    const heapStats = v8.getHeapStatistics();
+    return {
+        rss: `${Math.round(mem.rss / 1024 / 1024)}MB`,
+        heapUsed: `${Math.round(mem.heapUsed / 1024 / 1024)}MB`,
+        heapTotal: `${Math.round(mem.heapTotal / 1024 / 1024)}MB`,
+        external: `${Math.round(mem.external / 1024 / 1024)}MB`,
+        // V8 heap stats give more detail
+        v8TotalHeapSize: `${Math.round(heapStats.total_heap_size / 1024 / 1024)}MB`,
+        v8UsedHeapSize: `${Math.round(heapStats.used_heap_size / 1024 / 1024)}MB`,
+        v8HeapSizeLimit: `${Math.round(heapStats.heap_size_limit / 1024 / 1024)}MB`,
+        v8MallocedMemory: `${Math.round(heapStats.malloced_memory / 1024 / 1024)}MB`,
+    };
+}
+/**
+ * Get ts.Program stats for understanding TypeScript memory usage
+ */
+function getTsProgramStats(pa) {
+    const sourceFiles = pa.program.getSourceFiles();
+    const projectSourceFiles = sourceFiles.filter((sf) => !sf.fileName.includes('node_modules'));
+    const nodeModulesFiles = sourceFiles.filter((sf) => sf.fileName.includes('node_modules'));
+    // Estimate AST size by sampling text length (rough proxy)
+    let totalProjectTextLength = 0;
+    for (const sf of projectSourceFiles) {
+        totalProjectTextLength += sf.getFullText().length;
+    }
+    return {
+        totalSourceFiles: sourceFiles.length,
+        projectSourceFiles: projectSourceFiles.length,
+        nodeModulesFiles: nodeModulesFiles.length,
+        projectTextSizeMB: `${(totalProjectTextLength / 1024 / 1024).toFixed(1)}MB`,
+        // Note: Actual AST memory is typically 10-20x the source text size
+        estimatedAstSizeMB: `${((totalProjectTextLength * 15) / 1024 / 1024).toFixed(0)}MB`,
+    };
+}
+/**
+ * Write a heap snapshot for detailed profiling.
+ * The snapshot can be loaded in Chrome DevTools for analysis.
+ *
+ * TODO: Once we understand memory usage, change this back to env-var controlled
+ */
+const HEAP_SNAPSHOT_MODE = 'off';
+function maybeWriteHeapSnapshot(label) {
+    if (HEAP_SNAPSHOT_MODE === 'off')
+        return;
+    try {
+        const filename = v8.writeHeapSnapshot();
+        console.log(`CodeYam Worker: Heap snapshot written (${label}): ${filename}`);
+    }
+    catch (error) {
+        console.error('CodeYam Worker: Failed to write heap snapshot:', error);
+    }
+}
+/**
+ * Main worker function called by Piscina
+ */
+export default async function runAnalyzeScope(input) {
+    // Reset metrics counters at the start of each entity analysis to prevent
+    // false "infinite loop" detection when processing multiple entities
+    resetScopeDataStructureMetrics();
+    try {
+        const { entity, project, dirPath, filePath, file, scope, model } = input;
+        // Create or reuse ProjectAnalyzer
+        const paKey = `${dirPath}`;
+        if (!projectAnalyzer || projectAnalyzerKey !== paKey) {
+            const memBefore = process.memoryUsage();
+            console.log('CodeYam Worker: Creating ProjectAnalyzer (ts.Program)...', {
+                memBefore: getWorkerMemoryStats(),
+            });
+            projectAnalyzer = ProjectAnalyzer.from({ project, dirPath });
+            projectAnalyzerKey = paKey;
+            const memAfter = process.memoryUsage();
+            const tsProgramStats = getTsProgramStats(projectAnalyzer);
+            console.log('CodeYam Worker: ProjectAnalyzer created', {
+                memAfter: getWorkerMemoryStats(),
+                rssDelta: `+${Math.round((memAfter.rss - memBefore.rss) / 1024 / 1024)}MB`,
+                heapDelta: `+${Math.round((memAfter.heapUsed - memBefore.heapUsed) / 1024 / 1024)}MB`,
+                tsProgramStats,
+            });
+            // Optionally write heap snapshot after ts.Program creation
+            maybeWriteHeapSnapshot('after-ts-program');
+        }
+        // Create or reuse FileAnalyzer
+        if (!fileAnalyzer || fileAnalyzerPath !== filePath) {
+            fileAnalyzer = new FileAnalyzer({
+                projectAnalyzer,
+                file,
+            });
+            fileAnalyzerPath = filePath;
+        }
+        // Reconstruct the scope with proper structure
+        const fullScope = {
+            ...scope,
+            // dataStructure will be created during analyzeScopeLocal
+        };
+        // Run analyzeScopeLocal - this will populate fullScope.dataStructure
+        const dataStructure = await analyzeScopeLocal({
+            entity,
+            fileAnalyzer,
+            scope: fullScope,
+            model,
+        });
+        // Extract the serializable data structure
+        if (!dataStructure) {
+            throw new Error(`CodeYam Worker: analyzeScopeLocal returned null for ${entity.name} in ${entity.filePath}`);
+        }
+        // Periodic memory logging
+        entitiesProcessed++;
+        const now = Date.now();
+        const shouldLogMemory = entitiesProcessed % MEMORY_LOG_INTERVAL === 0 ||
+            now - lastMemoryLogAt > MEMORY_LOG_TIME_INTERVAL;
+        if (shouldLogMemory) {
+            lastMemoryLogAt = now;
+            console.log('CodeYam Worker: Memory stats', {
+                entitiesProcessed,
+                ...getWorkerMemoryStats(),
+            });
+            // Optionally write periodic heap snapshots
+            if (HEAP_SNAPSHOT_MODE === 'periodic' && entitiesProcessed % 500 === 0) {
+                maybeWriteHeapSnapshot(`after-${entitiesProcessed}-entities`);
+            }
+        }
+        return { dataStructure: dataStructure.toSerializable() };
+    }
+    catch (error) {
+        console.error(`CodeYam Worker: Fatal error analyzing ${input.entity.name} in ${input.entity.filePath}:`, error);
+        throw error; // Re-throw so Piscina can handle it
+    }
+}
+//# sourceMappingURL=analyzeScopeWorker.js.map

package/packages/ai/src/lib/worker/analyzeScopeWorker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"analyzeScopeWorker.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/worker/analyzeScopeWorker.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EACL,8BAA8B,GAE/B,MAAM,qCAAqC,CAAC;AAG7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,MAAM,IAAI,CAAC;AAEpB,iCAAiC;AACjC,gBAAgB,CAAC,WAAW,QAAQ,GAAG,CAAC,CAAC;AAoBzC,qCAAqC;AACrC,IAAI,eAAe,GAA2B,IAAI,CAAC;AACnD,IAAI,kBAAkB,GAAkB,IAAI,CAAC;AAE7C,IAAI,YAAY,GAAwB,IAAI,CAAC;AAC7C,IAAI,gBAAgB,GAAkB,IAAI,CAAC;AAE3C,kBAAkB;AAClB,IAAI,iBAAiB,GAAG,CAAC,CAAC;AAC1B,IAAI,eAAe,GAAG,CAAC,CAAC;AACxB,MAAM,mBAAmB,GAAG,GAAG,CAAC,CAAC,uBAAuB;AACxD,MAAM,wBAAwB,GAAG,KAAM,CAAC,CAAC,sBAAsB;AAE/D;;GAEG;AACH,SAAS,oBAAoB;IAC3B,MAAM,GAAG,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAClC,MAAM,SAAS,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC;IACzC,OAAO;QACL,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI;QAC7C,QAAQ,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI;QACvD,SAAS,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI;QACzD,QAAQ,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI;QACvD,iCAAiC;QACjC,eAAe,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,eAAe,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI;QAC3E,cAAc,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,cAAc,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI;QACzE,eAAe,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,eAAe,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI;QAC3E,gBAAgB,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,eAAe,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI;KAC7E,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,EAAmB;IAC5C,MAAM,WAAW,GAAG,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;IAChD,MAAM,kBAAkB,GAAG,WAAW,CAAC,MAAM,CAC3C,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,cAAc,CAAC,CAC9C,CAAC;IACF,MAAM,gBAAgB,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CACjD,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,cAAc,CAAC,CACrC,CAAC;IAEF,0DAA0D;IAC1D,IAAI,sBAAsB,GAAG,CAAC,CAAC;IAC/B,KAAK,MAAM,EAAE,IAAI,kBAAkB,EAAE,CAAC;QACpC,sBAAsB,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC;IACpD,CAAC;IAED,OAAO;QACL,gBAAgB,EAAE,WAAW,CAAC,MAAM;QACpC,kBAAkB,EAAE,kBAAkB,CAAC,MAAM;QAC7C,gBAAgB,EAAE,gBAAgB,CAAC,MAAM;QACzC,iBAAiB,EAAE,GAAG,CAAC,sBAAsB,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;QAC3E,mEAAmE;QACnE,kBAAkB,EAAE,GAAG,CAAC,CAAC,sBAAsB,GAAG,EAAE,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;KACpF,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,kBAAkB,GAAgC,KAAK,CAAC;AAE9D,SAAS,sBAAsB,CAAC,KAAa;IAC3C,IAAI,kBAAkB,KAAK,KAAK;QAAE,OAAO;IAEzC,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC;QACxC,OAAO,CAAC,GAAG,CACT,0CAA0C,KAAK,MAAM,QAAQ,EAAE,CAChE,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,gDAAgD,EAAE,KAAK,CAAC,CAAC;IACzE,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,OAAO,CAAC,KAAK,UAAU,eAAe,CAC3C,KAA8B;IAE9B,yEAAyE;IACzE,oEAAoE;IACpE,8BAA8B,EAAE,CAAC;IAEjC,IAAI,CAAC;QACH,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;QAEzE,kCAAkC;QAClC,MAAM,KAAK,GAAG,GAAG,OAAO,EAAE,CAAC;QAC3B,IAAI,CAAC,eAAe,IAAI,kBAAkB,KAAK,KAAK,EAAE,CAAC;YACrD,MAAM,SAAS,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;YACxC,OAAO,CAAC,GAAG,CAAC,0DAA0D,EAAE;gBACtE,SAAS,EAAE,oBAAoB,EAAE;aAClC,CAAC,CAAC;YAEH,eAAe,GAAG,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7D,kBAAkB,GAAG,KAAK,CAAC;YAE3B,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;YACvC,MAAM,cAAc,GAAG,iBAAiB,CAAC,eAAe,CAAC,CAAC;YAC1D,OAAO,CAAC,GAAG,CAAC,yCAAyC,EAAE;gBACrD,QAAQ,EAAE,oBAAoB,EAAE;gBAChC,QAAQ,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI;gBAC1E,SAAS,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,QAAQ,GAAG,SAAS,CAAC,QAAQ,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,IAAI;gBACrF,cAAc;aACf,CAAC,CAAC;YAEH,2DAA2D;YAC3D,sBAAsB,CAAC,kBAAkB,CAAC,CAAC;QAC7C,CAAC;QAED,+BAA+B;QAC/B,IAAI,CAAC,YAAY,IAAI,gBAAgB,KAAK,QAAQ,EAAE,CAAC;YACnD,YAAY,GAAG,IAAI,YAAY,CAAC;gBAC9B,eAAe;gBACf,IAAI;aACL,CAAC,CAAC;YACH,gBAAgB,GAAG,QAAQ,CAAC;QAC9B,CAAC;QAED,8CAA8C;QAC9C,MAAM,SAAS,GAAc;YAC3B,GAAG,KAAK;YACR,yDAAyD;SAC1D,CAAC;QAEF,qEAAqE;QACrE,MAAM,aAAa,GAAG,MAAM,iBAAiB,CAAC;YAC5C,MAAM;YACN,YAAY;YACZ,KAAK,EAAE,SAAS;YAChB,KAAK;SACN,CAAC,CAAC;QAEH,0CAA0C;QAC1C,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CACb,uDAAuD,MAAM,CAAC,IAAI,OAAO,MAAM,CAAC,QAAQ,EAAE,CAC3F,CAAC;QACJ,CAAC;QAED,0BAA0B;QAC1B,iBAAiB,EAAE,CAAC;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,eAAe,GACnB,iBAAiB,GAAG,mBAAmB,KAAK,CAAC;YAC7C,GAAG,GAAG,eAAe,GAAG,wBAAwB,CAAC;QAEnD,IAAI,eAAe,EAAE,CAAC;YACpB,eAAe,GAAG,GAAG,CAAC;YACtB,OAAO,CAAC,GAAG,CAAC,8BAA8B,EAAE;gBAC1C,iBAAiB;gBACjB,GAAG,oBAAoB,EAAE;aAC1B,CAAC,CAAC;YAEH,2CAA2C;YAC3C,IAAI,kBAAkB,KAAK,UAAU,IAAI,iBAAiB,GAAG,GAAG,KAAK,CAAC,EAAE,CAAC;gBACvE,sBAAsB,CAAC,SAAS,iBAAiB,WAAW,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAED,OAAO,EAAE,aAAa,EAAE,aAAa,CAAC,cAAc,EAAE,EAAE,CAAC;IAC3D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CACX,yCAAyC,KAAK,CAAC,MAAM,CAAC,IAAI,OAAO,KAAK,CAAC,MAAM,CAAC,QAAQ,GAAG,EACzF,KAAK,CACN,CAAC;QACF,MAAM,KAAK,CAAC,CAAC,oCAAoC;IACnD,CAAC;AACH,CAAC"}

package/packages/ai/src/lib/worker/analyzeScopeWorkerPaths.js

@@ -0,0 +1,23 @@
+import { dirname, join } from 'path';
+import { fileURLToPath } from 'url';
+/**
+ * This module breaks out the path configuration for analyzeScope.ts
+ * mainly so we can mock it out for Jest, which hates import.meta.url
+ */
+export default function analyzeScopeWorkerPaths() {
+    // Only import Node.js modules when actually running in Node.js
+    // This allows the file to be bundled for browser without errors
+    if (typeof process !== 'undefined' &&
+        process.versions &&
+        process.versions.node) {
+        const __dirname = dirname(fileURLToPath(import.meta.url));
+        return {
+            WORKER_PATH: join(__dirname, 'analyzeScopeWorker.js'),
+        };
+    }
+    // Return empty path for browser environments (should never be called)
+    return {
+        WORKER_PATH: '',
+    };
+}
+//# sourceMappingURL=analyzeScopeWorkerPaths.js.map

package/packages/ai/src/lib/worker/analyzeScopeWorkerPaths.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"analyzeScopeWorkerPaths.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/worker/analyzeScopeWorkerPaths.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AAEpC;;;GAGG;AACH,MAAM,CAAC,OAAO,UAAU,uBAAuB;IAC7C,+DAA+D;IAC/D,gEAAgE;IAChE,IACE,OAAO,OAAO,KAAK,WAAW;QAC9B,OAAO,CAAC,QAAQ;QAChB,OAAO,CAAC,QAAQ,CAAC,IAAI,EACrB,CAAC;QACD,MAAM,SAAS,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QAC1D,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,SAAS,EAAE,uBAAuB,CAAC;SACtD,CAAC;IACJ,CAAC;IAED,sEAAsE;IACtE,OAAO;QACL,WAAW,EAAE,EAAE;KAChB,CAAC;AACJ,CAAC"}

package/packages/ai/src/lib/worker/fuzzyMatchFunctionName.js

@@ -0,0 +1,64 @@
+/**
+ * Fuzzy matching for function names to handle generics and nested calls.
+ *
+ * IMPORTANT: This is a simplified version for LOOKUP ONLY, not for building/computing.
+ *
+ * The original ScopeDataStructure uses sophisticated logic (splitOutsideParenthesesAndArrays,
+ * functionOrScopeName) when COMPUTING results. That complex logic ensures correct behavior
+ * for edge cases like nested arrays, complex paths, etc.
+ *
+ * This simplified version is used AFTER computation, to LOOKUP pre-computed results in the
+ * serialized data structure. Since toSerializable() already ran the sophisticated logic,
+ * we only need simple string matching to find the right key.
+ *
+ * The architecture:
+ * - Build time (ScopeDataStructure): Sophisticated matching computes results
+ * - Runtime (SerializableDataStructure helpers): Simple matching looks up results
+ * - Tests verify: Both produce identical results
+ *
+ * DO NOT replace the original logic in ScopeDataStructure with this simplified version.
+ *
+ * Examples this handles:
+ * - "useLoaderData" matches "useLoaderData<typeof loader>"
+ * - "supabase.from" matches "supabase.from(\"table\")"
+ * - "foo" matches "foo<T>"
+ * - "EventSource" matches "EventSource('/api/events')"
+ *
+ * @param searchName - The name to search for (e.g., "useLoaderData")
+ * @param candidateName - The name to match against (e.g., "useLoaderData<typeof loader>")
+ * @returns true if the names match (with fuzzy matching)
+ */
+export function fuzzyMatchFunctionName(searchName, candidateName) {
+    // Exact match
+    if (candidateName === searchName) {
+        return true;
+    }
+    // Strip generics: "useLoaderData<typeof loader>" -> "useLoaderData"
+    if (candidateName.split('<')[0] === searchName) {
+        return true;
+    }
+    // Strip function call arguments: "EventSource('/api/events')" -> "EventSource"
+    if (candidateName.split('(')[0] === searchName) {
+        return true;
+    }
+    // Match partial path: "from" matches "supabase.from"
+    if (candidateName === searchName.split('.')[0]) {
+        return true;
+    }
+    // Match partial path with generics stripped
+    if (candidateName.split('<')[0] === searchName.split('.')[0]) {
+        return true;
+    }
+    return false;
+}
+/**
+ * Find a key in a record that fuzzy-matches the given name.
+ *
+ * @param record - Record to search in
+ * @param searchName - Name to search for
+ * @returns The matching key, or undefined if not found
+ */
+export function findFuzzyMatchingKey(record, searchName) {
+    return Object.keys(record).find((key) => fuzzyMatchFunctionName(searchName, key));
+}
+//# sourceMappingURL=fuzzyMatchFunctionName.js.map

package/packages/ai/src/lib/worker/fuzzyMatchFunctionName.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fuzzyMatchFunctionName.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/worker/fuzzyMatchFunctionName.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,MAAM,UAAU,sBAAsB,CACpC,UAAkB,EAClB,aAAqB;IAErB,cAAc;IACd,IAAI,aAAa,KAAK,UAAU,EAAE,CAAC;QACjC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,oEAAoE;IACpE,IAAI,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,EAAE,CAAC;QAC/C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,+EAA+E;IAC/E,IAAI,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,EAAE,CAAC;QAC/C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,qDAAqD;IACrD,IAAI,aAAa,KAAK,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,4CAA4C;IAC5C,IAAI,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,oBAAoB,CAClC,MAAyB,EACzB,UAAkB;IAElB,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CACtC,sBAAsB,CAAC,UAAU,EAAE,GAAG,CAAC,CACxC,CAAC;AACJ,CAAC"}

package/packages/ai/src/lib/wrapperDetection/detectWrapperRequirements.js

@@ -0,0 +1,103 @@
+import ts from 'typescript';
+import detectUseContextCalls from "./patterns/detectUseContextCalls.js";
+import detectThrowOnMissingPatterns from "./patterns/detectThrowOnMissingPatterns.js";
+import { KNOWN_LIBRARY_HOOKS } from "./knownLibraryHooks.js";
+/**
+ * Detects wrapper requirements for an entity by analyzing AST patterns.
+ *
+ * Detects:
+ * 1. Direct useContext calls - components/hooks calling useContext(SomeContext)
+ * 2. Throw-on-missing patterns - explicit "must be used within XProvider" errors
+ * 3. Known library hooks - hooks like useQuery, useDispatch that require providers
+ */
+export default function detectWrapperRequirements({ entity, fileAnalyzer, }) {
+    const sourceFile = fileAnalyzer.sourceFile;
+    if (!sourceFile) {
+        return [];
+    }
+    // Find the entity's node in the AST
+    const entityNode = findEntityNode(sourceFile, entity.name);
+    if (!entityNode) {
+        return [];
+    }
+    const allRequirements = [];
+    // 1. Detect useContext calls
+    const contextCalls = detectUseContextCalls(entityNode, entity);
+    allRequirements.push(...contextCalls);
+    // 2. Detect throw-on-missing patterns
+    const throwPatterns = detectThrowOnMissingPatterns(entityNode);
+    allRequirements.push(...throwPatterns);
+    // 3. Detect known library hooks from imports
+    const libraryHooks = detectKnownLibraryHooks(entity);
+    allRequirements.push(...libraryHooks);
+    // Deduplicate by hookName + contextName
+    return deduplicateRequirements(allRequirements);
+}
+/**
+ * Find the AST node for the entity (function, component, hook)
+ */
+function findEntityNode(sourceFile, entityName) {
+    let foundNode;
+    function visit(node) {
+        if (foundNode)
+            return;
+        // Function declaration: function Foo() {}
+        if (ts.isFunctionDeclaration(node) && node.name?.text === entityName) {
+            foundNode = node;
+            return;
+        }
+        // Variable declaration: const Foo = () => {}
+        if (ts.isVariableDeclaration(node) && ts.isIdentifier(node.name)) {
+            if (node.name.text === entityName) {
+                foundNode = node;
+                return;
+            }
+        }
+        // Class declaration: class Foo {}
+        if (ts.isClassDeclaration(node) && node.name?.text === entityName) {
+            foundNode = node;
+            return;
+        }
+        ts.forEachChild(node, visit);
+    }
+    visit(sourceFile);
+    return foundNode;
+}
+/**
+ * Detect known library hooks from entity's nodeModuleImports
+ */
+function detectKnownLibraryHooks(entity) {
+    const requirements = [];
+    const nodeModuleImports = entity.metadata?.nodeModuleImports ?? {};
+    for (const [modulePath, imports] of Object.entries(nodeModuleImports)) {
+        for (const imp of imports) {
+            const hookInfo = KNOWN_LIBRARY_HOOKS[imp.name];
+            // Check if the hook is from the expected package
+            if (hookInfo && modulePath.includes(hookInfo.package)) {
+                requirements.push({
+                    importPath: modulePath,
+                    hookName: imp.name,
+                    patternType: 'library-hook',
+                    confidence: 'high',
+                    suggestedProvider: hookInfo.provider,
+                    libraryPackage: hookInfo.package,
+                });
+            }
+        }
+    }
+    return requirements;
+}
+/**
+ * Deduplicate requirements by hookName + contextName/libraryPackage
+ */
+function deduplicateRequirements(requirements) {
+    const seen = new Set();
+    return requirements.filter((req) => {
+        const key = `${req.hookName}:${req.contextName ?? req.libraryPackage ?? ''}`;
+        if (seen.has(key))
+            return false;
+        seen.add(key);
+        return true;
+    });
+}
+//# sourceMappingURL=detectWrapperRequirements.js.map

package/packages/ai/src/lib/wrapperDetection/detectWrapperRequirements.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"detectWrapperRequirements.js","sourceRoot":"","sources":["../../../../../../../packages/ai/src/lib/wrapperDetection/detectWrapperRequirements.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,YAAY,CAAC;AAI5B,OAAO,qBAAqB,MAAM,kCAAkC,CAAC;AACrE,OAAO,4BAA4B,MAAM,yCAAyC,CAAC;AACnF,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAO1D;;;;;;;GAOG;AACH,MAAM,CAAC,OAAO,UAAU,yBAAyB,CAAC,EAChD,MAAM,EACN,YAAY,GACkB;IAC9B,MAAM,UAAU,GAAG,YAAY,CAAC,UAAU,CAAC;IAC3C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,oCAAoC;IACpC,MAAM,UAAU,GAAG,cAAc,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IAC3D,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,eAAe,GAAyB,EAAE,CAAC;IAEjD,6BAA6B;IAC7B,MAAM,YAAY,GAAG,qBAAqB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC/D,eAAe,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;IAEtC,sCAAsC;IACtC,MAAM,aAAa,GAAG,4BAA4B,CAAC,UAAU,CAAC,CAAC;IAC/D,eAAe,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;IAEvC,6CAA6C;IAC7C,MAAM,YAAY,GAAG,uBAAuB,CAAC,MAAM,CAAC,CAAC;IACrD,eAAe,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;IAEtC,wCAAwC;IACxC,OAAO,uBAAuB,CAAC,eAAe,CAAC,CAAC;AAClD,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CACrB,UAAyB,EACzB,UAAkB;IAElB,IAAI,SAA8B,CAAC;IAEnC,SAAS,KAAK,CAAC,IAAa;QAC1B,IAAI,SAAS;YAAE,OAAO;QAEtB,0CAA0C;QAC1C,IAAI,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;YACrE,SAAS,GAAG,IAAI,CAAC;YACjB,OAAO;QACT,CAAC;QAED,6CAA6C;QAC7C,IAAI,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACjE,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAClC,SAAS,GAAG,IAAI,CAAC;gBACjB,OAAO;YACT,CAAC;QACH,CAAC;QAED,kCAAkC;QAClC,IAAI,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;YAClE,SAAS,GAAG,IAAI,CAAC;YACjB,OAAO;QACT,CAAC;QAED,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,CAAC;IAClB,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,SAAS,uBAAuB,CAAC,MAAc;IAC7C,MAAM,YAAY,GAAyB,EAAE,CAAC;IAC9C,MAAM,iBAAiB,GAAG,MAAM,CAAC,QAAQ,EAAE,iBAAiB,IAAI,EAAE,CAAC;IAEnE,KAAK,MAAM,CAAC,UAAU,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACtE,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC/C,iDAAiD;YACjD,IAAI,QAAQ,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBACtD,YAAY,CAAC,IAAI,CAAC;oBAChB,UAAU,EAAE,UAAU;oBACtB,QAAQ,EAAE,GAAG,CAAC,IAAI;oBAClB,WAAW,EAAE,cAAc;oBAC3B,UAAU,EAAE,MAAM;oBAClB,iBAAiB,EAAE,QAAQ,CAAC,QAAQ;oBACpC,cAAc,EAAE,QAAQ,CAAC,OAAO;iBACjC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,SAAS,uBAAuB,CAC9B,YAAkC;IAElC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,OAAO,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE;QACjC,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,cAAc,IAAI,EAAE,EAAE,CAAC;QAC7E,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,OAAO,KAAK,CAAC;QAChC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACd,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;AACL,CAAC"}