@codeyam/codeyam-cli 0.1.0-staging.596f0eb → 0.1.0-staging.6e699e5

package/packages/analyze/src/lib/files/scenarios/enrichUnknownTypesFromSourceEquivalencies.js

@@ -0,0 +1,85 @@
+/**
+ * Enriches mocked dependency return value types using sourceEquivalencies.
+ *
+ * When a mocked dependency returns a value with an `unknown` type, and that value
+ * is passed to a child component, we can infer the type from the child's signature.
+ * The sourceEquivalencies track this data flow.
+ *
+ * Example:
+ * - trpc.quote.getQuoteQuestionAnswers.useQuery().functionCallReturnValue.data is `unknown`
+ * - sourceEquivalencies shows this data flows to QuoteHeaderExcel().signature[0].questionAnswers
+ * - QuoteHeaderExcel's signatureSchema says signature[0].questionAnswers is `array | undefined`
+ * - Therefore, data should be `array | undefined`
+ *
+ * Without this enrichment, fillInDirectSchemaGapsAndUnknowns will guess the type
+ * based on the property name, often incorrectly (e.g., 'data' becomes 'number'
+ * because pluralize.isPlural('data') returns true).
+ */
+export default function enrichUnknownTypesFromSourceEquivalencies(isolatedDependencySchemas, mergedDependencySchemas, mockedImports) {
+    // Build a map of mocked dependency schemas for quick lookup
+    const mockedSchemas = new Map();
+    for (const imp of mockedImports) {
+        if (!imp.isMocked || !imp.filePath)
+            continue;
+        const schema = mergedDependencySchemas?.[imp.filePath]?.[imp.name]?.returnValueSchema;
+        if (schema) {
+            mockedSchemas.set(`${imp.filePath}:${imp.name}`, schema);
+        }
+    }
+    if (mockedSchemas.size === 0)
+        return;
+    // Iterate over all sourceEquivalencies in all dependency schemas
+    for (const filePath in isolatedDependencySchemas) {
+        for (const entityName in isolatedDependencySchemas[filePath]) {
+            const depSchema = isolatedDependencySchemas[filePath][entityName];
+            const sourceEquivs = depSchema.sourceEquivalencies;
+            if (!sourceEquivs)
+                continue;
+            // For each sourceEquivalency: key is destination, value contains sources
+            for (const [destinationPath, sources] of Object.entries(sourceEquivs)) {
+                // Parse the destination to get the child entity name and signature path
+                // e.g., "QuoteHeaderExcel().signature[0].questionAnswers" ->
+                //   entityName: "QuoteHeaderExcel", sigPath: "signature[0].questionAnswers"
+                const destMatch = destinationPath.match(/^(\w+)\(\)\.(.+)$/);
+                if (!destMatch)
+                    continue;
+                const [, destEntityName, destSigPath] = destMatch;
+                // Find the child's signatureSchema to get the destination type
+                let destType;
+                for (const depFilePath in mergedDependencySchemas) {
+                    const childSchema = mergedDependencySchemas[depFilePath][destEntityName];
+                    if (childSchema?.signatureSchema?.[destSigPath]) {
+                        destType = childSchema.signatureSchema[destSigPath];
+                        break;
+                    }
+                }
+                if (!destType || destType === 'unknown')
+                    continue;
+                // Now check each source to see if it's a mocked dependency with unknown type
+                for (const source of sources) {
+                    const { scopeNodeName, schemaPath } = source;
+                    // Find which mocked dependency this belongs to
+                    for (const [key, schema] of mockedSchemas) {
+                        const [mockedFilePath, mockedName] = key.split(':');
+                        // Check if this source path is in the mocked dependency's schema
+                        // The schemaPath contains the full path like:
+                        // "trpc.quote.getQuoteQuestionAnswers.useQuery(...).functionCallReturnValue.data"
+                        // We need to find it in the returnValueSchema
+                        if (schema[schemaPath] === 'unknown') {
+                            // Found an unknown type that flows to a known type!
+                            schema[schemaPath] = destType;
+                        }
+                        // Also check without the functionCallReturnValue part
+                        // (some schemas might have the transformed path)
+                        const transformedPath = schemaPath.replace(/\.functionCallReturnValue/g, '');
+                        if (transformedPath !== schemaPath &&
+                            schema[transformedPath] === 'unknown') {
+                            schema[transformedPath] = destType;
+                        }
+                    }
+                }
+            }
+        }
+    }
+}
+//# sourceMappingURL=enrichUnknownTypesFromSourceEquivalencies.js.map

package/packages/analyze/src/lib/files/scenarios/enrichUnknownTypesFromSourceEquivalencies.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"enrichUnknownTypesFromSourceEquivalencies.js","sourceRoot":"","sources":["../../../../../../../../packages/analyze/src/lib/files/scenarios/enrichUnknownTypesFromSourceEquivalencies.ts"],"names":[],"mappings":"AAEA;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,CAAC,OAAO,UAAU,yCAAyC,CAC/D,yBAA6D,EAC7D,uBAA2D,EAC3D,aAAwE;IAExE,4DAA4D;IAC5D,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkC,CAAC;IAChE,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;QAChC,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,CAAC,GAAG,CAAC,QAAQ;YAAE,SAAS;QAC7C,MAAM,MAAM,GACV,uBAAuB,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,iBAAiB,CAAC;QACzE,IAAI,MAAM,EAAE,CAAC;YACX,aAAa,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAED,IAAI,aAAa,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO;IAErC,iEAAiE;IACjE,KAAK,MAAM,QAAQ,IAAI,yBAAyB,EAAE,CAAC;QACjD,KAAK,MAAM,UAAU,IAAI,yBAAyB,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7D,MAAM,SAAS,GAAG,yBAAyB,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,CAAC;YAClE,MAAM,YAAY,GAAG,SAAS,CAAC,mBAAmB,CAAC;YACnD,IAAI,CAAC,YAAY;gBAAE,SAAS;YAE5B,yEAAyE;YACzE,KAAK,MAAM,CAAC,eAAe,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;gBACtE,wEAAwE;gBACxE,6DAA6D;gBAC7D,4EAA4E;gBAC5E,MAAM,SAAS,GAAG,eAAe,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;gBAC7D,IAAI,CAAC,SAAS;oBAAE,SAAS;gBAEzB,MAAM,CAAC,EAAE,cAAc,EAAE,WAAW,CAAC,GAAG,SAAS,CAAC;gBAElD,+DAA+D;gBAC/D,IAAI,QAA4B,CAAC;gBACjC,KAAK,MAAM,WAAW,IAAI,uBAAuB,EAAE,CAAC;oBAClD,MAAM,WAAW,GACf,uBAAuB,CAAC,WAAW,CAAC,CAAC,cAAc,CAAC,CAAC;oBACvD,IAAI,WAAW,EAAE,eAAe,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;wBAChD,QAAQ,GAAG,WAAW,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;wBACpD,MAAM;oBACR,CAAC;gBACH,CAAC;gBAED,IAAI,CAAC,QAAQ,IAAI,QAAQ,KAAK,SAAS;oBAAE,SAAS;gBAElD,6EAA6E;gBAC7E,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;oBAC7B,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC;oBAE7C,+CAA+C;oBAC/C,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;wBAC1C,MAAM,CAAC,cAAc,EAAE,UAAU,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;wBAEpD,iEAAiE;wBACjE,8CAA8C;wBAC9C,kFAAkF;wBAClF,8CAA8C;wBAC9C,IAAI,MAAM,CAAC,UAAU,CAAC,KAAK,SAAS,EAAE,CAAC;4BACrC,oDAAoD;4BACpD,MAAM,CAAC,UAAU,CAAC,GAAG,QAAQ,CAAC;wBAChC,CAAC;wBAED,sDAAsD;wBACtD,iDAAiD;wBACjD,MAAM,eAAe,GAAG,UAAU,CAAC,OAAO,CACxC,4BAA4B,EAC5B,EAAE,CACH,CAAC;wBACF,IACE,eAAe,KAAK,UAAU;4BAC9B,MAAM,CAAC,eAAe,CAAC,KAAK,SAAS,EACrC,CAAC;4BACD,MAAM,CAAC,eAAe,CAAC,GAAG,QAAQ,CAAC;wBACrC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC"}