feat: support transform import xx from element-ui/lib/.. (#94)

Author: wangsongc

transformations/__tests__/element-plus-upgrade.spec.ts

@@ -24,3 +24,11 @@ defineInlineTest(
   `import ElementUI, { ElMessageBox as MessageBox } from "element-plus";`,
   'correctly transform multiple imports from element-plus'
 )
+
+defineInlineTest(
+  transform,
+  {},
+  `import locale from "element-ui/lib/locale/lang/zh-cn";`,
+  `import locale from "element-plus/lib/locale/lang/zh-cn";`,
+  'correctly transform import from element-plus'
+)

transformations/element-plus/element-plus-component-name.ts

@@ -11,6 +11,7 @@ import { getCntFunc } from '../../src/report'
  * @param content
  */
 export const transformAST: ASTTransformation = ({ root, j }) => {
+  const cntFunc = getCntFunc('element-plus-upgrade', subRules)
   // find element-ui import
   const elementPlusImport = root.find(j.ImportDeclaration, {
     source: {
@@ -22,6 +23,7 @@ export const transformAST: ASTTransformation = ({ root, j }) => {
     elementPlusImport.forEach(({ node }) => {
       let newSpecifier: (ImportSpecifier | ImportDefaultSpecifier)[] = []
       node.specifiers.forEach(importNode => {
+        cntFunc()
         if (importNode.type === 'ImportSpecifier') {
           newSpecifier.push(
             j.importSpecifier(
@@ -38,10 +40,6 @@ export const transformAST: ASTTransformation = ({ root, j }) => {
         }
       })
     })
-
-    // stats
-    const cntFunc = getCntFunc('observable', subRules)
-    cntFunc()
   }
 }
 

transformations/element-plus/element-plus-import.ts

@@ -7,20 +7,17 @@ import { getCntFunc } from '../../src/report'
  * @param content
  */
 export const transformAST: ASTTransformation = ({ root, j }) => {
+  const cntFunc = getCntFunc('element-plus-import', subRules)
   // find element-ui import
-  const elementPlusImport = root.find(j.ImportDeclaration, {
-    source: {
-      value: 'element-ui'
-    }
+  const elementPlusImport = root.find(j.ImportDeclaration, node => {
+    return node.source.value.toString().startsWith('element-ui')
   })
-
   if (elementPlusImport.length) {
     elementPlusImport.forEach(({ node }) => {
-      node.source.value = 'element-plus'
+      node.source.value =
+        'element-plus' +
+        node.source.value?.toString().substring('element-ui'.length)
     })
-
-    // stats
-    const cntFunc = getCntFunc('observable', subRules)
     cntFunc()
   }
 }