@workday/canvas-kit-codemod 7.1.4 → 7.2.0-427-next.2

package/dist/es6/v5/recategorizeButtons.js

@@ -1,24 +1,24 @@
 import { getImportRenameMap } from './getImportRenameMap';
-var updateJSXTag = function (nodePath, newTag) {
+const updateJSXTag = (nodePath, newTag) => {
     nodePath.value.openingElement.name.name = newTag;
     if (nodePath.value.closingElement) {
         nodePath.value.closingElement.name.name = newTag;
     }
 };
-var findBtnTag = function (elementName, node) {
+const findBtnTag = (elementName, node) => {
     if (elementName && node.openingElement.name.name === elementName) {
         return true;
     }
     return false;
 };
 export default function transformer(file, api, options) {
-    var j = api.jscodeshift;
-    var root = j(file.source);
-    var requiredImportSpecifiers = [];
+    const j = api.jscodeshift;
+    const root = j(file.source);
+    const requiredImportSpecifiers = [];
     /**
      * 1. Find button imports
      */
-    var _a = getImportRenameMap(j, root, '@workday/canvas-kit-react/button'), containsCanvasImports = _a.containsCanvasImports, importMap = _a.importMap;
+    const { containsCanvasImports, importMap } = getImportRenameMap(j, root, '@workday/canvas-kit-react/button');
     if (!containsCanvasImports) {
         return file.source;
     }
@@ -33,17 +33,17 @@ export default function transformer(file, api, options) {
      */
     root
         .find(j.JSXElement, findBtnTag.bind(undefined, importMap.Button))
-        .forEach(function (nodePath) {
+        .forEach((nodePath) => {
         var _a, _b;
-        var attrs = nodePath.value.openingElement.attributes;
-        var isPrimary = false;
-        var variantProp = attrs === null || attrs === void 0 ? void 0 : attrs.find(function (attr) { return attr.type === 'JSXAttribute' && attr.name.name === 'variant'; });
+        const attrs = nodePath.value.openingElement.attributes;
+        let isPrimary = false;
+        const variantProp = attrs === null || attrs === void 0 ? void 0 : attrs.find(attr => attr.type === 'JSXAttribute' && attr.name.name === 'variant');
         if (variantProp) {
             isPrimary = ((_a = variantProp.value) === null || _a === void 0 ? void 0 : _a.value) === 'primary';
             // remove variant prop
             (_b = nodePath.value.openingElement.attributes) === null || _b === void 0 ? void 0 : _b.splice(attrs === null || attrs === void 0 ? void 0 : attrs.indexOf(variantProp), 1);
         }
-        var buttonType = isPrimary ? 'PrimaryButton' : 'SecondaryButton';
+        const buttonType = isPrimary ? 'PrimaryButton' : 'SecondaryButton';
         // Swap to `PrimaryButton` or `SecondaryButton`
         updateJSXTag(nodePath, buttonType);
         requiredImportSpecifiers.push(buttonType);
@@ -55,7 +55,7 @@ export default function transformer(file, api, options) {
      */
     root
         .find(j.JSXElement, findBtnTag.bind(undefined, importMap.HighlightButton))
-        .forEach(function (nodePath) {
+        .forEach(nodePath => {
         updateJSXTag(nodePath, 'SecondaryButton');
         requiredImportSpecifiers.push('SecondaryButton');
     });
@@ -65,12 +65,12 @@ export default function transformer(file, api, options) {
      *  - If variant is 'inverse', keep it. Otherwise, remove the variant prop
      *  - Replace import with `SecondaryButton` if it doesn't exist
      */
-    root.find(j.JSXElement, findBtnTag.bind(undefined, importMap.OutlineButton)).forEach(function (nodePath) {
+    root.find(j.JSXElement, findBtnTag.bind(undefined, importMap.OutlineButton)).forEach(nodePath => {
         var _a;
         // Swap to `SecondaryButton`
         updateJSXTag(nodePath, 'SecondaryButton');
-        var attrs = nodePath.value.openingElement.attributes;
-        var variantProp = attrs === null || attrs === void 0 ? void 0 : attrs.find(function (attr) { return attr.type === 'JSXAttribute' && attr.name.name === 'variant'; });
+        const attrs = nodePath.value.openingElement.attributes;
+        const variantProp = attrs === null || attrs === void 0 ? void 0 : attrs.find(attr => attr.type === 'JSXAttribute' && attr.name.name === 'variant');
         if (variantProp && variantProp.type === 'JSXAttribute') {
             if (variantProp.value.value !== 'inverse') {
                 // remove variant prop
@@ -84,12 +84,12 @@ export default function transformer(file, api, options) {
      *  - Replace with `TertiaryButton`
      *  - If variant="text", remove variant prop
      */
-    root.find(j.JSXElement, findBtnTag.bind(undefined, importMap.TextButton)).forEach(function (nodePath) {
+    root.find(j.JSXElement, findBtnTag.bind(undefined, importMap.TextButton)).forEach(nodePath => {
         var _a, _b, _c;
         // Swap to `TertiaryButton`
         updateJSXTag(nodePath, 'TertiaryButton');
-        var attrs = nodePath.value.openingElement.attributes;
-        var variantProp = attrs === null || attrs === void 0 ? void 0 : attrs.find(function (attr) { return attr.type === 'JSXAttribute' && attr.name.name === 'variant'; });
+        const attrs = nodePath.value.openingElement.attributes;
+        const variantProp = attrs === null || attrs === void 0 ? void 0 : attrs.find(attr => attr.type === 'JSXAttribute' && attr.name.name === 'variant');
         if (variantProp && ((_b = (_a = variantProp) === null || _a === void 0 ? void 0 : _a.value) === null || _b === void 0 ? void 0 : _b.value) === 'text') {
             // remove variant prop
             (_c = nodePath.value.openingElement.attributes) === null || _c === void 0 ? void 0 : _c.splice(attrs === null || attrs === void 0 ? void 0 : attrs.indexOf(variantProp), 1);
@@ -106,17 +106,17 @@ export default function transformer(file, api, options) {
      */
     root
         .find(j.JSXElement, findBtnTag.bind(undefined, importMap.DropdownButton))
-        .forEach(function (nodePath) {
+        .forEach(nodePath => {
         var _a, _b;
-        var attrs = nodePath.value.openingElement.attributes;
-        var isPrimary = false;
-        var variantProp = attrs === null || attrs === void 0 ? void 0 : attrs.find(function (attr) { return attr.type === 'JSXAttribute' && attr.name.name === 'variant'; });
+        const attrs = nodePath.value.openingElement.attributes;
+        let isPrimary = false;
+        const variantProp = attrs === null || attrs === void 0 ? void 0 : attrs.find(attr => attr.type === 'JSXAttribute' && attr.name.name === 'variant');
         if (variantProp) {
             isPrimary = ((_a = variantProp.value) === null || _a === void 0 ? void 0 : _a.value) === 'primary';
             // remove variant prop
             (_b = nodePath.value.openingElement.attributes) === null || _b === void 0 ? void 0 : _b.splice(attrs === null || attrs === void 0 ? void 0 : attrs.indexOf(variantProp), 1);
         }
-        var buttonType = isPrimary ? 'PrimaryButton' : 'SecondaryButton';
+        const buttonType = isPrimary ? 'PrimaryButton' : 'SecondaryButton';
         // Swap to `PrimaryButton` or `SecondaryButton`
         updateJSXTag(nodePath, buttonType);
         // Add `icon="caretDownIcon" iconPosition="right"`
@@ -124,7 +124,7 @@ export default function transformer(file, api, options) {
         nodePath.value.openingElement.attributes.push(j.jsxAttribute(j.jsxIdentifier('iconPosition'), j.stringLiteral('right')));
         requiredImportSpecifiers.push(buttonType);
         // Add `import {caretDownIcon} from '@workday/canvas-system-icons-web'`
-        var lastImport = root.find(j.ImportDeclaration).at(-1);
+        const lastImport = root.find(j.ImportDeclaration).at(-1);
         if (lastImport) {
             lastImport.insertAfter(j.importDeclaration([j.importSpecifier(j.identifier('caretDownIcon'))], j.stringLiteral('@workday/canvas-system-icons-web')));
         }
@@ -133,22 +133,21 @@ export default function transformer(file, api, options) {
      * Remove old imports: `DropdownButton`, `TextButton`, `OutlineButton`, `HighlightButton`, `Button`.
      * Add new required imports
      */
-    var buttonImports = root.find(j.ImportDeclaration, {
+    const buttonImports = root.find(j.ImportDeclaration, {
         source: { value: '@workday/canvas-kit-react/button' },
     });
     if (!buttonImports.length) {
         // Add new specifiers to a new import
-        var allImports = root.find(j.ImportDeclaration);
-        var lastImport = allImports.at(allImports.length);
+        const allImports = root.find(j.ImportDeclaration);
+        const lastImport = allImports.at(allImports.length);
         if (lastImport) {
-            lastImport.insertAfter(j.importDeclaration(requiredImportSpecifiers.map(function (specifier) { return j.importSpecifier(j.identifier(specifier)); }), j.stringLiteral('@workday/canvas-kit-react/button')));
+            lastImport.insertAfter(j.importDeclaration(requiredImportSpecifiers.map(specifier => j.importSpecifier(j.identifier(specifier))), j.stringLiteral('@workday/canvas-kit-react/button')));
         }
     }
     else {
-        buttonImports.forEach(function (_a) {
-            var _b;
-            var node = _a.node;
-            var specifiersToRemove = [
+        buttonImports.forEach(({ node }) => {
+            var _a;
+            const specifiersToRemove = [
                 'DropdownButton',
                 'TextButton',
                 'OutlineButton',
@@ -158,7 +157,7 @@ export default function transformer(file, api, options) {
             // Remove old specifiers
             if (typeof node.source.value === 'string' &&
                 node.source.value === '@workday/canvas-kit-react/button') {
-                (_b = node.specifiers) === null || _b === void 0 ? void 0 : _b.forEach(function (specifier) {
+                (_a = node.specifiers) === null || _a === void 0 ? void 0 : _a.forEach(specifier => {
                     var _a, _b;
                     if (specifier.type === 'ImportSpecifier' &&
                         specifiersToRemove.includes(specifier.imported.name)) {
@@ -168,9 +167,9 @@ export default function transformer(file, api, options) {
                 });
             }
             //  Add new specifiers to existing import
-            requiredImportSpecifiers.forEach(function (specifier) {
+            requiredImportSpecifiers.forEach(specifier => {
                 var _a, _b;
-                if (!((_a = node.specifiers) === null || _a === void 0 ? void 0 : _a.find(function (existing) { return existing.type === 'ImportSpecifier' && existing.imported.name === specifier; }))) {
+                if (!((_a = node.specifiers) === null || _a === void 0 ? void 0 : _a.find(existing => existing.type === 'ImportSpecifier' && existing.imported.name === specifier))) {
                     (_b = node.specifiers) === null || _b === void 0 ? void 0 : _b.push(j.importSpecifier(j.identifier(specifier)));
                 }
             });

package/dist/es6/v5/removeButtonEnums.js

@@ -1,6 +1,6 @@
 import { getImportRenameMap } from './getImportRenameMap';
 // List of imports that will be removed from an import statement
-var importsToRemove = [
+const importsToRemove = [
     'ButtonVariant',
     'ButtonIconPosition',
     'ButtonSize',
@@ -15,7 +15,7 @@ var importsToRemove = [
 // List of enums used in code that will be translated to string literals
 // before: <Button variant={ButtonVariant.Primary} />
 // after:  <Button variant={"primary"} />
-var enumsToMap = {
+const enumsToMap = {
     'ButtonVariant.Primary': 'primary',
     'ButtonVariant.Secondary': 'secondary',
     'Button.Variant.Primary': 'primary',
@@ -89,7 +89,7 @@ var enumsToMap = {
 // List of enum types that will be converted to union types of string literals
 // before: function foo (variant: ButtonVariant) {}
 // after:  function foo (variant: 'primary' | 'secondary') {}
-var enumToLiteralUnionMap = {
+const enumToLiteralUnionMap = {
     ButtonVariant: ['primary', 'secondary'],
     ButtonIconPosition: ['left', 'right'],
     ButtonIconSize: ['small', 'medium', 'large'],
@@ -108,63 +108,61 @@ var enumToLiteralUnionMap = {
     DeprecatedButtonVariant: ['primary', 'secondary', 'delete'],
 };
 export default function transformer(file, api, options) {
-    var j = api.jscodeshift;
-    var root = j(file.source);
-    var _a = getImportRenameMap(j, root, '@workday/canvas-kit-react/button'), containsCanvasImports = _a.containsCanvasImports, importMap = _a.importMap;
+    const j = api.jscodeshift;
+    const root = j(file.source);
+    const { containsCanvasImports, importMap } = getImportRenameMap(j, root, '@workday/canvas-kit-react/button');
     if (!containsCanvasImports) {
         return file.source;
     }
     // Remove Variant imports
     root
-        .find(j.ImportSpecifier, function (node) {
+        .find(j.ImportSpecifier, (node) => {
         return importsToRemove.includes(node.imported.name);
     })
         .remove();
-    var reverseImportMap = Object.keys(importMap).reduce(function (result, key) {
+    const reverseImportMap = Object.keys(importMap).reduce((result, key) => {
         result[importMap[key]] = key;
         return result;
     }, {});
     // The following will replace JSX attributes that match something in `enumsToMap`
     // Example: variant={Button.Variant.Primary} => variant="primary"
-    function findEnumReplacement(node, candidates, matched) {
-        if (candidates === void 0) { candidates = enumsToMap; }
-        if (matched === void 0) { matched = ''; }
-        var candidate;
+    function findEnumReplacement(node, candidates = enumsToMap, matched = '') {
+        let candidate;
         // eslint-disable-next-line guard-for-in
         for (candidate in candidates) {
             // base case
             if (node.object.type === 'Identifier' &&
                 node.property.type === 'Identifier' &&
-                reverseImportMap[node.object.name] + "." + node.property.name + matched === candidate) {
+                `${reverseImportMap[node.object.name]}.${node.property.name}${matched}` === candidate) {
                 return enumsToMap[candidate];
             }
             // We didn't match yet, we'll try splitting the candidate string up into a left and right
             // portion and try again
-            var parts = candidate.split('.');
-            var last = parts.splice(-1, 1).join('');
+            const parts = candidate.split('.');
+            const last = parts.splice(-1, 1).join('');
             if (node.property.type === 'Identifier' &&
                 node.property.name === last &&
                 node.object.type === 'MemberExpression') {
                 // we have a partial match
-                return findEnumReplacement(node.object, enumsToMap, "." + last);
+                return findEnumReplacement(node.object, enumsToMap, `.${last}`);
             }
         }
         return null;
     }
     root
-        .find(j.MemberExpression, function (node) {
-        var matched = findEnumReplacement(node);
+        .find(j.MemberExpression, (node) => {
+        const matched = findEnumReplacement(node);
         return matched !== null;
     })
-        .replaceWith(function (nodePath) {
-        var matched = findEnumReplacement(nodePath.value);
+        .replaceWith(nodePath => {
+        const matched = findEnumReplacement(nodePath.value);
         return j.literal(matched);
     });
     // Rewrite `import { beta_Button as Button } from `@workday/canvas-kit-react/button` to
     // `import { Button } from '@workday/canvas-kit-react/button'
-    root.find(j.ImportSpecifier, { imported: { name: 'beta_Button' } }).forEach(function (nodePath) {
+    root.find(j.ImportSpecifier, { imported: { name: 'beta_Button' } }).forEach(nodePath => {
         var _a;
-        var parent = nodePath.parent.value;
+        const parent = nodePath.parent.value;
         if (parent.source &&
             typeof parent.source.value === 'string' &&
             parent.source.value.includes('@workday/canvas-kit-react')) {
@@ -190,7 +188,7 @@ export default function transformer(file, api, options) {
      */
     function findTypeRename(node) {
         if (node.type === 'TSTypeReference' && node.typeName.type === 'Identifier') {
-            var identifier = void 0;
+            let identifier;
             for (identifier in enumToLiteralUnionMap) {
                 if (node.typeName.name === importMap[identifier]) {
                     return enumToLiteralUnionMap[identifier];
@@ -200,12 +198,12 @@ export default function transformer(file, api, options) {
         return null;
     }
     root
-        .find(j.TSTypeReference, function (node) { return !!findTypeRename(node); })
-        .replaceWith(function (nodePath) {
-        var node = nodePath.value;
-        var matches = findTypeRename(node);
+        .find(j.TSTypeReference, (node) => !!findTypeRename(node))
+        .replaceWith(nodePath => {
+        const node = nodePath.value;
+        const matches = findTypeRename(node);
         if (matches) {
-            return j.tsUnionType(matches.map(function (literal) { return j.tsLiteralType({ type: 'StringLiteral', value: literal }); }));
+            return j.tsUnionType(matches.map(literal => j.tsLiteralType({ type: 'StringLiteral', value: literal })));
         }
         return nodePath.value;
     });

package/dist/es6/v5/removeDefaultImports.js

@@ -1,13 +1,13 @@
 // List of import paths and the named export
 // before: import Button from '@workday/canvas-kit-react/button
 // after:  import { Button } from '@workday/canvas-kit-react/button
-var renameDefaultMap = {
+const renameDefaultMap = {
     '@workday/canvas-kit-react/button': 'Button',
     '@workday/canvas-kit-react/card': 'Card',
 };
 export default function transformer(file, api, options) {
-    var j = api.jscodeshift;
-    var root = j(file.source);
+    const j = api.jscodeshift;
+    const root = j(file.source);
     // Renames default imports from `renameDefaultMap` and changes them to named exports
     // Example:
     // before:
@@ -16,11 +16,11 @@ export default function transformer(file, api, options) {
     // after:
     // import { Button } from '@workday/canvas-kit-react/button';
     // import { Button as StyledButton } from '@workday/canvas-kit-react/button';
-    root.find(j.ImportDefaultSpecifier).forEach(function (nodePath) {
+    root.find(j.ImportDefaultSpecifier).forEach(nodePath => {
         var _a;
-        var parent = nodePath.parent;
-        var importSource = String(parent.node.source.value);
-        var localName = (_a = nodePath.value.local) === null || _a === void 0 ? void 0 : _a.name;
+        const parent = nodePath.parent;
+        const importSource = String(parent.node.source.value);
+        const localName = (_a = nodePath.value.local) === null || _a === void 0 ? void 0 : _a.name;
         if (Object.keys(renameDefaultMap).includes(importSource) && localName) {
             nodePath.replace(j.importSpecifier(j.identifier(renameDefaultMap[importSource]), j.identifier(localName)));
         }

package/dist/es6/v5/renameButtonRefs.js

@@ -1,5 +1,5 @@
 import { getImportRenameMap } from './getImportRenameMap';
-var buttonNames = [
+const buttonNames = [
     'Button',
     'DeleteButton',
     'deprecated_Button',
@@ -13,19 +13,19 @@ var buttonNames = [
     'ToolbarIconButton',
 ];
 export default function transformer(file, api, options) {
-    var j = api.jscodeshift;
-    var root = j(file.source);
-    var _a = getImportRenameMap(j, root, '@workday/canvas-kit-react/button'), containsCanvasImports = _a.containsCanvasImports, importMap = _a.importMap;
+    const j = api.jscodeshift;
+    const root = j(file.source);
+    const { containsCanvasImports, importMap } = getImportRenameMap(j, root, '@workday/canvas-kit-react/button');
     if (!containsCanvasImports) {
         return file.source;
     }
     root
-        .find(j.JSXIdentifier, function (value) { return value.name === 'buttonRef'; })
-        .replaceWith(function (nodePath) {
+        .find(j.JSXIdentifier, (value) => value.name === 'buttonRef')
+        .replaceWith(nodePath => {
         var _a, _b, _c, _d;
         // only replace via one of our Button imports
-        var elementName = (_d = (_c = (_b = (_a = nodePath.parent) === null || _a === void 0 ? void 0 : _a.parent) === null || _b === void 0 ? void 0 : _b.value) === null || _c === void 0 ? void 0 : _c.name) === null || _d === void 0 ? void 0 : _d.name;
-        if (buttonNames.map(function (n) { return importMap[n]; }).includes(elementName)) {
+        const elementName = (_d = (_c = (_b = (_a = nodePath.parent) === null || _a === void 0 ? void 0 : _a.parent) === null || _b === void 0 ? void 0 : _b.value) === null || _c === void 0 ? void 0 : _c.name) === null || _d === void 0 ? void 0 : _d.name;
+        if (buttonNames.map(n => importMap[n]).includes(elementName)) {
             return j.jsxIdentifier('ref');
         }
         return nodePath.value;

package/dist/es6/v5/renameCoreImports.js

@@ -1,17 +1,17 @@
-var sourceMap = {
+const sourceMap = {
     '@workday/canvas-kit-react/core': '@workday/canvas-kit-react/tokens',
     '@workday/canvas-kit-labs-react/core': '@workday/canvas-kit-labs-react/tokens',
 };
 // Rename "/core" imports to "/tokens"
 export default function transformer(file, api) {
-    var j = api.jscodeshift;
-    var root = j(file.source);
+    const j = api.jscodeshift;
+    const root = j(file.source);
     // transform import {} from "@workday/canvas-kit-react/core"
     // to import {} from "@workday/canvas-kit-react/tokens"
     root
         .find(j.ImportDeclaration, { source: { value: '@workday/canvas-kit-react/core' } })
-        .forEach(function (nodePath) {
-        var source = nodePath.value.source.value;
+        .forEach(nodePath => {
+        const source = nodePath.value.source.value;
         nodePath.value.source.value = sourceMap[source];
         return nodePath;
     });
@@ -19,8 +19,8 @@ export default function transformer(file, api) {
     // to "@workday/canvas-kit-labs-react/tokens"
     root
         .find(j.ImportDeclaration, { source: { value: '@workday/canvas-kit-labs-react/core' } })
-        .forEach(function (nodePath) {
-        var source = nodePath.value.source.value;
+        .forEach(nodePath => {
+        const source = nodePath.value.source.value;
         nodePath.value.source.value = sourceMap[source];
         return nodePath;
     });
@@ -28,10 +28,10 @@ export default function transformer(file, api) {
     // to `import {type} from "@workday/canvas-kit-labs-react/tokens";`
     root
         .find(j.ImportDeclaration, { source: { value: '@workday/canvas-kit-labs-react/tokens' } })
-        .forEach(function (nodePath) {
+        .forEach(nodePath => {
         var _a;
-        var newSpecifiers = [];
-        (_a = nodePath.value.specifiers) === null || _a === void 0 ? void 0 : _a.forEach(function (specifier) {
+        const newSpecifiers = [];
+        (_a = nodePath.value.specifiers) === null || _a === void 0 ? void 0 : _a.forEach(specifier => {
             var _a;
             if (specifier.type === 'ImportDefaultSpecifier' && ((_a = specifier.local) === null || _a === void 0 ? void 0 : _a.name) === 'type') {
                 return newSpecifiers.push({

package/dist/es6/v5/renameInputRefs.js

@@ -1,5 +1,5 @@
 import { getImportRenameMap } from './getImportRenameMap';
-var inputsMap = {
+const inputsMap = {
     '@workday/canvas-kit-react/checkbox': ['Checkbox'],
     '@workday/canvas-kit-react/color-picker': ['ColorInput', 'ColorPreview'],
     '@workday/canvas-kit-react/radio': ['Radio'],
@@ -8,8 +8,8 @@ var inputsMap = {
     '@workday/canvas-kit-react/text-input': ['TextInput'],
 };
 export default function transformer(file, api, options) {
-    var j = api.jscodeshift;
-    var root = j(file.source);
+    const j = api.jscodeshift;
+    const root = j(file.source);
     // defaultImports should be an array of local names that were imported using
     // a default import from any of the modules listed in inputsMap. For example,
     // given...
@@ -22,26 +22,26 @@ export default function transformer(file, api, options) {
     // ```
     //
     // ... defaultImports should be set to `['Checkbox', 'CanvasTextArea']`
-    var defaultImports = [];
-    root.find(j.ImportDefaultSpecifier).forEach(function (nodePath) {
+    const defaultImports = [];
+    root.find(j.ImportDefaultSpecifier).forEach(nodePath => {
         var _a;
-        var packageName = nodePath.parent.node.source.value;
-        var localName = (_a = nodePath.value.local) === null || _a === void 0 ? void 0 : _a.name;
+        const packageName = nodePath.parent.node.source.value;
+        const localName = (_a = nodePath.value.local) === null || _a === void 0 ? void 0 : _a.name;
         if (packageName in inputsMap && localName) {
             defaultImports.push(localName);
         }
     });
-    var runningSource = file.source;
-    var _loop_1 = function (inputPackageName) {
-        var currentRoot = j(runningSource);
-        var _a = getImportRenameMap(j, currentRoot, inputPackageName), containsCanvasImports = _a.containsCanvasImports, importMap = _a.importMap, styledMap = _a.styledMap;
+    let runningSource = file.source;
+    for (const inputPackageName of Object.keys(inputsMap)) {
+        const currentRoot = j(runningSource);
+        const { containsCanvasImports, importMap, styledMap } = getImportRenameMap(j, currentRoot, inputPackageName);
         if (containsCanvasImports) {
-            var inputNames_1 = inputsMap[inputPackageName];
+            const inputNames = inputsMap[inputPackageName];
             currentRoot
-                .find(j.JSXIdentifier, function (value) { return value.name === 'inputRef'; })
-                .replaceWith(function (nodePath) {
+                .find(j.JSXIdentifier, (value) => value.name === 'inputRef')
+                .replaceWith(nodePath => {
                 var _a, _b, _c, _d;
-                var elementName = (_d = (_c = (_b = (_a = nodePath.parent) === null || _a === void 0 ? void 0 : _a.parent) === null || _b === void 0 ? void 0 : _b.value) === null || _c === void 0 ? void 0 : _c.name) === null || _d === void 0 ? void 0 : _d.name;
+                const elementName = (_d = (_c = (_b = (_a = nodePath.parent) === null || _a === void 0 ? void 0 : _a.parent) === null || _b === void 0 ? void 0 : _b.value) === null || _c === void 0 ? void 0 : _c.name) === null || _d === void 0 ? void 0 : _d.name;
                 // Rewrite inputRef to ref if any of the following are true:
                 // (a) elementName was the name used in a default import of the current
                 //     input being processed (e.g., `import TextInput ...`)
@@ -51,18 +51,14 @@ export default function transformer(file, api, options) {
                 // (c) elementName was created by calling `styled` on the current input
                 //     being processed (e.g., `const StyledTextInput = styled(TextInput)`)
                 if (defaultImports.includes(elementName) ||
-                    inputNames_1.map(function (n) { return importMap[n]; }).includes(elementName) ||
-                    inputNames_1.map(function (n) { return styledMap[n]; }).includes(elementName)) {
+                    inputNames.map(n => importMap[n]).includes(elementName) ||
+                    inputNames.map(n => styledMap[n]).includes(elementName)) {
                     return j.jsxIdentifier('ref');
                 }
                 return nodePath.value;
             });
             runningSource = currentRoot.toSource();
         }
-    };
-    for (var _i = 0, _a = Object.keys(inputsMap); _i < _a.length; _i++) {
-        var inputPackageName = _a[_i];
-        _loop_1(inputPackageName);
     }
     return runningSource;
 }

package/dist/es6/v5/renamePreviewTokenImports.js

@@ -1,12 +1,12 @@
 export default function transformer(file, api) {
-    var j = api.jscodeshift;
-    var root = j(file.source);
+    const j = api.jscodeshift;
+    const root = j(file.source);
     // find react/tokens package import declarations
-    var tokenImport = root.find(j.ImportDeclaration, {
+    const tokenImport = root.find(j.ImportDeclaration, {
         source: { value: '@workday/canvas-kit-react/tokens' },
     });
     // find react/common package import declarations
-    var commonImport = root.find(j.ImportDeclaration, {
+    const commonImport = root.find(j.ImportDeclaration, {
         source: { value: '@workday/canvas-kit-react/common' },
     });
     function transformTypeImports(nodePath, specifier) {
@@ -15,7 +15,7 @@ export default function transformer(file, api) {
             // push this specifier into those node specifiers.
             // This prevents possible duplicate import declarations.
             if (tokenImport.paths().length) {
-                tokenImport.forEach(function (tokenImportPath) {
+                tokenImport.forEach(tokenImportPath => {
                     var _a;
                     (_a = tokenImportPath.value.specifiers) === null || _a === void 0 ? void 0 : _a.push(specifier);
                 });
@@ -35,7 +35,7 @@ export default function transformer(file, api) {
             // push this specifier into those node specifiers.
             // This prevents possible duplicate import declarations.
             if (commonImport.paths().length) {
-                commonImport.forEach(function (tokenImportPath) {
+                commonImport.forEach(tokenImportPath => {
                     var _a;
                     (_a = tokenImportPath.value.specifiers) === null || _a === void 0 ? void 0 : _a.push(specifier);
                 });
@@ -49,9 +49,9 @@ export default function transformer(file, api) {
     // Transform labs/tokens imports
     root
         .find(j.ImportDeclaration, { source: { value: '@workday/canvas-kit-labs-react/tokens' } })
-        .forEach(function (nodePath) {
+        .forEach(nodePath => {
         var _a;
-        (_a = nodePath.value.specifiers) === null || _a === void 0 ? void 0 : _a.forEach(function (specifier) {
+        (_a = nodePath.value.specifiers) === null || _a === void 0 ? void 0 : _a.forEach(specifier => {
             if (specifier.type === 'ImportSpecifier') {
                 // Update type import node to use tokens package as the source
                 transformTypeImports(nodePath, specifier);
@@ -64,9 +64,9 @@ export default function transformer(file, api) {
     // Transform preview/tokens imports
     root
         .find(j.ImportDeclaration, { source: { value: '@workday/canvas-kit-preview-react/tokens' } })
-        .forEach(function (nodePath) {
+        .forEach(nodePath => {
         var _a;
-        (_a = nodePath.value.specifiers) === null || _a === void 0 ? void 0 : _a.forEach(function (specifier) {
+        (_a = nodePath.value.specifiers) === null || _a === void 0 ? void 0 : _a.forEach(specifier => {
             if (specifier.type === 'ImportSpecifier') {
                 // Update type import node to use tokens package as the source
                 transformTypeImports(nodePath, specifier);