feat: add emits declaration

ygj6 · ygj6 · commit 2714c6b55b82 · 2021-05-10T19:37:04.000+08:00
diff --git a/generator/codemods/vue/add-emit-declaration.js b/generator/codemods/vue/add-emit-declaration.js
@@ -0,0 +1,48 @@
+/**
+ * @param {Object} context
+ * @param {import('jscodeshift').JSCodeshift} context.j
+ * @param {ReturnType<import('jscodeshift').Core>} context.root
+ */
+module.exports = function addEmitDeclaration(context) {
+  const { j, root } = context
+
+  // this.$emit('xxx') => emits: ['xxx']
+  const this$emits = root.find(j.CallExpression, {
+    callee: {
+      type: 'MemberExpression',
+      object: { type: 'ThisExpression' },
+      property: {
+        type: 'Identifier',
+        name: '$emit'
+      }
+    }
+  })
+
+  const emits = []
+  for (let i = 0; i < this$emits.length; i++) {
+    const arg = this$emits.at(i).get().node.arguments[0]
+    if (arg.type === 'StringLiteral') {
+      emits.push(arg.value)
+    }
+  }
+
+  if (emits.length === 0) {
+    return
+  }
+
+  const properties = root
+    .find(j.ExportDefaultDeclaration)
+    .at(0)
+    .find(j.ObjectExpression)
+    .at(0)
+
+  properties.replaceWith(nodePath => {
+    nodePath.node.properties.unshift(
+      j.objectProperty(
+        j.identifier('emits'),
+        j.arrayExpression(emits.map(el => j.stringLiteral(el)))
+      )
+    )
+    return nodePath.node
+  })
+}
diff --git a/generator/codemods/vue/index.js b/generator/codemods/vue/index.js
@@ -0,0 +1,12 @@
+/** @type {import('jscodeshift').Transform} */
+module.exports = function(fileInfo, api) {
+  const j = api.jscodeshift
+  const root = j(fileInfo.source)
+  const context = { j, root }
+
+  require('./add-emit-declaration')(context)
+
+  return root.toSource({ lineTerminator: '\n' })
+}
+
+module.exports.parser = 'babylon'
diff --git a/generator/index.js b/generator/index.js
@@ -22,6 +22,14 @@ module.exports = (api) => {
   const globalAPITransform = require('./codemods/global-api')
   api.transformScript(api.entryFile, globalAPITransform)
 
+  const vueTransform = require('./codemods/vue')
+  const vueFiles = Object.keys(api.generator.files).filter(el =>
+    el.endsWith('.vue')
+  )
+  for (let i = 0; i < vueFiles.length; i++) {
+    api.transformScript(vueFiles[i], vueTransform)
+  }
+
   if (api.hasPlugin('eslint')) {
     api.extendPackage({
       devDependencies: {
