add class method case

vercel · Apr 12, 2024 · 26728a2 · 26728a2
1 parent 4ee6e1d
commit 26728a2
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 3 deletions.
diff --git a/packages/next-swc/crates/next-custom-transforms/src/transforms/server_actions.rs b/packages/next-swc/crates/next-custom-transforms/src/transforms/server_actions.rs
@@ -582,6 +582,16 @@ impl<C: Comments> VisitMut for ServerActions<C> {
         self.in_default_export_decl = old_in_default_export_decl;
     }
 
+    fn visit_mut_class_method(&mut self, m: &mut ClassMethod) {
+        let old_in_export_decl = self.in_export_decl;
+        let old_in_default_export_decl = self.in_default_export_decl;
+        self.in_export_decl = false;
+        self.in_default_export_decl = false;
+        m.visit_mut_children_with(self);
+        self.in_export_decl = old_in_export_decl;
+        self.in_default_export_decl = old_in_default_export_decl;
+    }
+
     fn visit_mut_arrow_expr(&mut self, a: &mut ArrowExpr) {
         // Arrow expressions need to be visited in prepass to determine if it's
         // an action function or not.

diff --git a/...es/next-swc/crates/next-custom-transforms/tests/fixture/server-actions/server/31/input.js b/...es/next-swc/crates/next-custom-transforms/tests/fixture/server-actions/server/31/input.js
@@ -7,3 +7,11 @@ export const action = {
     })()
   },
 }.f
+
+export const action2 = new (class X {
+  async f(x) {
+    ;(() => {
+      console.log(x)
+    })()
+  }
+})().f
diff --git a/...s/next-swc/crates/next-custom-transforms/tests/fixture/server-actions/server/31/output.js b/...s/next-swc/crates/next-custom-transforms/tests/fixture/server-actions/server/31/output.js
@@ -1,4 +1,4 @@
-/* __next_internal_action_entry_do_not_use__ {"f14702b5a021dd117f7ec7a3c838f397c2046d3b":"action"} */ import { registerServerReference } from "private-next-rsc-server-reference";
+/* __next_internal_action_entry_do_not_use__ {"abf760c735ba66c4c26a2913864dd7e28fb88a91":"action2","f14702b5a021dd117f7ec7a3c838f397c2046d3b":"action"} */ import { registerServerReference } from "private-next-rsc-server-reference";
 import { encryptActionBoundArgs, decryptActionBoundArgs } from "private-next-rsc-action-encryption";
 export const action = {
     async f (x) {
@@ -8,8 +8,18 @@ export const action = {
         })();
     }
 }.f;
+export const action2 = new class X {
+    async f(x) {
+        ;
+        (()=>{
+            console.log(x);
+        })();
+    }
+}().f;
 import { ensureServerEntryExports } from "private-next-rsc-action-validate";
 ensureServerEntryExports([
-    action
+    action,
+    action2
 ]);
-registerServerReference("f14702b5a021dd117f7ec7a3c838f397c2046d3b", action);
+registerServerReference("f14702b5a021dd117f7ec7a3c838f397c2046d3b", action);
+registerServerReference("abf760c735ba66c4c26a2913864dd7e28fb88a91", action2);