fix: rename function declaration

j4k0xb · Jan 3, 2024 · aa26838 · aa26838
1 parent ed75f2b
commit aa26838
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/packages/webcrack/src/ast-utils/rename.ts b/packages/webcrack/src/ast-utils/rename.ts
@@ -42,6 +42,8 @@ export function renameFast(binding: Binding, newName: string): void {
         },
         noScope: true,
       });
+    } else if (ref.isFunctionDeclaration() && t.isIdentifier(ref.node.id)) {
+      ref.node.id.name = newName;
     } else {
       throw new Error(
         `Unexpected constant violation (${ref.type}): ${codePreview(ref.node)}`,

diff --git a/packages/webcrack/src/ast-utils/test/rename.test.ts b/packages/webcrack/src/ast-utils/test/rename.test.ts
@@ -20,6 +20,20 @@ describe('rename variable', () => {
     `);
   });
 
+  test('duplicate function binding', () => {
+    const ast = parse('var a; function a() {}');
+    traverse(ast, {
+      Program(path) {
+        const binding = path.scope.getBinding('a')!;
+        renameFast(binding, 'b');
+      },
+    });
+    expect(ast).toMatchInlineSnapshot(`
+      var b;
+      function b() {}
+    `);
+  });
+
   test('different types of assignments', () => {
     const ast = parse(`
       var a = 1;