refactor: cleanup

varletjs · Jun 24, 2024 · ee73126 · ee73126
1 parent 5d60dfa
commit ee73126
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 21 deletions.
diff --git a/packages/varlet-icon-builder/src/commands/generate.ts b/packages/varlet-icon-builder/src/commands/generate.ts
@@ -114,7 +114,13 @@ export async function generateModule(options: GenerateModuleOptions) {
         content = content.replace(/\.vue/g, outputExtname)
       }
 
-      return getTransformResult(content, framework, format, filename, outputExtname)
+      return getTransformResult({
+        filename,
+        content,
+        loader: framework === 'vue3' ? 'ts' : 'tsx',
+        format,
+        outputExtname,
+      })
     }),
   )
 

diff --git a/packages/varlet-icon-builder/src/utils/esbuild.ts b/packages/varlet-icon-builder/src/utils/esbuild.ts
@@ -1,27 +1,21 @@
-import { GenerateFramework } from './config.js'
 import esbuild from 'esbuild'
 
-export function getEsbuildLoader(framework: GenerateFramework) {
-  switch (framework) {
-    case GenerateFramework.vue3:
-      return 'ts'
-    case GenerateFramework.react:
-      return 'tsx'
-    default:
-      return 'ts'
-  }
-}
-
-export function getTransformResult(
-  content: string,
-  framework: GenerateFramework,
-  format: 'cjs' | 'esm',
-  filename: string,
-  outputExtname: string,
-) {
+export function getTransformResult({
+  filename,
+  content,
+  loader,
+  format,
+  outputExtname,
+}: {
+  filename: string
+  content: string
+  loader: 'ts' | 'tsx'
+  format: 'cjs' | 'esm'
+  outputExtname: string
+}) {
   return esbuild
     .transform(content, {
-      loader: getEsbuildLoader(framework),
+      loader,
       target: 'es2016',
       format,
     })