refactor(bezier-codemod): add bezier context to enum transform function

channel-io · Dec 22, 2023 · 4a70aab · 4a70aab
1 parent bcf4880
commit 4a70aab
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 18 deletions.
diff --git a/packages/bezier-codemod/src/shared/enum.ts b/packages/bezier-codemod/src/shared/enum.ts
@@ -11,7 +11,7 @@ type Member = string
 type Value = string
 export type EnumTransformMap = Record<Name, Record<Member, Value>>
 
-export const transformEnumMemberToStringLiteral = (sourceFile: SourceFile, enumTransforms: EnumTransformMap) => {
+export const transformEnumTostringLiteralInBezier = (sourceFile: SourceFile, enumTransforms: EnumTransformMap) => {
   const transformedEnumNames: string[] = []
 
   Object

diff --git a/packages/bezier-codemod/src/transforms/enum-member-to-string-literal/transform.ts b/packages/bezier-codemod/src/transforms/enum-member-to-string-literal/transform.ts
@@ -2,7 +2,7 @@ import { type SourceFile } from 'ts-morph'
 
 import {
   type EnumTransformMap,
-  transformEnumMemberToStringLiteral,
+  transformEnumTostringLiteralInBezier,
 } from '../../shared/enum.js'
 
 const ENUM_TRANSFORM_MAP: EnumTransformMap = {
@@ -18,7 +18,7 @@ const ENUM_TRANSFORM_MAP: EnumTransformMap = {
 }
 
 function enumMemberToStringLiteral(sourceFile: SourceFile): true | void {
-  return transformEnumMemberToStringLiteral(sourceFile, ENUM_TRANSFORM_MAP)
+  return transformEnumTostringLiteralInBezier(sourceFile, ENUM_TRANSFORM_MAP)
 }
 
 export default enumMemberToStringLiteral
diff --git a/packages/bezier-codemod/src/transforms/v2-z-index-interpolation-to-css-variable/transform.ts b/packages/bezier-codemod/src/transforms/v2-z-index-interpolation-to-css-variable/transform.ts
@@ -1,11 +1,11 @@
 /* eslint-disable no-template-curly-in-string */
 import { type SourceFile } from 'ts-morph'
 
-import { transformEnumMemberToStringLiteral } from '../../shared/enum.js'
+import { transformEnumTostringLiteralInBezier } from '../../shared/enum.js'
 import { interpolationTransform } from '../../shared/interpolation.js'
 import { removeUnusedNamedImport } from '../../utils/import.js'
 
-const cssVariableByInterpolation = {
+const CSS_VARIABLE_TRANSFORM_MAP = {
   'ZIndex.Hide': 'var(--z-index-hidden);',
   'ZIndex.Auto': 'var(--z-index-auto);',
   'ZIndex.Base': 'var(--z-index-base);',
@@ -17,22 +17,24 @@ const cssVariableByInterpolation = {
   'ZIndex.Important': 'var(--z-index-important);',
 }
 
+const ENUM_TRANSFORM_MAP = {
+  ZIndex: {
+    Hide: 'var(--z-index-hidden)',
+    Base: 'var(--z-index-base)',
+    Float: 'var(--z-index-float)',
+    Overlay: 'var(--z-index-overlay)',
+    Modal: 'var(--z-index-modal)',
+    Toast: 'var(--z-index-toast)',
+    Tooltip: 'var(--z-index-tooltip)',
+    Important: 'var(--z-index-important)',
+  },
+}
+
 const replaceZIndexInterpolation = (sourceFile: SourceFile) => {
   const oldSourceFileText = sourceFile.getText()
 
-  interpolationTransform(sourceFile, cssVariableByInterpolation)
-  transformEnumMemberToStringLiteral(sourceFile, {
-    ZIndex: {
-      Hide: 'var(--z-index-hidden)',
-      Base: 'var(--z-index-base)',
-      Float: 'var(--z-index-float)',
-      Overlay: 'var(--z-index-overlay)',
-      Modal: 'var(--z-index-modal)',
-      Toast: 'var(--z-index-toast)',
-      Tooltip: 'var(--z-index-tooltip)',
-      Important: 'var(--z-index-important)',
-    },
-  })
+  interpolationTransform(sourceFile, CSS_VARIABLE_TRANSFORM_MAP)
+  transformEnumTostringLiteralInBezier(sourceFile, ENUM_TRANSFORM_MAP)
 
   const isChanged = sourceFile.getText() !== oldSourceFileText
   if (isChanged) {