feat: detect CallExpression and AssignmentExpression at root of setup block

Author: zcf0508

src/analyze/setupScript.ts

@@ -693,15 +693,52 @@ export function processSetup(
     ExpressionStatement(path) {
       if (path.type === 'ExpressionStatement'
         && path.node.expression.type === 'CallExpression'
-        && path.node.expression.callee.type === 'Identifier') {
-        traverseHooks(path.node.expression, path.scope);
+        && path.node.expression.callee.type === 'Identifier'
+      ) {
+        const name = path.node.expression.callee.name;
+        if (
+          graph.nodes.has(name)
+          && (path.scope.block.type === 'Program')
+        ) {
+          const _node = nodeCollection.getNode(name);
+          if (_node?.info?.used) {
+            _node?.info?.used?.add('Call Expression');
+          }
+          else if (_node) {
+            _node.info = {
+              ..._node?.info,
+              used: new Set(['Call Expression']),
+            };
+          }
+        }
+        else {
+          traverseHooks(path.node.expression, path.scope);
+        }
       }
       if (path.type === 'ExpressionStatement'
         && path.node.expression.type === 'AssignmentExpression'
         && path.node.expression.right.type === 'CallExpression'
-        && path.node.expression.right.callee.type === 'Identifier') {
+        && path.node.expression.right.callee.type === 'Identifier'
+      ) {
         traverseHooks(path.node.expression.right, path.scope);
       }
+      if (path.type === 'ExpressionStatement'
+        && path.node.expression.type === 'AssignmentExpression'
+        && path.node.expression.left.type === 'Identifier'
+        && graph.nodes.has(path.node.expression.left.name)
+        && (path.scope.block.type === 'Program')
+      ) {
+        const _node = nodeCollection.getNode(path.node.expression.left.name);
+        if (_node?.info?.used) {
+          _node?.info?.used?.add('Assignment Expression');
+        }
+        else if (_node) {
+          _node.info = {
+            ..._node?.info,
+            used: new Set(['Assignment Expression']),
+          };
+        }
+      }
     },
   }, parentScope, parentPath);
 

src/vis.ts

@@ -5,6 +5,14 @@ type CustomNode = Node & {
   info: TypedNode['info']
 };
 
+function filterNodeUserd(used: Set<string> | undefined) {
+  const usedArray = Array.from(used || []);
+  return new Set(usedArray.filter(u => ![
+    'Assignment Expression',
+    'Call Expression',
+  ].includes(u)));
+}
+
 export function getVisData(
   graph: {
     nodes: Set<TypedNode>
@@ -29,8 +37,8 @@ export function getVisData(
         ? 'used'
         : 'normal',
       title: `${
-        node.info?.used?.size
-          ? `used by ${Array.from(node.info?.used || [])?.map(i => `\`${i}\``).join(',')}\n\n`
+        filterNodeUserd(node.info?.used).size
+          ? `used by ${Array.from(filterNodeUserd(node.info?.used))?.map(i => `\`${i}\``).join(',')}\n\n`
           : ''
       }${
         usedNodes.has(node.label)

test/output/suggent-gen.txt

@@ -311,4 +311,16 @@
     },
     "type": "info",
   },
+  {
+    "message": "Node [stop1] is not used, perhaps you can remove it.",
+    "nodeInfo": {
+      "info": {
+        "column": 6,
+        "line": 107,
+      },
+      "label": "stop1",
+      "type": "var",
+    },
+    "type": "info",
+  },
 ]

test/output/vue/setup-block.vue.graph.txt

@@ -455,6 +455,7 @@
           "line": 111,
           "used": Set {
             "watch",
+            "Assignment Expression",
           },
         },
         "label": "stop2",
@@ -465,6 +466,9 @@
       "info": {
         "column": 2,
         "line": 104,
+        "used": Set {
+          "Call Expression",
+        },
       },
       "label": "fun1c",
       "type": "var",
@@ -495,6 +499,7 @@
         "line": 111,
         "used": Set {
           "watch",
+          "Assignment Expression",
         },
       },
       "label": "stop2",
@@ -758,6 +763,9 @@
       "info": {
         "column": 2,
         "line": 104,
+        "used": Set {
+          "Call Expression",
+        },
       },
       "label": "fun1c",
       "type": "var",
@@ -776,6 +784,7 @@
         "line": 111,
         "used": Set {
           "watch",
+          "Assignment Expression",
         },
       },
       "label": "stop2",