feat: implement remove-vue-use

Author: haoqunjiang

generator/codemods/global-api/__testfixtures__/vue-use.input.js

@@ -0,0 +1,6 @@
+import Vue from 'vue'
+import VueRouter from 'vue-router'
+import Vuex from 'Vuex'
+
+Vue.use(VueRouter)
+Vue.use(Vuex)

generator/codemods/global-api/__testfixtures__/vue-use.output.js

@@ -7,3 +7,4 @@ defineTest(__dirname, 'index', null, 'custom-root-option')
 // defineTest(__dirname, 'index', null, 'el')
 defineTest(__dirname, 'index', null, 'vue-router')
 defineTest(__dirname, 'index', null, 'vuex')
+defineTest(__dirname, 'index', null, 'vue-use')

generator/codemods/global-api/__tests__/global-api-test.js

@@ -9,13 +9,36 @@ module.exports = function(fileInfo, api) {
   require('./root-option-to-use')(context, 'router')
   require('./remove-trivial-root')(context)
   require('./remove-production-tip')(context)
+  require('./remove-vue-use')(context)
   require('./remove-contextual-h')(context)
 
-  // remove extraneous Vue import
-  const localVueUsages = root.find(j.Identifier, { name: 'Vue' })
-  if (localVueUsages.length === 1) {
-    localVueUsages.closest(j.ImportDefaultSpecifier).remove()
-  }
+  // remove extraneous imports
+  removeExtraneousImport(context, 'Vue')
+  removeExtraneousImport(context, 'Vuex')
+  removeExtraneousImport(context, 'VueRouter')
 
   return root.toSource({ lineTerminator: '\n' })
 }
+
+/**
+ * @param {Object} context
+ * @param {import('jscodeshift').JSCodeshift} context.j
+ * @param {ReturnType<import('jscodeshift').Core>} context.root
+ */
+function removeExtraneousImport({ root, j }, name) {
+  const localUsages = root.find(j.Identifier, { name })
+  if (localUsages.length === 1) {
+    const importDecl = localUsages.closest(j.ImportDeclaration)
+    
+    if (!importDecl.length) {
+      return
+    }
+
+    if (importDecl.get(0).node.specifiers.length === 1) {
+      importDecl.remove()
+    } else {
+      localUsages.closest(j.ImportSpecifier).remove()
+      localUsages.closest(j.ImportDefaultSpecifier).remove()
+    }
+  }
+}

generator/codemods/global-api/index.js

@@ -0,0 +1,26 @@
+/**
+ * Remove `Vue.use()` calls
+ * Per current design, `Vue.use` is replaced by `app.use`.
+ * But in library implementations like `vue-router` and `vuex`,
+ * the new `app.use` does not reuse the same argument passed to `Vue.use()`,
+ * but expects instantiated instances that are used to pass to the root components instead.
+ * So we now expect the migration to be done in the `root-option-to-use` transformation,
+ * and the `Vue.use` statements can be just abandoned.
+ * @param {Object} context
+ * @param {import('jscodeshift').JSCodeshift} context.j
+ * @param {ReturnType<import('jscodeshift').Core>} context.root
+ */
+module.exports = function removeVueUse({ j, root }) {
+  const vueUseCalls = root.find(j.CallExpression, {
+    callee: {
+      type: 'MemberExpression',
+      object: {
+        name: 'Vue'
+      },
+      property: {
+        name: 'use'
+      }
+    }
+  })
+  vueUseCalls.remove()
+}