feat: add emits declaration

Author: ygj6

Diff for: generator/codemods/vue/add-emit-declaration.js

@@ -0,0 +1,74 @@
+/**
+ * @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 defaultObject = root
+    .find(j.ExportDefaultDeclaration)
+    .at(0)
+    .find(j.ObjectExpression)
+    .at(0)
+
+  let oldEmits = emits
+  let emitsProperty = defaultObject.find(j.ObjectProperty, {
+    key: {
+      type: 'Identifier',
+      name: 'emits'
+    }
+  })
+  if (emitsProperty.length > 0) {
+    oldEmits = emitsProperty
+      .at(0)
+      .get()
+      .node.value.elements.map(el => el.value)
+
+    let hasChange = false
+    for (const el of emits) {
+      if (!oldEmits.includes(el)) {
+        oldEmits.push(el)
+        hasChange = true
+      }
+    }
+    if (!hasChange) {
+      return
+    }
+    emitsProperty.remove()
+  }
+
+  defaultObject.replaceWith(({ node }) => {
+    node.properties.unshift(
+      j.objectProperty(
+        j.identifier('emits'),
+        j.arrayExpression(oldEmits.map(el => j.stringLiteral(el)))
+      )
+    )
+    return node
+  })
+}

Diff for: 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 for: 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: {