feat: support var or fun rename

Author: zcf0508

src/analyze/setupScript.ts

@@ -293,7 +293,7 @@ export function processSetup(ast: t.Node, parentScope?: Scope, parentPath?: t.No
             }
           });
         }
-        if(path.node.id.type === 'ObjectPattern'){
+        else if(path.node.id.type === 'ObjectPattern'){
           path.node.id.properties.forEach((property) => {
             if(property.type === 'ObjectProperty' && property.value.type === 'Identifier') {
               const name = property.value.name;
@@ -336,7 +336,7 @@ export function processSetup(ast: t.Node, parentScope?: Scope, parentPath?: t.No
             }
           });
         }
-        if([
+        else if([
           'CallExpression', 
           'ArrowFunctionExpression', 
           'FunctionDeclaration',
@@ -382,6 +382,38 @@ export function processSetup(ast: t.Node, parentScope?: Scope, parentPath?: t.No
             }, path.scope, path);
           }
         }
+        else if(path.node.id.type === 'Identifier') {
+          const name = path.node.id.name;
+          if(path.node.init.type === 'Identifier') {
+            const binding = path.scope.getBinding(path.node.init.name);
+            if(
+              graph.nodes.has(path.node.init.name) 
+              && (binding?.scope.block.type === 'Program'
+                || (parentScope === binding?.scope)
+              )
+            ) {
+              graph.edges.get(name)?.add(path.node.init.name);
+            }
+          } else {
+            traverse(path.node.init, {
+              Identifier(path1) {
+                const binding = path1.scope.getBinding(path1.node.name);
+                if(
+                  graph.nodes.has(path1.node.name) 
+                  && (
+                    path1.parent.type !== 'MemberExpression'
+                    || path1.parent.object === path1.node
+                  )
+                  && (binding?.scope.block.type === 'Program'
+                    || (parentScope === binding?.scope)
+                  )
+                ) {
+                  graph.edges.get(name)?.add(path1.node.name);
+                }
+              },
+            }, path.scope, path);
+          }
+        }
       }
     },
 

test/options-setup/TestComponent.graph.ts

@@ -6,15 +6,19 @@ const number: TypedNode = {label: 'number', type: NodeType.var, info: {line: 65,
 const count: TypedNode = {label: 'count', type: NodeType.var, info: {line: 67, column: 10}};
 const plus: TypedNode = {label: 'plus', type: NodeType.fun, info: {line: 70, column: 6}};
 const add: TypedNode = {label: 'add', type: NodeType.fun, info: {line: 74, column: 6}};
-const a: TypedNode = {label: 'a', type: NodeType.var, info: {line: 83, column: 6}};
+const a: TypedNode = {label: 'a', type: NodeType.var, info: {line: 81, column: 10}};
+const b: TypedNode = {label: 'b', type: NodeType.var, info: {line: 82, column: 10}};
+const c: TypedNode = {label: 'c', type: NodeType.var, info: {line: 87, column: 6}};
 
 edges.set(number, new Set([]));
+edges.set(b, new Set([count]));
+edges.set(c, new Set([count]));
 edges.set(count, new Set([]));
 edges.set(plus, new Set([plus]));
 edges.set(add, new Set([number, add, count]));
-edges.set(a, new Set([count]));
+// edges.set(a, new Set([count]));
 
 export const graph = {
-  nodes: new Set([number, count, plus, add, a]),
+  nodes: new Set([number, b, count, c, plus, add]),
   edges,
 };

test/options-setup/TestComponent.vue

@@ -77,11 +77,15 @@ export default {
         methods.add();
         console.log(count)
       }
-    }
+    };
+
+    const a = {count};
+    const b = count;
 
     return {
       ...toRefs(data),
-      a: count,
+      b,
+      c: count,
       ...methods,
     }
   }