phantom-pr 0.4.19 → 0.5.1

package/dist/core/context/contextSelector.js

@@ -0,0 +1,293 @@
+/**
+ * Intelligent context selection using AST analysis.
+ * Minimizes token usage while maximizing test generation quality.
+ *
+ * Pure functions only - no file I/O (receives file contents as input).
+ */
+/** Known external packages that should never be included in context. */
+const EXTERNAL_PACKAGES = new Set([
+    'react',
+    'react-dom',
+    'next',
+    'next/router',
+    'next/link',
+    'next/image',
+    'next/head',
+    'next/script',
+    'lodash',
+    'underscore',
+    'axios',
+    'moment',
+    'dayjs',
+    'date-fns',
+    'classnames',
+    'clsx',
+    'uuid',
+    'zod',
+    'yup',
+    'formik',
+    'react-hook-form',
+    'react-query',
+    '@tanstack/react-query',
+    'swr',
+    'redux',
+    'react-redux',
+    '@reduxjs/toolkit',
+    'zustand',
+    'jotai',
+    'recoil',
+    'framer-motion',
+    'styled-components',
+    '@emotion/react',
+    '@emotion/styled',
+    'tailwindcss',
+    '@mui/material',
+    '@chakra-ui/react',
+    'antd',
+    'prop-types',
+]);
+/**
+ * Estimate token count from content length.
+ * Rough approximation: ~4 chars per token for English/code.
+ */
+function estimateTokens(content) {
+    return Math.ceil(content.length / 4);
+}
+/**
+ * Extract mocked module paths from a test file.
+ * Looks for jest.mock('path') patterns.
+ */
+function extractMockedModules(testContent) {
+    const mocked = new Set();
+    // jest.mock('...')
+    const jestMockRe = /jest\.mock\(\s*['"]([^'"]+)['"]/g;
+    for (const match of testContent.matchAll(jestMockRe)) {
+        const modulePath = match[1];
+        if (modulePath)
+            mocked.add(modulePath);
+    }
+    // vi.mock('...')
+    const viMockRe = /vi\.mock\(\s*['"]([^'"]+)['"]/g;
+    for (const match of testContent.matchAll(viMockRe)) {
+        const modulePath = match[1];
+        if (modulePath)
+            mocked.add(modulePath);
+    }
+    return mocked;
+}
+/**
+ * Check if a path is an external package (not a relative import).
+ */
+function isExternalPackage(importPath) {
+    // Relative imports start with . or /
+    if (importPath.startsWith('.') || importPath.startsWith('/')) {
+        return false;
+    }
+    // Check against known external packages
+    if (EXTERNAL_PACKAGES.has(importPath)) {
+        return true;
+    }
+    // Scoped packages like @org/package
+    if (importPath.startsWith('@')) {
+        const parts = importPath.split('/');
+        const scopedName = parts.slice(0, 2).join('/');
+        if (EXTERNAL_PACKAGES.has(scopedName)) {
+            return true;
+        }
+    }
+    // Non-relative, non-scoped package names are likely external
+    // But we can't be sure without checking node_modules, so be conservative
+    return !importPath.includes('/') && !importPath.startsWith('.');
+}
+/**
+ * Resolve a relative import path to match against availableFiles.
+ * Handles extensions and index files.
+ */
+function resolveImportPath(fromPath, importPath) {
+    if (!importPath.startsWith('.')) {
+        return [importPath];
+    }
+    // Get directory of the source file
+    const fromDir = fromPath.replace(/\/[^/]+$/, '');
+    // Join paths
+    let resolved = importPath;
+    if (importPath.startsWith('./')) {
+        resolved = `${fromDir}/${importPath.slice(2)}`;
+    }
+    else if (importPath.startsWith('../')) {
+        const parts = fromDir.split('/');
+        let upCount = 0;
+        let remaining = importPath;
+        while (remaining.startsWith('../')) {
+            upCount++;
+            remaining = remaining.slice(3);
+        }
+        const baseParts = parts.slice(0, -upCount);
+        resolved = [...baseParts, remaining].join('/');
+    }
+    // Normalize
+    resolved = resolved.replace(/\/+/g, '/').replace(/^\//, '');
+    // Generate candidates with different extensions
+    const exts = ['.ts', '.tsx', '.js', '.jsx', '.mjs', '.cjs', '.json'];
+    const candidates = [resolved];
+    // Add extension variants
+    if (!exts.some(ext => resolved.endsWith(ext))) {
+        for (const ext of exts) {
+            candidates.push(`${resolved}${ext}`);
+            candidates.push(`${resolved}/index${ext}`);
+        }
+    }
+    return candidates;
+}
+/**
+ * Check if a path looks like a type definition file.
+ */
+function isTypeDefinitionPath(path) {
+    const lower = path.toLowerCase();
+    return (lower.includes('/types') ||
+        lower.includes('.types.') ||
+        lower.includes('.d.ts') ||
+        lower.endsWith('types.ts') ||
+        lower.endsWith('types.tsx'));
+}
+/**
+ * Check if a path looks like a constants file.
+ */
+function isConstantsPath(path) {
+    const lower = path.toLowerCase();
+    return (lower.includes('/constants') ||
+        lower.includes('.constants.') ||
+        lower.endsWith('constants.ts') ||
+        lower.endsWith('constants.js'));
+}
+/**
+ * Select context files based on analysis and heuristics.
+ * Returns files sorted by priority with reasons for selection.
+ */
+export function selectContext(input) {
+    const selected = [];
+    let tokenBudgetUsed = 0;
+    // Track what we've included
+    const includedPaths = new Set();
+    // Get mocked modules from existing test
+    const mockedModules = input.existingTestContent
+        ? extractMockedModules(input.existingTestContent)
+        : new Set();
+    // Helper to add a file if within budget
+    const addFile = (path, priority, reason) => {
+        if (includedPaths.has(path))
+            return true;
+        const content = input.availableFiles.get(path);
+        if (!content)
+            return false;
+        const tokens = estimateTokens(content);
+        // Required files always included
+        if (priority !== 'required') {
+            if (tokenBudgetUsed + tokens > input.maxTokens) {
+                return false;
+            }
+        }
+        includedPaths.add(path);
+        tokenBudgetUsed += tokens;
+        selected.push({ path, content, priority, reason });
+        return true;
+    };
+    // 1. REQUIRED: Target file - always include
+    addFile(input.targetPath, 'required', 'Target file under test');
+    // 2. REQUIRED: Existing test file - if exists, include for pattern learning
+    if (input.existingTestPath && input.existingTestContent) {
+        addFile(input.existingTestPath, 'required', 'Existing test file for pattern learning');
+    }
+    // 3. Process child components - they should be mocked, not included
+    // But we note them as excluded for clarity
+    const childrenToMock = [];
+    for (const child of input.analysis.children) {
+        if (child.importPath && child.importPath !== 'unknown') {
+            childrenToMock.push(child.name);
+        }
+    }
+    // 4. Process external calls to find type definitions and constants
+    const typesNeeded = [];
+    const constantsNeeded = [];
+    for (const call of input.analysis.externalCalls) {
+        // Skip external packages
+        if (isExternalPackage(call.module))
+            continue;
+        // Skip mocked modules
+        if (mockedModules.has(call.module))
+            continue;
+        // Check if it's a type or constant
+        const candidates = resolveImportPath(input.targetPath, call.module);
+        for (const candidate of candidates) {
+            if (input.availableFiles.has(candidate)) {
+                if (isTypeDefinitionPath(candidate)) {
+                    typesNeeded.push(candidate);
+                }
+                else if (isConstantsPath(candidate)) {
+                    constantsNeeded.push(candidate);
+                }
+                break;
+            }
+        }
+    }
+    // 5. RECOMMENDED: Type definitions
+    const seenTypes = new Set();
+    for (const typePath of typesNeeded.sort()) {
+        if (seenTypes.has(typePath))
+            continue;
+        seenTypes.add(typePath);
+        addFile(typePath, 'recommended', 'Type definitions used by component');
+    }
+    // 6. RECOMMENDED: Constants
+    const seenConstants = new Set();
+    for (const constPath of constantsNeeded.sort()) {
+        if (seenConstants.has(constPath))
+            continue;
+        seenConstants.add(constPath);
+        addFile(constPath, 'recommended', 'Constants imported by component');
+    }
+    // 7. OPTIONAL: Other non-child, non-mocked, non-external imports
+    // Find all imports from target content
+    const importRe = /(?:import|from)\s+['"]([^'"]+)['"]/g;
+    const allImports = [];
+    for (const match of input.targetContent.matchAll(importRe)) {
+        const imp = match[1];
+        if (imp)
+            allImports.push(imp);
+    }
+    for (const imp of [...new Set(allImports)].sort()) {
+        // Skip external packages
+        if (isExternalPackage(imp))
+            continue;
+        // Skip mocked modules
+        if (mockedModules.has(imp))
+            continue;
+        // Skip child components (they should be mocked)
+        const isChildImport = input.analysis.children.some(c => c.importPath === imp || c.importPath.includes(imp));
+        if (isChildImport)
+            continue;
+        // Try to find the file
+        const candidates = resolveImportPath(input.targetPath, imp);
+        for (const candidate of candidates) {
+            if (input.availableFiles.has(candidate) && !includedPaths.has(candidate)) {
+                // Skip if already processed as type or constant
+                if (seenTypes.has(candidate) || seenConstants.has(candidate))
+                    continue;
+                addFile(candidate, 'optional', `Imported module: ${imp}`);
+                break;
+            }
+        }
+    }
+    // Sort by priority order: required → recommended → optional
+    // Then alphabetically within each priority
+    const priorityOrder = { required: 0, recommended: 1, optional: 2 };
+    selected.sort((a, b) => {
+        const pa = priorityOrder[a.priority];
+        const pb = priorityOrder[b.priority];
+        if (pa !== pb)
+            return pa - pb;
+        return a.path.localeCompare(b.path);
+    });
+    return selected;
+}
+//# sourceMappingURL=contextSelector.js.map

package/dist/core/context/contextSelector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextSelector.js","sourceRoot":"","sources":["../../../src/core/context/contextSelector.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAuBH,wEAAwE;AACxE,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC;IAChC,OAAO;IACP,WAAW;IACX,MAAM;IACN,aAAa;IACb,WAAW;IACX,YAAY;IACZ,WAAW;IACX,aAAa;IACb,QAAQ;IACR,YAAY;IACZ,OAAO;IACP,QAAQ;IACR,OAAO;IACP,UAAU;IACV,YAAY;IACZ,MAAM;IACN,MAAM;IACN,KAAK;IACL,KAAK;IACL,QAAQ;IACR,iBAAiB;IACjB,aAAa;IACb,uBAAuB;IACvB,KAAK;IACL,OAAO;IACP,aAAa;IACb,kBAAkB;IAClB,SAAS;IACT,OAAO;IACP,QAAQ;IACR,eAAe;IACf,mBAAmB;IACnB,gBAAgB;IAChB,iBAAiB;IACjB,aAAa;IACb,eAAe;IACf,kBAAkB;IAClB,MAAM;IACN,YAAY;CACb,CAAC,CAAC;AAEH;;;GAGG;AACH,SAAS,cAAc,CAAC,OAAe;IACrC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACvC,CAAC;AAED;;;GAGG;AACH,SAAS,oBAAoB,CAAC,WAAmB;IAC/C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;IAEjC,mBAAmB;IACnB,MAAM,UAAU,GAAG,kCAAkC,CAAC;IACtD,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QACrD,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,UAAU;YAAE,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC;IAED,iBAAiB;IACjB,MAAM,QAAQ,GAAG,gCAAgC,CAAC;IAClD,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnD,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,UAAU;YAAE,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,UAAkB;IAC3C,qCAAqC;IACrC,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC7D,OAAO,KAAK,CAAC;IACf,CAAC;IAED,wCAAwC;IACxC,IAAI,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;QACtC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,oCAAoC;IACpC,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC/B,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC/C,IAAI,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,6DAA6D;IAC7D,yEAAyE;IACzE,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;AAClE,CAAC;AAED;;;GAGG;AACH,SAAS,iBAAiB,CAAC,QAAgB,EAAE,UAAkB;IAC7D,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAChC,OAAO,CAAC,UAAU,CAAC,CAAC;IACtB,CAAC;IAED,mCAAmC;IACnC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;IAEjD,aAAa;IACb,IAAI,QAAQ,GAAG,UAAU,CAAC;IAC1B,IAAI,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QAChC,QAAQ,GAAG,GAAG,OAAO,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IACjD,CAAC;SAAM,IAAI,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QACxC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,SAAS,GAAG,UAAU,CAAC;QAC3B,OAAO,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACnC,OAAO,EAAE,CAAC;YACV,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC;QACD,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;QAC3C,QAAQ,GAAG,CAAC,GAAG,SAAS,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACjD,CAAC;IAED,YAAY;IACZ,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAE5D,gDAAgD;IAChD,MAAM,IAAI,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IACrE,MAAM,UAAU,GAAG,CAAC,QAAQ,CAAC,CAAC;IAE9B,yBAAyB;IACzB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC9C,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,GAAG,GAAG,EAAE,CAAC,CAAC;YACrC,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,SAAS,GAAG,EAAE,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;GAEG;AACH,SAAS,oBAAoB,CAAC,IAAY;IACxC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACjC,OAAO,CACL,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACxB,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;QACzB,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;QACvB,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC;QAC1B,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,CAC5B,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,IAAY;IACnC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACjC,OAAO,CACL,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC;QAC5B,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC;QAC7B,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC9B,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,CAC/B,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,KAA4B;IACxD,MAAM,QAAQ,GAA0B,EAAE,CAAC;IAC3C,IAAI,eAAe,GAAG,CAAC,CAAC;IAExB,4BAA4B;IAC5B,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IAExC,wCAAwC;IACxC,MAAM,aAAa,GAAG,KAAK,CAAC,mBAAmB;QAC7C,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,mBAAmB,CAAC;QACjD,CAAC,CAAC,IAAI,GAAG,EAAU,CAAC;IAEtB,wCAAwC;IACxC,MAAM,OAAO,GAAG,CACd,IAAY,EACZ,QAAyC,EACzC,MAAc,EACL,EAAE;QACX,IAAI,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QAEzC,MAAM,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO;YAAE,OAAO,KAAK,CAAC;QAE3B,MAAM,MAAM,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;QAEvC,iCAAiC;QACjC,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;YAC5B,IAAI,eAAe,GAAG,MAAM,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;gBAC/C,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QAED,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxB,eAAe,IAAI,MAAM,CAAC;QAC1B,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QACnD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IAEF,4CAA4C;IAC5C,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,UAAU,EAAE,wBAAwB,CAAC,CAAC;IAEhE,4EAA4E;IAC5E,IAAI,KAAK,CAAC,gBAAgB,IAAI,KAAK,CAAC,mBAAmB,EAAE,CAAC;QACxD,OAAO,CACL,KAAK,CAAC,gBAAgB,EACtB,UAAU,EACV,yCAAyC,CAC1C,CAAC;IACJ,CAAC;IAED,oEAAoE;IACpE,2CAA2C;IAC3C,MAAM,cAAc,GAAa,EAAE,CAAC;IACpC,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;QAC5C,IAAI,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACvD,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,mEAAmE;IACnE,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,MAAM,eAAe,GAAa,EAAE,CAAC;IAErC,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QAChD,yBAAyB;QACzB,IAAI,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC;YAAE,SAAS;QAE7C,sBAAsB;QACtB,IAAI,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC;YAAE,SAAS;QAE7C,mCAAmC;QACnC,MAAM,UAAU,GAAG,iBAAiB,CAAC,KAAK,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACpE,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,IAAI,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;gBACxC,IAAI,oBAAoB,CAAC,SAAS,CAAC,EAAE,CAAC;oBACpC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC9B,CAAC;qBAAM,IAAI,eAAe,CAAC,SAAS,CAAC,EAAE,CAAC;oBACtC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAClC,CAAC;gBACD,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAED,mCAAmC;IACnC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,KAAK,MAAM,QAAQ,IAAI,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;QAC1C,IAAI,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC;YAAE,SAAS;QACtC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxB,OAAO,CAAC,QAAQ,EAAE,aAAa,EAAE,oCAAoC,CAAC,CAAC;IACzE,CAAC;IAED,4BAA4B;IAC5B,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IACxC,KAAK,MAAM,SAAS,IAAI,eAAe,CAAC,IAAI,EAAE,EAAE,CAAC;QAC/C,IAAI,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC;YAAE,SAAS;QAC3C,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7B,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,iCAAiC,CAAC,CAAC;IACvE,CAAC;IAED,iEAAiE;IACjE,uCAAuC;IACvC,MAAM,QAAQ,GAAG,qCAAqC,CAAC;IACvD,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC3D,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACrB,IAAI,GAAG;YAAE,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,MAAM,GAAG,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;QAClD,yBAAyB;QACzB,IAAI,iBAAiB,CAAC,GAAG,CAAC;YAAE,SAAS;QAErC,sBAAsB;QACtB,IAAI,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,SAAS;QAErC,gDAAgD;QAChD,MAAM,aAAa,GAAG,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAChD,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,GAAG,IAAI,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CACxD,CAAC;QACF,IAAI,aAAa;YAAE,SAAS;QAE5B,uBAAuB;QACvB,MAAM,UAAU,GAAG,iBAAiB,CAAC,KAAK,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QAC5D,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,IAAI,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;gBACzE,gDAAgD;gBAChD,IAAI,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC;oBAAE,SAAS;gBAEvE,OAAO,CAAC,SAAS,EAAE,UAAU,EAAE,oBAAoB,GAAG,EAAE,CAAC,CAAC;gBAC1D,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAED,4DAA4D;IAC5D,2CAA2C;IAC3C,MAAM,aAAa,GAAG,EAAE,QAAQ,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;IACnE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACrB,MAAM,EAAE,GAAG,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QACrC,MAAM,EAAE,GAAG,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QACrC,IAAI,EAAE,KAAK,EAAE;YAAE,OAAO,EAAE,GAAG,EAAE,CAAC;QAC9B,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/dist/core/context/index.d.ts

@@ -0,0 +1,4 @@
+export type { ContextBudgets, ContextFile, ContextPack } from './types.js';
+export { buildContextPack } from './packer.js';
+export type { SelectedContextFile, ContextSelectionInput } from './contextSelector.js';
+export { selectContext } from './contextSelector.js';

package/dist/core/context/index.js

@@ -0,0 +1,3 @@
+export { buildContextPack } from './packer.js';
+export { selectContext } from './contextSelector.js';
+//# sourceMappingURL=index.js.map

package/dist/core/context/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/context/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAG/C,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC"}

package/dist/core/generator/enhancedContext.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Enhanced context building that incorporates AST analysis.
+ * Wraps existing context packer with analysis-aware selection.
+ *
+ * Maintains backward compatibility - new features are additive.
+ */
+import { type ComponentAnalysis } from '../analysis/index.js';
+import { type SelectedContextFile } from '../context/contextSelector.js';
+export type EnhancedContextOptions = {
+    rootAbs: string;
+    targetPathPosix: string;
+    existingTestPathPosix?: string;
+    maxTokens?: number;
+    enabled?: boolean;
+};
+export type EnhancedContextResult = {
+    analysis: ComponentAnalysis;
+    selectedFiles: SelectedContextFile[];
+    analysisJson: string;
+    enabled: boolean;
+};
+/**
+ * Build enhanced context using AST analysis.
+ * Returns analysis and selected files for inclusion in LLM prompt.
+ *
+ * @param opts - Context options
+ * @returns Enhanced context result with analysis and selected files
+ */
+export declare function buildEnhancedContext(opts: EnhancedContextOptions): Promise<EnhancedContextResult>;
+/**
+ * Format analysis for inclusion in LLM prompt.
+ * Returns a prompt section with key insights from the analysis.
+ */
+export declare function formatAnalysisForPrompt(analysis: ComponentAnalysis): string;

package/dist/core/generator/enhancedContext.js

@@ -0,0 +1,144 @@
+/**
+ * Enhanced context building that incorporates AST analysis.
+ * Wraps existing context packer with analysis-aware selection.
+ *
+ * Maintains backward compatibility - new features are additive.
+ */
+import fs from 'node:fs';
+import path from 'node:path';
+import { extractComponentAnalysis, emptyAnalysis } from '../analysis/index.js';
+import { selectContext } from '../context/contextSelector.js';
+import { stableStringify } from '../stableJson.js';
+/**
+ * Build enhanced context using AST analysis.
+ * Returns analysis and selected files for inclusion in LLM prompt.
+ *
+ * @param opts - Context options
+ * @returns Enhanced context result with analysis and selected files
+ */
+export async function buildEnhancedContext(opts) {
+    const enabled = opts.enabled ?? true;
+    if (!enabled) {
+        return {
+            analysis: emptyAnalysis(opts.targetPathPosix),
+            selectedFiles: [],
+            analysisJson: '{}',
+            enabled: false,
+        };
+    }
+    // Read target file
+    let targetContent = '';
+    try {
+        const targetAbs = path.resolve(opts.rootAbs, ...opts.targetPathPosix.split('/'));
+        targetContent = await fs.promises.readFile(targetAbs, 'utf8');
+    }
+    catch {
+        // If we can't read the target, return empty analysis
+        return {
+            analysis: emptyAnalysis(opts.targetPathPosix),
+            selectedFiles: [],
+            analysisJson: '{}',
+            enabled: true,
+        };
+    }
+    // Extract component analysis
+    const analysis = extractComponentAnalysis(opts.targetPathPosix, targetContent);
+    // Read existing test file if provided
+    let existingTestContent;
+    if (opts.existingTestPathPosix) {
+        try {
+            const testAbs = path.resolve(opts.rootAbs, ...opts.existingTestPathPosix.split('/'));
+            existingTestContent = await fs.promises.readFile(testAbs, 'utf8');
+        }
+        catch {
+            // Test file doesn't exist yet - that's fine
+        }
+    }
+    // For now, just use the target file in available files
+    // A full implementation would scan the directory, but we keep it simple
+    const availableFiles = new Map();
+    availableFiles.set(opts.targetPathPosix, targetContent);
+    if (existingTestContent && opts.existingTestPathPosix) {
+        availableFiles.set(opts.existingTestPathPosix, existingTestContent);
+    }
+    // Select context using analysis
+    const contextInput = {
+        targetPath: opts.targetPathPosix,
+        targetContent,
+        availableFiles,
+        analysis,
+        maxTokens: opts.maxTokens ?? 8000,
+    };
+    // Add optional properties only if defined
+    if (opts.existingTestPathPosix !== undefined) {
+        contextInput.existingTestPath = opts.existingTestPathPosix;
+    }
+    if (existingTestContent !== undefined) {
+        contextInput.existingTestContent = existingTestContent;
+    }
+    const selectedFiles = selectContext(contextInput);
+    // Create deterministic JSON representation
+    const analysisJson = stableStringify({
+        filePath: analysis.filePath,
+        branchCount: analysis.branches.length,
+        branches: analysis.branches.map(b => ({
+            line: b.line,
+            kind: b.kind,
+            condition: b.condition.slice(0, 50), // Truncate long conditions
+        })),
+        effectCount: analysis.effects.length,
+        effects: analysis.effects.map(e => ({
+            line: e.line,
+            kind: e.kind,
+            hasCleanup: e.hasCleanup,
+            depCount: e.dependencies.length,
+        })),
+        childCount: analysis.children.length,
+        children: analysis.children.map(c => ({
+            name: c.name,
+            importPath: c.importPath,
+            isConditional: c.isConditional,
+        })),
+    });
+    return {
+        analysis,
+        selectedFiles,
+        analysisJson,
+        enabled: true,
+    };
+}
+/**
+ * Format analysis for inclusion in LLM prompt.
+ * Returns a prompt section with key insights from the analysis.
+ */
+export function formatAnalysisForPrompt(analysis) {
+    if (analysis.branches.length === 0 &&
+        analysis.effects.length === 0 &&
+        analysis.children.length === 0) {
+        return '';
+    }
+    const lines = [];
+    lines.push('## PRE-ANALYZED STRUCTURE (use this to guide test coverage)');
+    if (analysis.branches.length > 0) {
+        lines.push(`- BRANCHES (${analysis.branches.length}): Test BOTH paths for each`);
+        for (const b of analysis.branches.slice(0, 5)) {
+            lines.push(`  - Line ${b.line}: ${b.kind} on \`${b.condition.slice(0, 40)}\``);
+        }
+    }
+    if (analysis.effects.length > 0) {
+        lines.push(`- EFFECTS (${analysis.effects.length}):`);
+        for (const e of analysis.effects) {
+            const cleanup = e.hasCleanup ? ' [HAS CLEANUP - test unmount]' : '';
+            lines.push(`  - Line ${e.line}: ${e.kind}${cleanup}`);
+        }
+    }
+    if (analysis.children.length > 0) {
+        lines.push(`- CHILDREN TO MOCK (${analysis.children.length}):`);
+        for (const c of analysis.children.slice(0, 5)) {
+            lines.push(`  - ${c.name} from '${c.importPath}'`);
+        }
+    }
+    lines.push('');
+    return lines.join('\n');
+}
+//# sourceMappingURL=enhancedContext.js.map

package/dist/core/generator/enhancedContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"enhancedContext.js","sourceRoot":"","sources":["../../../src/core/generator/enhancedContext.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,wBAAwB,EAAE,aAAa,EAA0B,MAAM,sBAAsB,CAAC;AACvG,OAAO,EAAE,aAAa,EAA4B,MAAM,+BAA+B,CAAC;AACxF,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAiBnD;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,IAA4B;IAE5B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC;IAErC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO;YACL,QAAQ,EAAE,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC;YAC7C,aAAa,EAAE,EAAE;YACjB,YAAY,EAAE,IAAI;YAClB,OAAO,EAAE,KAAK;SACf,CAAC;IACJ,CAAC;IAED,mBAAmB;IACnB,IAAI,aAAa,GAAG,EAAE,CAAC;IACvB,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;QACjF,aAAa,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAChE,CAAC;IAAC,MAAM,CAAC;QACP,qDAAqD;QACrD,OAAO;YACL,QAAQ,EAAE,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC;YAC7C,aAAa,EAAE,EAAE;YACjB,YAAY,EAAE,IAAI;YAClB,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,6BAA6B;IAC7B,MAAM,QAAQ,GAAG,wBAAwB,CAAC,IAAI,CAAC,eAAe,EAAE,aAAa,CAAC,CAAC;IAE/E,sCAAsC;IACtC,IAAI,mBAAuC,CAAC;IAC5C,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC/B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;YACrF,mBAAmB,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACpE,CAAC;QAAC,MAAM,CAAC;YACP,4CAA4C;QAC9C,CAAC;IACH,CAAC;IAED,uDAAuD;IACvD,wEAAwE;IACxE,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;IACjD,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,aAAa,CAAC,CAAC;IACxD,IAAI,mBAAmB,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;QACtD,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,qBAAqB,EAAE,mBAAmB,CAAC,CAAC;IACtE,CAAC;IAED,gCAAgC;IAChC,MAAM,YAAY,GAAwC;QACxD,UAAU,EAAE,IAAI,CAAC,eAAe;QAChC,aAAa;QACb,cAAc;QACd,QAAQ;QACR,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI;KAClC,CAAC;IAEF,0CAA0C;IAC1C,IAAI,IAAI,CAAC,qBAAqB,KAAK,SAAS,EAAE,CAAC;QAC7C,YAAY,CAAC,gBAAgB,GAAG,IAAI,CAAC,qBAAqB,CAAC;IAC7D,CAAC;IACD,IAAI,mBAAmB,KAAK,SAAS,EAAE,CAAC;QACtC,YAAY,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;IACzD,CAAC;IAED,MAAM,aAAa,GAAG,aAAa,CAAC,YAAY,CAAC,CAAC;IAElD,2CAA2C;IAC3C,MAAM,YAAY,GAAG,eAAe,CAAC;QACnC,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,WAAW,EAAE,QAAQ,CAAC,QAAQ,CAAC,MAAM;QACrC,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACpC,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,2BAA2B;SACjE,CAAC,CAAC;QACH,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM;QACpC,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAClC,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,UAAU,EAAE,CAAC,CAAC,UAAU;YACxB,QAAQ,EAAE,CAAC,CAAC,YAAY,CAAC,MAAM;SAChC,CAAC,CAAC;QACH,UAAU,EAAE,QAAQ,CAAC,QAAQ,CAAC,MAAM;QACpC,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACpC,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,UAAU,EAAE,CAAC,CAAC,UAAU;YACxB,aAAa,EAAE,CAAC,CAAC,aAAa;SAC/B,CAAC,CAAC;KACJ,CAAC,CAAC;IAEH,OAAO;QACL,QAAQ;QACR,aAAa;QACb,YAAY;QACZ,OAAO,EAAE,IAAI;KACd,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CAAC,QAA2B;IACjE,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC;QAC9B,QAAQ,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC;QAC7B,QAAQ,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,CAAC,IAAI,CAAC,6DAA6D,CAAC,CAAC;IAE1E,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,eAAe,QAAQ,CAAC,QAAQ,CAAC,MAAM,6BAA6B,CAAC,CAAC;QACjF,KAAK,MAAM,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YAC9C,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,SAAS,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;QACjF,CAAC;IACH,CAAC;IAED,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,KAAK,CAAC,IAAI,CAAC,cAAc,QAAQ,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC;QACtD,KAAK,MAAM,CAAC,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,+BAA+B,CAAC,CAAC,CAAC,EAAE,CAAC;YACpE,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,GAAG,OAAO,EAAE,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAED,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,uBAAuB,QAAQ,CAAC,QAAQ,CAAC,MAAM,IAAI,CAAC,CAAC;QAChE,KAAK,MAAM,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YAC9C,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,UAAU,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}