refactor(rsc): normalize return value of transformServerActionServer (#1313)

Co-authored-by: Hiroshi Ogawa <4232207+hi-ogawa@users.noreply.github.com>
Co-authored-by: OpenCode <noreply@opencode.ai>

Author: hi-ogawa

packages/plugin-rsc/src/plugin.ts

@@ -2081,8 +2081,7 @@ function vitePluginUseServer(
             manager.serverReferenceMetaMap[id] = {
               importId: id,
               referenceKey: getNormalizedId(),
-              exportNames:
-                'names' in result ? result.names : result.exportNames,
+              exportNames: result.referenceNames,
             }
             const importSource = resolvePackage(`${PKG_NAME}/react/rsc/server`)
             output.prepend(

packages/plugin-rsc/src/transforms/server-action.test.ts

@@ -0,0 +1,32 @@
+import { parseAstAsync } from 'vite'
+import { expect, test } from 'vitest'
+import { transformServerActionServer } from './server-action'
+
+async function transform(input: string) {
+  const ast = await parseAstAsync(input)
+  return transformServerActionServer(input, ast, {
+    runtime: (value) => value,
+  })
+}
+
+test('normalizes top-level server reference names', async () => {
+  const result = await transform(
+    `'use server'; export async function action() {}`,
+  )
+
+  expect(result).toMatchObject({
+    exportNames: ['action'],
+    referenceNames: ['action'],
+  })
+})
+
+test('normalizes inline server reference names', async () => {
+  const result = await transform(
+    `export function App() { return async function action() { 'use server' } }`,
+  )
+
+  expect(result).toMatchObject({
+    names: ['$$hoist_0_anonymous_server_function'],
+    referenceNames: ['$$hoist_0_anonymous_server_function'],
+  })
+})

packages/plugin-rsc/src/transforms/server-action.ts

@@ -20,17 +20,21 @@ export function transformServerActionServer(
   | {
       exportNames: string[]
       output: MagicString
+      referenceNames: string[]
     }
   | {
       output: MagicString
       names: string[]
+      referenceNames: string[]
     } {
   // TODO: unify (generalize transformHoistInlineDirective to support top-level directive cases)
   if (hasDirective(ast.body, 'use server')) {
-    return transformWrapExport(input, ast, options)
+    const result = transformWrapExport(input, ast, options)
+    return { ...result, referenceNames: result.exportNames }
   }
-  return transformHoistInlineDirective(input, ast, {
+  const result = transformHoistInlineDirective(input, ast, {
     ...options,
     directive: 'use server',
   })
+  return { ...result, referenceNames: result.names }
 }