@figma/code-connect 1.1.4 → 1.2.1

package/dist/connect/wizard/prop_mapping.js

@@ -1,17 +1,30 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.extractSignatureAndGeneratePropMapping = exports.generatePropMapping = exports.buildMatchableNamesMap = exports.MatchableNameTypes = exports.generateValueMapping = exports.PROPERTY_PLACEHOLDER = void 0;
+exports.generatePropMapping = exports.buildMatchableNamesMap = exports.MatchableNameTypes = exports.generateValueMapping = exports.PROPERTY_PLACEHOLDER = void 0;
 const figma_rest_api_1 = require("../figma_rest_api");
 const fast_fuzzy_1 = require("fast-fuzzy");
-const logging_1 = require("../../common/logging");
 const intrinsics_1 = require("../../connect/intrinsics");
-const signature_extraction_1 = require("./signature_extraction");
 const create_1 = require("../../react/create");
+/**
+ * These thresholds were come up with by running the benchmarking
+ * and tweaking until high positives + acceptable false positives
+ * were achieved. It's worth repeating this process whenever making
+ * changes to the mapping algorithm.
+ */
+const MININUM_MATCH_SCORES = {
+    fuzzy: {
+        property: 0.65,
+        variantValue: 0.8,
+    },
+    embeddings: {
+        property: 0.84,
+        variantValue: 0.87,
+    },
+};
 /**
  * Used when we should output a placeholder for an unknown value in prop mapping.
  */
 exports.PROPERTY_PLACEHOLDER = 'PROPERTY_PLACEHOLDER';
-const PROP_MINIMUM_MATCH_THRESHOLD = 0.8;
 function generateValueMapping(propSignature, figmaPropDef) {
     const codeEnumOptions = propSignature.split(' | ').map((str) => str.substring(1, str.length - 1)); // remove quote marks
     const searcher = new fast_fuzzy_1.Searcher(figmaPropDef.variantOptions);
@@ -24,6 +37,11 @@ function generateValueMapping(propSignature, figmaPropDef) {
     }, {});
 }
 exports.generateValueMapping = generateValueMapping;
+function signatureIsJsxLike(signature) {
+    return (signature.includes('ElementType') ||
+        signature.includes('ReactElement') ||
+        signature.includes('ReactNode'));
+}
 /**
  * Attempts to create a mapping between a code prop and figma prop.
  * These props have been matched by name and aren't guaranteed to
@@ -65,42 +83,53 @@ function generateIntrinsic({ propSignature, figmaPropName: figmaPropNameWithNode
             };
         }
     }
+    if (signatureIsJsxLike(propSignature) &&
+        figmaPropDef.type === figma_rest_api_1.FigmaRestApi.ComponentPropertyType.InstanceSwap) {
+        return {
+            kind: intrinsics_1.IntrinsicKind.Instance,
+            args: {
+                figmaPropName,
+            },
+        };
+    }
     return null;
 }
 function stripNodeIdFromPropertyName(propertyName) {
     return propertyName.replace(/#\d+:\d+/, '');
 }
-const DELIMITERS_REGEX = /[\s-_]/g;
 var MatchableNameTypes;
 (function (MatchableNameTypes) {
     MatchableNameTypes[MatchableNameTypes["Property"] = 0] = "Property";
     MatchableNameTypes[MatchableNameTypes["VariantValue"] = 1] = "VariantValue";
     // ChildLayer, // TODO
 })(MatchableNameTypes || (exports.MatchableNameTypes = MatchableNameTypes = {}));
+const MATCHABLE_NAME_TYPES_PRIORITY = [
+    MatchableNameTypes.Property,
+    MatchableNameTypes.VariantValue,
+];
 /**
- * Builds a map of all properties and enum values, indexed by matchable name.
+ * Builds a map of all properties and enum values, indexed by name.
  * @param componentPropertyDefinitions
  * @returns A map of {name: values[]}. Each value is an array to avoid
  * collisions between properties / enum values
  */
 function buildMatchableNamesMap(componentPropertyDefinitions) {
     const matchableValues = {};
-    function add(matchableStr, definition) {
-        matchableValues[matchableStr] = matchableValues[matchableStr] || [];
-        matchableValues[matchableStr].push(definition);
+    function add(name, definition) {
+        matchableValues[name] = matchableValues[name] || [];
+        matchableValues[name].push(definition);
     }
     Object.entries(componentPropertyDefinitions || {}).forEach(([propName, propDef]) => {
-        const matchableStr = stripNodeIdFromPropertyName(propName).replace(DELIMITERS_REGEX, '');
-        add(matchableStr, {
+        const name = stripNodeIdFromPropertyName(propName);
+        add(name, {
             type: MatchableNameTypes.Property,
             name: propName,
         });
         if (propDef.type === figma_rest_api_1.FigmaRestApi.ComponentPropertyType.Variant) {
-            propDef.variantOptions?.forEach((variantOption) => {
-                const variantMatchableStr = variantOption.replace(DELIMITERS_REGEX, '');
-                add(variantMatchableStr, {
+            propDef.variantOptions?.forEach((variantValue) => {
+                add(variantValue, {
                     type: MatchableNameTypes.VariantValue,
-                    name: variantOption,
+                    name: variantValue,
                     variantProperty: propName,
                 });
             });
@@ -109,110 +138,86 @@ function buildMatchableNamesMap(componentPropertyDefinitions) {
     return matchableValues;
 }
 exports.buildMatchableNamesMap = buildMatchableNamesMap;
-function generatePropMapping({ componentPropertyDefinitions, signature, }) {
+function generatePropMapping({ componentPropertyDefinitions, signature, componentMatchResults, matchableNamesMap, }) {
     const propMapping = {};
-    const matchableNames = buildMatchableNamesMap(componentPropertyDefinitions);
-    const searchSpace = Object.keys(matchableNames);
+    const searchSpace = Object.keys(matchableNamesMap);
     const searcher = new fast_fuzzy_1.Searcher(searchSpace);
-    function findBestMatch(matchData, { score, where }) {
-        for (const nameMatches of matchData) {
-            if (nameMatches.score < score) {
-                return null;
+    let minimumMatchScores = MININUM_MATCH_SCORES.fuzzy;
+    const useEmbeddings = !!componentMatchResults;
+    if (useEmbeddings) {
+        minimumMatchScores = MININUM_MATCH_SCORES.embeddings;
+    }
+    function attemptGetIntrinsicForProp({ propMatch, propSignature, }) {
+        const itemsInPriorityOrder = [...matchableNamesMap[propMatch.item]].sort((a, b) => MATCHABLE_NAME_TYPES_PRIORITY.indexOf(a.type) -
+            MATCHABLE_NAME_TYPES_PRIORITY.indexOf(b.type));
+        for (const item of itemsInPriorityOrder) {
+            /**
+             * First, look for matching property names with compatible types
+             */
+            if (item.type === MatchableNameTypes.Property &&
+                propMatch.score > minimumMatchScores.property) {
+                const intrinsic = generateIntrinsic({
+                    propSignature,
+                    figmaPropName: item.name,
+                    figmaPropDef: componentPropertyDefinitions[item.name],
+                });
+                if (intrinsic) {
+                    return intrinsic;
+                }
+                /**
+                 * Then if no match AND a boolean prop, look for matching variant values, e.g:
+                 *
+                 * disabled: figma.enum('State', {
+                 *   Disabled: true,
+                 * })
+                 */
             }
-            const items = matchableNames[nameMatches.item];
-            const match = items.find(where);
-            if (match) {
-                return match;
+            else if (item.type === MatchableNameTypes.VariantValue &&
+                propSignature === 'false | true' &&
+                propMatch.score > minimumMatchScores.variantValue) {
+                return {
+                    kind: intrinsics_1.IntrinsicKind.Enum,
+                    args: {
+                        figmaPropName: item.variantProperty,
+                        valueMapping: {
+                            [item.name]: true,
+                        },
+                    },
+                };
             }
         }
         return null;
     }
+    /**
+     * Attempt to generate an intrinsic for a given property name by looking
+     * for name matches with a minimum threshold
+     *
+     * Embeddings are priorized over fuzzy matching if present (they
+     * may be missing if gated or missing permissions).
+     */
     for (const [propName, propSignatureWithOptionalModifier] of Object.entries(signature)) {
         const propSignature = propSignatureWithOptionalModifier.startsWith('?')
             ? propSignatureWithOptionalModifier.substring(1)
             : propSignatureWithOptionalModifier;
-        const results = searcher.search(propName, { returnMatchData: true });
-        if (results.length === 0) {
+        let matches = searcher.search(propName, { returnMatchData: true });
+        if (useEmbeddings) {
+            matches = componentMatchResults[propName];
+        }
+        if (matches.length === 0) {
             continue;
         }
-        /**
-         * First, look for matching property names with compatible types
-         */
-        const matchingProperty = findBestMatch(results, {
-            score: PROP_MINIMUM_MATCH_THRESHOLD,
-            where: (item) => item.type === MatchableNameTypes.Property,
-        });
-        if (matchingProperty) {
-            const { name: figmaPropName } = matchingProperty;
-            const intrinsic = generateIntrinsic({
+        for (const match of matches) {
+            const intrinsic = attemptGetIntrinsicForProp({
+                propMatch: match,
                 propSignature,
-                figmaPropName,
-                figmaPropDef: componentPropertyDefinitions[figmaPropName],
             });
             if (intrinsic) {
                 propMapping[propName] = intrinsic;
-            }
-        }
-        /**
-         * Then if no match AND a boolean prop, look for matching variant values, e.g:
-         *
-         * disabled: figma.enum('State', {
-         *   Disabled: true,
-         * })
-         */
-        if (!propMapping[propName] && propSignature === 'false | true') {
-            const matchingProperty = findBestMatch(results, {
-                score: PROP_MINIMUM_MATCH_THRESHOLD,
-                where: (item) => item.type === MatchableNameTypes.VariantValue,
-            });
-            if (matchingProperty) {
-                const { name: variantValue, variantProperty } = matchingProperty;
-                if (!variantProperty) {
-                    throw new Error('Expected variant property'); // satisfying TS
-                }
-                propMapping[propName] = {
-                    kind: intrinsics_1.IntrinsicKind.Enum,
-                    args: {
-                        figmaPropName: variantProperty,
-                        valueMapping: {
-                            [variantValue]: true,
-                        },
-                    },
-                };
+                break;
             }
         }
     }
     return propMapping;
 }
 exports.generatePropMapping = generatePropMapping;
-function extractSignatureAndGeneratePropMapping({ exportName, filepath, projectInfo, componentPropertyDefinitions, cmd, }) {
-    let signature;
-    try {
-        signature = (0, signature_extraction_1.extractSignature)({
-            nameToFind: exportName,
-            sourceFilePath: filepath,
-            projectInfo,
-        });
-        if (cmd.verbose && Object.keys(signature).length === 0) {
-            logging_1.logger.warn(`No TS signature found for "${exportName}" in ${filepath}`);
-        }
-    }
-    catch (e) {
-        if (cmd.verbose) {
-            logging_1.logger.warn(`Could not extract TS signature for "${exportName}" in ${filepath}`);
-        }
-        return {
-            propMapping: undefined,
-            signature: undefined,
-        };
-    }
-    return {
-        propMapping: generatePropMapping({
-            componentPropertyDefinitions,
-            signature,
-        }),
-        signature: signature,
-    };
-}
-exports.extractSignatureAndGeneratePropMapping = extractSignatureAndGeneratePropMapping;
 //# sourceMappingURL=prop_mapping.js.map

package/dist/connect/wizard/prop_mapping.js.map

@@ -1 +1 @@
-{"version":3,"file":"prop_mapping.js","sourceRoot":"","sources":["../../../src/connect/wizard/prop_mapping.ts"],"names":[],"mappings":";;;AAEA,sDAAgD;AAEhD,2CAAgD;AAEhD,kDAA6C;AAC7C,yDAAmG;AACnG,iEAAyD;AACzD,+CAAkD;AAElD;;GAEG;AACU,QAAA,oBAAoB,GAAG,sBAAsB,CAAA;AAE1D,MAAM,4BAA4B,GAAG,GAAG,CAAA;AAExC,SAAgB,oBAAoB,CAClC,aAAqB,EACrB,YAAsD;IAEtD,MAAM,eAAe,GAAG,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAA,CAAC,qBAAqB;IACvH,MAAM,QAAQ,GAAG,IAAI,qBAAQ,CAAC,YAAY,CAAC,cAAc,CAAC,CAAA;IAC1D,OAAO,eAAe,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,aAAa,EAAE,EAAE;QAC5D,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAA;QACzE,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,GAAG,EAAE,CAAC;YAC7C,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,aAAa,CAAA;QAC/C,CAAC;QACD,OAAO,YAAY,CAAA;IACrB,CAAC,EAAE,EAAkB,CAAC,CAAA;AACxB,CAAC;AAbD,oDAaC;AAED;;;;GAIG;AACH,SAAS,iBAAiB,CAAC,EACzB,aAAa,EACb,aAAa,EAAE,uBAAuB,EACtC,YAAY,GAKb;IACC,MAAM,aAAa,GAAG,2BAA2B,CAAC,uBAAuB,CAAC,CAAA;IAC1E,IAAI,aAAa,KAAK,QAAQ,IAAI,YAAY,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,IAAI,EAAE,CAAC;QAChG,OAAO;YACL,IAAI,EAAE,0BAAa,CAAC,MAAM;YAC1B,IAAI,EAAE;gBACJ,aAAa;aACd;SACF,CAAA;IACH,CAAC;IACD,IACE,aAAa,KAAK,cAAc;QAChC,CAAC,YAAY,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,OAAO;YAC/D,CAAC,YAAY,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,OAAO;gBAC/D,YAAY,CAAC,cAAc,EAAE,MAAM,KAAK,CAAC;gBACzC,YAAY,CAAC,cAAc,EAAE,KAAK,CAAC,sBAAa,CAAC,CAAC,CAAC,EACvD,CAAC;QACD,OAAO;YACL,IAAI,EAAE,0BAAa,CAAC,OAAO;YAC3B,IAAI,EAAE;gBACJ,aAAa;aACd;SACF,CAAA;IACH,CAAC;IACD,IACE,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC;QAC7B,YAAY,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,OAAO,EAChE,CAAC;QACD,MAAM,YAAY,GAAG,oBAAoB,CAAC,aAAa,EAAE,YAAY,CAAC,CAAA;QACtE,2CAA2C;QAC3C,IAAI,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzC,OAAO;gBACL,IAAI,EAAE,0BAAa,CAAC,IAAI;gBACxB,IAAI,EAAE;oBACJ,aAAa;oBACb,YAAY;iBACb;aACF,CAAA;QACH,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAA;AACb,CAAC;AAED,SAAS,2BAA2B,CAAC,YAAoB;IACvD,OAAO,YAAY,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAA;AAC7C,CAAC;AAED,MAAM,gBAAgB,GAAG,SAAS,CAAA;AAElC,IAAY,kBAIX;AAJD,WAAY,kBAAkB;IAC5B,mEAAQ,CAAA;IACR,2EAAY,CAAA;IACZ,sBAAsB;AACxB,CAAC,EAJW,kBAAkB,kCAAlB,kBAAkB,QAI7B;AAQD;;;;;GAKG;AACH,SAAgB,sBAAsB,CACpC,4BAAqF;IAErF,MAAM,eAAe,GAAoC,EAAE,CAAA;IAE3D,SAAS,GAAG,CAAC,YAAoB,EAAE,UAAyB;QAC1D,eAAe,CAAC,YAAY,CAAC,GAAG,eAAe,CAAC,YAAY,CAAC,IAAI,EAAE,CAAA;QACnE,eAAe,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IAChD,CAAC;IAED,MAAM,CAAC,OAAO,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,EAAE;QACjF,MAAM,YAAY,GAAG,2BAA2B,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAA;QAExF,GAAG,CAAC,YAAY,EAAE;YAChB,IAAI,EAAE,kBAAkB,CAAC,QAAQ;YACjC,IAAI,EAAE,QAAQ;SACf,CAAC,CAAA;QAEF,IAAI,OAAO,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,OAAO,EAAE,CAAC;YAChE,OAAO,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,aAAa,EAAE,EAAE;gBAChD,MAAM,mBAAmB,GAAG,aAAa,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAA;gBAEvE,GAAG,CAAC,mBAAmB,EAAE;oBACvB,IAAI,EAAE,kBAAkB,CAAC,YAAY;oBACrC,IAAI,EAAE,aAAa;oBACnB,eAAe,EAAE,QAAQ;iBAC1B,CAAC,CAAA;YACJ,CAAC,CAAC,CAAA;QACJ,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,OAAO,eAAe,CAAA;AACxB,CAAC;AAhCD,wDAgCC;AAED,SAAgB,mBAAmB,CAAC,EAClC,4BAA4B,EAC5B,SAAS,GAIV;IACC,MAAM,WAAW,GAAgB,EAAE,CAAA;IAEnC,MAAM,cAAc,GAAG,sBAAsB,CAAC,4BAA4B,CAAC,CAAA;IAE3E,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;IAC/C,MAAM,QAAQ,GAAG,IAAI,qBAAQ,CAAC,WAAW,CAAC,CAAA;IAE1C,SAAS,aAAa,CACpB,SAA8B,EAC9B,EAAE,KAAK,EAAE,KAAK,EAA8D;QAE5E,KAAK,MAAM,WAAW,IAAI,SAAS,EAAE,CAAC;YACpC,IAAI,WAAW,CAAC,KAAK,GAAG,KAAK,EAAE,CAAC;gBAC9B,OAAO,IAAI,CAAA;YACb,CAAC;YACD,MAAM,KAAK,GAAG,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,CAAA;YAC9C,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC/B,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,KAAK,CAAA;YACd,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAA;IACb,CAAC;IAED,KAAK,MAAM,CAAC,QAAQ,EAAE,iCAAiC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;QACtF,MAAM,aAAa,GAAG,iCAAiC,CAAC,UAAU,CAAC,GAAG,CAAC;YACrE,CAAC,CAAC,iCAAiC,CAAC,SAAS,CAAC,CAAC,CAAC;YAChD,CAAC,CAAC,iCAAiC,CAAA;QAErC,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAA;QAEpE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,SAAQ;QACV,CAAC;QAED;;WAEG;QACH,MAAM,gBAAgB,GAAG,aAAa,CAAC,OAAO,EAAE;YAC9C,KAAK,EAAE,4BAA4B;YACnC,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,kBAAkB,CAAC,QAAQ;SAC3D,CAAC,CAAA;QACF,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,gBAAgB,CAAA;YAEhD,MAAM,SAAS,GAAG,iBAAiB,CAAC;gBAClC,aAAa;gBACb,aAAa;gBACb,YAAY,EAAE,4BAA4B,CAAC,aAAa,CAAC;aAC1D,CAAC,CAAA;YACF,IAAI,SAAS,EAAE,CAAC;gBACd,WAAW,CAAC,QAAQ,CAAC,GAAG,SAAS,CAAA;YACnC,CAAC;QACH,CAAC;QAED;;;;;;WAMG;QACH,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,aAAa,KAAK,cAAc,EAAE,CAAC;YAC/D,MAAM,gBAAgB,GAAG,aAAa,CAAC,OAAO,EAAE;gBAC9C,KAAK,EAAE,4BAA4B;gBACnC,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,kBAAkB,CAAC,YAAY;aAC/D,CAAC,CAAA;YACF,IAAI,gBAAgB,EAAE,CAAC;gBACrB,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,GAAG,gBAAgB,CAAA;gBAChE,IAAI,CAAC,eAAe,EAAE,CAAC;oBACrB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAA,CAAC,gBAAgB;gBAC/D,CAAC;gBACD,WAAW,CAAC,QAAQ,CAAC,GAAG;oBACtB,IAAI,EAAE,0BAAa,CAAC,IAAI;oBACxB,IAAI,EAAE;wBACJ,aAAa,EAAE,eAAe;wBAC9B,YAAY,EAAE;4BACZ,CAAC,YAAY,CAAC,EAAE,IAAI;yBACrB;qBACF;iBACF,CAAA;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAA;AACpB,CAAC;AA7FD,kDA6FC;AAED,SAAgB,sCAAsC,CAAC,EACrD,UAAU,EACV,QAAQ,EACR,WAAW,EACX,4BAA4B,EAC5B,GAAG,GAOJ;IAIC,IAAI,SAAiC,CAAA;IAErC,IAAI,CAAC;QACH,SAAS,GAAG,IAAA,uCAAgB,EAAC;YAC3B,UAAU,EAAE,UAAU;YACtB,cAAc,EAAE,QAAQ;YACxB,WAAW;SACZ,CAAC,CAAA;QACF,IAAI,GAAG,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvD,gBAAM,CAAC,IAAI,CAAC,8BAA8B,UAAU,QAAQ,QAAQ,EAAE,CAAC,CAAA;QACzE,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAChB,gBAAM,CAAC,IAAI,CAAC,uCAAuC,UAAU,QAAQ,QAAQ,EAAE,CAAC,CAAA;QAClF,CAAC;QACD,OAAO;YACL,WAAW,EAAE,SAAS;YACtB,SAAS,EAAE,SAAS;SACrB,CAAA;IACH,CAAC;IAED,OAAO;QACL,WAAW,EAAE,mBAAmB,CAAC;YAC/B,4BAA4B;YAC5B,SAAS;SACV,CAAC;QACF,SAAS,EAAE,SAAS;KACrB,CAAA;AACH,CAAC;AA5CD,wFA4CC"}
+{"version":3,"file":"prop_mapping.js","sourceRoot":"","sources":["../../../src/connect/wizard/prop_mapping.ts"],"names":[],"mappings":";;;AACA,sDAAgD;AAEhD,2CAAqC;AACrC,yDAAiF;AACjF,+CAAkD;AAGlD;;;;;GAKG;AACH,MAAM,oBAAoB,GAAG;IAC3B,KAAK,EAAE;QACL,QAAQ,EAAE,IAAI;QACd,YAAY,EAAE,GAAG;KAClB;IACD,UAAU,EAAE;QACV,QAAQ,EAAE,IAAI;QACd,YAAY,EAAE,IAAI;KACnB;CACF,CAAA;AAED;;GAEG;AACU,QAAA,oBAAoB,GAAG,sBAAsB,CAAA;AAE1D,SAAgB,oBAAoB,CAClC,aAAqB,EACrB,YAAsD;IAEtD,MAAM,eAAe,GAAG,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAA,CAAC,qBAAqB;IACvH,MAAM,QAAQ,GAAG,IAAI,qBAAQ,CAAC,YAAY,CAAC,cAAc,CAAC,CAAA;IAC1D,OAAO,eAAe,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,aAAa,EAAE,EAAE;QAC5D,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAA;QACzE,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,GAAG,EAAE,CAAC;YAC7C,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,aAAa,CAAA;QAC/C,CAAC;QACD,OAAO,YAAY,CAAA;IACrB,CAAC,EAAE,EAAkB,CAAC,CAAA;AACxB,CAAC;AAbD,oDAaC;AAED,SAAS,kBAAkB,CAAC,SAAiB;IAC3C,OAAO,CACL,SAAS,CAAC,QAAQ,CAAC,aAAa,CAAC;QACjC,SAAS,CAAC,QAAQ,CAAC,cAAc,CAAC;QAClC,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC,CAChC,CAAA;AACH,CAAC;AAED;;;;GAIG;AACH,SAAS,iBAAiB,CAAC,EACzB,aAAa,EACb,aAAa,EAAE,uBAAuB,EACtC,YAAY,GAKb;IACC,MAAM,aAAa,GAAG,2BAA2B,CAAC,uBAAuB,CAAC,CAAA;IAC1E,IAAI,aAAa,KAAK,QAAQ,IAAI,YAAY,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,IAAI,EAAE,CAAC;QAChG,OAAO;YACL,IAAI,EAAE,0BAAa,CAAC,MAAM;YAC1B,IAAI,EAAE;gBACJ,aAAa;aACd;SACF,CAAA;IACH,CAAC;IACD,IACE,aAAa,KAAK,cAAc;QAChC,CAAC,YAAY,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,OAAO;YAC/D,CAAC,YAAY,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,OAAO;gBAC/D,YAAY,CAAC,cAAc,EAAE,MAAM,KAAK,CAAC;gBACzC,YAAY,CAAC,cAAc,EAAE,KAAK,CAAC,sBAAa,CAAC,CAAC,CAAC,EACvD,CAAC;QACD,OAAO;YACL,IAAI,EAAE,0BAAa,CAAC,OAAO;YAC3B,IAAI,EAAE;gBACJ,aAAa;aACd;SACF,CAAA;IACH,CAAC;IACD,IACE,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC;QAC7B,YAAY,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,OAAO,EAChE,CAAC;QACD,MAAM,YAAY,GAAG,oBAAoB,CAAC,aAAa,EAAE,YAAY,CAAC,CAAA;QACtE,2CAA2C;QAC3C,IAAI,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzC,OAAO;gBACL,IAAI,EAAE,0BAAa,CAAC,IAAI;gBACxB,IAAI,EAAE;oBACJ,aAAa;oBACb,YAAY;iBACb;aACF,CAAA;QACH,CAAC;IACH,CAAC;IACD,IACE,kBAAkB,CAAC,aAAa,CAAC;QACjC,YAAY,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,YAAY,EACrE,CAAC;QACD,OAAO;YACL,IAAI,EAAE,0BAAa,CAAC,QAAQ;YAC5B,IAAI,EAAE;gBACJ,aAAa;aACd;SACF,CAAA;IACH,CAAC;IACD,OAAO,IAAI,CAAA;AACb,CAAC;AAED,SAAS,2BAA2B,CAAC,YAAoB;IACvD,OAAO,YAAY,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAA;AAC7C,CAAC;AAED,IAAY,kBAIX;AAJD,WAAY,kBAAkB;IAC5B,mEAAQ,CAAA;IACR,2EAAY,CAAA;IACZ,sBAAsB;AACxB,CAAC,EAJW,kBAAkB,kCAAlB,kBAAkB,QAI7B;AAED,MAAM,6BAA6B,GAAG;IACpC,kBAAkB,CAAC,QAAQ;IAC3B,kBAAkB,CAAC,YAAY;CACvB,CAAA;AAQV;;;;;GAKG;AACH,SAAgB,sBAAsB,CACpC,4BAAqF;IAErF,MAAM,eAAe,GAAsB,EAAE,CAAA;IAE7C,SAAS,GAAG,CAAC,IAAY,EAAE,UAAyB;QAClD,eAAe,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,CAAA;QACnD,eAAe,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IACxC,CAAC;IAED,MAAM,CAAC,OAAO,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,EAAE;QACjF,MAAM,IAAI,GAAG,2BAA2B,CAAC,QAAQ,CAAC,CAAA;QAElD,GAAG,CAAC,IAAI,EAAE;YACR,IAAI,EAAE,kBAAkB,CAAC,QAAQ;YACjC,IAAI,EAAE,QAAQ;SACf,CAAC,CAAA;QAEF,IAAI,OAAO,CAAC,IAAI,KAAK,6BAAY,CAAC,qBAAqB,CAAC,OAAO,EAAE,CAAC;YAChE,OAAO,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE;gBAC/C,GAAG,CAAC,YAAY,EAAE;oBAChB,IAAI,EAAE,kBAAkB,CAAC,YAAY;oBACrC,IAAI,EAAE,YAAY;oBAClB,eAAe,EAAE,QAAQ;iBAC1B,CAAC,CAAA;YACJ,CAAC,CAAC,CAAA;QACJ,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,OAAO,eAAe,CAAA;AACxB,CAAC;AA9BD,wDA8BC;AAED,SAAgB,mBAAmB,CAAC,EAClC,4BAA4B,EAC5B,SAAS,EACT,qBAAqB,EACrB,iBAAiB,GAMlB;IACC,MAAM,WAAW,GAAgB,EAAE,CAAA;IAEnC,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;IAClD,MAAM,QAAQ,GAAG,IAAI,qBAAQ,CAAC,WAAW,CAAC,CAAA;IAE1C,IAAI,kBAAkB,GAAG,oBAAoB,CAAC,KAAK,CAAA;IAEnD,MAAM,aAAa,GAAG,CAAC,CAAC,qBAAqB,CAAA;IAC7C,IAAI,aAAa,EAAE,CAAC;QAClB,kBAAkB,GAAG,oBAAoB,CAAC,UAAU,CAAA;IACtD,CAAC;IAED,SAAS,0BAA0B,CAAC,EAClC,SAAS,EACT,aAAa,GAId;QACC,MAAM,oBAAoB,GAAG,CAAC,GAAG,iBAAiB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CACtE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACP,6BAA6B,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;YAC7C,6BAA6B,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAChD,CAAA;QAED,KAAK,MAAM,IAAI,IAAI,oBAAoB,EAAE,CAAC;YACxC;;eAEG;YACH,IACE,IAAI,CAAC,IAAI,KAAK,kBAAkB,CAAC,QAAQ;gBACzC,SAAS,CAAC,KAAK,GAAG,kBAAkB,CAAC,QAAQ,EAC7C,CAAC;gBACD,MAAM,SAAS,GAAG,iBAAiB,CAAC;oBAClC,aAAa;oBACb,aAAa,EAAE,IAAI,CAAC,IAAI;oBACxB,YAAY,EAAE,4BAA4B,CAAC,IAAI,CAAC,IAAI,CAAC;iBACtD,CAAC,CAAA;gBACF,IAAI,SAAS,EAAE,CAAC;oBACd,OAAO,SAAS,CAAA;gBAClB,CAAC;gBACD;;;;;;mBAMG;YACL,CAAC;iBAAM,IACL,IAAI,CAAC,IAAI,KAAK,kBAAkB,CAAC,YAAY;gBAC7C,aAAa,KAAK,cAAc;gBAChC,SAAS,CAAC,KAAK,GAAG,kBAAkB,CAAC,YAAY,EACjD,CAAC;gBACD,OAAO;oBACL,IAAI,EAAE,0BAAa,CAAC,IAAI;oBACxB,IAAI,EAAE;wBACJ,aAAa,EAAE,IAAI,CAAC,eAAe;wBACnC,YAAY,EAAE;4BACZ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI;yBAClB;qBACF;iBACW,CAAA;YAChB,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAA;IACb,CAAC;IAED;;;;;;OAMG;IACH,KAAK,MAAM,CAAC,QAAQ,EAAE,iCAAiC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;QACtF,MAAM,aAAa,GAAG,iCAAiC,CAAC,UAAU,CAAC,GAAG,CAAC;YACrE,CAAC,CAAC,iCAAiC,CAAC,SAAS,CAAC,CAAC,CAAC;YAChD,CAAC,CAAC,iCAAiC,CAAA;QAErC,IAAI,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAsB,CAAA;QAEvF,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAA;QAC3C,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,SAAQ;QACV,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,SAAS,GAAG,0BAA0B,CAAC;gBAC3C,SAAS,EAAE,KAAK;gBAChB,aAAa;aACd,CAAC,CAAA;YACF,IAAI,SAAS,EAAE,CAAC;gBACd,WAAW,CAAC,QAAQ,CAAC,GAAG,SAAS,CAAA;gBACjC,MAAK;YACP,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAA;AACpB,CAAC;AAlHD,kDAkHC"}

package/dist/connect/wizard/prop_mapping_helpers.d.ts

@@ -0,0 +1,71 @@
+import { BaseCommand } from '../../commands/connect';
+import { ComponentTypeSignature } from '../../react/parser';
+import { FigmaRestApi } from '../figma_rest_api';
+import { ProjectInfo } from '../project';
+import { MatchableName } from './prop_mapping';
+import { PropMapping } from '../parser_executable_types';
+export type MatchableNamesMap = Record<string, MatchableName[]>;
+export type PropMappingData = {
+    [filepathExport: string]: {
+        signature: ComponentTypeSignature;
+        matchableNamesMap: MatchableNamesMap;
+        componentPropertyDefinitions: FigmaRestApi.Component['componentPropertyDefinitions'];
+    };
+};
+/**
+ * Preprocess signatures and matchable names for all components
+ */
+export declare function getPropMappingData({ filepathExportsToComponents, projectInfo, cmd, }: {
+    filepathExportsToComponents: Record<string, FigmaRestApi.Component>;
+    projectInfo: ProjectInfo;
+    cmd: BaseCommand;
+}): PropMappingData;
+export declare function getUniqueMatchableNames(propMappingData: PropMappingData): string[];
+export type PropMatchResult = {
+    item: string;
+    score: number;
+};
+export type ComponentMatchResults = {
+    [propName: string]: PropMatchResult[];
+};
+type AllMatchResults = {
+    [filepathExport: string]: ComponentMatchResults;
+};
+type MatchableNameEmbeddings = {
+    [matchableName: string]: number[];
+};
+export declare function buildAllEmbeddingsMatchResults(propMappingData: PropMappingData, matchableNameEmbeddings: MatchableNameEmbeddings): AllMatchResults;
+export declare function generateAllPropsMappings({ propMappingData, accessToken, figmaUrl, useAi, mockResponseName, }: {
+    propMappingData: PropMappingData;
+    accessToken: string;
+    figmaUrl: string;
+    useAi: boolean;
+    mockResponseName?: string;
+}): Promise<{
+    [filepathExport: string]: PropMapping;
+}>;
+/**
+ * This is the top level function that takes a map of filepathExports to components and generates prop mappings.
+ * It does the following:
+ *
+ * 1. For each component we want to match, extract their TS signature and all figma properties
+ * 2. Make an array of all strings we want embeddings for (react props, figma properties, variant values)
+ * 3. Call embeddings endpoint with above and create a map of names => embeddings
+ * 4. For each component, build a map of code props to an list of matchable names + scores, sorted by their calculated embedding distance to the code prop
+ * 5. Finally, pass those name matches and component data to the prop mapping algorithm to generate the mapping
+ */
+export declare function extractDataAndGenerateAllPropsMappings({ filepathExportsToComponents, projectInfo, cmd, figmaUrl, accessToken, useAi, }: {
+    filepathExportsToComponents: Record<string, FigmaRestApi.Component>;
+    projectInfo: ProjectInfo;
+    cmd: BaseCommand;
+    figmaUrl: string;
+    accessToken: string;
+    useAi: boolean;
+}): Promise<{
+    propMappingData: PropMappingData;
+    propMappings: {
+        [filepathExport: string]: PropMapping;
+    };
+}>;
+export {};
+//# sourceMappingURL=prop_mapping_helpers.d.ts.map

package/dist/connect/wizard/prop_mapping_helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prop_mapping_helpers.d.ts","sourceRoot":"","sources":["../../../src/connect/wizard/prop_mapping_helpers.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAA;AAEpD,OAAO,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAA;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAA;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAA;AAExC,OAAO,EAAE,aAAa,EAA+C,MAAM,gBAAgB,CAAA;AAE3F,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAA;AAIxD,MAAM,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,EAAE,aAAa,EAAE,CAAC,CAAA;AAE/D,MAAM,MAAM,eAAe,GAAG;IAC5B,CAAC,cAAc,EAAE,MAAM,GAAG;QACxB,SAAS,EAAE,sBAAsB,CAAA;QACjC,iBAAiB,EAAE,iBAAiB,CAAA;QACpC,4BAA4B,EAAE,YAAY,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAA;KACrF,CAAA;CACF,CAAA;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,EACjC,2BAA2B,EAC3B,WAAW,EACX,GAAG,GACJ,EAAE;IACD,2BAA2B,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,SAAS,CAAC,CAAA;IACnE,WAAW,EAAE,WAAW,CAAA;IACxB,GAAG,EAAE,WAAW,CAAA;CACjB,mBA+BA;AAED,wBAAgB,uBAAuB,CAAC,eAAe,EAAE,eAAe,YAMvE;AAED,MAAM,MAAM,eAAe,GAAG;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAA;AAE7D,MAAM,MAAM,qBAAqB,GAAG;IAClC,CAAC,QAAQ,EAAE,MAAM,GAAG,eAAe,EAAE,CAAA;CACtC,CAAA;AAED,KAAK,eAAe,GAAG;IACrB,CAAC,cAAc,EAAE,MAAM,GAAG,qBAAqB,CAAA;CAChD,CAAA;AAED,KAAK,uBAAuB,GAAG;IAC7B,CAAC,aAAa,EAAE,MAAM,GAAG,MAAM,EAAE,CAAA;CAClC,CAAA;AASD,wBAAgB,8BAA8B,CAC5C,eAAe,EAAE,eAAe,EAChC,uBAAuB,EAAE,uBAAuB,mBAiBjD;AA8DD,wBAAsB,wBAAwB,CAAC,EAC7C,eAAe,EACf,WAAW,EACX,QAAQ,EACR,KAAK,EACL,gBAAgB,GACjB,EAAE;IACD,eAAe,EAAE,eAAe,CAAA;IAChC,WAAW,EAAE,MAAM,CAAA;IACnB,QAAQ,EAAE,MAAM,CAAA;IAChB,KAAK,EAAE,OAAO,CAAA;IACd,gBAAgB,CAAC,EAAE,MAAM,CAAA;CAC1B;;GAmCA;AAED;;;;;;;;;GASG;AACH,wBAAsB,sCAAsC,CAAC,EAC3D,2BAA2B,EAC3B,WAAW,EACX,GAAG,EACH,QAAQ,EACR,WAAW,EACX,KAAK,GACN,EAAE;IACD,2BAA2B,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,SAAS,CAAC,CAAA;IACnE,WAAW,EAAE,WAAW,CAAA;IACxB,GAAG,EAAE,WAAW,CAAA;IAChB,QAAQ,EAAE,MAAM,CAAA;IAChB,WAAW,EAAE,MAAM,CAAA;IACnB,KAAK,EAAE,OAAO,CAAA;CACf;;;;;GAeA"}

package/dist/connect/wizard/prop_mapping_helpers.js

@@ -0,0 +1,176 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.extractDataAndGenerateAllPropsMappings = exports.generateAllPropsMappings = exports.buildAllEmbeddingsMatchResults = exports.getUniqueMatchableNames = exports.getPropMappingData = void 0;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const logging_1 = require("../../common/logging");
+const helpers_1 = require("./helpers");
+const prop_mapping_1 = require("./prop_mapping");
+const signature_extraction_1 = require("./signature_extraction");
+const embeddings_1 = require("./embeddings");
+const fetch_1 = require("../../common/fetch");
+/**
+ * Preprocess signatures and matchable names for all components
+ */
+function getPropMappingData({ filepathExportsToComponents, projectInfo, cmd, }) {
+    const propMappingData = {};
+    for (const [filepathExport, { componentPropertyDefinitions }] of Object.entries(filepathExportsToComponents)) {
+        const { filepath, exportName } = (0, helpers_1.parseFilepathExport)(filepathExport);
+        if (projectInfo.config.parser === 'react' && filepath && exportName) {
+            try {
+                const signature = (0, signature_extraction_1.extractSignature)({
+                    nameToFind: exportName,
+                    sourceFilePath: filepath,
+                });
+                if (cmd.verbose && Object.keys(signature).length === 0) {
+                    logging_1.logger.warn(`No TS signature found for "${exportName}" in ${filepath}`);
+                }
+                propMappingData[filepathExport] = {
+                    signature,
+                    componentPropertyDefinitions,
+                    matchableNamesMap: (0, prop_mapping_1.buildMatchableNamesMap)(componentPropertyDefinitions),
+                };
+            }
+            catch (e) {
+                if (cmd.verbose) {
+                    logging_1.logger.warn(`Could not extract signature for "${exportName}" in ${filepath}`);
+                }
+            }
+        }
+    }
+    return propMappingData;
+}
+exports.getPropMappingData = getPropMappingData;
+function getUniqueMatchableNames(propMappingData) {
+    const allNames = Object.values(propMappingData).flatMap((d) => [
+        ...Object.keys(d.signature),
+        ...Object.keys(d.matchableNamesMap),
+    ]);
+    return Array.from(new Set(allNames));
+}
+exports.getUniqueMatchableNames = getUniqueMatchableNames;
+function cosineSimilarity(a, b) {
+    const dot_product = a.reduce((acc, val, i) => acc + val * b[i], 0);
+    const magnitude_a = Math.sqrt(a.reduce((acc, val) => acc + val * val, 0));
+    const magnitude_b = Math.sqrt(b.reduce((acc, val) => acc + val * val, 0));
+    return dot_product / (magnitude_a * magnitude_b);
+}
+function buildAllEmbeddingsMatchResults(propMappingData, matchableNameEmbeddings) {
+    const allMatchResults = {};
+    Object.entries(propMappingData).forEach(([filepathExport, { signature, matchableNamesMap }]) => {
+        allMatchResults[filepathExport] = {};
+        Object.keys(signature).forEach((propName) => {
+            allMatchResults[filepathExport][propName] = Object.keys(matchableNamesMap)
+                .map((item) => ({
+                item,
+                score: cosineSimilarity(matchableNameEmbeddings[propName], matchableNameEmbeddings[item]),
+            }))
+                .sort((a, b) => b.score - a.score);
+        });
+    });
+    return allMatchResults;
+}
+exports.buildAllEmbeddingsMatchResults = buildAllEmbeddingsMatchResults;
+async function getMockEmbeddingsResponse(uniqueMatchableNames, mockResponseName) {
+    /**
+     * Refetch and write local mock responses.
+     * This should be done whenever any upstream changes are made to e.g. TS signature extraction
+     */
+    const updateMockFiles = false;
+    // Return mock response or update local mocks
+    const mockResponsePath = mockResponseName &&
+        path_1.default.join(__dirname, `__test__/prop_mapping/test_cases/embeddings_responses/${mockResponseName}.json`);
+    if (updateMockFiles) {
+        if (!process.env.FIGMA_ACCESS_TOKEN) {
+            throw new Error('process.env.FIGMA_ACCESS_TOKEN required to fetch embeddings');
+        }
+        if (!process.env.FILE_URL) {
+            throw new Error("process.env.FILE_URL required to fetch embeddings (note: contents of file don't matter)");
+        }
+        const res = await (0, embeddings_1.fetchEmbeddings)({
+            uniqueMatchableNames,
+            accessToken: process.env.FIGMA_ACCESS_TOKEN,
+            figmaUrl: process.env.FILE_URL,
+        });
+        fs_1.default.writeFileSync(mockResponsePath, JSON.stringify(res));
+        return res;
+    }
+    else {
+        return JSON.parse(fs_1.default.readFileSync(mockResponsePath, 'utf-8'));
+    }
+}
+async function getEmbeddingsMatchResults({ propMappingData, accessToken, figmaUrl, mockResponseName, }) {
+    const uniqueMatchableNames = getUniqueMatchableNames(propMappingData);
+    const res = mockResponseName
+        ? await getMockEmbeddingsResponse(uniqueMatchableNames, mockResponseName)
+        : await (0, embeddings_1.fetchEmbeddings)({ uniqueMatchableNames, accessToken, figmaUrl });
+    const matchableNamesEmbeddings = {};
+    res?.meta.embeddings.forEach((embedding, index) => {
+        matchableNamesEmbeddings[uniqueMatchableNames[index]] = embedding;
+    });
+    return buildAllEmbeddingsMatchResults(propMappingData, matchableNamesEmbeddings);
+}
+async function generateAllPropsMappings({ propMappingData, accessToken, figmaUrl, useAi, mockResponseName, }) {
+    let allMatchResults = {};
+    if (useAi) {
+        try {
+            allMatchResults = await getEmbeddingsMatchResults({
+                propMappingData,
+                accessToken,
+                figmaUrl,
+                mockResponseName,
+            });
+        }
+        catch (e) {
+            if ((0, fetch_1.isFetchError)(e)) {
+                logging_1.logger.error(`Failed to fetch embeddings: ${e.data?.message || e.response?.status}`);
+            }
+            else {
+                logging_1.logger.error(`Failed to compute embeddings: ${e}`);
+            }
+            logging_1.logger.info('Falling back to using fuzzy matching');
+        }
+    }
+    const propMappings = {};
+    Object.entries(propMappingData).forEach(([filepathExport, { signature, componentPropertyDefinitions, matchableNamesMap }]) => {
+        propMappings[filepathExport] = (0, prop_mapping_1.generatePropMapping)({
+            matchableNamesMap,
+            componentPropertyDefinitions,
+            signature,
+            componentMatchResults: allMatchResults[filepathExport],
+        });
+    });
+    return propMappings;
+}
+exports.generateAllPropsMappings = generateAllPropsMappings;
+/**
+ * This is the top level function that takes a map of filepathExports to components and generates prop mappings.
+ * It does the following:
+ *
+ * 1. For each component we want to match, extract their TS signature and all figma properties
+ * 2. Make an array of all strings we want embeddings for (react props, figma properties, variant values)
+ * 3. Call embeddings endpoint with above and create a map of names => embeddings
+ * 4. For each component, build a map of code props to an list of matchable names + scores, sorted by their calculated embedding distance to the code prop
+ * 5. Finally, pass those name matches and component data to the prop mapping algorithm to generate the mapping
+ */
+async function extractDataAndGenerateAllPropsMappings({ filepathExportsToComponents, projectInfo, cmd, figmaUrl, accessToken, useAi, }) {
+    const propMappingData = getPropMappingData({
+        filepathExportsToComponents,
+        projectInfo,
+        cmd,
+    });
+    return {
+        propMappingData,
+        propMappings: await generateAllPropsMappings({
+            propMappingData,
+            accessToken,
+            figmaUrl,
+            useAi,
+        }),
+    };
+}
+exports.extractDataAndGenerateAllPropsMappings = extractDataAndGenerateAllPropsMappings;
+//# sourceMappingURL=prop_mapping_helpers.js.map

package/dist/connect/wizard/prop_mapping_helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prop_mapping_helpers.js","sourceRoot":"","sources":["../../../src/connect/wizard/prop_mapping_helpers.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAmB;AACnB,gDAAuB;AAEvB,kDAA6C;AAI7C,uCAA+C;AAC/C,iDAA2F;AAC3F,iEAAyD;AAEzD,6CAAkE;AAClE,8CAAiD;AAYjD;;GAEG;AACH,SAAgB,kBAAkB,CAAC,EACjC,2BAA2B,EAC3B,WAAW,EACX,GAAG,GAKJ;IACC,MAAM,eAAe,GAAoB,EAAE,CAAA;IAE3C,KAAK,MAAM,CAAC,cAAc,EAAE,EAAE,4BAA4B,EAAE,CAAC,IAAI,MAAM,CAAC,OAAO,CAC7E,2BAA2B,CAC5B,EAAE,CAAC;QACF,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,IAAA,6BAAmB,EAAC,cAAc,CAAC,CAAA;QAEpE,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,KAAK,OAAO,IAAI,QAAQ,IAAI,UAAU,EAAE,CAAC;YACpE,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,IAAA,uCAAgB,EAAC;oBACjC,UAAU,EAAE,UAAU;oBACtB,cAAc,EAAE,QAAQ;iBACzB,CAAC,CAAA;gBACF,IAAI,GAAG,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACvD,gBAAM,CAAC,IAAI,CAAC,8BAA8B,UAAU,QAAQ,QAAQ,EAAE,CAAC,CAAA;gBACzE,CAAC;gBACD,eAAe,CAAC,cAAc,CAAC,GAAG;oBAChC,SAAS;oBACT,4BAA4B;oBAC5B,iBAAiB,EAAE,IAAA,qCAAsB,EAAC,4BAA4B,CAAC;iBACxE,CAAA;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBAChB,gBAAM,CAAC,IAAI,CAAC,oCAAoC,UAAU,QAAQ,QAAQ,EAAE,CAAC,CAAA;gBAC/E,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,eAAe,CAAA;AACxB,CAAC;AAvCD,gDAuCC;AAED,SAAgB,uBAAuB,CAAC,eAAgC;IACtE,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QAC7D,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3B,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,iBAAiB,CAAC;KACpC,CAAC,CAAA;IACF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAA;AACtC,CAAC;AAND,0DAMC;AAgBD,SAAS,gBAAgB,CAAC,CAAW,EAAE,CAAW;IAChD,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;IAClE,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,CAAA;IACzE,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,CAAA;IACzE,OAAO,WAAW,GAAG,CAAC,WAAW,GAAG,WAAW,CAAC,CAAA;AAClD,CAAC;AAED,SAAgB,8BAA8B,CAC5C,eAAgC,EAChC,uBAAgD;IAEhD,MAAM,eAAe,GAAoB,EAAE,CAAA;IAE3C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,EAAE,EAAE,SAAS,EAAE,iBAAiB,EAAE,CAAC,EAAE,EAAE;QAC7F,eAAe,CAAC,cAAc,CAAC,GAAG,EAAE,CAAA;QACpC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC1C,eAAe,CAAC,cAAc,CAAC,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC;iBACvE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACd,IAAI;gBACJ,KAAK,EAAE,gBAAgB,CAAC,uBAAuB,CAAC,QAAQ,CAAC,EAAE,uBAAuB,CAAC,IAAI,CAAC,CAAC;aAC1F,CAAC,CAAC;iBACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAA;QACtC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,OAAO,eAAe,CAAA;AACxB,CAAC;AAnBD,wEAmBC;AAED,KAAK,UAAU,yBAAyB,CAAC,oBAA8B,EAAE,gBAAwB;IAC/F;;;OAGG;IACH,MAAM,eAAe,GAAG,KAAK,CAAA;IAE7B,6CAA6C;IAC7C,MAAM,gBAAgB,GACpB,gBAAgB;QAChB,cAAI,CAAC,IAAI,CACP,SAAS,EACT,yDAAyD,gBAAgB,OAAO,CACjF,CAAA;IACH,IAAI,eAAe,EAAE,CAAC;QACpB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,6DAA6D,CAAC,CAAA;QAChF,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CACb,yFAAyF,CAC1F,CAAA;QACH,CAAC;QACD,MAAM,GAAG,GAAG,MAAM,IAAA,4BAAe,EAAC;YAChC,oBAAoB;YACpB,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,kBAAkB;YAC3C,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,QAAQ;SAC/B,CAAC,CAAA;QACF,YAAE,CAAC,aAAa,CAAC,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAA;QACvD,OAAO,GAAG,CAAA;IACZ,CAAC;SAAM,CAAC;QACN,OAAO,IAAI,CAAC,KAAK,CAAC,YAAE,CAAC,YAAY,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAuB,CAAA;IACrF,CAAC;AACH,CAAC;AAED,KAAK,UAAU,yBAAyB,CAAC,EACvC,eAAe,EACf,WAAW,EACX,QAAQ,EACR,gBAAgB,GAMjB;IACC,MAAM,oBAAoB,GAAG,uBAAuB,CAAC,eAAe,CAAC,CAAA;IAErE,MAAM,GAAG,GAAG,gBAAgB;QAC1B,CAAC,CAAC,MAAM,yBAAyB,CAAC,oBAAoB,EAAE,gBAAgB,CAAC;QACzE,CAAC,CAAC,MAAM,IAAA,4BAAe,EAAC,EAAE,oBAAoB,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAA;IAE1E,MAAM,wBAAwB,GAA4B,EAAE,CAAA;IAE5D,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,SAAmB,EAAE,KAAa,EAAE,EAAE;QAClE,wBAAwB,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAA;IACnE,CAAC,CAAC,CAAA;IACF,OAAO,8BAA8B,CAAC,eAAe,EAAE,wBAAwB,CAAC,CAAA;AAClF,CAAC;AAEM,KAAK,UAAU,wBAAwB,CAAC,EAC7C,eAAe,EACf,WAAW,EACX,QAAQ,EACR,KAAK,EACL,gBAAgB,GAOjB;IACC,IAAI,eAAe,GAAoB,EAAE,CAAA;IAEzC,IAAI,KAAK,EAAE,CAAC;QACV,IAAI,CAAC;YACH,eAAe,GAAG,MAAM,yBAAyB,CAAC;gBAChD,eAAe;gBACf,WAAW;gBACX,QAAQ;gBACR,gBAAgB;aACjB,CAAC,CAAA;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,IAAA,oBAAY,EAAC,CAAC,CAAC,EAAE,CAAC;gBACpB,gBAAM,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC,IAAI,EAAE,OAAO,IAAI,CAAC,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAA;YACtF,CAAC;iBAAM,CAAC;gBACN,gBAAM,CAAC,KAAK,CAAC,iCAAiC,CAAC,EAAE,CAAC,CAAA;YACpD,CAAC;YACD,gBAAM,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAA;QACrD,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAA8C,EAAE,CAAA;IAElE,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,OAAO,CACrC,CAAC,CAAC,cAAc,EAAE,EAAE,SAAS,EAAE,4BAA4B,EAAE,iBAAiB,EAAE,CAAC,EAAE,EAAE;QACnF,YAAY,CAAC,cAAc,CAAC,GAAG,IAAA,kCAAmB,EAAC;YACjD,iBAAiB;YACjB,4BAA4B;YAC5B,SAAS;YACT,qBAAqB,EAAE,eAAe,CAAC,cAAc,CAAC;SACvD,CAAC,CAAA;IACJ,CAAC,CACF,CAAA;IAED,OAAO,YAAY,CAAA;AACrB,CAAC;AA/CD,4DA+CC;AAED;;;;;;;;;GASG;AACI,KAAK,UAAU,sCAAsC,CAAC,EAC3D,2BAA2B,EAC3B,WAAW,EACX,GAAG,EACH,QAAQ,EACR,WAAW,EACX,KAAK,GAQN;IACC,MAAM,eAAe,GAAG,kBAAkB,CAAC;QACzC,2BAA2B;QAC3B,WAAW;QACX,GAAG;KACJ,CAAC,CAAA;IACF,OAAO;QACL,eAAe;QACf,YAAY,EAAE,MAAM,wBAAwB,CAAC;YAC3C,eAAe;YACf,WAAW;YACX,QAAQ;YACR,KAAK;SACN,CAAC;KACH,CAAA;AACH,CAAC;AA7BD,wFA6BC"}

package/dist/connect/wizard/run_wizard.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"run_wizard.d.ts","sourceRoot":"","sources":["../../../src/connect/wizard/run_wizard.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAiD,MAAM,wBAAwB,CAAA;AACnG,OAAO,OAAO,MAAM,SAAS,CAAA;AAG7B,OAAO,EAAE,YAAY,EAAa,MAAM,mBAAmB,CAAA;AAG3D,OAAO,EACL,gBAAgB,EAKhB,iBAAiB,EACjB,WAAW,EAEZ,MAAM,uBAAuB,CAAA;AA6B9B,KAAK,0BAA0B,GAAG;IAAE,aAAa,EAAE,MAAM,CAAC;IAAC,cAAc,EAAE,MAAM,CAAA;CAAE,EAAE,CAAA;AA0JrF,wBAAgB,4BAA4B,CAC1C,UAAU,EAAE,YAAY,CAAC,SAAS,EAAE,EACpC,8BAA8B,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EACtD,2BAA2B,EAAE,0BAA0B,EACvD,GAAG,EAAE,MAAM,GACV,OAAO,CAAC,MAAM,EAAE,CA2ClB;AAsRD,wBAAgB,kCAAkC,CAAC,EACjD,aAAa,EACb,WAAW,EACX,GAAG,GACJ,EAAE;IACD,aAAa,EAAE,MAAM,CAAA;IACrB,WAAW,EAAE,MAAM,CAAA;IACnB,GAAG,EAAE,MAAM,CAAA;CACZ,iBAYA;AAED,wBAAsB,qDAAqD,CACzE,GAAG,EAAE,WAAW,EAChB,YAAY,EAAE,MAAM,EACpB,kBAAkB,EAAE,YAAY,CAAC,SAAS,EAAE,EAC5C,WAAW,EAAE,WAAW,CAAC,iBAAiB,CAAC,GAAG,gBAAgB;;;GA8C/D;AAsFD,wBAAsB,SAAS,CAAC,GAAG,EAAE,WAAW,iBAiK/C"}
+{"version":3,"file":"run_wizard.d.ts","sourceRoot":"","sources":["../../../src/connect/wizard/run_wizard.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAiD,MAAM,wBAAwB,CAAA;AACnG,OAAO,OAAO,MAAM,SAAS,CAAA;AAG7B,OAAO,EAAE,YAAY,EAAa,MAAM,mBAAmB,CAAA;AAE3D,OAAO,EACL,gBAAgB,EAKhB,iBAAiB,EACjB,WAAW,EAEZ,MAAM,uBAAuB,CAAA;AA0B9B,KAAK,0BAA0B,GAAG;IAAE,aAAa,EAAE,MAAM,CAAC;IAAC,cAAc,EAAE,MAAM,CAAA;CAAE,EAAE,CAAA;AAkKrF,wBAAgB,4BAA4B,CAC1C,UAAU,EAAE,YAAY,CAAC,SAAS,EAAE,EACpC,8BAA8B,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EACtD,2BAA2B,EAAE,0BAA0B,EACvD,GAAG,EAAE,MAAM,GACV,OAAO,CAAC,MAAM,EAAE,CA2ClB;AAySD,wBAAgB,kCAAkC,CAAC,EACjD,aAAa,EACb,WAAW,EACX,GAAG,GACJ,EAAE;IACD,aAAa,EAAE,MAAM,CAAA;IACrB,WAAW,EAAE,MAAM,CAAA;IACnB,GAAG,EAAE,MAAM,CAAA;CACZ,iBAYA;AAED,wBAAsB,qDAAqD,CACzE,GAAG,EAAE,WAAW,EAChB,YAAY,EAAE,MAAM,EACpB,kBAAkB,EAAE,YAAY,CAAC,SAAS,EAAE,EAC5C,WAAW,EAAE,WAAW,CAAC,iBAAiB,CAAC,GAAG,gBAAgB;;;GA8C/D;AAsFD,wBAAsB,SAAS,CAAC,GAAG,EAAE,WAAW,iBAsL/C"}