@codeyam/codeyam-cli 0.1.0-staging.1669d45 → 0.1.0-staging.323686

package/packages/ai/src/lib/astScopes/processExpression.js

@@ -6,6 +6,7 @@ import { methodRegistry, ArrayPushSemantics } from "./methodSemantics.js";
 import { isArithmeticOperator, isAssignmentOperator, isBitwiseCompoundOperator, isComparisonOperator, isDefinedType, isNumericCompoundOperator, leftOrRightType, unwrapExpression, } from "./sharedPatterns.js";
 import { processBindingPattern } from "./processBindings.js";
 import { extractConditionalEffectsFromTernary, findUseStateSetters, } from "./conditionalEffectsExtractor.js";
+import { detectArrayDerivedPattern } from "./arrayDerivationDetector.js";
 /**
  * Extracts the component name from a JSX element.
  * Returns null for intrinsic elements (div, span, etc.) since we only care about
@@ -273,6 +274,123 @@ function getSourceLocation(node, sourceFile) {
             : codeSnippet,
     };
 }
+/**
+ * Extracts the root array path from an expression that ends with .map().
+ * Handles chained methods like .filter().map(), .slice().map(), etc.
+ *
+ * Examples:
+ * - items.map(...) → "items"
+ * - data.users.map(...) → "data.users"
+ * - items.filter(...).map(...) → "items"
+ * - items.slice(0, 5).map(...) → "items"
+ */
+function extractArrayPathFromMapCall(expr, sourceFile) {
+    // Walk up the chain to find the root array
+    let current = expr.expression;
+    while (ts.isPropertyAccessExpression(current)) {
+        const methodName = current.name.getText(sourceFile);
+        // Common array methods that return arrays (so we keep going up)
+        const arrayReturningMethods = [
+            'map',
+            'filter',
+            'slice',
+            'concat',
+            'flat',
+            'flatMap',
+            'reverse',
+            'sort',
+            'toReversed',
+            'toSorted',
+            'toSpliced',
+        ];
+        if (arrayReturningMethods.includes(methodName)) {
+            const objectExpr = current.expression;
+            // If the object is a call expression (chained method), keep going
+            if (ts.isCallExpression(objectExpr)) {
+                current = objectExpr.expression;
+            }
+            else {
+                // Found the root - it's an identifier or property access
+                const path = StructuredPath.fromNode(objectExpr, sourceFile);
+                return path ? path.toString() : null;
+            }
+        }
+        else {
+            // Not an array method we recognize
+            break;
+        }
+    }
+    return null;
+}
+/**
+ * Extracts JSX rendering usages from a JSX expression.
+ * Detects:
+ * - array.map() calls → 'array-map' type
+ * - string interpolations (identifiers/property access) → 'text-interpolation' type
+ *
+ * Recursively searches inside && chains and ternary expressions.
+ *
+ * @param expr The expression inside {expr}
+ * @param context The analysis context
+ */
+function extractJsxRenderingUsage(expr, context) {
+    const unwrapped = unwrapExpression(expr);
+    const sourceLocation = getSourceLocation(expr, context.sourceFile);
+    // Detect array.map() calls
+    if (ts.isCallExpression(unwrapped)) {
+        const calleeExpr = unwrapped.expression;
+        if (ts.isPropertyAccessExpression(calleeExpr)) {
+            const methodName = calleeExpr.name.getText(context.sourceFile);
+            if (methodName === 'map') {
+                const arrayPath = extractArrayPathFromMapCall(unwrapped, context.sourceFile);
+                if (arrayPath) {
+                    context.addJsxRenderingUsage({
+                        path: arrayPath,
+                        renderingType: 'array-map',
+                        valueType: 'array',
+                        sourceLocation,
+                    });
+                }
+            }
+        }
+    }
+    // Detect simple string interpolations: {title} or {user.name}
+    else if (ts.isIdentifier(unwrapped) ||
+        ts.isPropertyAccessExpression(unwrapped)) {
+        const path = StructuredPath.fromNode(unwrapped, context.sourceFile);
+        if (path) {
+            const pathStr = path.toString();
+            const typeInfo = context.getTypeInfo(path);
+            // Only track as text interpolation if it's a string type
+            // Check for 'string' type, or types that contain 'string' (but not 'string[]')
+            if (typeInfo === 'string' ||
+                (typeInfo &&
+                    typeInfo.includes('string') &&
+                    !typeInfo.includes('string[]'))) {
+                context.addJsxRenderingUsage({
+                    path: pathStr,
+                    renderingType: 'text-interpolation',
+                    valueType: 'string',
+                    sourceLocation,
+                });
+            }
+        }
+    }
+    // Recursively search inside && chains: {showList && items.map(...)}
+    else if (ts.isBinaryExpression(unwrapped) &&
+        unwrapped.operatorToken.kind === ts.SyntaxKind.AmpersandAmpersandToken) {
+        // Check the right side of the && chain (where .map() typically appears)
+        const rightSide = unwrapExpression(unwrapped.right);
+        extractJsxRenderingUsage(rightSide, context);
+        // Also check nested && chains on the left
+        extractJsxRenderingUsage(unwrapped.left, context);
+    }
+    // Recursively search inside ternaries: {isEmpty ? null : items.map(...)}
+    else if (ts.isConditionalExpression(unwrapped)) {
+        extractJsxRenderingUsage(unwrapped.whenTrue, context);
+        extractJsxRenderingUsage(unwrapped.whenFalse, context);
+    }
+}
 /**
  * Counts the number of conditions in an && chain (excluding JSX consequence)
  */
@@ -330,6 +448,9 @@ function extractConditionalsFromJsx(node, context, parentConditions = []) {
         // Process JSX expressions: {expr}
         if (ts.isJsxExpression(child) && child.expression) {
             const expr = unwrapExpression(child.expression);
+            // Extract JSX rendering usages (array.map, text interpolation)
+            // This handles direct usages like {items.map(...)} or {user.name}
+            extractJsxRenderingUsage(expr, context);
             // If the expression is an && chain, extract its conditional usages
             if (ts.isBinaryExpression(expr) &&
                 expr.operatorToken.kind === ts.SyntaxKind.AmpersandAmpersandToken) {
@@ -568,8 +689,42 @@ function extractConditionalsFromJsx(node, context, parentConditions = []) {
         }
         // Recursively process nested JSX elements - Fix 32: pass parent conditions
         else if (ts.isJsxElement(child)) {
+            // Check if this is a user-defined component (vs intrinsic element like div)
+            // If it's a component AND there are parent conditions, record the gating conditions
+            const componentName = getComponentNameFromJsx(child);
+            if (componentName && parentConditions.length > 0) {
+                for (const condition of parentConditions) {
+                    context.addChildBoundaryGatingCondition(componentName, {
+                        path: condition.path,
+                        conditionType: 'truthiness',
+                        location: 'ternary',
+                        sourceLocation: condition.sourceLocation,
+                        controlsJsxRendering: true,
+                        isNegated: condition.isNegated,
+                    });
+                }
+            }
             extractConditionalsFromJsx(child, context, parentConditions);
         }
+        // Handle self-closing JSX elements (e.g., <ScenarioViewer />)
+        else if (ts.isJsxSelfClosingElement(child)) {
+            // Check if this is a user-defined component (vs intrinsic element like div)
+            // If it's a component AND there are parent conditions, record the gating conditions
+            const componentName = getComponentNameFromJsx(child);
+            if (componentName && parentConditions.length > 0) {
+                for (const condition of parentConditions) {
+                    context.addChildBoundaryGatingCondition(componentName, {
+                        path: condition.path,
+                        conditionType: 'truthiness',
+                        location: 'ternary',
+                        sourceLocation: condition.sourceLocation,
+                        controlsJsxRendering: true,
+                        isNegated: condition.isNegated,
+                    });
+                }
+            }
+            // Self-closing elements have no children, so no recursion needed
+        }
         // Recursively process nested JSX fragments - Fix 32: pass parent conditions
         else if (ts.isJsxFragment(child)) {
             extractConditionalsFromJsx(child, context, parentConditions);
@@ -640,12 +795,33 @@ export function extractConditionalUsage(condition, context, location, options =
             return;
         }
         // Handle binary expressions with || (logical OR)
-        // OR breaks the chain - each side is independent
+        // When OR is inside an && chain, we need to continue chain tracking
+        // and mark conditions as OR alternatives
         if (ts.isBinaryExpression(unwrapped) &&
             unwrapped.operatorToken.kind === ts.SyntaxKind.BarBarToken) {
-            // Both sides of || are independent conditional checks (no chain tracking)
-            extractWithChainTracking(unwrapped.left, null, false);
-            extractWithChainTracking(unwrapped.right, null, false);
+            if (chainInfo) {
+                // We're inside an && chain - continue tracking but mark as OR alternatives
+                // Generate an orGroupId so conditions from both sides can be grouped
+                const orGroupId = chainInfo.currentOrGroupId ?? `or_${crypto.randomUUID().slice(0, 8)}`;
+                // Process left side with OR group tracking
+                const leftChainInfo = {
+                    ...chainInfo,
+                    currentOrGroupId: orGroupId,
+                };
+                extractWithChainTracking(unwrapped.left, leftChainInfo, false);
+                // Process right side with same OR group
+                // Note: we use leftChainInfo's currentPosition which may have been updated
+                const rightChainInfo = {
+                    ...leftChainInfo,
+                    currentPosition: chainInfo.currentPosition,
+                };
+                extractWithChainTracking(unwrapped.right, rightChainInfo, false);
+            }
+            else {
+                // Not inside a chain - OR breaks into independent conditional checks
+                extractWithChainTracking(unwrapped.left, null, false);
+                extractWithChainTracking(unwrapped.right, null, false);
+            }
             return;
         }
         // Handle comparison operators (===, !==, <, >, <=, >=)
@@ -671,7 +847,7 @@ export function extractConditionalUsage(condition, context, location, options =
                 return literalValue;
             };
             // Helper to add a condition
-            const addCondition = (path, conditionType, comparedValues, requiredValue) => {
+            const addCondition = (path, conditionType, comparedValues, requiredValue, sourceExpr) => {
                 const usage = {
                     path,
                     conditionType,
@@ -681,6 +857,16 @@ export function extractConditionalUsage(condition, context, location, options =
                     isNegated,
                     controlsJsxRendering,
                 };
+                // Check for inline array-derived patterns (.length) on the source expression
+                if (sourceExpr) {
+                    const arrayDerived = detectArrayDerivedPattern(sourceExpr);
+                    if (arrayDerived) {
+                        usage.derivedFrom = {
+                            operation: arrayDerived.operation,
+                            sourcePath: arrayDerived.sourcePath,
+                        };
+                    }
+                }
                 // Add chain info if part of a compound conditional
                 if (chainInfo) {
                     usage.chainId = chainInfo.chainId;
@@ -695,6 +881,9 @@ export function extractConditionalUsage(condition, context, location, options =
                         comparedValues,
                         isNegated,
                         requiredValue,
+                        ...(chainInfo.currentOrGroupId && {
+                            orGroupId: chainInfo.currentOrGroupId,
+                        }),
                     });
                 }
                 context.addConditionalUsage(usage);
@@ -702,21 +891,21 @@ export function extractConditionalUsage(condition, context, location, options =
             // Check if left is a variable and right is a literal
             if (leftPath && isLiteralExpression(unwrapped.right)) {
                 const literalValue = getLiteralValue(unwrapped.right, context);
-                addCondition(leftPath.toLeftHandSideString(), 'comparison', literalValue !== undefined ? [literalValue] : undefined, getRequiredValue(literalValue, isNegated));
+                addCondition(leftPath.toLeftHandSideString(), 'comparison', literalValue !== undefined ? [literalValue] : undefined, getRequiredValue(literalValue, isNegated), unwrapped.left);
                 return;
             }
             // Check if right is a variable and left is a literal
             if (rightPath && isLiteralExpression(unwrapped.left)) {
                 const literalValue = getLiteralValue(unwrapped.left, context);
-                addCondition(rightPath.toLeftHandSideString(), 'comparison', literalValue !== undefined ? [literalValue] : undefined, getRequiredValue(literalValue, isNegated));
+                addCondition(rightPath.toLeftHandSideString(), 'comparison', literalValue !== undefined ? [literalValue] : undefined, getRequiredValue(literalValue, isNegated), unwrapped.right);
                 return;
             }
             // Both sides are variables - record both as comparisons without specific values
             if (leftPath) {
-                addCondition(leftPath.toLeftHandSideString(), 'comparison');
+                addCondition(leftPath.toLeftHandSideString(), 'comparison', undefined, undefined, unwrapped.left);
             }
             if (rightPath) {
-                addCondition(rightPath.toLeftHandSideString(), 'comparison');
+                addCondition(rightPath.toLeftHandSideString(), 'comparison', undefined, undefined, unwrapped.right);
             }
             return;
         }
@@ -740,6 +929,15 @@ export function extractConditionalUsage(condition, context, location, options =
                 isNegated,
                 controlsJsxRendering,
             };
+            // Check for inline array-derived patterns (.some(), .every(), .includes(), .length)
+            // This populates derivedFrom so downstream code can resolve to the base array path
+            const arrayDerived = detectArrayDerivedPattern(unwrapped);
+            if (arrayDerived) {
+                usage.derivedFrom = {
+                    operation: arrayDerived.operation,
+                    sourcePath: arrayDerived.sourcePath,
+                };
+            }
             // Add chain info if part of a compound conditional
             if (chainInfo) {
                 usage.chainId = chainInfo.chainId;
@@ -754,6 +952,9 @@ export function extractConditionalUsage(condition, context, location, options =
                     conditionType: 'truthiness',
                     isNegated,
                     requiredValue: !isNegated,
+                    ...(chainInfo.currentOrGroupId && {
+                        orGroupId: chainInfo.currentOrGroupId,
+                    }),
                 });
             }
             context.addConditionalUsage(usage);
@@ -1229,12 +1430,28 @@ export function processExpression({ node, context, targetPath, typeHint, }) {
                     }
                 }
                 else if (operatorKind === ts.SyntaxKind.BarBarToken) {
-                    // For ||, also create an equivalence to the left side
+                    // For ||, create equivalences to BOTH sides
                     // This enables data flow tracing through fallback expressions
+                    // e.g., `const item = items.find(...) || null` should trace to both:
+                    //   - items[] (from the find result)
+                    //   - null (from the fallback)
                     if (targetPath) {
+                        // Process left side recursively to capture its full equivalency chain
+                        processExpression({
+                            node: unwrappedNode.left,
+                            context,
+                            targetPath,
+                        });
+                        // Process right side recursively for completeness
+                        processExpression({
+                            node: unwrappedNode.right,
+                            context,
+                            targetPath,
+                        });
+                        // Set resultPath to left side for type inference
                         resultPath = StructuredPath.fromNode(unwrappedNode.left, context.sourceFile);
                     }
-                    // Note: Unlike ??, we don't set targetPath when there's no target
+                    // Note: When there's no targetPath, we don't recursively process
                     // because || is often used in boolean contexts where the full expression matters
                 }
             }