next_custom_transforms/transforms/

server_actions.rs

use std::{
    cell::RefCell,
    collections::{BTreeMap, hash_map},
    convert::{TryFrom, TryInto},
    mem::{replace, take},
    path::{Path, PathBuf},
    rc::Rc,
    sync::Arc,
};

use base64::{display::Base64Display, prelude::BASE64_STANDARD};
use hex::encode as hex_encode;
use indoc::formatdoc;
use pathdiff::diff_paths;
use rustc_hash::{FxHashMap, FxHashSet};
use serde::Deserialize;
use sha1::{Digest, Sha1};
use swc_core::{
    atoms::{Atom, Wtf8Atom, atom},
    common::{
        BytePos, DUMMY_SP, FileName, Mark, SourceMap, Span, SyntaxContext,
        comments::{Comment, CommentKind, Comments, SingleThreadedComments},
        errors::HANDLER,
        source_map::{PURE_SP, SourceMapGenConfig},
        util::take::Take,
    },
    ecma::{
        ast::*,
        codegen::{self, Emitter, text_writer::JsWriter},
        utils::{ExprFactory, private_ident, quote_ident},
        visit::{VisitMut, VisitMutWith, noop_visit_mut_type, visit_mut_pass},
    },
    quote,
};
use turbo_rcstr::{RcStr, rcstr};

use crate::FxIndexMap;

#[derive(Clone, Copy, Debug, Deserialize)]
pub enum ServerActionsMode {
    Webpack,
    Turbopack,
}

#[derive(Clone, Debug, Deserialize)]
#[serde(deny_unknown_fields, rename_all = "camelCase")]
pub struct Config {
    pub is_react_server_layer: bool,
    pub is_development: bool,
    pub use_cache_enabled: bool,
    pub hash_salt: String,
    pub cache_kinds: FxHashSet<RcStr>,
}

#[derive(Clone, Debug)]
enum Directive {
    UseServer,
    UseCache { cache_kind: RcStr },
}

#[derive(Clone, Debug)]
enum DirectiveLocation {
    Module,
    FunctionBody,
}

#[derive(Clone, Debug)]
enum ThisStatus {
    Allowed,
    Forbidden { directive: Directive },
}

#[derive(Clone)]
struct ServerReferenceExport {
    ident: Ident,
    export_name: ModuleExportName,
    reference_id: Atom,
    needs_cache_runtime_wrapper: bool,
}

/// Export info for serialization
#[derive(Clone, Debug, serde::Serialize)]
struct ServerReferenceExportInfo {
    name: Atom,
}

#[derive(Clone, Debug)]
enum ServerActionsErrorKind {
    ExportedSyncFunction {
        span: Span,
        in_action_file: bool,
    },
    ForbiddenExpression {
        span: Span,
        expr: String,
        directive: Directive,
    },
    InlineSyncFunction {
        span: Span,
        directive: Directive,
    },
    InlineUseCacheInClassInstanceMethod {
        span: Span,
    },
    InlineUseCacheInClientComponent {
        span: Span,
    },
    InlineUseServerInClassInstanceMethod {
        span: Span,
    },
    InlineUseServerInClientComponent {
        span: Span,
    },
    MisplacedDirective {
        span: Span,
        directive: String,
        location: DirectiveLocation,
    },
    MisplacedWrappedDirective {
        span: Span,
        directive: String,
        location: DirectiveLocation,
    },
    MisspelledDirective {
        span: Span,
        directive: String,
        expected_directive: String,
    },
    MultipleDirectives {
        span: Span,
        location: DirectiveLocation,
    },
    UnknownCacheKind {
        span: Span,
        cache_kind: RcStr,
    },
    UseCacheWithoutCacheComponents {
        span: Span,
        directive: String,
    },
    WrappedDirective {
        span: Span,
        directive: String,
    },
}

#[allow(clippy::too_many_arguments)]
#[tracing::instrument(level = tracing::Level::TRACE, skip_all)]
pub fn server_actions<C: Comments>(
    file_name: &FileName,
    file_query: Option<RcStr>,
    config: Config,
    comments: C,
    unresolved_mark: Mark,
    cm: Arc<SourceMap>,
    use_cache_telemetry_tracker: Rc<RefCell<FxHashMap<String, usize>>>,
    mode: ServerActionsMode,
) -> impl Pass + use<C> {
    visit_mut_pass(ServerActions {
        config,
        mode,
        comments,
        cm,
        file_name: file_name.to_string(),
        file_query,
        start_pos: BytePos(0),
        file_directive: None,
        current_export_name: None,
        fn_decl_ident: None,
        in_callee: false,
        has_action: false,
        has_cache: false,
        this_status: ThisStatus::Allowed,

        reference_index: 0,
        in_module_level: true,
        should_track_names: false,
        has_server_reference_with_bound_args: false,

        names: Default::default(),
        declared_idents: Default::default(),

        // This flag allows us to rewrite `function foo() {}` to `const foo = createProxy(...)`.
        rewrite_fn_decl_to_proxy_decl: None,
        rewrite_default_fn_expr_to_proxy_expr: None,
        rewrite_expr_to_proxy_expr: None,

        annotations: Default::default(),
        extra_items: Default::default(),
        hoisted_extra_items: Default::default(),
        reference_ids_by_export_name: Default::default(),
        server_reference_exports: Default::default(),

        private_ctxt: SyntaxContext::empty().apply_mark(Mark::new()),
        unresolved_ctxt: SyntaxContext::empty().apply_mark(unresolved_mark),

        arrow_or_fn_expr_ident: None,
        export_name_by_local_id: Default::default(),
        local_ids_that_need_cache_runtime_wrapper_if_exported: FxHashSet::default(),

        use_cache_telemetry_tracker,
    })
}

/// Serializes the Server References into a magic comment prefixed by
/// `__next_internal_action_entry_do_not_use__`.
fn generate_server_references_comment(
    export_infos_ordered_by_reference_id: &BTreeMap<&Atom, ServerReferenceExportInfo>,
    entry_path_query: Option<(&str, &str)>,
) -> String {
    format!(
        " __next_internal_action_entry_do_not_use__ {} ",
        if let Some(entry_path_query) = entry_path_query {
            serde_json::to_string(&(
                &export_infos_ordered_by_reference_id,
                entry_path_query.0,
                entry_path_query.1,
            ))
        } else {
            serde_json::to_string(&export_infos_ordered_by_reference_id)
        }
        .unwrap()
    )
}

struct ServerActions<C: Comments> {
    #[allow(unused)]
    config: Config,
    file_name: String,
    file_query: Option<RcStr>,
    comments: C,
    cm: Arc<SourceMap>,
    mode: ServerActionsMode,

    start_pos: BytePos,
    file_directive: Option<Directive>,
    current_export_name: Option<ModuleExportName>,
    fn_decl_ident: Option<Ident>,
    in_callee: bool,
    has_action: bool,
    has_cache: bool,
    this_status: ThisStatus,

    reference_index: u32,
    in_module_level: bool,
    should_track_names: bool,
    has_server_reference_with_bound_args: bool,

    names: Vec<Name>,
    declared_idents: Vec<Ident>,

    // This flag allows us to rewrite `function foo() {}` to `const foo = createProxy(...)`.
    rewrite_fn_decl_to_proxy_decl: Option<VarDecl>,
    rewrite_default_fn_expr_to_proxy_expr: Option<Box<Expr>>,
    rewrite_expr_to_proxy_expr: Option<Box<Expr>>,

    annotations: Vec<Stmt>,
    extra_items: Vec<ModuleItem>,
    hoisted_extra_items: Vec<ModuleItem>,

    /// A map of all server references (inline + exported): export_name -> reference_id
    reference_ids_by_export_name: FxIndexMap<ModuleExportName, Atom>,

    /// A list of server references for originally exported server functions only.
    server_reference_exports: Vec<ServerReferenceExport>,

    private_ctxt: SyntaxContext,
    unresolved_ctxt: SyntaxContext,

    arrow_or_fn_expr_ident: Option<Ident>,
    export_name_by_local_id: FxIndexMap<Id, ModuleExportName>,

    /// Tracks which local IDs need cache runtime wrappers if exported (collected during pre-pass).
    /// Includes imports, destructured identifiers, and variables with unknown-type init
    /// expressions (calls, identifiers, etc.). Excludes known functions (arrow/fn
    /// declarations) and known non-functions (object/array/literals). When these IDs are
    /// exported, their exports are stripped and replaced with conditional wrappers.
    local_ids_that_need_cache_runtime_wrapper_if_exported: FxHashSet<Id>,

    use_cache_telemetry_tracker: Rc<RefCell<FxHashMap<String, usize>>>,
}

impl<C: Comments> ServerActions<C> {
    fn generate_server_reference_id(
        &self,
        export_name: &ModuleExportName,
        is_cache: bool,
        params: Option<&Vec<Param>>,
    ) -> Atom {
        // Attach a checksum to the action using sha1:
        // $$id = special_byte + sha1('hash_salt' + 'file_name' + ':' + 'export_name');
        // Currently encoded as hex.

        let mut hasher = Sha1::new();
        hasher.update(self.config.hash_salt.as_bytes());
        hasher.update(self.file_name.as_bytes());
        hasher.update(b":");

        let export_name_bytes = match export_name {
            ModuleExportName::Ident(ident) => &ident.sym.as_bytes(),
            ModuleExportName::Str(s) => &s.value.as_bytes(),
        };

        hasher.update(export_name_bytes);

        let mut result = hasher.finalize().to_vec();

        // Prepend an extra byte to the ID, with the following format:
        // 0     000000    0
        // ^type ^arg mask ^rest args
        //
        // The type bit represents if the action is a cache function or not.
        // For cache functions, the type bit is set to 1. Otherwise, it's 0.
        //
        // The arg mask bit is used to determine which arguments are used by
        // the function itself, up to 6 arguments. The bit is set to 1 if the
        // argument is used, or being spread or destructured (so it can be
        // indirectly or partially used). The bit is set to 0 otherwise.
        //
        // The rest args bit is used to determine if there's a ...rest argument
        // in the function signature. If there is, the bit is set to 1.
        //
        //  For example:
        //
        //   async function foo(a, foo, b, bar, ...baz) {
        //     'use cache';
        //     return a + b;
        //   }
        //
        // will have it encoded as [1][101011][1]. The first bit is set to 1
        // because it's a cache function. The second part has 1010 because the
        // only arguments used are `a` and `b`. The subsequent 11 bits are set
        // to 1 because there's a ...rest argument starting from the 5th. The
        // last bit is set to 1 as well for the same reason.
        let type_bit = if is_cache { 1u8 } else { 0u8 };
        let mut arg_mask = 0u8;
        let mut rest_args = 0u8;

        if let Some(params) = params {
            // TODO: For the current implementation, we don't track if an
            // argument ident is actually referenced in the function body.
            // Instead, we go with the easy route and assume defined ones are
            // used. This can be improved in the future.
            for (i, param) in params.iter().enumerate() {
                if let Pat::Rest(_) = param.pat {
                    // If there's a ...rest argument, we set the rest args bit
                    // to 1 and set the arg mask to 0b111111.
                    arg_mask = 0b111111;
                    rest_args = 0b1;
                    break;
                }
                if i < 6 {
                    arg_mask |= 0b1 << (5 - i);
                } else {
                    // More than 6 arguments, we set the rest args bit to 1.
                    // This is rare for a Server Action, usually.
                    rest_args = 0b1;
                    break;
                }
            }
        } else {
            // If we can't determine the arguments (e.g. not statically analyzable),
            // we assume all arguments are used.
            arg_mask = 0b111111;
            rest_args = 0b1;
        }

        result.push((type_bit << 7) | (arg_mask << 1) | rest_args);
        result.rotate_right(1);

        Atom::from(hex_encode(result))
    }

    fn is_default_export(&self) -> bool {
        matches!(
            self.current_export_name,
            Some(ModuleExportName::Ident(ref i)) if i.sym == *"default"
        )
    }

    fn gen_action_ident(&mut self) -> Atom {
        let id: Atom = format!("$$RSC_SERVER_ACTION_{0}", self.reference_index).into();
        self.reference_index += 1;
        id
    }

    fn gen_cache_ident(&mut self) -> Atom {
        let id: Atom = format!("$$RSC_SERVER_CACHE_{0}", self.reference_index).into();
        self.reference_index += 1;
        id
    }

    fn create_bound_action_args_array_pat(&mut self, arg_len: usize) -> Pat {
        Pat::Array(ArrayPat {
            span: DUMMY_SP,
            elems: (0..arg_len)
                .map(|i| {
                    Some(Pat::Ident(
                        Ident::new(
                            format!("$$ACTION_ARG_{i}").into(),
                            DUMMY_SP,
                            self.private_ctxt,
                        )
                        .into(),
                    ))
                })
                .collect(),
            optional: false,
            type_ann: None,
        })
    }

    // Check if the function or arrow function is an action or cache function,
    // and remove any server function directive.
    fn get_directive_for_function(
        &mut self,
        maybe_body: Option<&mut BlockStmt>,
    ) -> Option<Directive> {
        let mut directive: Option<Directive> = None;

        // Even if it's a file-level action or cache module, the function body
        // might still have directives that override the module-level annotations.
        if let Some(body) = maybe_body {
            let directive_visitor = &mut DirectiveVisitor {
                config: &self.config,
                directive: None,
                has_file_directive: self.file_directive.is_some(),
                is_allowed_position: true,
                location: DirectiveLocation::FunctionBody,
                use_cache_telemetry_tracker: self.use_cache_telemetry_tracker.clone(),
            };

            body.stmts.retain(|stmt| {
                let has_directive = directive_visitor.visit_stmt(stmt);

                !has_directive
            });

            directive = directive_visitor.directive.clone();
        }

        // All exported functions inherit the file directive if they don't have their own directive.
        if self.current_export_name.is_some()
            && directive.is_none()
            && self.file_directive.is_some()
        {
            return self.file_directive.clone();
        }

        directive
    }

    fn get_directive_for_module(&mut self, stmts: &mut Vec<ModuleItem>) -> Option<Directive> {
        let directive_visitor = &mut DirectiveVisitor {
            config: &self.config,
            directive: None,
            has_file_directive: false,
            is_allowed_position: true,
            location: DirectiveLocation::Module,
            use_cache_telemetry_tracker: self.use_cache_telemetry_tracker.clone(),
        };

        stmts.retain(|item| {
            if let ModuleItem::Stmt(stmt) = item {
                let has_directive = directive_visitor.visit_stmt(stmt);

                !has_directive
            } else {
                directive_visitor.is_allowed_position = false;
                true
            }
        });

        directive_visitor.directive.clone()
    }

    fn maybe_hoist_and_create_proxy_for_server_action_arrow_expr(
        &mut self,
        ids_from_closure: Vec<Name>,
        arrow: &mut ArrowExpr,
    ) -> Box<Expr> {
        let mut new_params: Vec<Param> = vec![];

        let closure_bound_ident =
            Ident::new(atom!("$$ACTION_CLOSURE_BOUND"), DUMMY_SP, self.private_ctxt);

        if !ids_from_closure.is_empty() {
            // First param is the encrypted closure variables.
            new_params.push(Param {
                span: DUMMY_SP,
                decorators: vec![],
                pat: Pat::Ident(closure_bound_ident.clone().into()),
            });
        }

        for p in arrow.params.iter() {
            new_params.push(Param::from(p.clone()));
        }

        let action_name = self.gen_action_ident();
        let action_ident = Ident::new(action_name.clone(), arrow.span, self.private_ctxt);
        let action_id = self.generate_server_reference_id(
            &ModuleExportName::Ident(action_ident.clone()),
            false,
            Some(&new_params),
        );

        self.has_action = true;
        self.reference_ids_by_export_name.insert(
            ModuleExportName::Ident(action_ident.clone()),
            action_id.clone(),
        );

        // If this is an exported arrow, remove it from export_name_by_local_id so the
        // post-pass doesn't register it again (it's already registered above).
        if self.current_export_name.is_some()
            && let Some(arrow_ident) = &self.arrow_or_fn_expr_ident
        {
            self.export_name_by_local_id
                .swap_remove(&arrow_ident.to_id());
        }

        if let BlockStmtOrExpr::BlockStmt(block) = &mut *arrow.body {
            block.visit_mut_with(&mut ClosureReplacer {
                used_ids: &ids_from_closure,
                private_ctxt: self.private_ctxt,
            });
        }

        let mut new_body: BlockStmtOrExpr = *arrow.body.clone();

        if !ids_from_closure.is_empty() {
            // Prepend the decryption declaration to the body.
            // var [arg1, arg2, arg3] = await decryptActionBoundArgs(actionId,
            // $$ACTION_CLOSURE_BOUND)
            let decryption_decl = VarDecl {
                span: DUMMY_SP,
                kind: VarDeclKind::Var,
                declare: false,
                decls: vec![VarDeclarator {
                    span: DUMMY_SP,
                    name: self.create_bound_action_args_array_pat(ids_from_closure.len()),
                    init: Some(Box::new(Expr::Await(AwaitExpr {
                        span: DUMMY_SP,
                        arg: Box::new(Expr::Call(CallExpr {
                            span: DUMMY_SP,
                            callee: quote_ident!("decryptActionBoundArgs").as_callee(),
                            args: vec![action_id.clone().as_arg(), closure_bound_ident.as_arg()],
                            ..Default::default()
                        })),
                    }))),
                    definite: Default::default(),
                }],
                ..Default::default()
            };

            match &mut new_body {
                BlockStmtOrExpr::BlockStmt(body) => {
                    body.stmts.insert(0, decryption_decl.into());
                }
                BlockStmtOrExpr::Expr(body_expr) => {
                    new_body = BlockStmtOrExpr::BlockStmt(BlockStmt {
                        span: DUMMY_SP,
                        stmts: vec![
                            decryption_decl.into(),
                            Stmt::Return(ReturnStmt {
                                span: DUMMY_SP,
                                arg: Some(body_expr.take()),
                            }),
                        ],
                        ..Default::default()
                    });
                }
            }
        }

        // Create the action export decl from the arrow function
        // export const $$RSC_SERVER_ACTION_0 = async function action($$ACTION_CLOSURE_BOUND) {}
        self.hoisted_extra_items
            .push(ModuleItem::ModuleDecl(ModuleDecl::ExportDecl(ExportDecl {
                span: DUMMY_SP,
                decl: VarDecl {
                    kind: VarDeclKind::Const,
                    span: DUMMY_SP,
                    decls: vec![VarDeclarator {
                        span: DUMMY_SP,
                        name: Pat::Ident(action_ident.clone().into()),
                        definite: false,
                        init: Some(Box::new(Expr::Fn(FnExpr {
                            ident: self.arrow_or_fn_expr_ident.clone(),
                            function: Box::new(Function {
                                params: new_params,
                                body: match new_body {
                                    BlockStmtOrExpr::BlockStmt(body) => Some(body),
                                    BlockStmtOrExpr::Expr(expr) => Some(BlockStmt {
                                        span: DUMMY_SP,
                                        stmts: vec![Stmt::Return(ReturnStmt {
                                            span: DUMMY_SP,
                                            arg: Some(expr),
                                        })],
                                        ..Default::default()
                                    }),
                                },
                                is_async: true,
                                ..Default::default()
                            }),
                        }))),
                    }],
                    declare: Default::default(),
                    ctxt: self.private_ctxt,
                }
                .into(),
            })));

        self.hoisted_extra_items
            .push(ModuleItem::Stmt(Stmt::Expr(ExprStmt {
                span: DUMMY_SP,
                expr: Box::new(annotate_ident_as_server_reference(
                    action_ident.clone(),
                    action_id.clone(),
                    arrow.span,
                )),
            })));

        if ids_from_closure.is_empty() {
            Box::new(action_ident.clone().into())
        } else {
            self.has_server_reference_with_bound_args = true;
            Box::new(bind_args_to_ident(
                action_ident.clone(),
                ids_from_closure
                    .iter()
                    .cloned()
                    .map(|id| Some(id.as_arg()))
                    .collect(),
                action_id.clone(),
            ))
        }
    }

    fn maybe_hoist_and_create_proxy_for_server_action_function(
        &mut self,
        ids_from_closure: Vec<Name>,
        function: &mut Function,
        fn_name: Option<Ident>,
    ) -> Box<Expr> {
        let mut new_params: Vec<Param> = vec![];

        let closure_bound_ident =
            Ident::new(atom!("$$ACTION_CLOSURE_BOUND"), DUMMY_SP, self.private_ctxt);

        if !ids_from_closure.is_empty() {
            // First param is the encrypted closure variables.
            new_params.push(Param {
                span: DUMMY_SP,
                decorators: vec![],
                pat: Pat::Ident(closure_bound_ident.clone().into()),
            });
        }

        new_params.append(&mut function.params);

        let action_name: Atom = self.gen_action_ident();
        let mut action_ident = Ident::new(action_name.clone(), function.span, self.private_ctxt);
        if action_ident.span.lo == self.start_pos {
            action_ident.span = Span::dummy_with_cmt();
        }

        let action_id = self.generate_server_reference_id(
            &ModuleExportName::Ident(action_ident.clone()),
            false,
            Some(&new_params),
        );

        self.has_action = true;
        self.reference_ids_by_export_name.insert(
            ModuleExportName::Ident(action_ident.clone()),
            action_id.clone(),
        );

        // If this is an exported function, remove it from export_name_by_local_id so the
        // post-pass doesn't register it again (it's already registered above).
        if self.current_export_name.is_some()
            && let Some(ref fn_name) = fn_name
        {
            self.export_name_by_local_id.swap_remove(&fn_name.to_id());
        }

        function.body.visit_mut_with(&mut ClosureReplacer {
            used_ids: &ids_from_closure,
            private_ctxt: self.private_ctxt,
        });

        let mut new_body: Option<BlockStmt> = function.body.clone();

        if !ids_from_closure.is_empty() {
            // Prepend the decryption declaration to the body.
            // var [arg1, arg2, arg3] = await decryptActionBoundArgs(actionId,
            // $$ACTION_CLOSURE_BOUND)
            let decryption_decl = VarDecl {
                span: DUMMY_SP,
                kind: VarDeclKind::Var,
                decls: vec![VarDeclarator {
                    span: DUMMY_SP,
                    name: self.create_bound_action_args_array_pat(ids_from_closure.len()),
                    init: Some(Box::new(Expr::Await(AwaitExpr {
                        span: DUMMY_SP,
                        arg: Box::new(Expr::Call(CallExpr {
                            span: DUMMY_SP,
                            callee: quote_ident!("decryptActionBoundArgs").as_callee(),
                            args: vec![action_id.clone().as_arg(), closure_bound_ident.as_arg()],
                            ..Default::default()
                        })),
                    }))),
                    definite: Default::default(),
                }],
                ..Default::default()
            };

            if let Some(body) = &mut new_body {
                body.stmts.insert(0, decryption_decl.into());
            } else {
                new_body = Some(BlockStmt {
                    span: DUMMY_SP,
                    stmts: vec![decryption_decl.into()],
                    ..Default::default()
                });
            }
        }

        // Create the action export decl from the function
        // export const $$RSC_SERVER_ACTION_0 = async function action($$ACTION_CLOSURE_BOUND) {}
        self.hoisted_extra_items
            .push(ModuleItem::ModuleDecl(ModuleDecl::ExportDecl(ExportDecl {
                span: DUMMY_SP,
                decl: VarDecl {
                    kind: VarDeclKind::Const,
                    span: DUMMY_SP,
                    decls: vec![VarDeclarator {
                        span: DUMMY_SP, // TODO: need to map it to the original span?
                        name: Pat::Ident(action_ident.clone().into()),
                        definite: false,
                        init: Some(Box::new(Expr::Fn(FnExpr {
                            ident: fn_name,
                            function: Box::new(Function {
                                params: new_params,
                                body: new_body,
                                ..function.take()
                            }),
                        }))),
                    }],
                    declare: Default::default(),
                    ctxt: self.private_ctxt,
                }
                .into(),
            })));

        self.hoisted_extra_items
            .push(ModuleItem::Stmt(Stmt::Expr(ExprStmt {
                span: DUMMY_SP,
                expr: Box::new(annotate_ident_as_server_reference(
                    action_ident.clone(),
                    action_id.clone(),
                    function.span,
                )),
            })));

        if ids_from_closure.is_empty() {
            Box::new(action_ident.clone().into())
        } else {
            self.has_server_reference_with_bound_args = true;
            Box::new(bind_args_to_ident(
                action_ident.clone(),
                ids_from_closure
                    .iter()
                    .cloned()
                    .map(|id| Some(id.as_arg()))
                    .collect(),
                action_id.clone(),
            ))
        }
    }

    fn maybe_hoist_and_create_proxy_for_cache_arrow_expr(
        &mut self,
        ids_from_closure: Vec<Name>,
        cache_kind: RcStr,
        arrow: &mut ArrowExpr,
    ) -> Box<Expr> {
        let mut new_params: Vec<Param> = vec![];

        // Add the collected closure variables as the first parameter to the
        // function. They are unencrypted and passed into this function by the
        // cache wrapper.
        if !ids_from_closure.is_empty() {
            new_params.push(Param {
                span: DUMMY_SP,
                decorators: vec![],
                pat: self.create_bound_action_args_array_pat(ids_from_closure.len()),
            });
        }

        for p in arrow.params.iter() {
            new_params.push(Param::from(p.clone()));
        }

        let cache_name: Atom = self.gen_cache_ident();
        let export_name: Atom = cache_name.clone();

        let reference_id = self.generate_server_reference_id(
            &ModuleExportName::Ident(export_name.clone().into()),
            true,
            Some(&new_params),
        );

        self.has_cache = true;
        self.reference_ids_by_export_name.insert(
            ModuleExportName::Ident(export_name.clone().into()),
            reference_id.clone(),
        );

        // If this is an exported arrow, remove it from export_name_by_local_id so the
        // post-pass doesn't register it again (it's already registered above).
        if self.current_export_name.is_some()
            && let Some(arrow_ident) = &self.arrow_or_fn_expr_ident
        {
            self.export_name_by_local_id
                .swap_remove(&arrow_ident.to_id());
        }

        if let BlockStmtOrExpr::BlockStmt(block) = &mut *arrow.body {
            block.visit_mut_with(&mut ClosureReplacer {
                used_ids: &ids_from_closure,
                private_ctxt: self.private_ctxt,
            });
        }

        let inner_fn_body = match *arrow.body.take() {
            BlockStmtOrExpr::BlockStmt(body) => Some(body),
            BlockStmtOrExpr::Expr(expr) => Some(BlockStmt {
                stmts: vec![Stmt::Return(ReturnStmt {
                    span: DUMMY_SP,
                    arg: Some(expr),
                })],
                ..Default::default()
            }),
        };

        let cache_ident = create_and_hoist_cache_function(
            cache_kind.as_str(),
            reference_id.clone(),
            ids_from_closure.len(),
            cache_name,
            self.arrow_or_fn_expr_ident.clone(),
            new_params.clone(),
            inner_fn_body,
            arrow.span,
            &mut self.hoisted_extra_items,
            self.unresolved_ctxt,
        );

        if let Some(Ident { sym, .. }) = &self.arrow_or_fn_expr_ident {
            self.hoisted_extra_items
                .push(ModuleItem::Stmt(assign_name_to_ident(
                    &cache_ident,
                    sym.as_str(),
                    self.unresolved_ctxt,
                )));
        }

        let bound_args: Vec<_> = ids_from_closure
            .iter()
            .cloned()
            .map(|id| Some(id.as_arg()))
            .collect();

        if bound_args.is_empty() {
            Box::new(cache_ident.clone().into())
        } else {
            self.has_server_reference_with_bound_args = true;
            Box::new(bind_args_to_ident(
                cache_ident.clone(),
                bound_args,
                reference_id.clone(),
            ))
        }
    }

    fn maybe_hoist_and_create_proxy_for_cache_function(
        &mut self,
        ids_from_closure: Vec<Name>,
        fn_name: Option<Ident>,
        cache_kind: RcStr,
        function: &mut Function,
    ) -> Box<Expr> {
        let mut new_params: Vec<Param> = vec![];

        // Add the collected closure variables as the first parameter to the
        // function. They are unencrypted and passed into this function by the
        // cache wrapper.
        if !ids_from_closure.is_empty() {
            new_params.push(Param {
                span: DUMMY_SP,
                decorators: vec![],
                pat: self.create_bound_action_args_array_pat(ids_from_closure.len()),
            });
        }

        for p in function.params.iter() {
            new_params.push(p.clone());
        }

        let cache_name: Atom = self.gen_cache_ident();

        let reference_id = self.generate_server_reference_id(
            &ModuleExportName::Ident(cache_name.clone().into()),
            true,
            Some(&new_params),
        );

        self.has_cache = true;
        self.reference_ids_by_export_name.insert(
            ModuleExportName::Ident(cache_name.clone().into()),
            reference_id.clone(),
        );

        // If this is an exported function, remove it from export_name_by_local_id so the
        // post-pass doesn't register it again (it's already registered above).
        if self.current_export_name.is_some()
            && let Some(ref fn_name) = fn_name
        {
            self.export_name_by_local_id.swap_remove(&fn_name.to_id());
        }

        function.body.visit_mut_with(&mut ClosureReplacer {
            used_ids: &ids_from_closure,
            private_ctxt: self.private_ctxt,
        });

        let function_body = function.body.take();
        let function_span = function.span;

        let cache_ident = create_and_hoist_cache_function(
            cache_kind.as_str(),
            reference_id.clone(),
            ids_from_closure.len(),
            cache_name,
            fn_name.clone(),
            new_params.clone(),
            function_body,
            function_span,
            &mut self.hoisted_extra_items,
            self.unresolved_ctxt,
        );

        if let Some(Ident { ref sym, .. }) = fn_name {
            self.hoisted_extra_items
                .push(ModuleItem::Stmt(assign_name_to_ident(
                    &cache_ident,
                    sym.as_str(),
                    self.unresolved_ctxt,
                )));
        } else if self.is_default_export() {
            self.hoisted_extra_items
                .push(ModuleItem::Stmt(assign_name_to_ident(
                    &cache_ident,
                    "default",
                    self.unresolved_ctxt,
                )));
        }

        let bound_args: Vec<_> = ids_from_closure
            .iter()
            .cloned()
            .map(|id| Some(id.as_arg()))
            .collect();

        if bound_args.is_empty() {
            Box::new(cache_ident.clone().into())
        } else {
            self.has_server_reference_with_bound_args = true;
            Box::new(bind_args_to_ident(
                cache_ident.clone(),
                bound_args,
                reference_id.clone(),
            ))
        }
    }

    /// Validates that a function is async, emitting an error if not.
    /// Returns true if async, false otherwise.
    fn validate_async_function(
        &self,
        is_async: bool,
        span: Span,
        fn_name: Option<&Ident>,
        directive: &Directive,
    ) -> bool {
        if is_async {
            true
        } else {
            emit_error(ServerActionsErrorKind::InlineSyncFunction {
                span: fn_name.as_ref().map_or(span, |ident| ident.span),
                directive: directive.clone(),
            });
            false
        }
    }

    /// Registers a server action export (for a 'use server' file directive).
    fn register_server_action_export(
        &mut self,
        export_name: &ModuleExportName,
        fn_name: Option<&Ident>,
        params: Option<&Vec<Param>>,
        span: Span,
        take_fn_or_arrow_expr: &mut dyn FnMut() -> Box<Expr>,
    ) {
        if let Some(fn_name) = fn_name {
            let reference_id = self.generate_server_reference_id(export_name, false, params);

            self.has_action = true;
            self.reference_ids_by_export_name
                .insert(export_name.clone(), reference_id.clone());

            self.server_reference_exports.push(ServerReferenceExport {
                ident: fn_name.clone(),
                export_name: export_name.clone(),
                reference_id: reference_id.clone(),
                needs_cache_runtime_wrapper: false,
            });
        } else if self.is_default_export() {
            let action_ident = Ident::new(self.gen_action_ident(), span, self.private_ctxt);
            let reference_id = self.generate_server_reference_id(export_name, false, params);

            self.has_action = true;
            self.reference_ids_by_export_name
                .insert(export_name.clone(), reference_id.clone());

            self.server_reference_exports.push(ServerReferenceExport {
                ident: action_ident.clone(),
                export_name: export_name.clone(),
                reference_id: reference_id.clone(),
                needs_cache_runtime_wrapper: false,
            });

            // For the server layer, also hoist the function and rewrite the default export.
            if self.config.is_react_server_layer {
                self.hoisted_extra_items
                    .push(ModuleItem::Stmt(Stmt::Decl(Decl::Var(Box::new(VarDecl {
                        kind: VarDeclKind::Const,
                        decls: vec![VarDeclarator {
                            span: DUMMY_SP,
                            name: Pat::Ident(action_ident.clone().into()),
                            init: Some(take_fn_or_arrow_expr()),
                            definite: false,
                        }],
                        ..Default::default()
                    })))));

                self.hoisted_extra_items
                    .push(ModuleItem::Stmt(assign_name_to_ident(
                        &action_ident,
                        "default",
                        self.unresolved_ctxt,
                    )));

                self.rewrite_default_fn_expr_to_proxy_expr =
                    Some(Box::new(Expr::Ident(action_ident)));
            }
        }
    }

    /// Registers a cache export for the client layer (for 'use cache' directive).
    fn register_cache_export_on_client(
        &mut self,
        export_name: &ModuleExportName,
        fn_name: Option<&Ident>,
        params: Option<&Vec<Param>>,
        span: Span,
    ) {
        if let Some(fn_name) = fn_name {
            let reference_id = self.generate_server_reference_id(export_name, true, params);

            self.has_cache = true;
            self.reference_ids_by_export_name
                .insert(export_name.clone(), reference_id.clone());

            self.server_reference_exports.push(ServerReferenceExport {
                ident: fn_name.clone(),
                export_name: export_name.clone(),
                reference_id: reference_id.clone(),
                needs_cache_runtime_wrapper: false,
            });
        } else if self.is_default_export() {
            let cache_ident = Ident::new(self.gen_cache_ident(), span, self.private_ctxt);
            let reference_id = self.generate_server_reference_id(export_name, true, params);

            self.has_cache = true;
            self.reference_ids_by_export_name
                .insert(export_name.clone(), reference_id.clone());

            self.server_reference_exports.push(ServerReferenceExport {
                ident: cache_ident.clone(),
                export_name: export_name.clone(),
                reference_id: reference_id.clone(),
                needs_cache_runtime_wrapper: false,
            });
        }
    }
}

impl<C: Comments> VisitMut for ServerActions<C> {
    fn visit_mut_export_decl(&mut self, decl: &mut ExportDecl) {
        // For inline exports like `export function foo() {}` or `export const bar = ...`,
        // the export name is looked up from export_name_by_local_id and set as current_export_name
        // in visit_mut_fn_decl or visit_mut_var_declarator.
        decl.decl.visit_mut_with(self);
    }

    fn visit_mut_export_default_decl(&mut self, decl: &mut ExportDefaultDecl) {
        let old_current_export_name = self.current_export_name.take();
        self.current_export_name = Some(ModuleExportName::Ident(atom!("default").into()));
        self.rewrite_default_fn_expr_to_proxy_expr = None;
        decl.decl.visit_mut_with(self);
        self.current_export_name = old_current_export_name;
    }

    fn visit_mut_export_default_expr(&mut self, expr: &mut ExportDefaultExpr) {
        let old_current_export_name = self.current_export_name.take();
        self.current_export_name = Some(ModuleExportName::Ident(atom!("default").into()));
        expr.expr.visit_mut_with(self);
        self.current_export_name = old_current_export_name;

        // For 'use server' or 'use cache' files with call expressions as default exports,
        // hoist the call expression to a const declarator.
        if matches!(&*expr.expr, Expr::Call(_)) {
            if matches!(self.file_directive, Some(Directive::UseServer)) {
                let export_name = ModuleExportName::Ident(atom!("default").into());
                let action_ident =
                    Ident::new(self.gen_action_ident(), expr.span, self.private_ctxt);
                let action_id = self.generate_server_reference_id(&export_name, false, None);

                self.has_action = true;
                self.reference_ids_by_export_name
                    .insert(export_name.clone(), action_id.clone());

                self.server_reference_exports.push(ServerReferenceExport {
                    ident: action_ident.clone(),
                    export_name: export_name.clone(),
                    reference_id: action_id.clone(),
                    needs_cache_runtime_wrapper: false,
                });

                self.hoisted_extra_items
                    .push(ModuleItem::Stmt(Stmt::Decl(Decl::Var(Box::new(VarDecl {
                        kind: VarDeclKind::Const,
                        decls: vec![VarDeclarator {
                            span: DUMMY_SP,
                            name: Pat::Ident(action_ident.clone().into()),
                            init: Some(expr.expr.take()),
                            definite: false,
                        }],
                        ..Default::default()
                    })))));

                self.rewrite_default_fn_expr_to_proxy_expr =
                    Some(Box::new(Expr::Ident(action_ident)));
            } else if matches!(self.file_directive, Some(Directive::UseCache { .. })) {
                let cache_ident = Ident::new(self.gen_cache_ident(), expr.span, self.private_ctxt);

                self.export_name_by_local_id.insert(
                    cache_ident.to_id(),
                    ModuleExportName::Ident(atom!("default").into()),
                );

                self.local_ids_that_need_cache_runtime_wrapper_if_exported
                    .insert(cache_ident.to_id());

                self.hoisted_extra_items
                    .push(ModuleItem::Stmt(Stmt::Decl(Decl::Var(Box::new(VarDecl {
                        kind: VarDeclKind::Const,
                        decls: vec![VarDeclarator {
                            span: DUMMY_SP,
                            name: Pat::Ident(cache_ident.into()),
                            init: Some(expr.expr.take()),
                            definite: false,
                        }],
                        ..Default::default()
                    })))));

                // Note: We don't set rewrite_default_fn_expr_to_proxy_expr here. The export will be
                // removed via the should_remove_statement flag in the main pass.
            }
        }
    }

    fn visit_mut_fn_expr(&mut self, f: &mut FnExpr) {
        let old_this_status = replace(&mut self.this_status, ThisStatus::Allowed);
        let old_arrow_or_fn_expr_ident = self.arrow_or_fn_expr_ident.clone();
        if let Some(ident) = &f.ident {
            self.arrow_or_fn_expr_ident = Some(ident.clone());
        }
        f.visit_mut_children_with(self);
        self.this_status = old_this_status;
        self.arrow_or_fn_expr_ident = old_arrow_or_fn_expr_ident;
    }

    fn visit_mut_function(&mut self, f: &mut Function) {
        let directive = self.get_directive_for_function(f.body.as_mut());
        let declared_idents_until = self.declared_idents.len();
        let old_names = take(&mut self.names);

        if let Some(directive) = &directive {
            self.this_status = ThisStatus::Forbidden {
                directive: directive.clone(),
            };
        }

        // Visit children
        {
            let old_in_module = replace(&mut self.in_module_level, false);
            let should_track_names = directive.is_some() || self.should_track_names;
            let old_should_track_names = replace(&mut self.should_track_names, should_track_names);
            let old_current_export_name = self.current_export_name.take();
            let old_fn_decl_ident = self.fn_decl_ident.take();
            f.visit_mut_children_with(self);
            self.in_module_level = old_in_module;
            self.should_track_names = old_should_track_names;
            self.current_export_name = old_current_export_name;
            self.fn_decl_ident = old_fn_decl_ident;
        }

        let mut child_names = take(&mut self.names);

        if self.should_track_names {
            self.names = [old_names, child_names.clone()].concat();
        }

        if let Some(directive) = directive {
            let fn_name = self
                .fn_decl_ident
                .as_ref()
                .or(self.arrow_or_fn_expr_ident.as_ref())
                .cloned();

            if !self.validate_async_function(f.is_async, f.span, fn_name.as_ref(), &directive) {
                // If this is an exported function that failed validation, remove it from
                // export_name_by_local_id so the post-pass doesn't register it.
                if self.current_export_name.is_some()
                    && let Some(fn_name) = fn_name
                {
                    self.export_name_by_local_id.swap_remove(&fn_name.to_id());
                }

                return;
            }

            // If this function is invalid, or any prior errors have been emitted, skip further
            // processing.
            if HANDLER.with(|handler| handler.has_errors()) {
                return;
            }

            // For server action files, register exports without hoisting (for both server and
            // client layers).
            if matches!(self.file_directive, Some(Directive::UseServer))
                && matches!(directive, Directive::UseServer)
                && let Some(export_name) = self.current_export_name.clone()
            {
                let params = f.params.clone();
                let span = f.span;

                self.register_server_action_export(
                    &export_name,
                    fn_name.as_ref(),
                    Some(&params),
                    span,
                    &mut || {
                        Box::new(Expr::Fn(FnExpr {
                            ident: fn_name.clone(),
                            function: Box::new(f.take()),
                        }))
                    },
                );

                return;
            }

            // For the client layer, register cache exports without hoisting.
            if !self.config.is_react_server_layer {
                if matches!(directive, Directive::UseCache { .. })
                    && let Some(export_name) = self.current_export_name.clone()
                {
                    self.register_cache_export_on_client(
                        &export_name,
                        fn_name.as_ref(),
                        Some(&f.params),
                        f.span,
                    );
                }

                return;
            }

            if let Directive::UseCache { cache_kind } = directive {
                // Collect all the identifiers defined inside the closure and used
                // in the cache function. With deduplication.
                retain_names_from_declared_idents(
                    &mut child_names,
                    &self.declared_idents[..declared_idents_until],
                );

                let new_expr = self.maybe_hoist_and_create_proxy_for_cache_function(
                    child_names.clone(),
                    self.fn_decl_ident
                        .as_ref()
                        .or(self.arrow_or_fn_expr_ident.as_ref())
                        .cloned(),
                    cache_kind,
                    f,
                );

                if self.is_default_export() {
                    // This function expression is also the default export:
                    // `export default async function() {}`
                    // This specific case (default export) isn't handled by `visit_mut_expr`.
                    // Replace the original function expr with a action proxy expr.
                    self.rewrite_default_fn_expr_to_proxy_expr = Some(new_expr);
                } else if let Some(ident) = &self.fn_decl_ident {
                    // Replace the original function declaration with a cache decl.
                    self.rewrite_fn_decl_to_proxy_decl = Some(VarDecl {
                        span: DUMMY_SP,
                        kind: VarDeclKind::Var,
                        decls: vec![VarDeclarator {
                            span: DUMMY_SP,
                            name: Pat::Ident(ident.clone().into()),
                            init: Some(new_expr),
                            definite: false,
                        }],
                        ..Default::default()
                    });
                } else {
                    self.rewrite_expr_to_proxy_expr = Some(new_expr);
                }
            } else {
                // Collect all the identifiers defined inside the closure and used
                // in the action function. With deduplication.
                retain_names_from_declared_idents(
                    &mut child_names,
                    &self.declared_idents[..declared_idents_until],
                );

                let new_expr = self.maybe_hoist_and_create_proxy_for_server_action_function(
                    child_names,
                    f,
                    fn_name,
                );

                if self.is_default_export() {
                    // This function expression is also the default export:
                    // `export default async function() {}`
                    // This specific case (default export) isn't handled by `visit_mut_expr`.
                    // Replace the original function expr with a action proxy expr.
                    self.rewrite_default_fn_expr_to_proxy_expr = Some(new_expr);
                } else if let Some(ident) = &self.fn_decl_ident {
                    // Replace the original function declaration with an action proxy
                    // declaration expr.
                    self.rewrite_fn_decl_to_proxy_decl = Some(VarDecl {
                        span: DUMMY_SP,
                        kind: VarDeclKind::Var,
                        decls: vec![VarDeclarator {
                            span: DUMMY_SP,
                            name: Pat::Ident(ident.clone().into()),
                            init: Some(new_expr),
                            definite: false,
                        }],
                        ..Default::default()
                    });
                } else {
                    self.rewrite_expr_to_proxy_expr = Some(new_expr);
                }
            }
        }
    }

    fn visit_mut_decl(&mut self, d: &mut Decl) {
        self.rewrite_fn_decl_to_proxy_decl = None;
        d.visit_mut_children_with(self);

        if let Some(decl) = &self.rewrite_fn_decl_to_proxy_decl {
            *d = (*decl).clone().into();
        }

        self.rewrite_fn_decl_to_proxy_decl = None;
    }

    fn visit_mut_fn_decl(&mut self, f: &mut FnDecl) {
        let old_this_status = replace(&mut self.this_status, ThisStatus::Allowed);
        let old_current_export_name = self.current_export_name.take();
        if self.in_module_level
            && let Some(export_name) = self.export_name_by_local_id.get(&f.ident.to_id())
        {
            self.current_export_name = Some(export_name.clone());
        }
        let old_fn_decl_ident = self.fn_decl_ident.replace(f.ident.clone());
        f.visit_mut_children_with(self);
        self.this_status = old_this_status;
        self.current_export_name = old_current_export_name;
        self.fn_decl_ident = old_fn_decl_ident;
    }

    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.
        let directive = self.get_directive_for_function(
            if let BlockStmtOrExpr::BlockStmt(block) = &mut *a.body {
                Some(block)
            } else {
                None
            },
        );

        if let Some(directive) = &directive {
            self.this_status = ThisStatus::Forbidden {
                directive: directive.clone(),
            };
        }

        let declared_idents_until = self.declared_idents.len();
        let old_names = take(&mut self.names);

        {
            // Visit children
            let old_in_module = replace(&mut self.in_module_level, false);
            let should_track_names = directive.is_some() || self.should_track_names;
            let old_should_track_names = replace(&mut self.should_track_names, should_track_names);
            let old_current_export_name = self.current_export_name.take();
            {
                for n in &mut a.params {
                    collect_idents_in_pat(n, &mut self.declared_idents);
                }
            }
            a.visit_mut_children_with(self);
            self.in_module_level = old_in_module;
            self.should_track_names = old_should_track_names;
            self.current_export_name = old_current_export_name;
        }

        let mut child_names = take(&mut self.names);

        if self.should_track_names {
            self.names = [old_names, child_names.clone()].concat();
        }

        if let Some(directive) = directive {
            let arrow_ident = self.arrow_or_fn_expr_ident.clone();

            if !self.validate_async_function(a.is_async, a.span, arrow_ident.as_ref(), &directive) {
                // If this is an exported arrow function that failed validation, remove it from
                // export_name_by_local_id so the post-pass doesn't register it.
                if self.current_export_name.is_some()
                    && let Some(arrow_ident) = arrow_ident
                {
                    self.export_name_by_local_id
                        .swap_remove(&arrow_ident.to_id());
                }

                return;
            }

            // If this function is invalid, or any prior errors have been emitted, skip further
            // processing.
            if HANDLER.with(|handler| handler.has_errors()) {
                return;
            }

            // For server action files, register exports without hoisting (for both server and
            // client layers).
            if matches!(self.file_directive, Some(Directive::UseServer))
                && matches!(directive, Directive::UseServer)
                && let Some(export_name) = self.current_export_name.clone()
            {
                let params: Vec<Param> = a.params.iter().map(|p| Param::from(p.clone())).collect();

                self.register_server_action_export(
                    &export_name,
                    arrow_ident.as_ref(),
                    Some(&params),
                    a.span,
                    &mut || Box::new(Expr::Arrow(a.take())),
                );

                return;
            }

            // For the client layer, register cache exports without hoisting.
            if !self.config.is_react_server_layer {
                if matches!(directive, Directive::UseCache { .. })
                    && let Some(export_name) = self.current_export_name.clone()
                {
                    let params: Vec<Param> =
                        a.params.iter().map(|p| Param::from(p.clone())).collect();

                    self.register_cache_export_on_client(
                        &export_name,
                        arrow_ident.as_ref(),
                        Some(&params),
                        a.span,
                    );
                }

                return;
            }

            // Collect all the identifiers defined inside the closure and used
            // in the action function. With deduplication.
            retain_names_from_declared_idents(
                &mut child_names,
                &self.declared_idents[..declared_idents_until],
            );

            if let Directive::UseCache { cache_kind } = directive {
                self.rewrite_expr_to_proxy_expr =
                    Some(self.maybe_hoist_and_create_proxy_for_cache_arrow_expr(
                        child_names,
                        cache_kind,
                        a,
                    ));
            } else {
                self.rewrite_expr_to_proxy_expr = Some(
                    self.maybe_hoist_and_create_proxy_for_server_action_arrow_expr(child_names, a),
                );
            }
        }
    }

    fn visit_mut_module(&mut self, m: &mut Module) {
        self.start_pos = m.span.lo;
        m.visit_mut_children_with(self);
    }

    fn visit_mut_stmt(&mut self, n: &mut Stmt) {
        n.visit_mut_children_with(self);

        if self.in_module_level {
            return;
        }

        // If it's a closure (not in the module level), we need to collect
        // identifiers defined in the closure.
        collect_decl_idents_in_stmt(n, &mut self.declared_idents);
    }

    fn visit_mut_param(&mut self, n: &mut Param) {
        n.visit_mut_children_with(self);

        if self.in_module_level {
            return;
        }

        collect_idents_in_pat(&n.pat, &mut self.declared_idents);
    }

    fn visit_mut_prop_or_spread(&mut self, n: &mut PropOrSpread) {
        let old_arrow_or_fn_expr_ident = self.arrow_or_fn_expr_ident.clone();
        let old_current_export_name = self.current_export_name.take();

        match n {
            PropOrSpread::Prop(box Prop::KeyValue(KeyValueProp {
                key: PropName::Ident(ident_name),
                value: box Expr::Arrow(_) | box Expr::Fn(_),
                ..
            })) => {
                self.current_export_name = None;
                self.arrow_or_fn_expr_ident = Some(ident_name.clone().into());
            }
            PropOrSpread::Prop(box Prop::Method(MethodProp { key, .. })) => {
                let key = key.clone();

                if let PropName::Ident(ident_name) = &key {
                    self.arrow_or_fn_expr_ident = Some(ident_name.clone().into());
                }

                let old_this_status = replace(&mut self.this_status, ThisStatus::Allowed);
                self.rewrite_expr_to_proxy_expr = None;
                self.current_export_name = None;
                n.visit_mut_children_with(self);
                self.current_export_name = old_current_export_name.clone();
                self.this_status = old_this_status;

                if let Some(expr) = self.rewrite_expr_to_proxy_expr.take() {
                    *n = PropOrSpread::Prop(Box::new(Prop::KeyValue(KeyValueProp {
                        key,
                        value: expr,
                    })));
                }

                return;
            }
            _ => {}
        }

        if !self.in_module_level
            && self.should_track_names
            && let PropOrSpread::Prop(box Prop::Shorthand(i)) = n
        {
            self.names.push(Name::from(&*i));
            self.should_track_names = false;
            n.visit_mut_children_with(self);
            self.should_track_names = true;
            return;
        }

        n.visit_mut_children_with(self);
        self.arrow_or_fn_expr_ident = old_arrow_or_fn_expr_ident;
        self.current_export_name = old_current_export_name;
    }

    fn visit_mut_class(&mut self, n: &mut Class) {
        let old_this_status = replace(&mut self.this_status, ThisStatus::Allowed);
        n.visit_mut_children_with(self);
        self.this_status = old_this_status;
    }

    fn visit_mut_class_member(&mut self, n: &mut ClassMember) {
        if let ClassMember::Method(ClassMethod {
            is_abstract: false,
            is_static: true,
            kind: MethodKind::Method,
            key,
            span,
            accessibility: None | Some(Accessibility::Public),
            ..
        }) = n
        {
            let key = key.clone();
            let span = *span;
            let old_arrow_or_fn_expr_ident = self.arrow_or_fn_expr_ident.clone();

            if let PropName::Ident(ident_name) = &key {
                self.arrow_or_fn_expr_ident = Some(ident_name.clone().into());
            }

            let old_this_status = replace(&mut self.this_status, ThisStatus::Allowed);
            let old_current_export_name = self.current_export_name.take();
            self.rewrite_expr_to_proxy_expr = None;
            self.current_export_name = None;
            n.visit_mut_children_with(self);
            self.this_status = old_this_status;
            self.current_export_name = old_current_export_name;
            self.arrow_or_fn_expr_ident = old_arrow_or_fn_expr_ident;

            if let Some(expr) = self.rewrite_expr_to_proxy_expr.take() {
                *n = ClassMember::ClassProp(ClassProp {
                    span,
                    key,
                    value: Some(expr),
                    is_static: true,
                    ..Default::default()
                });
            }
        } else {
            n.visit_mut_children_with(self);
        }
    }

    fn visit_mut_class_method(&mut self, n: &mut ClassMethod) {
        if n.is_static {
            n.visit_mut_children_with(self);
        } else {
            let (is_action_fn, is_cache_fn) = has_body_directive(&n.function.body);

            if is_action_fn {
                emit_error(
                    ServerActionsErrorKind::InlineUseServerInClassInstanceMethod { span: n.span },
                );
            } else if is_cache_fn {
                emit_error(
                    ServerActionsErrorKind::InlineUseCacheInClassInstanceMethod { span: n.span },
                );
            } else {
                n.visit_mut_children_with(self);
            }
        }
    }

    fn visit_mut_call_expr(&mut self, n: &mut CallExpr) {
        if let Callee::Expr(box Expr::Ident(Ident { sym, .. })) = &mut n.callee
            && (sym == "jsxDEV" || sym == "_jsxDEV")
        {
            // Do not visit the 6th arg in a generated jsxDEV call, which is a `this`
            // expression, to avoid emitting an error for using `this` if it's
            // inside of a server function. https://github.com/facebook/react/blob/9106107/packages/react/src/jsx/ReactJSXElement.js#L429
            if n.args.len() > 4 {
                for arg in &mut n.args[0..4] {
                    arg.visit_mut_with(self);
                }
                return;
            }
        }

        let old_current_export_name = self.current_export_name.take();
        n.visit_mut_children_with(self);
        self.current_export_name = old_current_export_name;
    }

    fn visit_mut_callee(&mut self, n: &mut Callee) {
        let old_in_callee = replace(&mut self.in_callee, true);
        n.visit_mut_children_with(self);
        self.in_callee = old_in_callee;
    }

    fn visit_mut_expr(&mut self, n: &mut Expr) {
        if !self.in_module_level
            && self.should_track_names
            && let Ok(mut name) = Name::try_from(&*n)
        {
            if self.in_callee {
                // This is a callee i.e. `foo.bar()`,
                // we need to track the actual value instead of the method name.
                if !name.1.is_empty() {
                    name.1.pop();
                }
            }

            self.names.push(name);
            self.should_track_names = false;
            n.visit_mut_children_with(self);
            self.should_track_names = true;
            return;
        }

        self.rewrite_expr_to_proxy_expr = None;
        n.visit_mut_children_with(self);
        if let Some(expr) = self.rewrite_expr_to_proxy_expr.take() {
            *n = *expr;
        }
    }

    fn visit_mut_module_items(&mut self, stmts: &mut Vec<ModuleItem>) {
        self.file_directive = self.get_directive_for_module(stmts);

        let in_cache_file = matches!(self.file_directive, Some(Directive::UseCache { .. }));
        let in_action_file = matches!(self.file_directive, Some(Directive::UseServer));

        // Only track exported identifiers in server action files or cache files.
        let should_track_exports = in_action_file || in_cache_file;

        // Pre-pass: Collect a mapping from local identifiers to export names for all exports
        // in server boundary files ('use server' or 'use cache'). This mapping is used to:
        // 1. Set current_export_name when visiting exported functions/variables during the main
        //    pass.
        // 2. Register any remaining exports in the post-pass that weren't handled by the visitor.
        if should_track_exports {
            for stmt in stmts.iter() {
                match stmt {
                    ModuleItem::ModuleDecl(ModuleDecl::ExportDefaultExpr(export_default_expr)) => {
                        if let Expr::Ident(ident) = &*export_default_expr.expr {
                            self.export_name_by_local_id.insert(
                                ident.to_id(),
                                ModuleExportName::Ident(atom!("default").into()),
                            );
                        }
                    }
                    ModuleItem::ModuleDecl(ModuleDecl::ExportDefaultDecl(export_default_decl)) => {
                        // export default function foo() {}
                        if let DefaultDecl::Fn(f) = &export_default_decl.decl
                            && let Some(ident) = &f.ident
                        {
                            self.export_name_by_local_id.insert(
                                ident.to_id(),
                                ModuleExportName::Ident(atom!("default").into()),
                            );
                        }
                    }
                    ModuleItem::ModuleDecl(ModuleDecl::ExportDecl(export_decl)) => {
                        // export function foo() {} or export const bar = ...
                        match &export_decl.decl {
                            Decl::Fn(f) => {
                                self.export_name_by_local_id.insert(
                                    f.ident.to_id(),
                                    ModuleExportName::Ident(f.ident.clone()),
                                );
                            }
                            Decl::Var(var) => {
                                for decl in &var.decls {
                                    // Collect all identifiers from the pattern and track which may
                                    // need cache runtime wrappers. For destructuring patterns, we
                                    // always need wrappers since we can't statically know if the
                                    // destructured values are functions. For simple identifiers,
                                    // check the init expression.
                                    let mut idents = vec![];
                                    collect_idents_in_pat(&decl.name, &mut idents);

                                    let is_destructuring = !matches!(&decl.name, Pat::Ident(_));
                                    let needs_wrapper = if is_destructuring {
                                        true
                                    } else if let Some(init) = &decl.init {
                                        may_need_cache_runtime_wrapper(init)
                                    } else {
                                        false
                                    };

                                    for ident in idents {
                                        self.export_name_by_local_id.insert(
                                            ident.to_id(),
                                            ModuleExportName::Ident(ident.clone()),
                                        );

                                        if needs_wrapper {
                                            self.local_ids_that_need_cache_runtime_wrapper_if_exported
                                                .insert(ident.to_id());
                                        }
                                    }
                                }
                            }
                            _ => {}
                        }
                    }
                    ModuleItem::ModuleDecl(ModuleDecl::ExportNamed(named_export))
                        if named_export.src.is_none() && !named_export.type_only =>
                    {
                        for spec in &named_export.specifiers {
                            match spec {
                                ExportSpecifier::Named(ExportNamedSpecifier {
                                    orig: ModuleExportName::Ident(orig),
                                    exported: Some(exported),
                                    is_type_only: false,
                                    ..
                                }) => {
                                    // export { foo as bar } or export { foo as "📙" }
                                    self.export_name_by_local_id
                                        .insert(orig.to_id(), exported.clone());
                                }
                                ExportSpecifier::Named(ExportNamedSpecifier {
                                    orig: ModuleExportName::Ident(orig),
                                    exported: None,
                                    is_type_only: false,
                                    ..
                                }) => {
                                    // export { foo }
                                    self.export_name_by_local_id.insert(
                                        orig.to_id(),
                                        ModuleExportName::Ident(orig.clone()),
                                    );
                                }
                                _ => {}
                            }
                        }
                    }
                    ModuleItem::Stmt(Stmt::Decl(Decl::Var(var_decl))) => {
                        // Track which declarations need cache runtime wrappers if exported.
                        for decl in &var_decl.decls {
                            if let Pat::Ident(ident_pat) = &decl.name
                                && let Some(init) = &decl.init
                                && may_need_cache_runtime_wrapper(init)
                            {
                                self.local_ids_that_need_cache_runtime_wrapper_if_exported
                                    .insert(ident_pat.id.to_id());
                            }
                        }
                    }
                    ModuleItem::Stmt(Stmt::Decl(Decl::Fn(_fn_decl))) => {
                        // Function declarations are known functions and don't need runtime
                        // wrappers.
                    }
                    ModuleItem::ModuleDecl(ModuleDecl::Import(import_decl)) => {
                        // Track all imports. We don't know if they're functions, so they need
                        // runtime wrappers if they end up getting re-exported.
                        for spec in &import_decl.specifiers {
                            match spec {
                                ImportSpecifier::Named(named) => {
                                    self.local_ids_that_need_cache_runtime_wrapper_if_exported
                                        .insert(named.local.to_id());
                                }
                                ImportSpecifier::Default(default) => {
                                    self.local_ids_that_need_cache_runtime_wrapper_if_exported
                                        .insert(default.local.to_id());
                                }
                                ImportSpecifier::Namespace(ns) => {
                                    self.local_ids_that_need_cache_runtime_wrapper_if_exported
                                        .insert(ns.local.to_id());
                                }
                            }
                        }
                    }
                    _ => {}
                }
            }
        }

        let old_annotations = self.annotations.take();
        let mut new = Vec::with_capacity(stmts.len());

        // Main pass: For each statement, validate exports in server boundary files,
        // visit and transform it, and add it to the output along with any hoisted items.
        for mut stmt in stmts.take() {
            let mut should_remove_statement = false;

            if should_track_exports {
                let mut disallowed_export_span = DUMMY_SP;

                match &mut stmt {
                    ModuleItem::ModuleDecl(ModuleDecl::ExportDecl(ExportDecl { decl, span })) => {
                        match decl {
                            Decl::Var(var) => {
                                let mut has_export_needing_wrapper = false;

                                for decl in &var.decls {
                                    if let Pat::Ident(_) = &decl.name
                                        && let Some(init) = &decl.init
                                    {
                                        // Disallow exporting literals. Admittedly, this is
                                        // pretty arbitrary. We don't disallow exporting object
                                        // and array literals, as that would be too restrictive,
                                        // especially for page and layout files with
                                        // 'use cache', that may want to export metadata or
                                        // viewport objects.
                                        if let Expr::Lit(_) = &**init {
                                            disallowed_export_span = *span;
                                        }
                                    }

                                    // For cache files, check if any identifier needs a runtime
                                    // wrapper.
                                    if in_cache_file {
                                        let mut idents: Vec<Ident> = Vec::new();
                                        collect_idents_in_pat(&decl.name, &mut idents);

                                        for ident in idents {
                                            let needs_cache_runtime_wrapper = self
                                                .local_ids_that_need_cache_runtime_wrapper_if_exported
                                                .contains(&ident.to_id());

                                            if needs_cache_runtime_wrapper {
                                                has_export_needing_wrapper = true;
                                            }
                                        }
                                    }
                                }

                                // If this export needs a cache runtime wrapper, convert to a
                                // regular declaration (remove export keyword).
                                if in_cache_file && has_export_needing_wrapper {
                                    stmt = ModuleItem::Stmt(Stmt::Decl(Decl::Var(var.clone())));
                                }
                            }
                            Decl::Fn(_)
                            | Decl::TsInterface(_)
                            | Decl::TsTypeAlias(_)
                            | Decl::TsEnum(_) => {}
                            _ => {
                                disallowed_export_span = *span;
                            }
                        }
                    }
                    ModuleItem::ModuleDecl(ModuleDecl::ExportNamed(named)) if !named.type_only => {
                        if let Some(src) = &named.src {
                            // export { x } from './module'
                            if in_cache_file {
                                // Transform re-exports into imports so we can wrap them with
                                // cache runtime wrappers.
                                let import_specs: Vec<ImportSpecifier> = named
                                        .specifiers
                                        .iter()
                                        .filter_map(|spec| {
                                            if let ExportSpecifier::Named(ExportNamedSpecifier {
                                                orig: ModuleExportName::Ident(orig),
                                                exported,
                                                is_type_only: false,
                                                ..
                                            }) = spec
                                            {
                                                    // Now that we're converting this to an import,
                                                    // track it as a local export so the post-pass
                                                    // can register it.
                                                    let export_name =
                                                        if let Some(exported) = exported {
                                                            exported.clone()
                                                        } else {
                                                            ModuleExportName::Ident(orig.clone())
                                                        };

                                                    self.export_name_by_local_id
                                                        .insert(orig.to_id(), export_name);

                                                    self.local_ids_that_need_cache_runtime_wrapper_if_exported
                                                        .insert(orig.to_id());

                                                    return Some(ImportSpecifier::Named(
                                                        ImportNamedSpecifier {
                                                            span: DUMMY_SP,
                                                            local: orig.clone(),
                                                            imported: None,
                                                            is_type_only: false,
                                                        },
                                                    ));
                                            }
                                            None
                                        })
                                        .collect();

                                if !import_specs.is_empty() {
                                    // Add import statement.
                                    self.extra_items.push(ModuleItem::ModuleDecl(
                                        ModuleDecl::Import(ImportDecl {
                                            span: named.span,
                                            specifiers: import_specs,
                                            src: src.clone(),
                                            type_only: false,
                                            with: named.with.clone(),
                                            phase: Default::default(),
                                        }),
                                    ));
                                }

                                // Remove value specifiers from the export statement, keeping
                                // only type-only specifiers.
                                named.specifiers.retain(|spec| {
                                    matches!(
                                        spec,
                                        ExportSpecifier::Named(ExportNamedSpecifier {
                                            is_type_only: true,
                                            ..
                                        })
                                    )
                                });

                                // If all specifiers were value specifiers (converted to
                                // imports), remove the entire statement.
                                if named.specifiers.is_empty() {
                                    should_remove_statement = true;
                                }
                            } else if named.specifiers.iter().any(|s| match s {
                                ExportSpecifier::Namespace(_) | ExportSpecifier::Default(_) => true,
                                ExportSpecifier::Named(s) => !s.is_type_only,
                            }) {
                                disallowed_export_span = named.span;
                            }
                        } else {
                            // For cache files, remove specifiers that need cache runtime
                            // wrappers. Keep type-only specifiers and value specifiers that
                            // don't need wrappers (like function declarations).
                            if in_cache_file {
                                named.specifiers.retain(|spec| {
                                    if let ExportSpecifier::Named(ExportNamedSpecifier {
                                        orig: ModuleExportName::Ident(ident),
                                        is_type_only: false,
                                        ..
                                    }) = spec
                                    {
                                        !self
                                            .local_ids_that_need_cache_runtime_wrapper_if_exported
                                            .contains(&ident.to_id())
                                    } else {
                                        true
                                    }
                                });

                                if named.specifiers.is_empty() {
                                    should_remove_statement = true;
                                }
                            }
                        }
                    }
                    ModuleItem::ModuleDecl(ModuleDecl::ExportDefaultDecl(ExportDefaultDecl {
                        decl,
                        span,
                    })) => match decl {
                        DefaultDecl::Fn(_) | DefaultDecl::TsInterfaceDecl(_) => {}
                        _ => {
                            disallowed_export_span = *span;
                        }
                    },
                    ModuleItem::ModuleDecl(ModuleDecl::ExportDefaultExpr(default_expr)) => {
                        match &mut *default_expr.expr {
                            Expr::Fn(_) | Expr::Arrow(_) => {}
                            Expr::Ident(ident) => {
                                // For cache files, remove the export if it needs a runtime wrapper
                                // (we'll generate a new export when adding the wrapper).
                                if in_cache_file {
                                    let needs_cache_runtime_wrapper = self
                                        .local_ids_that_need_cache_runtime_wrapper_if_exported
                                        .contains(&ident.to_id());

                                    if needs_cache_runtime_wrapper {
                                        should_remove_statement = true;
                                    }
                                }
                            }
                            Expr::Call(_call) => {
                                // For cache files, mark the statement for removal. After visiting
                                // (which transforms nested directives), we'll hoist the call to a
                                // const declaration that gets a cache runtime wrapper.
                                if in_cache_file {
                                    should_remove_statement = true;
                                }
                            }
                            _ => {
                                disallowed_export_span = default_expr.span;
                            }
                        }
                    }
                    ModuleItem::ModuleDecl(ModuleDecl::ExportAll(ExportAll {
                        span,
                        type_only,
                        ..
                    })) if !*type_only => {
                        disallowed_export_span = *span;
                    }
                    _ => {}
                }

                // Emit validation error if we found a disallowed export
                if disallowed_export_span != DUMMY_SP {
                    emit_error(ServerActionsErrorKind::ExportedSyncFunction {
                        span: disallowed_export_span,
                        in_action_file,
                    });
                    return;
                }
            }

            stmt.visit_mut_with(self);

            let new_stmt = if should_remove_statement {
                None
            } else if let Some(expr) = self.rewrite_default_fn_expr_to_proxy_expr.take() {
                Some(ModuleItem::ModuleDecl(ModuleDecl::ExportDefaultExpr(
                    ExportDefaultExpr {
                        span: DUMMY_SP,
                        expr,
                    },
                )))
            } else {
                Some(stmt)
            };

            if self.config.is_react_server_layer || self.file_directive.is_none() {
                new.append(&mut self.hoisted_extra_items);
                if let Some(stmt) = new_stmt {
                    new.push(stmt);
                }
                new.extend(self.annotations.drain(..).map(ModuleItem::Stmt));
                new.append(&mut self.extra_items);
            }
        }

        // Post-pass: For server boundary files, register any exports that weren't already
        // registered during the main pass.
        if should_track_exports {
            for (id, export_name) in &self.export_name_by_local_id {
                if self.reference_ids_by_export_name.contains_key(export_name) {
                    continue;
                }

                if in_cache_file
                    && !self
                        .local_ids_that_need_cache_runtime_wrapper_if_exported
                        .contains(id)
                {
                    continue;
                }

                self.server_reference_exports.push(ServerReferenceExport {
                    ident: Ident::from(id.clone()),
                    export_name: export_name.clone(),
                    reference_id: self.generate_server_reference_id(
                        export_name,
                        in_cache_file,
                        None,
                    ),
                    needs_cache_runtime_wrapper: in_cache_file,
                });
            }
        }

        if in_action_file || in_cache_file && !self.config.is_react_server_layer {
            self.reference_ids_by_export_name.extend(
                self.server_reference_exports
                    .iter()
                    .map(|e| (e.export_name.clone(), e.reference_id.clone())),
            );

            if !self.reference_ids_by_export_name.is_empty() {
                self.has_action |= in_action_file;
                self.has_cache |= in_cache_file;
            }
        };

        // If it's compiled in the client layer, each export field needs to be
        // wrapped by a reference creation call.
        let create_ref_ident = private_ident!("createServerReference");
        let call_server_ident = private_ident!("callServer");
        let find_source_map_url_ident = private_ident!("findSourceMapURL");

        let client_layer_import = ((self.has_action || self.has_cache)
            && !self.config.is_react_server_layer)
            .then(|| {
                // import {
                //   createServerReference,
                //   callServer,
                //   findSourceMapURL
                // } from 'private-next-rsc-action-client-wrapper'
                // createServerReference("action_id")
                ModuleItem::ModuleDecl(ModuleDecl::Import(ImportDecl {
                    span: DUMMY_SP,
                    specifiers: vec![
                        ImportSpecifier::Named(ImportNamedSpecifier {
                            span: DUMMY_SP,
                            local: create_ref_ident.clone(),
                            imported: None,
                            is_type_only: false,
                        }),
                        ImportSpecifier::Named(ImportNamedSpecifier {
                            span: DUMMY_SP,
                            local: call_server_ident.clone(),
                            imported: None,
                            is_type_only: false,
                        }),
                        ImportSpecifier::Named(ImportNamedSpecifier {
                            span: DUMMY_SP,
                            local: find_source_map_url_ident.clone(),
                            imported: None,
                            is_type_only: false,
                        }),
                    ],
                    src: Box::new(Str {
                        span: DUMMY_SP,
                        value: atom!("private-next-rsc-action-client-wrapper").into(),
                        raw: None,
                    }),
                    type_only: false,
                    with: None,
                    phase: Default::default(),
                }))
            });

        let mut client_layer_exports = FxIndexMap::default();

        // If it's a "use server" or a "use cache" file, all exports need to be annotated.
        if should_track_exports {
            let server_reference_exports = self.server_reference_exports.take();

            for ServerReferenceExport {
                ident,
                export_name,
                reference_id: ref_id,
                needs_cache_runtime_wrapper,
                ..
            } in &server_reference_exports
            {
                if !self.config.is_react_server_layer {
                    if matches!(export_name, ModuleExportName::Ident(i) if i.sym == *"default") {
                        let export_expr = ModuleItem::ModuleDecl(ModuleDecl::ExportDefaultExpr(
                            ExportDefaultExpr {
                                span: DUMMY_SP,
                                expr: Box::new(Expr::Call(CallExpr {
                                    // In development, we generate these spans for sourcemapping
                                    // with better logs/errors. For production, this is not
                                    // generated because it would leak server code to the browser.
                                    span: if self.config.is_react_server_layer
                                        || self.config.is_development
                                    {
                                        self.comments.add_pure_comment(ident.span.lo);
                                        ident.span
                                    } else {
                                        PURE_SP
                                    },
                                    callee: Callee::Expr(Box::new(Expr::Ident(
                                        create_ref_ident.clone(),
                                    ))),
                                    args: vec![
                                        ref_id.clone().as_arg(),
                                        call_server_ident.clone().as_arg(),
                                        Expr::undefined(DUMMY_SP).as_arg(),
                                        find_source_map_url_ident.clone().as_arg(),
                                        "default".as_arg(),
                                    ],
                                    ..Default::default()
                                })),
                            },
                        ));
                        client_layer_exports.insert(
                            atom!("default"),
                            (
                                vec![export_expr],
                                ModuleExportName::Ident(atom!("default").into()),
                                ref_id.clone(),
                            ),
                        );
                    } else {
                        let var_name = if in_cache_file {
                            self.gen_cache_ident()
                        } else {
                            self.gen_action_ident()
                        };

                        let var_ident = Ident::new(var_name.clone(), DUMMY_SP, self.private_ctxt);

                        // Determine span for the variable name. In development, we generate these
                        // spans for sourcemapping with better logs/errors. For production, this is
                        // not generated because it would leak server code to the browser.
                        let name_span =
                            if self.config.is_react_server_layer || self.config.is_development {
                                ident.span
                            } else {
                                DUMMY_SP
                            };

                        let export_name_str: Wtf8Atom = match export_name {
                            ModuleExportName::Ident(i) => i.sym.clone().into(),
                            ModuleExportName::Str(s) => s.value.clone(),
                        };

                        let var_decl = ModuleItem::Stmt(Stmt::Decl(Decl::Var(Box::new(VarDecl {
                            span: DUMMY_SP,
                            kind: VarDeclKind::Const,
                            decls: vec![VarDeclarator {
                                span: DUMMY_SP,
                                name: Pat::Ident(
                                    Ident::new(var_name.clone(), name_span, self.private_ctxt)
                                        .into(),
                                ),
                                init: Some(Box::new(Expr::Call(CallExpr {
                                    span: PURE_SP,
                                    callee: Callee::Expr(Box::new(Expr::Ident(
                                        create_ref_ident.clone(),
                                    ))),
                                    args: vec![
                                        ref_id.clone().as_arg(),
                                        call_server_ident.clone().as_arg(),
                                        Expr::undefined(DUMMY_SP).as_arg(),
                                        find_source_map_url_ident.clone().as_arg(),
                                        export_name_str.as_arg(),
                                    ],
                                    ..Default::default()
                                }))),
                                definite: false,
                            }],
                            ..Default::default()
                        }))));

                        // Determine the export name. In development, we generate these spans for
                        // sourcemapping with better logs/errors. For production, this is not
                        // generated because it would leak server code to the browser.
                        let exported_name =
                            if self.config.is_react_server_layer || self.config.is_development {
                                export_name.clone()
                            } else {
                                strip_export_name_span(export_name)
                            };

                        let export_named =
                            ModuleItem::ModuleDecl(ModuleDecl::ExportNamed(NamedExport {
                                span: DUMMY_SP,
                                specifiers: vec![ExportSpecifier::Named(ExportNamedSpecifier {
                                    span: DUMMY_SP,
                                    orig: ModuleExportName::Ident(var_ident),
                                    exported: Some(exported_name),
                                    is_type_only: false,
                                })],
                                src: None,
                                type_only: false,
                                with: None,
                            }));

                        client_layer_exports.insert(
                            var_name,
                            (
                                vec![var_decl, export_named],
                                export_name.clone(),
                                ref_id.clone(),
                            ),
                        );
                    }
                } else if in_cache_file {
                    // Generate cache runtime wrapper for exports that might not be functions
                    // at compile time (re-exported identifiers, HOC results, etc.)
                    // Pattern: let x = orig; if (typeof orig === "function") { x = wrapper(...) }

                    if !*needs_cache_runtime_wrapper {
                        continue;
                    }

                    // Generate wrapper ident: $$RSC_SERVER_CACHE_exportName
                    let wrapper_ident = Ident::new(
                        format!("$$RSC_SERVER_CACHE_{}", export_name.atom()).into(),
                        ident.span,
                        self.private_ctxt,
                    );

                    self.has_cache = true;
                    self.reference_ids_by_export_name
                        .insert(export_name.clone(), ref_id.clone());

                    // let $$RSC_SERVER_CACHE_exportName = exportName;
                    self.extra_items
                        .push(ModuleItem::Stmt(Stmt::Decl(Decl::Var(Box::new(VarDecl {
                            kind: VarDeclKind::Let,
                            decls: vec![VarDeclarator {
                                span: ident.span,
                                name: Pat::Ident(wrapper_ident.clone().into()),
                                init: Some(Box::new(Expr::Ident(ident.clone()))),
                                definite: false,
                            }],
                            ..Default::default()
                        })))));

                    let wrapper_stmts = {
                        let mut stmts = vec![
                            // $$RSC_SERVER_CACHE_exportName = $$reactCache__(...);
                            Stmt::Expr(ExprStmt {
                                span: DUMMY_SP,
                                expr: Box::new(Expr::Assign(AssignExpr {
                                    span: DUMMY_SP,
                                    op: op!("="),
                                    left: AssignTarget::Simple(SimpleAssignTarget::Ident(
                                        wrapper_ident.clone().into(),
                                    )),
                                    right: Box::new(create_cache_wrapper(
                                        "default",
                                        ref_id.clone(),
                                        0,
                                        // Don't use the same name as the original to avoid
                                        // shadowing. We don't need it here for call stacks.
                                        None,
                                        Expr::Ident(ident.clone()),
                                        ident.span,
                                        None,
                                        self.unresolved_ctxt,
                                    )),
                                })),
                            }),
                            // registerServerReference($$RSC_SERVER_CACHE_exportName, ...);
                            Stmt::Expr(ExprStmt {
                                span: DUMMY_SP,
                                expr: Box::new(annotate_ident_as_server_reference(
                                    wrapper_ident.clone(),
                                    ref_id.clone(),
                                    ident.span,
                                )),
                            }),
                        ];

                        // Only assign a name if the original ident is not a generated one.
                        if !ident.sym.starts_with("$$RSC_SERVER_") {
                            // Object.defineProperty($$RSC_SERVER_CACHE_exportName, "name", {...});
                            stmts.push(assign_name_to_ident(
                                &wrapper_ident,
                                &ident.sym,
                                self.unresolved_ctxt,
                            ));
                        }

                        stmts
                    };

                    // if (typeof ident === "function") { $$RSC_SERVER_CACHE_exportName = wrapper }
                    self.extra_items.push(ModuleItem::Stmt(Stmt::If(IfStmt {
                        test: Box::new(Expr::Bin(BinExpr {
                            span: DUMMY_SP,
                            op: op!("==="),
                            left: Box::new(Expr::Unary(UnaryExpr {
                                span: DUMMY_SP,
                                op: op!("typeof"),
                                arg: Box::new(Expr::Ident(ident.clone())),
                            })),
                            right: Box::new(Expr::Lit(Lit::Str(Str {
                                span: DUMMY_SP,
                                value: atom!("function").into(),
                                raw: None,
                            }))),
                        })),
                        cons: Box::new(Stmt::Block(BlockStmt {
                            stmts: wrapper_stmts,
                            ..Default::default()
                        })),
                        ..Default::default()
                    })));

                    // Generate export with rename: export { $$RSC_SERVER_CACHE_name as name }
                    if matches!(export_name, ModuleExportName::Ident(i) if i.sym == *"default") {
                        self.extra_items.push(ModuleItem::ModuleDecl(
                            ModuleDecl::ExportDefaultExpr(ExportDefaultExpr {
                                span: DUMMY_SP,
                                expr: Box::new(Expr::Ident(wrapper_ident)),
                            }),
                        ));
                    } else {
                        self.extra_items
                            .push(ModuleItem::ModuleDecl(ModuleDecl::ExportNamed(
                                NamedExport {
                                    span: DUMMY_SP,
                                    specifiers: vec![ExportSpecifier::Named(
                                        ExportNamedSpecifier {
                                            span: DUMMY_SP,
                                            orig: ModuleExportName::Ident(wrapper_ident),
                                            exported: Some(export_name.clone()),
                                            is_type_only: false,
                                        },
                                    )],
                                    src: None,
                                    type_only: false,
                                    with: None,
                                },
                            )));
                    }
                } else {
                    self.annotations.push(Stmt::Expr(ExprStmt {
                        span: DUMMY_SP,
                        expr: Box::new(annotate_ident_as_server_reference(
                            ident.clone(),
                            ref_id.clone(),
                            ident.span,
                        )),
                    }));
                }
            }

            // Ensure that the exports are functions by appending a runtime check:
            //
            //   import { ensureServerEntryExports } from 'private-next-rsc-action-validate'
            //   ensureServerEntryExports([action1, action2, ...])
            //
            // But it's only needed for the server layer, because on the client
            // layer they're transformed into references already.
            if (self.has_action || self.has_cache) && self.config.is_react_server_layer {
                new.append(&mut self.extra_items);

                // For "use cache" files, there's no need to do extra annotations.
                if !in_cache_file && !server_reference_exports.is_empty() {
                    let ensure_ident = private_ident!("ensureServerEntryExports");
                    new.push(ModuleItem::ModuleDecl(ModuleDecl::Import(ImportDecl {
                        span: DUMMY_SP,
                        specifiers: vec![ImportSpecifier::Named(ImportNamedSpecifier {
                            span: DUMMY_SP,
                            local: ensure_ident.clone(),
                            imported: None,
                            is_type_only: false,
                        })],
                        src: Box::new(Str {
                            span: DUMMY_SP,
                            value: atom!("private-next-rsc-action-validate").into(),
                            raw: None,
                        }),
                        type_only: false,
                        with: None,
                        phase: Default::default(),
                    })));
                    new.push(ModuleItem::Stmt(Stmt::Expr(ExprStmt {
                        span: DUMMY_SP,
                        expr: Box::new(Expr::Call(CallExpr {
                            span: DUMMY_SP,
                            callee: Callee::Expr(Box::new(Expr::Ident(ensure_ident))),
                            args: vec![ExprOrSpread {
                                spread: None,
                                expr: Box::new(Expr::Array(ArrayLit {
                                    span: DUMMY_SP,
                                    elems: server_reference_exports
                                        .iter()
                                        .map(|ServerReferenceExport { ident, .. }| {
                                            Some(ExprOrSpread {
                                                spread: None,
                                                expr: Box::new(Expr::Ident(ident.clone())),
                                            })
                                        })
                                        .collect(),
                                })),
                            }],
                            ..Default::default()
                        })),
                    })));
                }

                // Append annotations to the end of the file.
                new.extend(self.annotations.drain(..).map(ModuleItem::Stmt));
            }
        }

        // import { cache as $$cache__ } from "private-next-rsc-cache-wrapper";
        // import { cache as $$reactCache__ } from "react";
        if self.has_cache && self.config.is_react_server_layer {
            new.push(ModuleItem::ModuleDecl(ModuleDecl::Import(ImportDecl {
                span: DUMMY_SP,
                specifiers: vec![ImportSpecifier::Named(ImportNamedSpecifier {
                    span: DUMMY_SP,
                    local: quote_ident!("$$cache__").into(),
                    imported: Some(quote_ident!("cache").into()),
                    is_type_only: false,
                })],
                src: Box::new(Str {
                    span: DUMMY_SP,
                    value: atom!("private-next-rsc-cache-wrapper").into(),
                    raw: None,
                }),
                type_only: false,
                with: None,
                phase: Default::default(),
            })));

            new.push(ModuleItem::ModuleDecl(ModuleDecl::Import(ImportDecl {
                span: DUMMY_SP,
                specifiers: vec![ImportSpecifier::Named(ImportNamedSpecifier {
                    span: DUMMY_SP,
                    local: quote_ident!("$$reactCache__").into(),
                    imported: Some(quote_ident!("cache").into()),
                    is_type_only: false,
                })],
                src: Box::new(Str {
                    span: DUMMY_SP,
                    value: atom!("react").into(),
                    raw: None,
                }),
                type_only: false,
                with: None,
                phase: Default::default(),
            })));

            // Make them the first items
            new.rotate_right(2);
        }

        if (self.has_action || self.has_cache) && self.config.is_react_server_layer {
            // Inlined actions are only allowed on the server layer.
            // import { registerServerReference } from 'private-next-rsc-server-reference'
            new.push(ModuleItem::ModuleDecl(ModuleDecl::Import(ImportDecl {
                span: DUMMY_SP,
                specifiers: vec![ImportSpecifier::Named(ImportNamedSpecifier {
                    span: DUMMY_SP,
                    local: quote_ident!("registerServerReference").into(),
                    imported: None,
                    is_type_only: false,
                })],
                src: Box::new(Str {
                    span: DUMMY_SP,
                    value: atom!("private-next-rsc-server-reference").into(),
                    raw: None,
                }),
                type_only: false,
                with: None,
                phase: Default::default(),
            })));

            let mut import_count = 1;

            // Encryption and decryption only happens when there are bound arguments.
            if self.has_server_reference_with_bound_args {
                // import { encryptActionBoundArgs, decryptActionBoundArgs } from
                // 'private-next-rsc-action-encryption'
                new.push(ModuleItem::ModuleDecl(ModuleDecl::Import(ImportDecl {
                    span: DUMMY_SP,
                    specifiers: vec![
                        ImportSpecifier::Named(ImportNamedSpecifier {
                            span: DUMMY_SP,
                            local: quote_ident!("encryptActionBoundArgs").into(),
                            imported: None,
                            is_type_only: false,
                        }),
                        ImportSpecifier::Named(ImportNamedSpecifier {
                            span: DUMMY_SP,
                            local: quote_ident!("decryptActionBoundArgs").into(),
                            imported: None,
                            is_type_only: false,
                        }),
                    ],
                    src: Box::new(Str {
                        span: DUMMY_SP,
                        value: atom!("private-next-rsc-action-encryption").into(),
                        raw: None,
                    }),
                    type_only: false,
                    with: None,
                    phase: Default::default(),
                })));
                import_count += 1;
            }

            // Make them the first items
            new.rotate_right(import_count);
        }

        if self.has_action || self.has_cache {
            // Build a map of reference_id -> export info
            let export_infos_ordered_by_reference_id = self
                .reference_ids_by_export_name
                .iter()
                .map(|(export_name, reference_id)| {
                    let name_atom = export_name.atom().into_owned();
                    (reference_id, ServerReferenceExportInfo { name: name_atom })
                })
                .collect::<BTreeMap<_, _>>();

            if self.config.is_react_server_layer {
                // Prepend a special comment to the top of the file.
                self.comments.add_leading(
                    self.start_pos,
                    Comment {
                        span: DUMMY_SP,
                        kind: CommentKind::Block,
                        text: generate_server_references_comment(
                            &export_infos_ordered_by_reference_id,
                            match self.mode {
                                ServerActionsMode::Webpack => None,
                                ServerActionsMode::Turbopack => Some((
                                    &self.file_name,
                                    self.file_query.as_ref().map_or("", |v| v),
                                )),
                            },
                        )
                        .into(),
                    },
                );
            } else {
                match self.mode {
                    ServerActionsMode::Webpack => {
                        self.comments.add_leading(
                            self.start_pos,
                            Comment {
                                span: DUMMY_SP,
                                kind: CommentKind::Block,
                                text: generate_server_references_comment(
                                    &export_infos_ordered_by_reference_id,
                                    None,
                                )
                                .into(),
                            },
                        );
                        new.push(client_layer_import.unwrap());
                        new.rotate_right(1);
                        new.extend(
                            client_layer_exports
                                .into_iter()
                                .flat_map(|(_, (items, _, _))| items),
                        );
                    }
                    ServerActionsMode::Turbopack => {
                        new.push(ModuleItem::Stmt(Stmt::Expr(ExprStmt {
                            expr: Box::new(Expr::Lit(Lit::Str(
                                atom!("use turbopack no side effects").into(),
                            ))),
                            span: DUMMY_SP,
                        })));
                        new.rotate_right(1);
                        for (_, (items, export_name, ref_id)) in client_layer_exports {
                            let mut module_items = vec![
                                ModuleItem::Stmt(Stmt::Expr(ExprStmt {
                                    expr: Box::new(Expr::Lit(Lit::Str(
                                        atom!("use turbopack no side effects").into(),
                                    ))),
                                    span: DUMMY_SP,
                                })),
                                client_layer_import.clone().unwrap(),
                            ];
                            module_items.extend(items);

                            // For Turbopack, always strip spans from the main output file's
                            // re-exports since the actual source maps are in the data URLs.
                            let stripped_export_name = strip_export_name_span(&export_name);

                            let name_atom = export_name.atom().into_owned();
                            let export_info = ServerReferenceExportInfo { name: name_atom };

                            new.push(ModuleItem::ModuleDecl(ModuleDecl::ExportNamed(
                                NamedExport {
                                    specifiers: vec![ExportSpecifier::Named(
                                        ExportNamedSpecifier {
                                            span: DUMMY_SP,
                                            orig: stripped_export_name,
                                            exported: None,
                                            is_type_only: false,
                                        },
                                    )],
                                    src: Some(Box::new(
                                        program_to_data_url(
                                            &self.file_name,
                                            &self.cm,
                                            module_items,
                                            Comment {
                                                span: DUMMY_SP,
                                                kind: CommentKind::Block,
                                                text: generate_server_references_comment(
                                                    &std::iter::once((&ref_id, export_info))
                                                        .collect(),
                                                    Some((
                                                        &self.file_name,
                                                        self.file_query.as_ref().map_or("", |v| v),
                                                    )),
                                                )
                                                .into(),
                                            },
                                        )
                                        .into(),
                                    )),
                                    span: DUMMY_SP,
                                    type_only: false,
                                    with: None,
                                },
                            )));
                        }
                    }
                }
            }
        }

        *stmts = new;

        self.annotations = old_annotations;
    }

    fn visit_mut_stmts(&mut self, stmts: &mut Vec<Stmt>) {
        let old_annotations = self.annotations.take();

        let mut new = Vec::with_capacity(stmts.len());
        for mut stmt in stmts.take() {
            stmt.visit_mut_with(self);

            new.push(stmt);
            new.append(&mut self.annotations);
        }

        *stmts = new;

        self.annotations = old_annotations;
    }

    fn visit_mut_jsx_attr(&mut self, attr: &mut JSXAttr) {
        let old_arrow_or_fn_expr_ident = self.arrow_or_fn_expr_ident.take();

        if let (Some(JSXAttrValue::JSXExprContainer(container)), JSXAttrName::Ident(ident_name)) =
            (&attr.value, &attr.name)
        {
            match &container.expr {
                JSXExpr::Expr(box Expr::Arrow(_)) | JSXExpr::Expr(box Expr::Fn(_)) => {
                    self.arrow_or_fn_expr_ident = Some(ident_name.clone().into());
                }
                _ => {}
            }
        }

        attr.visit_mut_children_with(self);
        self.arrow_or_fn_expr_ident = old_arrow_or_fn_expr_ident;
    }

    fn visit_mut_var_declarator(&mut self, var_declarator: &mut VarDeclarator) {
        let old_current_export_name = self.current_export_name.take();
        let old_arrow_or_fn_expr_ident = self.arrow_or_fn_expr_ident.take();

        if let (Pat::Ident(ident), Some(box Expr::Arrow(_) | box Expr::Fn(_))) =
            (&var_declarator.name, &var_declarator.init)
        {
            if self.in_module_level
                && let Some(export_name) = self.export_name_by_local_id.get(&ident.to_id())
            {
                self.current_export_name = Some(export_name.clone());
            }

            self.arrow_or_fn_expr_ident = Some(ident.id.clone());
        }

        var_declarator.visit_mut_children_with(self);

        self.current_export_name = old_current_export_name;
        self.arrow_or_fn_expr_ident = old_arrow_or_fn_expr_ident;
    }

    fn visit_mut_assign_expr(&mut self, assign_expr: &mut AssignExpr) {
        let old_arrow_or_fn_expr_ident = self.arrow_or_fn_expr_ident.clone();

        if let (
            AssignTarget::Simple(SimpleAssignTarget::Ident(ident)),
            Expr::Arrow(_) | Expr::Fn(_),
        ) = (&assign_expr.left, &*assign_expr.right)
        {
            self.arrow_or_fn_expr_ident = Some(ident.id.clone());
        }

        assign_expr.visit_mut_children_with(self);
        self.arrow_or_fn_expr_ident = old_arrow_or_fn_expr_ident;
    }

    fn visit_mut_this_expr(&mut self, n: &mut ThisExpr) {
        if let ThisStatus::Forbidden { directive } = &self.this_status {
            emit_error(ServerActionsErrorKind::ForbiddenExpression {
                span: n.span,
                expr: "this".into(),
                directive: directive.clone(),
            });
        }
    }

    fn visit_mut_super(&mut self, n: &mut Super) {
        if let ThisStatus::Forbidden { directive } = &self.this_status {
            emit_error(ServerActionsErrorKind::ForbiddenExpression {
                span: n.span,
                expr: "super".into(),
                directive: directive.clone(),
            });
        }
    }

    fn visit_mut_ident(&mut self, n: &mut Ident) {
        if n.sym == *"arguments"
            && let ThisStatus::Forbidden { directive } = &self.this_status
        {
            emit_error(ServerActionsErrorKind::ForbiddenExpression {
                span: n.span,
                expr: "arguments".into(),
                directive: directive.clone(),
            });
        }
    }

    noop_visit_mut_type!();
}

fn retain_names_from_declared_idents(
    child_names: &mut Vec<Name>,
    current_declared_idents: &[Ident],
) {
    // Collect the names to retain in a separate vector
    let mut retained_names = Vec::new();

    for name in child_names.iter() {
        let mut should_retain = true;

        // Merge child_names. For example if both `foo.bar` and `foo.bar.baz` are used,
        // we only need to keep `foo.bar` as it covers the other.

        // Currently this is O(n^2) and we can potentially improve this to O(n log n)
        // by sorting or using a hashset.
        for another_name in child_names.iter() {
            if name != another_name
                && name.0 == another_name.0
                && name.1.len() >= another_name.1.len()
            {
                let mut is_prefix = true;
                for i in 0..another_name.1.len() {
                    if name.1[i] != another_name.1[i] {
                        is_prefix = false;
                        break;
                    }
                }
                if is_prefix {
                    should_retain = false;
                    break;
                }
            }
        }

        if should_retain
            && current_declared_idents
                .iter()
                .any(|ident| ident.to_id() == name.0)
            && !retained_names.contains(name)
        {
            retained_names.push(name.clone());
        }
    }

    // Replace the original child_names with the retained names
    *child_names = retained_names;
}

/// Returns true if the expression may need a cache runtime wrapper.
/// Known functions and known non-functions return false.
fn may_need_cache_runtime_wrapper(expr: &Expr) -> bool {
    match expr {
        // Known functions - don't need wrapper
        Expr::Arrow(_) | Expr::Fn(_) => false,
        // Known non-functions - don't need wrapper
        Expr::Object(_) | Expr::Array(_) | Expr::Lit(_) => false,
        // Unknown/might be function - needs runtime check
        _ => true,
    }
}

/// Creates a cache wrapper expression:
/// $$reactCache__(function name() { return $$cache__(...) })
#[allow(clippy::too_many_arguments)]
fn create_cache_wrapper(
    cache_kind: &str,
    reference_id: Atom,
    bound_args_length: usize,
    fn_ident: Option<Ident>,
    target_expr: Expr,
    original_span: Span,
    params: Option<&[Param]>,
    unresolved_ctxt: SyntaxContext,
) -> Expr {
    let cache_call = CallExpr {
        span: original_span,
        callee: quote_ident!("$$cache__").as_callee(),
        args: vec![
            Box::new(Expr::from(cache_kind)).as_arg(),
            Box::new(Expr::from(reference_id.as_str())).as_arg(),
            Box::new(Expr::Lit(Lit::Num(Number {
                span: DUMMY_SP,
                value: bound_args_length as f64,
                raw: None,
            })))
            .as_arg(),
            Box::new(target_expr).as_arg(),
            match params {
                // The params are statically known and rest params are not used.
                Some(params) if !params.iter().any(|p| matches!(p.pat, Pat::Rest(_))) => {
                    if params.is_empty() {
                        // No params are declared, we can pass an empty array to ignore unused
                        // arguments.
                        Box::new(Expr::Array(ArrayLit {
                            span: DUMMY_SP,
                            elems: vec![],
                        }))
                        .as_arg()
                    } else {
                        // Slice to declared params length to ignore unused arguments.
                        Box::new(quote!(
                            "$array.prototype.slice.call(arguments, 0, $end)" as Expr,
                            array = quote_ident!(unresolved_ctxt, "Array"),
                            end: Expr = params.len().into(),
                        ))
                        .as_arg()
                    }
                }
                // The params are statically unknown, or rest params are used.
                _ => {
                    // Pass all arguments as an array.
                    Box::new(quote!(
                        "$array.prototype.slice.call(arguments)" as Expr,
                        array = quote_ident!(unresolved_ctxt, "Array"),
                    ))
                    .as_arg()
                }
            },
        ],
        ..Default::default()
    };

    // This wrapper function ensures that we have a user-space call stack frame.
    let wrapper_fn_expr = Box::new(Expr::Fn(FnExpr {
        ident: fn_ident,
        function: Box::new(Function {
            body: Some(BlockStmt {
                stmts: vec![Stmt::Return(ReturnStmt {
                    span: DUMMY_SP,
                    arg: Some(Box::new(Expr::Call(cache_call))),
                })],
                ..Default::default()
            }),
            span: original_span,
            ..Default::default()
        }),
    }));

    Expr::Call(CallExpr {
        callee: quote_ident!("$$reactCache__").as_callee(),
        args: vec![wrapper_fn_expr.as_arg()],
        ..Default::default()
    })
}

#[allow(clippy::too_many_arguments)]
fn create_and_hoist_cache_function(
    cache_kind: &str,
    reference_id: Atom,
    bound_args_length: usize,
    cache_name: Atom,
    fn_ident: Option<Ident>,
    params: Vec<Param>,
    body: Option<BlockStmt>,
    original_span: Span,
    hoisted_extra_items: &mut Vec<ModuleItem>,
    unresolved_ctxt: SyntaxContext,
) -> Ident {
    let cache_ident = private_ident!(Span::dummy_with_cmt(), cache_name.clone());
    let inner_fn_name: Atom = format!("{}_INNER", cache_name).into();
    let inner_fn_ident = private_ident!(Span::dummy_with_cmt(), inner_fn_name);

    let wrapper_fn = Box::new(create_cache_wrapper(
        cache_kind,
        reference_id.clone(),
        bound_args_length,
        fn_ident.clone(),
        Expr::Ident(inner_fn_ident.clone()),
        original_span,
        Some(&params),
        unresolved_ctxt,
    ));

    let inner_fn_expr = FnExpr {
        ident: fn_ident.clone(),
        function: Box::new(Function {
            params,
            body,
            span: original_span,
            is_async: true,
            ..Default::default()
        }),
    };

    hoisted_extra_items.push(ModuleItem::Stmt(Stmt::Decl(Decl::Var(Box::new(VarDecl {
        span: original_span,
        kind: VarDeclKind::Const,
        decls: vec![VarDeclarator {
            span: original_span,
            name: Pat::Ident(BindingIdent {
                id: inner_fn_ident.clone(),
                type_ann: None,
            }),
            init: Some(Box::new(Expr::Fn(inner_fn_expr))),
            definite: false,
        }],
        ..Default::default()
    })))));

    // For anonymous functions, set the name property to an empty string to
    // avoid leaking the internal variable name in stack traces.
    if fn_ident.is_none() {
        hoisted_extra_items.push(ModuleItem::Stmt(assign_name_to_ident(
            &inner_fn_ident,
            "",
            unresolved_ctxt,
        )));
    }

    hoisted_extra_items.push(ModuleItem::ModuleDecl(ModuleDecl::ExportDecl(ExportDecl {
        span: DUMMY_SP,
        decl: VarDecl {
            kind: VarDeclKind::Var,
            decls: vec![VarDeclarator {
                span: original_span,
                name: Pat::Ident(cache_ident.clone().into()),
                init: Some(wrapper_fn),
                definite: false,
            }],
            ..Default::default()
        }
        .into(),
    })));

    hoisted_extra_items.push(ModuleItem::Stmt(Stmt::Expr(ExprStmt {
        span: DUMMY_SP,
        expr: Box::new(annotate_ident_as_server_reference(
            cache_ident.clone(),
            reference_id,
            original_span,
        )),
    })));

    cache_ident
}

fn assign_name_to_ident(ident: &Ident, name: &str, unresolved_ctxt: SyntaxContext) -> Stmt {
    // Assign a name with `Object.defineProperty($$ACTION_0, 'name', {value: 'default'})`
    quote!(
        // WORKAROUND for https://github.com/microsoft/TypeScript/issues/61165
        // This should just be
        //
        //   "Object.defineProperty($action, \"name\", { value: $name });"
        //
        // but due to the above typescript bug, `Object.defineProperty` calls are typechecked incorrectly
        // in js files, and it can cause false positives when typechecking our fixture files.
        "$object[\"defineProperty\"]($action, \"name\", { value: $name });"
            as Stmt,
        object = quote_ident!(unresolved_ctxt, "Object"),
        action: Ident = ident.clone(),
        name: Expr = name.into(),
    )
}

fn annotate_ident_as_server_reference(ident: Ident, action_id: Atom, original_span: Span) -> Expr {
    // registerServerReference(reference, id, null)
    Expr::Call(CallExpr {
        span: original_span,
        callee: quote_ident!("registerServerReference").as_callee(),
        args: vec![
            ExprOrSpread {
                spread: None,
                expr: Box::new(Expr::Ident(ident)),
            },
            ExprOrSpread {
                spread: None,
                expr: Box::new(action_id.clone().into()),
            },
            ExprOrSpread {
                spread: None,
                expr: Box::new(Expr::Lit(Lit::Null(Null { span: DUMMY_SP }))),
            },
        ],
        ..Default::default()
    })
}

fn bind_args_to_ident(ident: Ident, bound: Vec<Option<ExprOrSpread>>, action_id: Atom) -> Expr {
    // ident.bind(null, [encryptActionBoundArgs("id", arg1, arg2, ...)])
    Expr::Call(CallExpr {
        span: DUMMY_SP,
        callee: Expr::Member(MemberExpr {
            span: DUMMY_SP,
            obj: Box::new(ident.into()),
            prop: MemberProp::Ident(quote_ident!("bind")),
        })
        .as_callee(),
        args: vec![
            ExprOrSpread {
                spread: None,
                expr: Box::new(Expr::Lit(Lit::Null(Null { span: DUMMY_SP }))),
            },
            ExprOrSpread {
                spread: None,
                expr: Box::new(Expr::Call(CallExpr {
                    span: DUMMY_SP,
                    callee: quote_ident!("encryptActionBoundArgs").as_callee(),
                    args: std::iter::once(ExprOrSpread {
                        spread: None,
                        expr: Box::new(action_id.into()),
                    })
                    .chain(bound.into_iter().flatten())
                    .collect(),
                    ..Default::default()
                })),
            },
        ],
        ..Default::default()
    })
}

// Detects if two strings are similar (but not the same).
// This implementation is fast and simple as it allows only one
// edit (add, remove, edit, swap), instead of using a N^2 Levenshtein algorithm.
//
// Example of similar strings of "use server":
// "use servers",
// "use-server",
// "use sevrer",
// "use srever",
// "use servre",
// "user server",
//
// This avoids accidental typos as there's currently no other static analysis
// tool to help when these mistakes happen.
fn detect_similar_strings(a: &str, b: &str) -> bool {
    let mut a = a.chars().collect::<Vec<char>>();
    let mut b = b.chars().collect::<Vec<char>>();

    if a.len() < b.len() {
        (a, b) = (b, a);
    }

    if a.len() == b.len() {
        // Same length, get the number of character differences.
        let mut diff = 0;
        for i in 0..a.len() {
            if a[i] != b[i] {
                diff += 1;
                if diff > 2 {
                    return false;
                }
            }
        }

        // Should be 1 or 2, but not 0.
        diff != 0
    } else {
        if a.len() - b.len() > 1 {
            return false;
        }

        // A has one more character than B.
        for i in 0..b.len() {
            if a[i] != b[i] {
                // This should be the only difference, a[i+1..] should be equal to b[i..].
                // Otherwise, they're not considered similar.
                // A: "use srerver"
                // B: "use server"
                //          ^
                return a[i + 1..] == b[i..];
            }
        }

        // This happens when the last character of A is an extra character.
        true
    }
}

// Check if the function or arrow function has any action or cache directives,
// without mutating the function body or erroring out.
// This is used to quickly determine if we need to use the module-level
// directives for this function or not.
fn has_body_directive(maybe_body: &Option<BlockStmt>) -> (bool, bool) {
    let mut is_action_fn = false;
    let mut is_cache_fn = false;

    if let Some(body) = maybe_body {
        for stmt in body.stmts.iter() {
            match stmt {
                Stmt::Expr(ExprStmt {
                    expr: box Expr::Lit(Lit::Str(Str { value, .. })),
                    ..
                }) => {
                    if value == "use server" {
                        is_action_fn = true;
                        break;
                    } else if value == "use cache" || value.starts_with("use cache: ") {
                        is_cache_fn = true;
                        break;
                    }
                }
                _ => break,
            }
        }
    }

    (is_action_fn, is_cache_fn)
}

fn collect_idents_in_array_pat(elems: &[Option<Pat>], idents: &mut Vec<Ident>) {
    for elem in elems.iter().flatten() {
        match elem {
            Pat::Ident(ident) => {
                idents.push(ident.id.clone());
            }
            Pat::Array(array) => {
                collect_idents_in_array_pat(&array.elems, idents);
            }
            Pat::Object(object) => {
                collect_idents_in_object_pat(&object.props, idents);
            }
            Pat::Rest(rest) => {
                if let Pat::Ident(ident) = &*rest.arg {
                    idents.push(ident.id.clone());
                }
            }
            Pat::Assign(AssignPat { left, .. }) => {
                collect_idents_in_pat(left, idents);
            }
            Pat::Expr(..) | Pat::Invalid(..) => {}
        }
    }
}

fn collect_idents_in_object_pat(props: &[ObjectPatProp], idents: &mut Vec<Ident>) {
    for prop in props {
        match prop {
            ObjectPatProp::KeyValue(KeyValuePatProp { value, .. }) => {
                // For { foo: bar }, only collect 'bar' (the local binding), not 'foo' (the property
                // key).
                match &**value {
                    Pat::Ident(ident) => {
                        idents.push(ident.id.clone());
                    }
                    Pat::Array(array) => {
                        collect_idents_in_array_pat(&array.elems, idents);
                    }
                    Pat::Object(object) => {
                        collect_idents_in_object_pat(&object.props, idents);
                    }
                    _ => {}
                }
            }
            ObjectPatProp::Assign(AssignPatProp { key, .. }) => {
                // For { foo }, 'foo' is both the property key and local binding.
                idents.push(key.id.clone());
            }
            ObjectPatProp::Rest(RestPat { arg, .. }) => {
                if let Pat::Ident(ident) = &**arg {
                    idents.push(ident.id.clone());
                }
            }
        }
    }
}

fn collect_idents_in_var_decls(decls: &[VarDeclarator], idents: &mut Vec<Ident>) {
    for decl in decls {
        collect_idents_in_pat(&decl.name, idents);
    }
}

fn collect_idents_in_pat(pat: &Pat, idents: &mut Vec<Ident>) {
    match pat {
        Pat::Ident(ident) => {
            idents.push(ident.id.clone());
        }
        Pat::Array(array) => {
            collect_idents_in_array_pat(&array.elems, idents);
        }
        Pat::Object(object) => {
            collect_idents_in_object_pat(&object.props, idents);
        }
        Pat::Assign(AssignPat { left, .. }) => {
            collect_idents_in_pat(left, idents);
        }
        Pat::Rest(RestPat { arg, .. }) => {
            if let Pat::Ident(ident) = &**arg {
                idents.push(ident.id.clone());
            }
        }
        Pat::Expr(..) | Pat::Invalid(..) => {}
    }
}

fn collect_decl_idents_in_stmt(stmt: &Stmt, idents: &mut Vec<Ident>) {
    if let Stmt::Decl(decl) = stmt {
        match decl {
            Decl::Var(var) => {
                collect_idents_in_var_decls(&var.decls, idents);
            }
            Decl::Fn(fn_decl) => {
                idents.push(fn_decl.ident.clone());
            }
            _ => {}
        }
    }
}

struct DirectiveVisitor<'a> {
    config: &'a Config,
    location: DirectiveLocation,
    directive: Option<Directive>,
    has_file_directive: bool,
    is_allowed_position: bool,
    use_cache_telemetry_tracker: Rc<RefCell<FxHashMap<String, usize>>>,
}

impl DirectiveVisitor<'_> {
    /**
     * Returns `true` if the statement contains a server directive.
     * The found directive is assigned to `DirectiveVisitor::directive`.
     */
    fn visit_stmt(&mut self, stmt: &Stmt) -> bool {
        let in_fn_body = matches!(self.location, DirectiveLocation::FunctionBody);
        let allow_inline = self.config.is_react_server_layer || self.has_file_directive;

        match stmt {
            Stmt::Expr(ExprStmt {
                expr: box Expr::Lit(Lit::Str(Str { value, span, .. })),
                ..
            }) => {
                if value == "use server" {
                    if in_fn_body && !allow_inline {
                        emit_error(ServerActionsErrorKind::InlineUseServerInClientComponent {
                            span: *span,
                        })
                    } else if let Some(Directive::UseCache { .. }) = self.directive {
                        emit_error(ServerActionsErrorKind::MultipleDirectives {
                            span: *span,
                            location: self.location.clone(),
                        });
                    } else if self.is_allowed_position {
                        self.directive = Some(Directive::UseServer);

                        return true;
                    } else {
                        emit_error(ServerActionsErrorKind::MisplacedDirective {
                            span: *span,
                            directive: value.to_string_lossy().into_owned(),
                            location: self.location.clone(),
                        });
                    }
                } else if detect_similar_strings(&value.to_string_lossy(), "use server") {
                    // Detect typo of "use server"
                    emit_error(ServerActionsErrorKind::MisspelledDirective {
                        span: *span,
                        directive: value.to_string_lossy().into_owned(),
                        expected_directive: "use server".to_string(),
                    });
                } else if value == "use action" {
                    emit_error(ServerActionsErrorKind::MisspelledDirective {
                        span: *span,
                        directive: value.to_string_lossy().into_owned(),
                        expected_directive: "use server".to_string(),
                    });
                } else
                // `use cache` or `use cache: foo`
                if let Some(rest) = value.as_str().and_then(|s| s.strip_prefix("use cache"))
                {
                    // Increment telemetry counter tracking usage of "use cache" directives

                    if in_fn_body && !allow_inline {
                        emit_error(ServerActionsErrorKind::InlineUseCacheInClientComponent {
                            span: *span,
                        })
                    } else if let Some(Directive::UseServer) = self.directive {
                        emit_error(ServerActionsErrorKind::MultipleDirectives {
                            span: *span,
                            location: self.location.clone(),
                        });
                    } else if self.is_allowed_position {
                        if !self.config.use_cache_enabled {
                            emit_error(ServerActionsErrorKind::UseCacheWithoutCacheComponents {
                                span: *span,
                                directive: value.to_string_lossy().into_owned(),
                            });
                        }

                        if rest.is_empty() {
                            self.directive = Some(Directive::UseCache {
                                cache_kind: rcstr!("default"),
                            });

                            self.increment_cache_usage_counter("default");

                            return true;
                        }

                        if rest.starts_with(": ") {
                            let cache_kind = RcStr::from(rest.split_at(": ".len()).1.to_string());

                            if !cache_kind.is_empty() {
                                if !self.config.cache_kinds.contains(&cache_kind) {
                                    emit_error(ServerActionsErrorKind::UnknownCacheKind {
                                        span: *span,
                                        cache_kind: cache_kind.clone(),
                                    });
                                }

                                self.increment_cache_usage_counter(&cache_kind);
                                self.directive = Some(Directive::UseCache { cache_kind });

                                return true;
                            }
                        }

                        // Emit helpful errors for variants like "use cache:<cache-kind>",
                        // "use cache : <cache-kind>", and "use cache <cache-kind>" etc.
                        let expected_directive = if let Some(colon_pos) = rest.find(':') {
                            let kind = rest[colon_pos + 1..].trim();

                            if kind.is_empty() {
                                "use cache: <cache-kind>".to_string()
                            } else {
                                format!("use cache: {kind}")
                            }
                        } else {
                            let kind = rest.trim();

                            if kind.is_empty() {
                                "use cache".to_string()
                            } else {
                                format!("use cache: {kind}")
                            }
                        };

                        emit_error(ServerActionsErrorKind::MisspelledDirective {
                            span: *span,
                            directive: value.to_string_lossy().into_owned(),
                            expected_directive,
                        });

                        return true;
                    } else {
                        emit_error(ServerActionsErrorKind::MisplacedDirective {
                            span: *span,
                            directive: value.to_string_lossy().into_owned(),
                            location: self.location.clone(),
                        });
                    }
                } else {
                    // Detect typo of "use cache"
                    if detect_similar_strings(&value.to_string_lossy(), "use cache") {
                        emit_error(ServerActionsErrorKind::MisspelledDirective {
                            span: *span,
                            directive: value.to_string_lossy().into_owned(),
                            expected_directive: "use cache".to_string(),
                        });
                    }
                }
            }
            Stmt::Expr(ExprStmt {
                expr:
                    box Expr::Paren(ParenExpr {
                        expr: box Expr::Lit(Lit::Str(Str { value, .. })),
                        ..
                    }),
                span,
                ..
            }) => {
                // Match `("use server")`.
                if value == "use server"
                    || detect_similar_strings(&value.to_string_lossy(), "use server")
                {
                    if self.is_allowed_position {
                        emit_error(ServerActionsErrorKind::WrappedDirective {
                            span: *span,
                            directive: "use server".to_string(),
                        });
                    } else {
                        emit_error(ServerActionsErrorKind::MisplacedWrappedDirective {
                            span: *span,
                            directive: "use server".to_string(),
                            location: self.location.clone(),
                        });
                    }
                } else if value == "use cache"
                    || detect_similar_strings(&value.to_string_lossy(), "use cache")
                {
                    if self.is_allowed_position {
                        emit_error(ServerActionsErrorKind::WrappedDirective {
                            span: *span,
                            directive: "use cache".to_string(),
                        });
                    } else {
                        emit_error(ServerActionsErrorKind::MisplacedWrappedDirective {
                            span: *span,
                            directive: "use cache".to_string(),
                            location: self.location.clone(),
                        });
                    }
                }
            }
            _ => {
                // Directives must not be placed after other statements.
                self.is_allowed_position = false;
            }
        };

        false
    }

    // Increment telemetry counter tracking usage of "use cache" directives
    fn increment_cache_usage_counter(&mut self, cache_kind: &str) {
        let mut tracker_map = RefCell::borrow_mut(&self.use_cache_telemetry_tracker);
        let entry = tracker_map.entry(cache_kind.to_string());
        match entry {
            hash_map::Entry::Occupied(mut occupied) => {
                *occupied.get_mut() += 1;
            }
            hash_map::Entry::Vacant(vacant) => {
                vacant.insert(1);
            }
        }
    }
}

pub(crate) struct ClosureReplacer<'a> {
    used_ids: &'a [Name],
    private_ctxt: SyntaxContext,
}

impl ClosureReplacer<'_> {
    fn index(&self, e: &Expr) -> Option<usize> {
        let name = Name::try_from(e).ok()?;
        self.used_ids.iter().position(|used_id| *used_id == name)
    }
}

impl VisitMut for ClosureReplacer<'_> {
    fn visit_mut_expr(&mut self, e: &mut Expr) {
        e.visit_mut_children_with(self);

        if let Some(index) = self.index(e) {
            *e = Expr::Ident(Ident::new(
                // $$ACTION_ARG_0
                format!("$$ACTION_ARG_{index}").into(),
                DUMMY_SP,
                self.private_ctxt,
            ));
        }
    }

    fn visit_mut_prop_or_spread(&mut self, n: &mut PropOrSpread) {
        n.visit_mut_children_with(self);

        if let PropOrSpread::Prop(box Prop::Shorthand(i)) = n {
            let name = Name::from(&*i);
            if let Some(index) = self.used_ids.iter().position(|used_id| *used_id == name) {
                *n = PropOrSpread::Prop(Box::new(Prop::KeyValue(KeyValueProp {
                    key: PropName::Ident(i.clone().into()),
                    value: Box::new(Expr::Ident(Ident::new(
                        // $$ACTION_ARG_0
                        format!("$$ACTION_ARG_{index}").into(),
                        DUMMY_SP,
                        self.private_ctxt,
                    ))),
                })));
            }
        }
    }

    noop_visit_mut_type!();
}

#[derive(Debug, Clone, PartialEq, Eq)]
struct NamePart {
    prop: Atom,
    is_member: bool,
    optional: bool,
}

#[derive(Debug, Clone, PartialEq, Eq)]
struct Name(Id, Vec<NamePart>);

impl From<&'_ Ident> for Name {
    fn from(value: &Ident) -> Self {
        Name(value.to_id(), vec![])
    }
}

impl TryFrom<&'_ Expr> for Name {
    type Error = ();

    fn try_from(value: &Expr) -> Result<Self, Self::Error> {
        match value {
            Expr::Ident(i) => Ok(Name(i.to_id(), vec![])),
            Expr::Member(e) => e.try_into(),
            Expr::OptChain(e) => e.try_into(),
            _ => Err(()),
        }
    }
}

impl TryFrom<&'_ MemberExpr> for Name {
    type Error = ();

    fn try_from(value: &MemberExpr) -> Result<Self, Self::Error> {
        match &value.prop {
            MemberProp::Ident(prop) => {
                let mut obj: Name = value.obj.as_ref().try_into()?;
                obj.1.push(NamePart {
                    prop: prop.sym.clone(),
                    is_member: true,
                    optional: false,
                });
                Ok(obj)
            }
            _ => Err(()),
        }
    }
}

impl TryFrom<&'_ OptChainExpr> for Name {
    type Error = ();

    fn try_from(value: &OptChainExpr) -> Result<Self, Self::Error> {
        match &*value.base {
            OptChainBase::Member(m) => match &m.prop {
                MemberProp::Ident(prop) => {
                    let mut obj: Name = m.obj.as_ref().try_into()?;
                    obj.1.push(NamePart {
                        prop: prop.sym.clone(),
                        is_member: false,
                        optional: value.optional,
                    });
                    Ok(obj)
                }
                _ => Err(()),
            },
            OptChainBase::Call(_) => Err(()),
        }
    }
}

impl From<Name> for Box<Expr> {
    fn from(value: Name) -> Self {
        let mut expr = Box::new(Expr::Ident(value.0.into()));

        for NamePart {
            prop,
            is_member,
            optional,
        } in value.1.into_iter()
        {
            #[allow(clippy::replace_box)]
            if is_member {
                expr = Box::new(Expr::Member(MemberExpr {
                    span: DUMMY_SP,
                    obj: expr,
                    prop: MemberProp::Ident(IdentName::new(prop, DUMMY_SP)),
                }));
            } else {
                expr = Box::new(Expr::OptChain(OptChainExpr {
                    span: DUMMY_SP,
                    base: Box::new(OptChainBase::Member(MemberExpr {
                        span: DUMMY_SP,
                        obj: expr,
                        prop: MemberProp::Ident(IdentName::new(prop, DUMMY_SP)),
                    })),
                    optional,
                }));
            }
        }

        expr
    }
}

fn emit_error(error_kind: ServerActionsErrorKind) {
    let (span, msg) = match error_kind {
        ServerActionsErrorKind::ExportedSyncFunction {
            span,
            in_action_file,
        } => (
            span,
            formatdoc! {
                r#"
                    Only async functions are allowed to be exported in a {directive} file.
                "#,
                directive = if in_action_file {
                    "\"use server\""
                } else {
                    "\"use cache\""
                }
            },
        ),
        ServerActionsErrorKind::ForbiddenExpression {
            span,
            expr,
            directive,
        } => (
            span,
            formatdoc! {
                r#"
                    {subject} cannot use `{expr}`.
                "#,
                subject = if let Directive::UseServer = directive {
                    "Server Actions"
                } else {
                    "\"use cache\" functions"
                }
            },
        ),
        ServerActionsErrorKind::InlineUseCacheInClassInstanceMethod { span } => (
            span,
            formatdoc! {
                r#"
                    It is not allowed to define inline "use cache" annotated class instance methods.
                    To define cached functions, use functions, object method properties, or static class methods instead.
                "#
            },
        ),
        ServerActionsErrorKind::InlineUseCacheInClientComponent { span } => (
            span,
            formatdoc! {
                r#"
                    It is not allowed to define inline "use cache" annotated functions in Client Components.
                    To use "use cache" functions in a Client Component, you can either export them from a separate file with "use cache" or "use server" at the top, or pass them down through props from a Server Component.
                "#
            },
        ),
        ServerActionsErrorKind::InlineUseServerInClassInstanceMethod { span } => (
            span,
            formatdoc! {
                r#"
                    It is not allowed to define inline "use server" annotated class instance methods.
                    To define Server Actions, use functions, object method properties, or static class methods instead.
                "#
            },
        ),
        ServerActionsErrorKind::InlineUseServerInClientComponent { span } => (
            span,
            formatdoc! {
                r#"
                    It is not allowed to define inline "use server" annotated Server Actions in Client Components.
                    To use Server Actions in a Client Component, you can either export them from a separate file with "use server" at the top, or pass them down through props from a Server Component.

                    Read more: https://nextjs.org/docs/app/api-reference/directives/use-server#using-server-functions-in-a-client-component
                "#
            },
        ),
        ServerActionsErrorKind::InlineSyncFunction { span, directive } => (
            span,
            formatdoc! {
                r#"
                    {subject} must be async functions.
                "#,
                subject = if let Directive::UseServer = directive {
                    "Server Actions"
                } else {
                    "\"use cache\" functions"
                }
            },
        ),
        ServerActionsErrorKind::MisplacedDirective {
            span,
            directive,
            location,
        } => (
            span,
            formatdoc! {
                r#"
                    The "{directive}" directive must be at the top of the {location}.
                "#,
                location = match location {
                    DirectiveLocation::Module => "file",
                    DirectiveLocation::FunctionBody => "function body",
                }
            },
        ),
        ServerActionsErrorKind::MisplacedWrappedDirective {
            span,
            directive,
            location,
        } => (
            span,
            formatdoc! {
                r#"
                    The "{directive}" directive must be at the top of the {location}, and cannot be wrapped in parentheses.
                "#,
                location = match location {
                    DirectiveLocation::Module => "file",
                    DirectiveLocation::FunctionBody => "function body",
                }
            },
        ),
        ServerActionsErrorKind::MisspelledDirective {
            span,
            directive,
            expected_directive,
        } => (
            span,
            formatdoc! {
                r#"
                    Did you mean "{expected_directive}"? "{directive}" is not a supported directive name."
                "#
            },
        ),
        ServerActionsErrorKind::MultipleDirectives { span, location } => (
            span,
            formatdoc! {
                r#"
                    Conflicting directives "use server" and "use cache" found in the same {location}. You cannot place both directives at the top of a {location}. Please remove one of them.
                "#,
                location = match location {
                    DirectiveLocation::Module => "file",
                    DirectiveLocation::FunctionBody => "function body",
                }
            },
        ),
        ServerActionsErrorKind::UnknownCacheKind { span, cache_kind } => (
            span,
            formatdoc! {
                r#"
                    Unknown cache kind "{cache_kind}". Please configure a cache handler for this kind in the `cacheHandlers` object in your Next.js config.
                "#
            },
        ),
        ServerActionsErrorKind::UseCacheWithoutCacheComponents { span, directive } => (
            span,
            formatdoc! {
                r#"
                    To use "{directive}", please enable the feature flag `cacheComponents` in your Next.js config.

                    Read more: https://nextjs.org/docs/app/api-reference/directives/use-cache#usage
                "#
            },
        ),
        ServerActionsErrorKind::WrappedDirective { span, directive } => (
            span,
            formatdoc! {
                r#"
                    The "{directive}" directive cannot be wrapped in parentheses.
                "#
            },
        ),
    };

    HANDLER.with(|handler| handler.struct_span_err(span, &msg).emit());
}

/// Strips span information from a ModuleExportName, replacing all spans with DUMMY_SP.
/// Used in production builds to prevent leaking source code information in source maps to browsers.
fn strip_export_name_span(export_name: &ModuleExportName) -> ModuleExportName {
    match export_name {
        ModuleExportName::Ident(i) => {
            ModuleExportName::Ident(Ident::new(i.sym.clone(), DUMMY_SP, i.ctxt))
        }
        ModuleExportName::Str(s) => ModuleExportName::Str(Str {
            span: DUMMY_SP,
            value: s.value.clone(),
            raw: None,
        }),
    }
}

fn program_to_data_url(
    file_name: &str,
    cm: &Arc<SourceMap>,
    body: Vec<ModuleItem>,
    prepend_comment: Comment,
) -> String {
    let module_span = Span::dummy_with_cmt();
    let comments = SingleThreadedComments::default();
    comments.add_leading(module_span.lo, prepend_comment);

    let program = &Program::Module(Module {
        span: module_span,
        body,
        shebang: None,
    });

    let mut output = vec![];
    let mut mappings = vec![];
    let mut emitter = Emitter {
        cfg: codegen::Config::default().with_minify(true),
        cm: cm.clone(),
        wr: Box::new(JsWriter::new(
            cm.clone(),
            " ",
            &mut output,
            Some(&mut mappings),
        )),
        comments: Some(&comments),
    };

    emitter.emit_program(program).unwrap();
    drop(emitter);

    pub struct InlineSourcesContentConfig<'a> {
        folder_path: Option<&'a Path>,
    }
    // This module will be placed at `some/path/to/data:28a9d2` where the original input file lives
    // at `some/path/to/actions.js`. So we need to generate a relative path, usually `./actions.js`
    impl SourceMapGenConfig for InlineSourcesContentConfig<'_> {
        fn file_name_to_source(&self, file: &FileName) -> String {
            let FileName::Custom(file) = file else {
                // Turbopack uses FileName::Custom for the `[project]/...` paths
                return file.to_string();
            };
            let Some(folder_path) = &self.folder_path else {
                return file.to_string();
            };

            if let Some(rel_path) = diff_paths(file, folder_path) {
                format!("./{}", rel_path.display())
            } else {
                file.to_string()
            }
        }

        fn inline_sources_content(&self, _f: &FileName) -> bool {
            true
        }
    }

    let map = cm.build_source_map(
        &mappings,
        None,
        InlineSourcesContentConfig {
            folder_path: PathBuf::from(format!("[project]/{file_name}")).parent(),
        },
    );
    let map = {
        if map.get_token_count() > 0 {
            let mut buf = vec![];
            map.to_writer(&mut buf)
                .expect("failed to generate sourcemap");
            Some(buf)
        } else {
            None
        }
    };

    let mut output = String::from_utf8(output).expect("codegen generated non-utf8 output");
    if let Some(map) = map {
        output.extend(
            format!(
                "\n//# sourceMappingURL=data:application/json;base64,{}",
                Base64Display::new(&map, &BASE64_STANDARD)
            )
            .chars(),
        );
    }
    format!("data:text/javascript,{}", urlencoding::encode(&output))
}