@kaizen/components 1.67.10 → 1.67.12

package/codemods/migrateBrandMomentMoodToVariant/index.ts

@@ -2,9 +2,8 @@ import { transformComponentsInDir } from "../utils"
 import { transformBrandMomentMoodToVariant } from "./transformBrandMomentMoodToVariant"
 
 const migrateBrandMomentMoodToVariant = (): void => {
-  // eslint-disable-next-line no-console
   console.log(
-    " ~(-_- ~) Running BrandMoment mood to variant transformer (~ -_-)~"
+    "~(-_- ~) Running BrandMoment mood to variant transformer (~ -_-)~"
   )
   const targetDir = process.argv[2]
   if (!targetDir) {

package/codemods/migrateBrandMomentMoodToVariant/transformBrandMomentMoodToVariant.spec.ts

@@ -1,11 +1,15 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst, TransformConfig } from "../utils"
+import {
+  printAst,
+  transformSourceForTagName,
+  type TransformSourceForTagNameArgs,
+} from "../utils"
 import { transformBrandMomentMoodToVariant } from "./transformBrandMomentMoodToVariant"
 
 const transformBrandMoment = (
-  sourceFile: TransformConfig["sourceFile"]
+  sourceFile: TransformSourceForTagNameArgs["sourceFile"]
 ): string =>
-  transformSource({
+  transformSourceForTagName({
     sourceFile,
     astTransformer: transformBrandMomentMoodToVariant,
     tagName: "BrandMoment",

package/codemods/migrateCardVariantToColor/index.ts

@@ -2,8 +2,7 @@ import { transformComponentsInDir } from "../utils"
 import { transformCardVariantToColor } from "./transformCardVariantToColor"
 
 const migrateCardVariantToColor = (): void => {
-  // eslint-disable-next-line no-console
-  console.log(" ~(-_- ~) Running Card variant to color transformer (~ -_-)~")
+  console.log("~(-_- ~) Running Card variant to color transformer (~ -_-)~")
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)

package/codemods/migrateCardVariantToColor/transformCardVariantToColor.spec.ts

@@ -1,9 +1,15 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst, TransformConfig } from "../utils"
+import {
+  printAst,
+  transformSourceForTagName,
+  type TransformSourceForTagNameArgs,
+} from "../utils"
 import { transformCardVariantToColor } from "./transformCardVariantToColor"
 
-const transformCard = (sourceFile: TransformConfig["sourceFile"]): string =>
-  transformSource({
+const transformCard = (
+  sourceFile: TransformSourceForTagNameArgs["sourceFile"]
+): string =>
+  transformSourceForTagName({
     sourceFile,
     astTransformer: transformCardVariantToColor,
     tagName: "Card",

package/codemods/migrateConfirmationModalMoodsToVariant/index.ts

@@ -2,8 +2,7 @@ import { transformComponentsInDir } from "../utils"
 import { transformConfirmationModalMoodsToVariant } from "./transformConfirmationModalMoodsToVariant"
 
 const run = (): void => {
-  // eslint-disable-next-line no-console
-  console.log(" ~(-_- ~) Running ConfirmationModal transformer (~ -_-)~")
+  console.log("~(-_- ~) Running ConfirmationModal transformer (~ -_-)~")
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)

package/codemods/migrateConfirmationModalMoodsToVariant/transformConfirmationModalMoodsToVariant.spec.ts

@@ -1,5 +1,5 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst } from "../utils"
+import { transformSourceForTagName, printAst } from "../utils"
 import { transformConfirmationModalMoodsToVariant } from "./transformConfirmationModalMoodsToVariant"
 
 describe("transformConfirmationModalMoodsToVariant", () => {
@@ -10,7 +10,7 @@ describe("transformConfirmationModalMoodsToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <ConfirmationModal variant="success"/>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformConfirmationModalMoodsToVariant,
       tagName: "ConfirmationModal",
@@ -25,7 +25,7 @@ describe("transformConfirmationModalMoodsToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <ConfirmationModal variant="warning"/>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformConfirmationModalMoodsToVariant,
       tagName: "ConfirmationModal",
@@ -40,7 +40,7 @@ describe("transformConfirmationModalMoodsToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <ConfirmationModal variant="warning" id="123"/>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformConfirmationModalMoodsToVariant,
       tagName: "ConfirmationModal",
@@ -55,7 +55,7 @@ describe("transformConfirmationModalMoodsToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><ConfirmationModal variant="success"/><ConfirmationModal variant="warning"/></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformConfirmationModalMoodsToVariant,
       tagName: "ConfirmationModal",
@@ -70,7 +70,7 @@ describe("transformConfirmationModalMoodsToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><ConfirmationModal variant="success"/><ConfirmationModal variant="cautionary"/><ConfirmationModal variant="success"/></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformConfirmationModalMoodsToVariant,
       tagName: "ConfirmationModal",
@@ -85,7 +85,7 @@ describe("transformConfirmationModalMoodsToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><ConfirmationModal variant="success"/></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformConfirmationModalMoodsToVariant,
       tagName: "ConfirmationModal",
@@ -100,7 +100,7 @@ describe("transformConfirmationModalMoodsToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><ConfirmationModal variant={confirmationModalVariable}/></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformConfirmationModalMoodsToVariant,
       tagName: "ConfirmationModal",

package/codemods/migrateEmptyStateIllustrationTypeToVariant/index.ts

@@ -2,9 +2,8 @@ import { transformComponentsInDir } from "../utils"
 import { transformEmptyStateIllustrationTypeToVariant } from "./transformEmptyStateIllustrationTypeToVariant"
 
 const migrateEmptyStateIllustrationTypeToVariant = (): void => {
-  // eslint-disable-next-line no-console
   console.log(
-    " ~(-_- ~) Running EmptyState illustrationType to variant transformer (~ -_-)~"
+    "~(-_- ~) Running EmptyState illustrationType to variant transformer (~ -_-)~"
   )
   const targetDir = process.argv[2]
   if (!targetDir) {

package/codemods/migrateEmptyStateIllustrationTypeToVariant/transformEmptyStateIllustrationTypeToVariant.spec.ts

@@ -1,11 +1,15 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst, TransformConfig } from "../utils"
+import {
+  printAst,
+  transformSourceForTagName,
+  type TransformSourceForTagNameArgs,
+} from "../utils"
 import { transformEmptyStateIllustrationTypeToVariant } from "./transformEmptyStateIllustrationTypeToVariant"
 
 const transformEmptyState = (
-  sourceFile: TransformConfig["sourceFile"]
+  sourceFile: TransformSourceForTagNameArgs["sourceFile"]
 ): string =>
-  transformSource({
+  transformSourceForTagName({
     sourceFile,
     astTransformer: transformEmptyStateIllustrationTypeToVariant,
     tagName: "EmptyState",

package/codemods/migrateGlobalNotificationTypeToVariant/index.ts

@@ -1,9 +1,8 @@
 import { transformNotificationTypeToVariant } from "../migrateNotificationTypeToVariant"
 import { transformComponentsInDir } from "../utils"
-/** This is here as a script runner that migrates GlobalNotification variant to their color equivalent */
+
 const migrateGlobalNotificationTypeToVariant = (): void => {
-  // eslint-disable-next-line no-console
-  console.log(" ~(-_- ~) Running GlobalNotification transformer (~ -_-)~")
+  console.log("~(-_- ~) Running GlobalNotification transformer (~ -_-)~")
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)

package/codemods/migrateInformationTileMoodToVariant/index.ts

@@ -2,9 +2,8 @@ import { transformComponentsInDir } from "../utils"
 import { transformInformationTileMoodToVariant } from "./transformInformationTileMoodToVariant"
 
 const migrateInformationTileMoodToVariant = (): void => {
-  // eslint-disable-next-line no-console
   console.log(
-    " ~(-_- ~) Running InformationTile mood to variant transformer (~ -_-)~"
+    "~(-_- ~) Running InformationTile mood to variant transformer (~ -_-)~"
   )
   const targetDir = process.argv[2]
   if (!targetDir) {

package/codemods/migrateInformationTileMoodToVariant/transformInformationTileMoodToVariant.spec.ts

@@ -1,11 +1,15 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst, TransformConfig } from "../utils"
+import {
+  printAst,
+  transformSourceForTagName,
+  type TransformSourceForTagNameArgs,
+} from "../utils"
 import { transformInformationTileMoodToVariant } from "./transformInformationTileMoodToVariant"
 
 const transformInformationTile = (
-  sourceFile: TransformConfig["sourceFile"]
+  sourceFile: TransformSourceForTagNameArgs["sourceFile"]
 ): string =>
-  transformSource({
+  transformSourceForTagName({
     sourceFile,
     astTransformer: transformInformationTileMoodToVariant,
     tagName: "InformationTile",

package/codemods/migrateInlineNotificationTypeToVariant/index.ts

@@ -1,9 +1,8 @@
 import { transformNotificationTypeToVariant } from "../migrateNotificationTypeToVariant"
 import { transformComponentsInDir } from "../utils"
-/** This is here as a script runner that migrates InlineNotification variant to their color equivalent */
+
 const migrateInlineNotificationTypeToVariant = (): void => {
-  // eslint-disable-next-line no-console
-  console.log(" ~(-_- ~) Running InlineNotification transformer (~ -_-)~")
+  console.log("~(-_- ~) Running InlineNotification transformer (~ -_-)~")
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)

package/codemods/migrateMultiActionTileMoodToVariant/index.ts

@@ -2,7 +2,6 @@ import { transformComponentsInDir } from "../utils"
 import { transformMultiActionTileMoodToVariant } from "./transformMultiActionTileMoodToVariant"
 
 const migrateMultiActionTileMoodToVariant = (): void => {
-  // eslint-disable-next-line no-console
   console.log(
     " ~(-_- ~) Running MultiActionTile mood to variant transformer (~ -_-)~"
   )

package/codemods/migrateMultiActionTileMoodToVariant/transformMultiActionTileMoodToVariant.spec.ts

@@ -1,11 +1,15 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst, TransformConfig } from "../utils"
+import {
+  printAst,
+  transformSourceForTagName,
+  type TransformSourceForTagNameArgs,
+} from "../utils"
 import { transformMultiActionTileMoodToVariant } from "./transformMultiActionTileMoodToVariant"
 
 const transformMultiActionTile = (
-  sourceFile: TransformConfig["sourceFile"]
+  sourceFile: TransformSourceForTagNameArgs["sourceFile"]
 ): string =>
-  transformSource({
+  transformSourceForTagName({
     sourceFile,
     astTransformer: transformMultiActionTileMoodToVariant,
     tagName: "MultiActionTile",

package/codemods/migrateNotificationTypeToVariant/migrateNotificationTypeToVariant.spec.ts

@@ -1,5 +1,5 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst } from "../utils"
+import { transformSourceForTagName, printAst } from "../utils"
 import { transformNotificationTypeToVariant } from "./migrateNotificationTypeToVariant"
 
 describe("transformNotificationTypeToVariant", () => {
@@ -10,7 +10,7 @@ describe("transformNotificationTypeToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InlineNotification variant="success">Test</InlineNotification>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformNotificationTypeToVariant,
       tagName: "InlineNotification",
@@ -25,7 +25,7 @@ describe("transformNotificationTypeToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <GlobalNotification variant="success">Test</GlobalNotification>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformNotificationTypeToVariant,
       tagName: "GlobalNotification",
@@ -40,7 +40,7 @@ describe("transformNotificationTypeToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <ToastNotification variant="success">Test</ToastNotification>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformNotificationTypeToVariant,
       tagName: "ToastNotification",
@@ -55,7 +55,7 @@ describe("transformNotificationTypeToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InlineNotification variant="informative">Test</InlineNotification>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformNotificationTypeToVariant,
       tagName: "InlineNotification",
@@ -69,7 +69,7 @@ describe("transformNotificationTypeToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InlineNotification variant="cautionary">Test</InlineNotification>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformNotificationTypeToVariant,
       tagName: "InlineNotification",
@@ -83,7 +83,7 @@ describe("transformNotificationTypeToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InlineNotification variant="security">Test</InlineNotification>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformNotificationTypeToVariant,
       tagName: "InlineNotification",
@@ -97,7 +97,7 @@ describe("transformNotificationTypeToVariant", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InlineNotification variant="warning">Test</InlineNotification>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformNotificationTypeToVariant,
       tagName: "InlineNotification",

package/codemods/migrateProgressBarMoodToColor/index.ts

@@ -2,7 +2,6 @@ import { transformComponentsInDir } from "../utils"
 import { transformProgressBarMoodToColor } from "./transformProgressBarMoodToColor"
 
 const migrateProgressBarMoodToColor = (): void => {
-  // eslint-disable-next-line no-console
   console.log(
     " ~(-_- ~) Running ProgressBar mood to color transformer (~ -_-)~"
   )

package/codemods/migrateProgressBarMoodToColor/transformProgressBarMoodToColor.spec.ts

@@ -1,11 +1,15 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst, TransformConfig } from "../utils"
+import {
+  printAst,
+  transformSourceForTagName,
+  type TransformSourceForTagNameArgs,
+} from "../utils"
 import { transformProgressBarMoodToColor } from "./transformProgressBarMoodToColor"
 
 const transformProgressBar = (
-  sourceFile: TransformConfig["sourceFile"]
+  sourceFile: TransformSourceForTagNameArgs["sourceFile"]
 ): string =>
-  transformSource({
+  transformSourceForTagName({
     sourceFile,
     astTransformer: transformProgressBarMoodToColor,
     tagName: "ProgressBar",

package/codemods/migrateToastNotificationTypeToVariant/index.ts

@@ -1,8 +1,7 @@
 import { transformNotificationTypeToVariant } from "../migrateNotificationTypeToVariant"
 import { transformComponentsInDir } from "../utils"
-/** This is here as a script runner that migrates ToastNotification variant to their color equivalent */
+
 const migrateToastNotificationTypeToVariant = (): void => {
-  // eslint-disable-next-line no-console
   console.log(" ~(-_- ~) Running ToastNotification transformer (~ -_-)~")
   const targetDir = process.argv[2]
   if (!targetDir) {

package/codemods/migrateWellVariantToColor/index.ts

@@ -1,8 +1,7 @@
 import { transformComponentsInDir } from "../utils"
 import { transformWellVariantToColor } from "./transformWellVariantToColor"
-/** This is here as a script runner that migrates Well variant to their color equivalent */
+
 const migrateWellVariantToColor = (): void => {
-  // eslint-disable-next-line no-console
   console.log(" ~(-_- ~) Running Well transformer (~ -_-)~")
   const targetDir = process.argv[2]
   if (!targetDir) {

package/codemods/migrateWellVariantToColor/transformWellVariantToColor.spec.ts

@@ -1,5 +1,5 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst } from "../utils"
+import { transformSourceForTagName, printAst } from "../utils"
 import { transformWellVariantToColor } from "./transformWellVariantToColor"
 
 describe("transformWellVariantToColor", () => {
@@ -12,7 +12,7 @@ describe("transformWellVariantToColor", () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <Well color="gray">Test</Well>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "Well",
@@ -29,7 +29,7 @@ describe("transformWellVariantToColor", () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <Well color="blue">Test</Well>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "Well",
@@ -46,7 +46,7 @@ describe("transformWellVariantToColor", () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <Well color="gray">Test</Well>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "Well",
@@ -63,7 +63,7 @@ describe("transformWellVariantToColor", () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <Well color="blue" id="123">Test</Well>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "Well",
@@ -80,7 +80,7 @@ describe("transformWellVariantToColor", () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <Well id="123" color="gray">Test</Well>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "Well",
@@ -97,7 +97,7 @@ describe("transformWellVariantToColor", () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <div><Well id="123" color="gray">Test</Well></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "Well",
@@ -114,7 +114,7 @@ describe("transformWellVariantToColor", () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <div><Well color="blue">Test</Well><Well color="gray">Test 2</Well></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "Well",
@@ -131,7 +131,7 @@ describe("transformWellVariantToColor", () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <div><Well color="blue">Test</Well><Well color="orange">Test</Well><Well color="green">Test</Well></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "Well",
@@ -148,7 +148,7 @@ describe("transformWellVariantToColor", () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <div><Well color="blue">Test</Well></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "Well",
@@ -165,7 +165,7 @@ describe("transformWellVariantToColor", () => {
       import {Well} from "@kaizen/components"
       export const TestComponent = () => <div><Well variant={wellVariable}>Test</Well></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "Well",
@@ -182,7 +182,7 @@ describe("transformWellVariantToColor", () => {
       import {Well as KaizenWell} from "@kaizen/components"
       export const TestComponent = () => <div><KaizenWell color="blue">Test</KaizenWell></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: transformWellVariantToColor,
       tagName: "KaizenWell",

package/codemods/migrateWellVariantToColor/transformWellVariantToColor.ts

@@ -2,10 +2,9 @@ import ts from "typescript"
 import { updateJsxElementWithNewProps } from "../utils"
 import { getPropValueText } from "../utils/getPropValueText"
 
-/** Recurses through AST to find and update any jsx element that matched the tagName */
 export const transformWellVariantToColor =
   (context: ts.TransformationContext, tagName: string) =>
-  (rootNode: ts.Node): ts.Node => {
+  (rootNode: ts.SourceFile): ts.SourceFile => {
     function visit(node: ts.Node): ts.Node {
       if (ts.isJsxOpeningElement(node) || ts.isJsxSelfClosingElement(node)) {
         if (node.tagName.getText() === tagName) {
@@ -71,5 +70,5 @@ export const transformWellVariantToColor =
       }
       return ts.visitEachChild(node, visit, context)
     }
-    return ts.visitNode(rootNode, visit)
+    return ts.visitNode(rootNode, visit) as ts.SourceFile
   }

package/codemods/removeInputEditModalMood/index.ts

@@ -1,8 +1,7 @@
 import { transformComponentsInDir } from "../utils"
 import { removeInputEditModalMood } from "./removeInputEditModalMood"
-/** This is here as a script runner that migrates Well variant to their color equivalent */
+
 const run = (): void => {
-  // eslint-disable-next-line no-console
   console.log(" ~(-_- ~) Running InputEditModal transformer (~ -_-)~")
   const targetDir = process.argv[2]
   if (!targetDir) {

package/codemods/removeInputEditModalMood/removeInputEditModalMood.spec.ts

@@ -1,5 +1,5 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst } from "../utils"
+import { transformSourceForTagName, printAst } from "../utils"
 import { removeInputEditModalMood } from "./removeInputEditModalMood"
 
 describe("removeInputEditModalMood", () => {
@@ -10,7 +10,7 @@ describe("removeInputEditModalMood", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InputEditModal />
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: removeInputEditModalMood,
       tagName: "InputEditModal",
@@ -25,7 +25,7 @@ describe("removeInputEditModalMood", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InputEditModal id="123"/>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: removeInputEditModalMood,
       tagName: "InputEditModal",
@@ -40,7 +40,7 @@ describe("removeInputEditModalMood", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><InputEditModal /><InputEditModal /></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: removeInputEditModalMood,
       tagName: "InputEditModal",

package/codemods/removePopoverVariant/index.ts

@@ -2,8 +2,7 @@ import { transformComponentsInDir } from "../utils"
 import { removePopoverVariant } from "./removePopoverVariant"
 
 const run = (): void => {
-  // eslint-disable-next-line no-console
-  console.log(" ~(-_- ~) Running Popover transformer (~ -_-)~")
+  console.log("~(-_- ~) Running Popover transformer (~ -_-)~")
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)

package/codemods/removePopoverVariant/removePopoverVariant.spec.ts

@@ -1,5 +1,5 @@
 import { parseJsx } from "../__tests__/utils"
-import { transformSource, printAst } from "../utils"
+import { transformSourceForTagName, printAst } from "../utils"
 import { removePopoverVariant } from "./removePopoverVariant"
 
 describe("removePopoverVariant()", () => {
@@ -10,7 +10,7 @@ describe("removePopoverVariant()", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <Popover />
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: removePopoverVariant,
       tagName: "Popover",
@@ -25,7 +25,7 @@ describe("removePopoverVariant()", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <Popover />
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: removePopoverVariant,
       tagName: "Popover",
@@ -40,7 +40,7 @@ describe("removePopoverVariant()", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <Popover id="123"/>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: removePopoverVariant,
       tagName: "Popover",
@@ -55,7 +55,7 @@ describe("removePopoverVariant()", () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><Popover /><Popover /></div>
     `)
-    const transformed = transformSource({
+    const transformed = transformSourceForTagName({
       sourceFile: inputAst,
       astTransformer: removePopoverVariant,
       tagName: "Popover",

package/codemods/upgradeIconV1/index.ts

@@ -1,11 +1,11 @@
 import fs from "fs"
-import ts from "typescript"
 import { getKaioTagNamesByRegex, transformSource, traverseDir } from "../utils"
+import { createEncodedSourceFile } from "../utils/createEncodedSourceFile"
 import { upgradeIconV1 } from "./upgradeIconV1"
 
 const run = (): void => {
-  // eslint-disable-next-line no-console
   console.log("~(-_- ~) Running Icon v1 to Future upgrade (~ -_-)~")
+
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
@@ -13,14 +13,14 @@ const run = (): void => {
 
   const transformFile = (
     componentFilePath: string,
-    sourceFile: ts.SourceFile
+    sourceCode: string
   ): void => {
+    const sourceFile = createEncodedSourceFile(componentFilePath, sourceCode)
     const tagNames = getKaioTagNamesByRegex(sourceFile, "Icon$")
     if (tagNames) {
       const updatedSourceFile = transformSource({
         sourceFile,
-        astTransformer: upgradeIconV1,
-        tagName: tagNames,
+        transformers: [upgradeIconV1(tagNames)],
       })
 
       fs.writeFileSync(componentFilePath, updatedSourceFile, "utf8")