@kaizen/components 1.69.2 → 1.70.1

package/codemods/migrateBrandMomentMoodToVariant/index.ts

@@ -1,14 +1,16 @@
 import { transformComponentsInDir } from '../utils'
 import { transformBrandMomentMoodToVariant } from './transformBrandMomentMoodToVariant'
 
-const migrateBrandMomentMoodToVariant = (): void => {
+const run = (): void => {
   console.log('~(-_- ~) Running BrandMoment mood to variant transformer (~ -_-)~')
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformBrandMomentMoodToVariant, 'BrandMoment')
+  transformComponentsInDir(targetDir, ['BrandMoment'], (tagNames) => [
+    transformBrandMomentMoodToVariant(tagNames),
+  ])
 }
 
-migrateBrandMomentMoodToVariant()
+run()

package/codemods/migrateBrandMomentMoodToVariant/transformBrandMomentMoodToVariant.spec.ts

@@ -1,13 +1,24 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName, type TransformSourceForTagNameArgs } from '../utils'
+import { printAst, transformSource, type TransformSourceArgs } from '../utils'
 import { transformBrandMomentMoodToVariant } from './transformBrandMomentMoodToVariant'
 
-const transformBrandMoment = (sourceFile: TransformSourceForTagNameArgs['sourceFile']): string =>
-  transformSourceForTagName({
+const transformBrandMoment = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const tagsMap = new Map([
+    [
+      'BrandMoment',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'BrandMoment',
+        originalName: 'BrandMoment',
+      },
+    ],
+  ])
+
+  return transformSource({
     sourceFile,
-    astTransformer: transformBrandMomentMoodToVariant,
-    tagName: 'BrandMoment',
+    transformers: [transformBrandMomentMoodToVariant(tagsMap)],
   })
+}
 
 describe('transformBrandMomentMoodToVariant()', () => {
   it('replaces mood="informative" with variant="informative"', () => {

package/codemods/migrateCardVariantToColor/index.ts

@@ -1,14 +1,16 @@
 import { transformComponentsInDir } from '../utils'
 import { transformCardVariantToColor } from './transformCardVariantToColor'
 
-const migrateCardVariantToColor = (): void => {
+const run = (): void => {
   console.log('~(-_- ~) Running Card variant to color transformer (~ -_-)~')
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformCardVariantToColor, 'Card')
+  transformComponentsInDir(targetDir, ['Card'], (tagNames) => [
+    transformCardVariantToColor(tagNames),
+  ])
 }
 
-migrateCardVariantToColor()
+run()

package/codemods/migrateCardVariantToColor/transformCardVariantToColor.spec.ts

@@ -1,13 +1,24 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName, type TransformSourceForTagNameArgs } from '../utils'
+import { printAst, transformSource, type TransformSourceArgs } from '../utils'
 import { transformCardVariantToColor } from './transformCardVariantToColor'
 
-const transformCard = (sourceFile: TransformSourceForTagNameArgs['sourceFile']): string =>
-  transformSourceForTagName({
+const transformCard = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const tagsMap = new Map([
+    [
+      'Card',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'Card',
+        originalName: 'Card',
+      },
+    ],
+  ])
+
+  return transformSource({
     sourceFile,
-    astTransformer: transformCardVariantToColor,
-    tagName: 'Card',
+    transformers: [transformCardVariantToColor(tagsMap)],
   })
+}
 
 describe('transformCardVariantToColor()', () => {
   it('replaces variant="assertive" with color="orange"', () => {

package/codemods/migrateConfirmationModalMoodsToVariant/index.ts

@@ -8,7 +8,9 @@ const run = (): void => {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformConfirmationModalMoodsToVariant, 'ConfirmationModal')
+  transformComponentsInDir(targetDir, ['ConfirmationModal'], (tagNames) => [
+    transformConfirmationModalMoodsToVariant(tagNames),
+  ])
 }
 
 run()

package/codemods/migrateConfirmationModalMoodsToVariant/transformConfirmationModalMoodsToVariant.spec.ts

@@ -1,8 +1,26 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName } from '../utils'
+import { printAst, transformSource, type TransformSourceArgs } from '../utils'
 import { transformConfirmationModalMoodsToVariant } from './transformConfirmationModalMoodsToVariant'
 
-describe('transformConfirmationModalMoodsToVariant', () => {
+const transformConfirmationModal = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const tagsMap = new Map([
+    [
+      'ConfirmationModal',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'ConfirmationModal',
+        originalName: 'ConfirmationModal',
+      },
+    ],
+  ])
+
+  return transformSource({
+    sourceFile,
+    transformers: [transformConfirmationModalMoodsToVariant(tagsMap)],
+  })
+}
+
+describe('transformConfirmationModalMoodsToVariant()', () => {
   it('replaces mood="positive" with variant="success"', () => {
     const inputAst = parseJsx(`
       export const TestComponent = () => <ConfirmationModal mood="positive"/>
@@ -10,12 +28,7 @@ describe('transformConfirmationModalMoodsToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <ConfirmationModal variant="success"/>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformConfirmationModalMoodsToVariant,
-      tagName: 'ConfirmationModal',
-    })
-    expect(transformed).toEqual(printAst(outputAst))
+    expect(transformConfirmationModal(inputAst)).toEqual(printAst(outputAst))
   })
 
   it('replaces mood="negative" with variant="warning"', () => {
@@ -25,12 +38,7 @@ describe('transformConfirmationModalMoodsToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <ConfirmationModal variant="warning"/>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformConfirmationModalMoodsToVariant,
-      tagName: 'ConfirmationModal',
-    })
-    expect(transformed).toEqual(printAst(outputAst))
+    expect(transformConfirmationModal(inputAst)).toEqual(printAst(outputAst))
   })
 
   it('handles multiple attributes and replace only variant', () => {
@@ -40,12 +48,7 @@ describe('transformConfirmationModalMoodsToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <ConfirmationModal variant="warning" id="123"/>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformConfirmationModalMoodsToVariant,
-      tagName: 'ConfirmationModal',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformConfirmationModal(inputAst)).toBe(printAst(outputAst))
   })
 
   it('transforms multiple ConfirmationModals', () => {
@@ -55,12 +58,7 @@ describe('transformConfirmationModalMoodsToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><ConfirmationModal variant="success"/><ConfirmationModal variant="warning"/></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformConfirmationModalMoodsToVariant,
-      tagName: 'ConfirmationModal',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformConfirmationModal(inputAst)).toBe(printAst(outputAst))
   })
 
   it('transforms ConfirmationModal with arbitrary braces', () => {
@@ -70,12 +68,7 @@ describe('transformConfirmationModalMoodsToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><ConfirmationModal variant="success"/><ConfirmationModal variant="cautionary"/><ConfirmationModal variant="success"/></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformConfirmationModalMoodsToVariant,
-      tagName: 'ConfirmationModal',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformConfirmationModal(inputAst)).toBe(printAst(outputAst))
   })
 
   it("won't add variant if variant already exists", () => {
@@ -85,12 +78,7 @@ describe('transformConfirmationModalMoodsToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><ConfirmationModal variant="success"/></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformConfirmationModalMoodsToVariant,
-      tagName: 'ConfirmationModal',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformConfirmationModal(inputAst)).toBe(printAst(outputAst))
   })
 
   it("won't modify variants usings variables", () => {
@@ -100,11 +88,6 @@ describe('transformConfirmationModalMoodsToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <div><ConfirmationModal variant={confirmationModalVariable}/></div>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformConfirmationModalMoodsToVariant,
-      tagName: 'ConfirmationModal',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformConfirmationModal(inputAst)).toBe(printAst(outputAst))
   })
 })

package/codemods/migrateEmptyStateIllustrationTypeToVariant/index.ts

@@ -1,14 +1,16 @@
 import { transformComponentsInDir } from '../utils'
 import { transformEmptyStateIllustrationTypeToVariant } from './transformEmptyStateIllustrationTypeToVariant'
 
-const migrateEmptyStateIllustrationTypeToVariant = (): void => {
+const run = (): void => {
   console.log('~(-_- ~) Running EmptyState illustrationType to variant transformer (~ -_-)~')
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformEmptyStateIllustrationTypeToVariant, 'EmptyState')
+  transformComponentsInDir(targetDir, ['EmptyState'], (tagNames) => [
+    transformEmptyStateIllustrationTypeToVariant(tagNames),
+  ])
 }
 
-migrateEmptyStateIllustrationTypeToVariant()
+run()

package/codemods/migrateEmptyStateIllustrationTypeToVariant/transformEmptyStateIllustrationTypeToVariant.spec.ts

@@ -1,13 +1,24 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName, type TransformSourceForTagNameArgs } from '../utils'
+import { printAst, transformSource, type TransformSourceArgs } from '../utils'
 import { transformEmptyStateIllustrationTypeToVariant } from './transformEmptyStateIllustrationTypeToVariant'
 
-const transformEmptyState = (sourceFile: TransformSourceForTagNameArgs['sourceFile']): string =>
-  transformSourceForTagName({
+const transformEmptyState = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const tagsMap = new Map([
+    [
+      'EmptyState',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'EmptyState',
+        originalName: 'EmptyState',
+      },
+    ],
+  ])
+
+  return transformSource({
     sourceFile,
-    astTransformer: transformEmptyStateIllustrationTypeToVariant,
-    tagName: 'EmptyState',
+    transformers: [transformEmptyStateIllustrationTypeToVariant(tagsMap)],
   })
+}
 
 describe('transformEmptyStateIllustrationTypeToVariant()', () => {
   it('replaces illustrationType="positive" with variant="success"', () => {

package/codemods/migrateGlobalNotificationTypeToVariant/index.ts

@@ -1,14 +1,16 @@
 import { transformNotificationTypeToVariant } from '../migrateNotificationTypeToVariant'
 import { transformComponentsInDir } from '../utils'
 
-const migrateGlobalNotificationTypeToVariant = (): void => {
+const run = (): void => {
   console.log('~(-_- ~) Running GlobalNotification transformer (~ -_-)~')
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformNotificationTypeToVariant, 'GlobalNotification')
+  transformComponentsInDir(targetDir, ['GlobalNotification'], (tagNames) => [
+    transformNotificationTypeToVariant(tagNames),
+  ])
 }
 
-migrateGlobalNotificationTypeToVariant()
+run()

package/codemods/migrateInformationTileMoodToVariant/index.ts

@@ -1,14 +1,16 @@
 import { transformComponentsInDir } from '../utils'
 import { transformInformationTileMoodToVariant } from './transformInformationTileMoodToVariant'
 
-const migrateInformationTileMoodToVariant = (): void => {
+const run = (): void => {
   console.log('~(-_- ~) Running InformationTile mood to variant transformer (~ -_-)~')
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformInformationTileMoodToVariant, 'InformationTile')
+  transformComponentsInDir(targetDir, ['InformationTile'], (tagNames) => [
+    transformInformationTileMoodToVariant(tagNames),
+  ])
 }
 
-migrateInformationTileMoodToVariant()
+run()

package/codemods/migrateInformationTileMoodToVariant/transformInformationTileMoodToVariant.spec.ts

@@ -1,15 +1,24 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName, type TransformSourceForTagNameArgs } from '../utils'
+import { printAst, transformSource, type TransformSourceArgs } from '../utils'
 import { transformInformationTileMoodToVariant } from './transformInformationTileMoodToVariant'
 
-const transformInformationTile = (
-  sourceFile: TransformSourceForTagNameArgs['sourceFile'],
-): string =>
-  transformSourceForTagName({
+const transformInformationTile = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const tagsMap = new Map([
+    [
+      'InformationTile',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'InformationTile',
+        originalName: 'InformationTile',
+      },
+    ],
+  ])
+
+  return transformSource({
     sourceFile,
-    astTransformer: transformInformationTileMoodToVariant,
-    tagName: 'InformationTile',
+    transformers: [transformInformationTileMoodToVariant(tagsMap)],
   })
+}
 
 describe('transformInformationTileMoodToVariant()', () => {
   it('replaces mood="positive" with variant="default"', () => {

package/codemods/migrateInlineNotificationTypeToVariant/index.ts

@@ -1,14 +1,16 @@
 import { transformNotificationTypeToVariant } from '../migrateNotificationTypeToVariant'
 import { transformComponentsInDir } from '../utils'
 
-const migrateInlineNotificationTypeToVariant = (): void => {
+const run = (): void => {
   console.log('~(-_- ~) Running InlineNotification transformer (~ -_-)~')
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformNotificationTypeToVariant, 'InlineNotification')
+  transformComponentsInDir(targetDir, ['InlineNotification'], (tagNames) => [
+    transformNotificationTypeToVariant(tagNames),
+  ])
 }
 
-migrateInlineNotificationTypeToVariant()
+run()

package/codemods/migrateMultiActionTileMoodToVariant/index.ts

@@ -1,14 +1,16 @@
 import { transformComponentsInDir } from '../utils'
 import { transformMultiActionTileMoodToVariant } from './transformMultiActionTileMoodToVariant'
 
-const migrateMultiActionTileMoodToVariant = (): void => {
+const run = (): void => {
   console.log(' ~(-_- ~) Running MultiActionTile mood to variant transformer (~ -_-)~')
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformMultiActionTileMoodToVariant, 'MultiActionTile')
+  transformComponentsInDir(targetDir, ['MultiActionTile'], (tagNames) => [
+    transformMultiActionTileMoodToVariant(tagNames),
+  ])
 }
 
-migrateMultiActionTileMoodToVariant()
+run()

package/codemods/migrateMultiActionTileMoodToVariant/transformMultiActionTileMoodToVariant.spec.ts

@@ -1,15 +1,24 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName, type TransformSourceForTagNameArgs } from '../utils'
+import { printAst, transformSource, type TransformSourceArgs } from '../utils'
 import { transformMultiActionTileMoodToVariant } from './transformMultiActionTileMoodToVariant'
 
-const transformMultiActionTile = (
-  sourceFile: TransformSourceForTagNameArgs['sourceFile'],
-): string =>
-  transformSourceForTagName({
+const transformMultiActionTile = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const tagsMap = new Map([
+    [
+      'MultiActionTile',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'MultiActionTile',
+        originalName: 'MultiActionTile',
+      },
+    ],
+  ])
+
+  return transformSource({
     sourceFile,
-    astTransformer: transformMultiActionTileMoodToVariant,
-    tagName: 'MultiActionTile',
+    transformers: [transformMultiActionTileMoodToVariant(tagsMap)],
   })
+}
 
 describe('transformMultiActionTileMoodToVariant()', () => {
   it('replaces mood="positive" with variant="default"', () => {

package/codemods/migrateNotificationTypeToVariant/migrateNotificationTypeToVariant.spec.ts

@@ -1,7 +1,41 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName } from '../utils'
+import { printAst, transformSource, type TransformSourceArgs } from '../utils'
 import { transformNotificationTypeToVariant } from './migrateNotificationTypeToVariant'
 
+const transformNotifications = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const tagsMap = new Map([
+    [
+      'GlobalNotification',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'GlobalNotification',
+        originalName: 'GlobalNotification',
+      },
+    ],
+    [
+      'InlineNotification',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'InlineNotification',
+        originalName: 'InlineNotification',
+      },
+    ],
+    [
+      'ToastNotification',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'ToastNotification',
+        originalName: 'ToastNotification',
+      },
+    ],
+  ])
+
+  return transformSource({
+    sourceFile,
+    transformers: [transformNotificationTypeToVariant(tagsMap)],
+  })
+}
+
 describe('transformNotificationTypeToVariant', () => {
   it('replaces InlineNotifications type="positive" with variant="success"', () => {
     const inputAst = parseJsx(`
@@ -10,12 +44,7 @@ describe('transformNotificationTypeToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InlineNotification variant="success">Test</InlineNotification>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformNotificationTypeToVariant,
-      tagName: 'InlineNotification',
-    })
-    expect(transformed).toEqual(printAst(outputAst))
+    expect(transformNotifications(inputAst)).toEqual(printAst(outputAst))
   })
 
   it('replaces GlobalNotification type="positive" with variant="success"', () => {
@@ -25,12 +54,7 @@ describe('transformNotificationTypeToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <GlobalNotification variant="success">Test</GlobalNotification>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformNotificationTypeToVariant,
-      tagName: 'GlobalNotification',
-    })
-    expect(transformed).toEqual(printAst(outputAst))
+    expect(transformNotifications(inputAst)).toEqual(printAst(outputAst))
   })
 
   it('replaces ToastNotification type="positive" with variant="success"', () => {
@@ -40,12 +64,7 @@ describe('transformNotificationTypeToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <ToastNotification variant="success">Test</ToastNotification>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformNotificationTypeToVariant,
-      tagName: 'ToastNotification',
-    })
-    expect(transformed).toEqual(printAst(outputAst))
+    expect(transformNotifications(inputAst)).toEqual(printAst(outputAst))
   })
 
   it('replaces type="informative" with variant="informative"', () => {
@@ -55,12 +74,7 @@ describe('transformNotificationTypeToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InlineNotification variant="informative">Test</InlineNotification>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformNotificationTypeToVariant,
-      tagName: 'InlineNotification',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformNotifications(inputAst)).toBe(printAst(outputAst))
   })
   it('replaces type="cautionary" with variant="cautionary"', () => {
     const inputAst = parseJsx(`
@@ -69,12 +83,7 @@ describe('transformNotificationTypeToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InlineNotification variant="cautionary">Test</InlineNotification>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformNotificationTypeToVariant,
-      tagName: 'InlineNotification',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformNotifications(inputAst)).toBe(printAst(outputAst))
   })
   it('replaces type="security" with variant="security"', () => {
     const inputAst = parseJsx(`
@@ -83,12 +92,7 @@ describe('transformNotificationTypeToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InlineNotification variant="security">Test</InlineNotification>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformNotificationTypeToVariant,
-      tagName: 'InlineNotification',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformNotifications(inputAst)).toBe(printAst(outputAst))
   })
   it('replaces type="negative" with variant="warning"', () => {
     const inputAst = parseJsx(`
@@ -97,11 +101,6 @@ describe('transformNotificationTypeToVariant', () => {
     const outputAst = parseJsx(`
       export const TestComponent = () => <InlineNotification variant="warning">Test</InlineNotification>
     `)
-    const transformed = transformSourceForTagName({
-      sourceFile: inputAst,
-      astTransformer: transformNotificationTypeToVariant,
-      tagName: 'InlineNotification',
-    })
-    expect(transformed).toBe(printAst(outputAst))
+    expect(transformNotifications(inputAst)).toBe(printAst(outputAst))
   })
 })

package/codemods/migrateProgressBarMoodToColor/index.ts

@@ -1,14 +1,16 @@
 import { transformComponentsInDir } from '../utils'
 import { transformProgressBarMoodToColor } from './transformProgressBarMoodToColor'
 
-const migrateProgressBarMoodToColor = (): void => {
+const run = (): void => {
   console.log(' ~(-_- ~) Running ProgressBar mood to color transformer (~ -_-)~')
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformProgressBarMoodToColor, 'ProgressBar')
+  transformComponentsInDir(targetDir, ['ProgressBar'], (tagNames) => [
+    transformProgressBarMoodToColor(tagNames),
+  ])
 }
 
-migrateProgressBarMoodToColor()
+run()

package/codemods/migrateProgressBarMoodToColor/transformProgressBarMoodToColor.spec.ts

@@ -1,13 +1,24 @@
 import { parseJsx } from '../__tests__/utils'
-import { printAst, transformSourceForTagName, type TransformSourceForTagNameArgs } from '../utils'
+import { printAst, transformSource, type TransformSourceArgs } from '../utils'
 import { transformProgressBarMoodToColor } from './transformProgressBarMoodToColor'
 
-const transformProgressBar = (sourceFile: TransformSourceForTagNameArgs['sourceFile']): string =>
-  transformSourceForTagName({
+const transformProgressBar = (sourceFile: TransformSourceArgs['sourceFile']): string => {
+  const tagsMap = new Map([
+    [
+      'ProgressBar',
+      {
+        importModuleName: '@kaizen/components',
+        tagName: 'ProgressBar',
+        originalName: 'ProgressBar',
+      },
+    ],
+  ])
+
+  return transformSource({
     sourceFile,
-    astTransformer: transformProgressBarMoodToColor,
-    tagName: 'ProgressBar',
+    transformers: [transformProgressBarMoodToColor(tagsMap)],
   })
+}
 
 describe('transformProgressBarMoodToColor()', () => {
   it('replaces mood="cautionary" with color="yellow"', () => {

package/codemods/migrateToastNotificationTypeToVariant/index.ts

@@ -1,14 +1,16 @@
 import { transformNotificationTypeToVariant } from '../migrateNotificationTypeToVariant'
 import { transformComponentsInDir } from '../utils'
 
-const migrateToastNotificationTypeToVariant = (): void => {
-  console.log(' ~(-_- ~) Running ToastNotification transformer (~ -_-)~')
+const run = (): void => {
+  console.log('~(-_- ~) Running ToastNotification transformer (~ -_-)~')
   const targetDir = process.argv[2]
   if (!targetDir) {
     process.exit(1)
   }
 
-  transformComponentsInDir(targetDir, transformNotificationTypeToVariant, 'ToastNotification')
+  transformComponentsInDir(targetDir, ['ToastNotification'], (tagNames) => [
+    transformNotificationTypeToVariant(tagNames),
+  ])
 }
 
-migrateToastNotificationTypeToVariant()
+run()