chore: optimized flawed codes

qmhc · Sep 30, 2021 · 4792da3 · 4792da3
1 parent 109721e
commit 4792da3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 12 deletions.
diff --git a/src/compile.ts b/src/compile.ts
@@ -48,19 +48,10 @@ export function compileVueCode(code: string) {
       content = transferSetupPosition(content)
       content += '\nexport default _sfc_main\n'
 
-      if (scriptSetup.lang === 'ts' || scriptSetup.lang === 'tsx') {
-        ext = scriptSetup.lang
-      } else if (!scriptSetup.lang || scriptSetup.lang === 'js'|| scriptSetup.lang === 'jsx') {
-        ext = scriptSetup.lang || 'js'
-      }
+      ext = scriptSetup.lang || 'js'
     } else if (script && script.content) {
       content = script.content
-
-      if (script.lang === 'ts' || script.lang === 'tsx') {
-        ext = script.lang
-      } else if (!script.lang || script.lang === 'js' || script.lang === 'jsx') {
-        ext = script.lang || 'js'
-      }
+      ext = script.lang || 'js'
     }
   }
 

diff --git a/src/index.ts b/src/index.ts
@@ -160,7 +160,7 @@ export default function dtsPlugin(options: PluginOptions = {}): Plugin {
         if (content) {
           if (ext === 'js' || ext === 'jsx') hasJsVue = true
 
-          project.createSourceFile(id + `.${ext || 'js'}`, content, { overwrite: true })
+          project.createSourceFile(`${id}.${ext || 'js'}`, content, { overwrite: true })
         }
       } else if (
         !id.includes('.vue?vue') &&