feat: implement transition group transformation

Author: ygj6

vue-transformations/__test__/transition-group-root.spec.ts

@@ -0,0 +1,9 @@
+import { runTest } from '../../src/testUtils'
+
+runTest(
+  'Add tag attribute for default transition-group',
+  'transition-group-root',
+  'transition-group-root',
+  'vue',
+  'vue'
+)

vue-transformations/__testfixtures__/transition-group-root/transition-group-root.input.vue

@@ -0,0 +1,14 @@
+<template>
+  <div class="hello">
+    <transition-group tag="ul">
+      <li v-for="item in items" :key="item">
+        {{ item }}
+      </li>
+    </transition-group>
+    <transition-group>
+      <span v-for="item in items" v-bind:key="item" class="list-item">
+        {{ item }}
+      </span>
+    </transition-group>
+  </div>
+</template>

vue-transformations/__testfixtures__/transition-group-root/transition-group-root.output.vue

@@ -0,0 +1,14 @@
+<template>
+  <div class="hello">
+    <transition-group tag="ul">
+      <li v-for="item in items" :key="item">
+        {{ item }}
+      </li>
+    </transition-group>
+    <transition-group tag="span">
+      <span v-for="item in items" v-bind:key="item" class="list-item">
+        {{ item }}
+      </span>
+    </transition-group>
+  </div>
+</template>

vue-transformations/index.ts

@@ -9,6 +9,9 @@ const transformationMap: {
 } = {
   'slot-attribute': require('./slot-attribute'),
   'slot-default': require('./slot-default'),
+  'v-for-template-key': require('./v-for-template-key'),
+  'v-else-if-key': require('./v-else-if-key'),
+  'transition-group-root': require('./transition-group-root'),
 }
 
 export default transformationMap

vue-transformations/transition-group-root.ts

@@ -0,0 +1,78 @@
+import * as OperationUtils from '../src/operationUtils'
+import * as parser from 'vue-eslint-parser'
+import type { Node } from 'vue-eslint-parser/ast/nodes'
+import type { Operation } from '../src/operationUtils'
+import type { VueASTTransformation } from '../src/wrapVueTransformation'
+import wrap from '../src/wrapVueTransformation'
+
+/**
+ * 每一个实际的规则，需要做以下几件事：
+ * 1、findNodes(fileInfo, ast): 寻找匹配规则的节点
+ * 2、fix(nodes): 完善匹配节点的增删改逻辑
+ * 3、applyFix(fileInfo, tempFixes): 执行fixer，对源码进行增删改，并返回转换后的代码
+ * @param context
+ */
+
+export const transformAST: VueASTTransformation = (context) => {
+  let fixOperations: Operation[] = []
+  const toFixNodes: Node[] = findNodes(context)
+  toFixNodes.forEach((node) => {
+    // fix(node) 返回的为 Operation 数组，因此用 concat 合并多个数组
+    fixOperations = fixOperations.concat(fix(node))
+  })
+  return fixOperations
+}
+
+export default wrap(transformAST)
+/**
+ * 定位 slot attribute 节点
+ *
+ * @param context { file: FileInfo, api: API }
+ * @param templateBody
+ * @returns 所有的 slot attribute 节点
+ */
+function findNodes(context: any): Node[] {
+  const { file } = context
+  const source = file.source
+  const options = { sourceType: 'module' }
+  const ast = parser.parse(source, options)
+  let toFixNodes: Node[] = []
+  let root: Node = <Node>ast.templateBody // 强制类型转换
+  parser.AST.traverseNodes(root, {
+    enterNode(node: Node) {
+      if (node.type === 'VElement' && node.name === 'transition-group') {
+        toFixNodes.push(node)
+      }
+    },
+    leaveNode(node: Node) {},
+  })
+  return toFixNodes
+}
+/**
+ * The repair logic for
+ * @param node The Target Node
+ */
+function fix(node: any): Operation[] {
+  let fixOperations: Operation[] = []
+
+  // The current node has no attribute that is v-for
+  let havaTagAttr: boolean = false
+  node.startTag.attributes
+    .filter(
+      (attr: any) =>
+        attr.type === 'VAttribute' &&
+        attr.key.type === 'VIdentifier' &&
+        attr.key.name === 'tag'
+    )
+    .forEach((element: any) => {
+      havaTagAttr = true
+    })
+  if (havaTagAttr) {
+    return fixOperations
+  }
+
+  fixOperations.push(
+    OperationUtils.insertTextAt(node.startTag.range[1] - 1, ' tag="span"')
+  )
+  return fixOperations
+}