fix: ref rewrite

swc-project · Mar 31, 2023 · c09e926 · c09e926
1 parent bab3925
commit c09e926
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/packages/swc-helpers/scripts/ast_grep.js b/packages/swc-helpers/scripts/ast_grep.js
@@ -110,17 +110,28 @@ export function ast_grep() {
                         rule: {
                             pattern: import_binding,
                             kind: "identifier",
-                            inside: { kind: "call_expression", field: "function" },
+                            inside: {
+                                not: {
+                                    kind: "import_specifier",
+                                },
+                            },
                         },
                     })
                     .forEach((match) => {
                         const range = match.range();
-                        const call_name = match.text();
+                        const ref_name = match.text();
+
+                        const is_func_call = match.inside("$FUN($$$)");
+
                         source.update(
                             range.start.index,
                             range.end.index,
-                            `(0, ${import_binding}.${call_name})`,
+                            `${import_binding}.${ref_name}`,
                         );
+
+                        if (is_func_call) {
+                            source.appendLeft(range.start.index, "(0, ").appendRight(range.end.index, ")");
+                        }
                     });
             });