fix import

package.json

@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-lazy-import",
-  "version": "1.0.4",
+  "version": "1.0.5",
   "description": "Used for vite Lazy import js and style",
   "scripts": {
     "update": "npm install --legacy-peer-deps",
@@ -14,9 +14,9 @@
   "module": "./dist/index.mjs",
   "typings": "types/index.d.ts",
   "dependencies": {
-    "rollup": "^4.18.0",
-    "@rollup/pluginutils": "^5.1.0",
-    "es-module-lexer": "^1.5.3"
+    "rollup": "4.18.0",
+    "@rollup/pluginutils": "5.1.0",
+    "es-module-lexer": "1.5.3"
   },
   "devDependencies": {
     "@babel/core": "^7.4.4",

src/index.ts

@@ -50,10 +50,13 @@ export function lazyImport (options: LazyImportConfig): Plugin<any> {
   function transformImport (id:string, exports: readonly ExportSpecifier[], resOpts: LazyImportResolver) {
     const importCodes: string[] = []
     exports.forEach(item => {
-      const importName = item.ln || item.n
+      let importName = item.n
       let variableCode = item.n
-      if (item.ln !== item.n) {
-        variableCode = `${importName} as ${item.n}`
+      let asName = ''
+      if (item.ln && item.ln !== item.n) {
+        importName = item.ln
+        asName = item.n
+        variableCode = `${importName} as ${asName}`
       }
 
       let jsPath = ''
@@ -102,6 +105,14 @@ export function lazyImport (options: LazyImportConfig): Plugin<any> {
     return importCodes.join(';\n')
   }
 
+  function toVariableExport (impCode: string) {
+    const rest = impCode.replace(/\n/, ' ').match(/\{[a-zA-Z,_'"\s]+?\}/)
+    if (rest) {
+      return `export ${rest[0]}`
+    }
+    return ''
+  }
+
   return {
     name: 'vite:lazy-import',
     enforce: 'post',
@@ -120,7 +131,8 @@ export function lazyImport (options: LazyImportConfig): Plugin<any> {
         if (!syntaxItem) {
           return
         }
-        const exports = parse(restCode.slice(syntaxItem.ss, syntaxItem.se).replace('import', 'export'))[1]
+        const exportCode = toVariableExport(restCode.slice(syntaxItem.ss, syntaxItem.se))
+        const exports = parse(exportCode)[1]
         if (!exports.length) {
           return
         }