refactor: Use swc AST to determine use client and server directives

packages/next/src/build/analysis/get-page-static-info.ts

@@ -51,9 +51,13 @@ export interface PageStaticInfo {
 
 const CLIENT_MODULE_LABEL =
   /\/\* __next_internal_client_entry_do_not_use__ ([^ ]*) (cjs|auto) \*\//
+
 const ACTION_MODULE_LABEL =
   /\/\* __next_internal_action_entry_do_not_use__ ([^ ]+) \*\//
 
+const CLIENT_DIRECTIVE_REGEX = /^use client$/
+const SERVER_ACTION_DIRECTIVE_REGEX = /^use server$/
+
 export type RSCModuleType = 'server' | 'client'
 export function getRSCModuleInformation(
   source: string,
@@ -75,14 +79,12 @@ export function getRSCModuleInformation(
   const clientEntryType = clientInfoMatch?.[2] as 'cjs' | 'auto'
 
   const type = clientRefs ? RSC_MODULE_TYPES.client : RSC_MODULE_TYPES.server
-  const hasUseClientDirective = /^\s*['"]use client['"]/.test(source)
   return {
     type,
     actions,
     clientRefs,
     clientEntryType,
     isClientRef,
-    hasUseClientDirective,
   }
 }
 
@@ -124,6 +126,7 @@ function checkExports(
   generateSitemaps?: boolean
   generateStaticParams: boolean
   extraProperties?: Set<string>
+  directives?: Set<string>
 } {
   const exportsSet = new Set<string>([
     'getStaticProps',
@@ -142,6 +145,7 @@ function checkExports(
       let generateSitemaps: boolean = false
       let generateStaticParams = false
       let extraProperties = new Set<string>()
+      let directives = new Set<string>()
 
       for (const node of swcAST.body) {
         if (
@@ -229,6 +233,18 @@ function checkExports(
               )
           }
         }
+
+        if (node.type === 'ExpressionStatement') {
+          if (node.expression.type === 'StringLiteral') {
+            const directive = node.expression.value
+            if (CLIENT_DIRECTIVE_REGEX.test(directive)) {
+              directives.add('client')
+            }
+            if (SERVER_ACTION_DIRECTIVE_REGEX.test(directive)) {
+              directives.add('server')
+            }
+          }
+        }
       }
 
       return {
@@ -240,6 +256,7 @@ function checkExports(
         generateSitemaps,
         generateStaticParams,
         extraProperties,
+        directives,
       }
     } catch (err) {}
   }
@@ -253,6 +270,7 @@ function checkExports(
     generateSitemaps: false,
     generateStaticParams: false,
     extraProperties: undefined,
+    directives: undefined,
   }
 }
 
@@ -467,6 +485,7 @@ export async function getPageStaticInfo(params: {
       preferredRegion,
       generateStaticParams,
       extraProperties,
+      directives,
     } = checkExports(swcAST, pageFilePath)
     const rscInfo = getRSCModuleInformation(fileContent)
     const rsc = rscInfo.type
@@ -575,7 +594,7 @@ export async function getPageStaticInfo(params: {
 
     if (
       pageType === 'app' &&
-      rscInfo.hasUseClientDirective &&
+      directives?.has('client') &&
       generateStaticParams
     ) {
       throw new Error(

packages/next/src/build/webpack/loaders/get-module-build-info.ts

@@ -32,7 +32,6 @@ export interface RSCMeta {
   clientRefs?: string[]
   clientEntryType?: 'cjs' | 'auto'
   isClientRef?: boolean
-  hasUseClientDirective?: boolean
   requests?: string[] // client requests in flight client entry
 }