feat: implement vuex transform

haoqunjiang · haoqunjiang · commit 6719d0c4d583 · 2020-04-05T01:03:27.000+08:00
diff --git a/generator/codemods/router/index.js b/generator/codemods/router/index.js
@@ -3,7 +3,8 @@ module.exports = function(fileInfo, api) {
   const j = api.jscodeshift
   const root = j(fileInfo.source)
 
-  // TODO: Vue.use(router) might be in either`router/index.js` or `main.js`
+  // TODO: new Router() -> createRouter()
+  // TODO: mode: 'history' -> history: createWebHistory()
 
   return root.toSource({ lineTerminator: '\n' })
 }
diff --git a/generator/codemods/vuex/__testfixtures__/import-alias.input.js b/generator/codemods/vuex/__testfixtures__/import-alias.input.js
@@ -0,0 +1,11 @@
+import { Store as TheStore } from 'vuex';
+
+const store = new TheStore({
+  state () {
+    return {
+      count: 1
+    }
+  }
+});
+
+export default store;
diff --git a/generator/codemods/vuex/__testfixtures__/import-alias.output.js b/generator/codemods/vuex/__testfixtures__/import-alias.output.js
@@ -1,11 +1,11 @@
-import { createStore } from 'vuex'
+import { createStore } from 'vuex';
 
 const store = createStore({
   state () {
     return {
       count: 1
     }
   }
-})
+});
 
-export default store
+export default store;
diff --git a/generator/codemods/vuex/__testfixtures__/store.input.js b/generator/codemods/vuex/__testfixtures__/store.input.js
@@ -0,0 +1,11 @@
+import { Store } from 'vuex';
+
+const store = new Store({
+  state () {
+    return {
+      count: 1
+    }
+  }
+});
+
+export default store;
diff --git a/generator/codemods/vuex/__testfixtures__/store.output.js b/generator/codemods/vuex/__testfixtures__/store.output.js
@@ -0,0 +1,11 @@
+import { createStore } from 'vuex';
+
+const store = createStore({
+  state () {
+    return {
+      count: 1
+    }
+  }
+});
+
+export default store;
diff --git a/generator/codemods/vuex/__testfixtures__/vuex-dot-store.input.js b/generator/codemods/vuex/__testfixtures__/vuex-dot-store.input.js
@@ -0,0 +1,11 @@
+import Vuex from 'vuex';
+
+const store = new Vuex.Store({
+  state () {
+    return {
+      count: 1
+    }
+  }
+});
+
+export default store;
diff --git a/generator/codemods/vuex/__testfixtures__/vuex-dot-store.output.js b/generator/codemods/vuex/__testfixtures__/vuex-dot-store.output.js
@@ -0,0 +1,11 @@
+import Vuex from 'vuex';
+
+const store = Vuex.createStore({
+  state () {
+    return {
+      count: 1
+    }
+  }
+});
+
+export default store;
diff --git a/generator/codemods/vuex/__testfixtures__/vuex-store.input.js b/generator/codemods/vuex/__testfixtures__/vuex-store.input.js
diff --git a/generator/codemods/vuex/__tests__/vuex-test.js b/generator/codemods/vuex/__tests__/vuex-test.js
@@ -0,0 +1,7 @@
+jest.autoMockOff()
+
+const { defineTest } = require('jscodeshift/dist/testUtils')
+
+defineTest(__dirname, 'index', null, 'store')
+defineTest(__dirname, 'index', null, 'vuex-dot-store')
+defineTest(__dirname, 'index', null, 'import-alias')
diff --git a/generator/codemods/vuex/index.js b/generator/codemods/vuex/index.js
@@ -1,7 +1,68 @@
+const addImport = require('../utils/add-import')
+const removeExtraneousImport = require('../utils/remove-extraneous-import')
+
 /** @type {import('jscodeshift').Transform} */
 module.exports = function(fileInfo, api) {
   const j = api.jscodeshift
   const root = j(fileInfo.source)
+  const context = { j, root }
+
+  // TODO: new Store() -> createStore()
+
+  const vuexImportDecls = root.find(j.ImportDeclaration, {
+    source: {
+      value: 'vuex'
+    }
+  })
+
+  const importedVuex = vuexImportDecls.find(j.ImportDefaultSpecifier)
+  const importedStore = vuexImportDecls.find(j.ImportSpecifier, {
+    imported: {
+      name: 'Store'
+    }
+  })
+
+  if (importedVuex.length) {
+    const localVuex = importedVuex.get(0).node.local.name
+    const newVuexDotStore = root.find(j.NewExpression, {
+      callee: {
+        type: 'MemberExpression',
+        object: {
+          type: 'Identifier',
+          name: localVuex
+        },
+        property: {
+          name: 'Store'
+        }
+      }
+    })
+
+    newVuexDotStore.replaceWith(({ node }) => {
+      return j.callExpression(
+        j.memberExpression(
+          j.identifier(localVuex),
+          j.identifier('createStore')
+        ),
+        node.arguments
+      )
+    })
+  }
+
+  if (importedStore.length) {
+    const localStore = importedStore.get(0).node.local.name
+    const newStore = root.find(j.NewExpression, {
+      callee: {
+        type: 'Identifier',
+        name: localStore
+      }
+    })
+
+    addImport(context, { imported: 'createStore' }, 'vuex')
+    newStore.replaceWith(({ node }) => {
+      return j.callExpression(j.identifier('createStore'), node.arguments)
+    })
+    removeExtraneousImport(context, localStore)
+  }
 
   return root.toSource({ lineTerminator: '\n' })
 }

Original file line number	Diff line number	Diff line change
`@@ -3,7 +3,8 @@ module.exports = function(fileInfo, api) {`
`3`	`3`	`const j = api.jscodeshift`
`4`	`4`	`const root = j(fileInfo.source)`
`5`	`5`
`6`		- // TODO: Vue.use(router) might be in either`router/index.js` or `main.js`
	`6`	`+ // TODO: new Router() -> createRouter()`
	`7`	`+ // TODO: mode: 'history' -> history: createWebHistory()`
`7`	`8`
`8`	`9`	`return root.toSource({ lineTerminator: '\n' })`
`9`	`10`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1,11 +1,11 @@`
`1`		`-import { createStore } from 'vuex'`
	`1`	`+import { createStore } from 'vuex';`
`2`	`2`
`3`	`3`	`const store = createStore({`
`4`	`4`	`state () {`
`5`	`5`	`return {`
`6`	`6`	`count: 1`
`7`	`7`	`}`
`8`	`8`	`}`
`9`		`-})`
	`9`	`+});`
`10`	`10`
`11`		`-export default store`
	`11`	`+export default store;`