rwsdk 1.0.0-alpha.2 → 1.0.0-alpha.20

package/dist/vite/miniflareHMRPlugin.mjs

@@ -17,7 +17,7 @@ const hasDirective = async (filepath, directive) => {
     const content = await readFile(filepath, "utf-8");
     return sourceHasDirective(content, directive);
 };
-const hasEntryAsAncestor = (module, entryFile, seen = new Set()) => {
+export const hasEntryAsAncestor = ({ module, entryFile, seen = new Set(), }) => {
     // Prevent infinite recursion
     if (seen.has(module)) {
         return false;
@@ -28,7 +28,7 @@ const hasEntryAsAncestor = (module, entryFile, seen = new Set()) => {
         if (importer.file === entryFile)
             return true;
         // Recursively check importers
-        if (hasEntryAsAncestor(importer, entryFile, seen))
+        if (hasEntryAsAncestor({ module: importer, entryFile, seen }))
             return true;
     }
     return false;

package/dist/vite/miniflareHMRPlugin.test.d.mts

@@ -0,0 +1 @@
+export {};

package/dist/vite/miniflareHMRPlugin.test.mjs

@@ -0,0 +1,42 @@
+import { describe, it, expect } from "vitest";
+import { hasEntryAsAncestor } from "./miniflareHMRPlugin.mjs";
+const createModule = (file) => ({
+    file,
+    importers: new Set(),
+});
+describe("hasEntryAsAncestor", () => {
+    it("should return true if the entry file is a direct importer", () => {
+        const entry = createModule("entry.js");
+        const mod = createModule("mod.js");
+        mod.importers.add(entry);
+        expect(hasEntryAsAncestor({ module: mod, entryFile: "entry.js" })).toBe(true);
+    });
+    it("should return true if the entry file is an indirect importer", () => {
+        const entry = createModule("entry.js");
+        const importer1 = createModule("importer1.js");
+        const mod = createModule("mod.js");
+        importer1.importers.add(entry);
+        mod.importers.add(importer1);
+        expect(hasEntryAsAncestor({ module: mod, entryFile: "entry.js" })).toBe(true);
+    });
+    it("should return false if the entry file is not an importer", () => {
+        const entry = createModule("entry.js");
+        const other = createModule("other.js");
+        const mod = createModule("mod.js");
+        mod.importers.add(other);
+        expect(hasEntryAsAncestor({ module: mod, entryFile: "entry.js" })).toBe(false);
+    });
+    it("should handle circular dependencies", () => {
+        const entry = createModule("entry.js");
+        const modA = createModule("modA.js");
+        const modB = createModule("modB.js");
+        modA.importers.add(entry);
+        modA.importers.add(modB);
+        modB.importers.add(modA);
+        expect(hasEntryAsAncestor({ module: modB, entryFile: "entry.js" })).toBe(true);
+    });
+    it("should return false for a module with no importers", () => {
+        const mod = createModule("mod.js");
+        expect(hasEntryAsAncestor({ module: mod, entryFile: "entry.js" })).toBe(false);
+    });
+});

package/dist/vite/redwoodPlugin.d.mts

@@ -1,12 +1,21 @@
 import { InlineConfig } from "vite";
+import { unstable_readConfig } from "wrangler";
 export type RedwoodPluginOptions = {
     silent?: boolean;
     rootDir?: string;
     includeCloudflarePlugin?: boolean;
     includeReactPlugin?: boolean;
     configPath?: string;
+    forceClientPaths?: string[];
+    forceServerPaths?: string[];
     entry?: {
         worker?: string;
     };
 };
+export declare const determineWorkerEntryPathname: ({ projectRootDir, workerConfigPath, options, readConfig, }: {
+    projectRootDir: string;
+    workerConfigPath: string;
+    options: RedwoodPluginOptions;
+    readConfig?: typeof unstable_readConfig;
+}) => Promise<string>;
 export declare const redwoodPlugin: (options?: RedwoodPluginOptions) => Promise<InlineConfig["plugins"]>;

package/dist/vite/redwoodPlugin.mjs

@@ -14,7 +14,7 @@ import { miniflareHMRPlugin } from "./miniflareHMRPlugin.mjs";
 import { moveStaticAssetsPlugin } from "./moveStaticAssetsPlugin.mjs";
 import { configPlugin } from "./configPlugin.mjs";
 import { $ } from "../lib/$.mjs";
-import { reactConditionsResolverPlugin } from "./reactConditionsResolverPlugin.mjs";
+import { knownDepsResolverPlugin } from "./knownDepsResolverPlugin.mjs";
 import { findWranglerConfig } from "../lib/findWranglerConfig.mjs";
 import { pathExists } from "fs-extra";
 import { injectVitePreamble } from "./injectVitePreamblePlugin.mjs";
@@ -27,11 +27,12 @@ import { manifestPlugin } from "./manifestPlugin.mjs";
 import { linkerPlugin } from "./linkerPlugin.mjs";
 import { directiveModulesDevPlugin } from "./directiveModulesDevPlugin.mjs";
 import { directivesFilteringPlugin } from "./directivesFilteringPlugin.mjs";
-const determineWorkerEntryPathname = async (projectRootDir, workerConfigPath, options) => {
+import { resolveForcedPaths } from "./resolveForcedPaths.mjs";
+export const determineWorkerEntryPathname = async ({ projectRootDir, workerConfigPath, options, readConfig = unstable_readConfig, }) => {
     if (options.entry?.worker) {
         return resolve(projectRootDir, options.entry.worker);
     }
-    const workerConfig = unstable_readConfig({ config: workerConfigPath });
+    const workerConfig = readConfig({ config: workerConfigPath });
     return resolve(projectRootDir, workerConfig.main ?? "src/worker.tsx");
 };
 const clientFiles = new Set();
@@ -39,11 +40,33 @@ const serverFiles = new Set();
 const clientEntryPoints = new Set();
 export const redwoodPlugin = async (options = {}) => {
     const projectRootDir = process.cwd();
+    if (options.forceClientPaths) {
+        const clientPaths = await resolveForcedPaths({
+            patterns: options.forceClientPaths,
+            projectRootDir,
+        });
+        for (const p of clientPaths) {
+            clientFiles.add(p);
+        }
+    }
+    if (options.forceServerPaths) {
+        const serverPaths = await resolveForcedPaths({
+            patterns: options.forceServerPaths,
+            projectRootDir,
+        });
+        for (const p of serverPaths) {
+            serverFiles.add(p);
+        }
+    }
     const workerConfigPath = options.configPath ??
         (process.env.RWSDK_WRANGLER_CONFIG
             ? resolve(projectRootDir, process.env.RWSDK_WRANGLER_CONFIG)
             : await findWranglerConfig(projectRootDir));
-    const workerEntryPathname = await determineWorkerEntryPathname(projectRootDir, workerConfigPath, options);
+    const workerEntryPathname = await determineWorkerEntryPathname({
+        projectRootDir,
+        workerConfigPath,
+        options,
+    });
     const shouldIncludeCloudflarePlugin = options.includeCloudflarePlugin ??
         !(await hasOwnCloudflareVitePlugin({ rootProjectDir: projectRootDir }));
     const shouldIncludeReactPlugin = options.includeReactPlugin ??
@@ -68,6 +91,7 @@ export const redwoodPlugin = async (options = {}) => {
             clientFiles,
             serverFiles,
             projectRootDir,
+            workerEntryPathname,
         }),
         configPlugin({
             silent: options.silent ?? false,
@@ -82,7 +106,7 @@ export const redwoodPlugin = async (options = {}) => {
             serverFiles,
             projectRootDir,
         }),
-        reactConditionsResolverPlugin({ projectRootDir }),
+        knownDepsResolverPlugin({ projectRootDir }),
         tsconfigPaths({ root: projectRootDir }),
         shouldIncludeCloudflarePlugin
             ? cloudflare({

package/dist/vite/redwoodPlugin.test.d.mts

@@ -0,0 +1 @@
+export {};

package/dist/vite/redwoodPlugin.test.mjs

@@ -0,0 +1,34 @@
+import { describe, it, expect } from "vitest";
+import { determineWorkerEntryPathname } from "./redwoodPlugin.mjs";
+import path from "node:path";
+describe("determineWorkerEntryPathname", () => {
+    const projectRootDir = "/test/project";
+    it("should use the entry path from options if provided", async () => {
+        const result = await determineWorkerEntryPathname({
+            projectRootDir,
+            workerConfigPath: "/test/project/wrangler.toml",
+            options: { entry: { worker: "src/custom-worker.ts" } },
+        });
+        expect(result).toBe(path.join(projectRootDir, "src/custom-worker.ts"));
+    });
+    it("should use the main path from wrangler config if no entry option is provided", async () => {
+        const readConfig = () => ({ main: "src/wrangler-worker.tsx" });
+        const result = await determineWorkerEntryPathname({
+            projectRootDir,
+            workerConfigPath: "/test/project/wrangler.toml",
+            options: {},
+            readConfig: readConfig,
+        });
+        expect(result).toBe(path.join(projectRootDir, "src/wrangler-worker.tsx"));
+    });
+    it("should use the default path if wrangler config has no main property", async () => {
+        const readConfig = () => ({});
+        const result = await determineWorkerEntryPathname({
+            projectRootDir,
+            workerConfigPath: "/test/project/wrangler.toml",
+            options: {},
+            readConfig: readConfig,
+        });
+        expect(result).toBe(path.join(projectRootDir, "src/worker.tsx"));
+    });
+});

package/dist/vite/resolveForcedPaths.d.mts

@@ -0,0 +1,4 @@
+export declare function resolveForcedPaths({ patterns, projectRootDir, }: {
+    patterns: string[];
+    projectRootDir: string;
+}): Promise<string[]>;

package/dist/vite/resolveForcedPaths.mjs

@@ -0,0 +1,9 @@
+import { glob } from "glob";
+import { normalizeModulePath } from "../lib/normalizeModulePath.mjs";
+export async function resolveForcedPaths({ patterns, projectRootDir, }) {
+    return (await glob(patterns, {
+        cwd: projectRootDir,
+        absolute: true,
+        realpath: true,
+    })).map((filepath) => normalizeModulePath(filepath, projectRootDir));
+}

package/dist/vite/runDirectivesScan.d.mts

@@ -1,7 +1,28 @@
 import { Environment, ResolvedConfig } from "vite";
-export declare const runDirectivesScan: ({ rootConfig, environments, clientFiles, serverFiles, }: {
+type Resolver = (context: {}, path: string, request: string, resolveContext: {}, callback: (err: Error | null, result?: string | false) => void) => void;
+export declare function resolveModuleWithEnvironment({ path, importer, importerEnv, clientResolver, workerResolver, }: {
+    path: string;
+    importer?: string;
+    importerEnv: "client" | "worker";
+    clientResolver: Resolver;
+    workerResolver: Resolver;
+}): Promise<{
+    id: string;
+} | null>;
+export declare function classifyModule({ contents, inheritedEnv, }: {
+    contents: string;
+    inheritedEnv: "client" | "worker";
+}): {
+    moduleEnv: "client" | "worker";
+    isClient: boolean;
+    isServer: boolean;
+};
+export type EsbuildLoader = "js" | "jsx" | "ts" | "tsx" | "default";
+export declare const runDirectivesScan: ({ rootConfig, environments, clientFiles, serverFiles, entries: initialEntries, }: {
     rootConfig: ResolvedConfig;
     environments: Record<string, Environment>;
     clientFiles: Set<string>;
     serverFiles: Set<string>;
+    entries?: string[];
 }) => Promise<void>;
+export {};

package/dist/vite/runDirectivesScan.mjs

@@ -4,8 +4,10 @@ import path from "node:path";
 import debug from "debug";
 import { getViteEsbuild } from "./getViteEsbuild.mjs";
 import { normalizeModulePath } from "../lib/normalizeModulePath.mjs";
+import { INTERMEDIATES_OUTPUT_DIR } from "../lib/constants.mjs";
 import { externalModules } from "./constants.mjs";
-import { createViteAwareResolver, } from "./createViteAwareResolver.mjs";
+import { createViteAwareResolver } from "./createViteAwareResolver.mjs";
+import { compile } from "@mdx-js/mdx";
 const log = debug("rwsdk:vite:run-directives-scan");
 // Copied from Vite's source code.
 // https://github.com/vitejs/vite/blob/main/packages/vite/src/shared/utils.ts
@@ -14,13 +16,47 @@ const isObject = (value) => Object.prototype.toString.call(value) === "[object O
 // https://github.com/vitejs/vite/blob/main/packages/vite/src/node/utils.ts
 const externalRE = /^(https?:)?\/\//;
 const isExternalUrl = (url) => externalRE.test(url);
-export const runDirectivesScan = async ({ rootConfig, environments, clientFiles, serverFiles, }) => {
-    console.log("\n🔍 Scanning for 'use client' and 'use server' directives...");
+export async function resolveModuleWithEnvironment({ path, importer, importerEnv, clientResolver, workerResolver, }) {
+    const resolver = importerEnv === "client" ? clientResolver : workerResolver;
+    return new Promise((resolvePromise) => {
+        resolver({}, importer || "", path, {}, (err, result) => {
+            if (!err && result) {
+                resolvePromise({ id: result });
+            }
+            else {
+                if (err) {
+                    const errorMessage = err.message || String(err);
+                    if (errorMessage.includes("Package path . is not exported")) {
+                        log("Package exports error for %s, marking as external", path);
+                    }
+                    else {
+                        log("Resolution failed for %s: %s", path, errorMessage);
+                    }
+                }
+                resolvePromise(null);
+            }
+        });
+    });
+}
+export function classifyModule({ contents, inheritedEnv, }) {
+    let moduleEnv = inheritedEnv;
+    const isClient = hasDirective(contents, "use client");
+    const isServer = hasDirective(contents, "use server");
+    if (isClient) {
+        moduleEnv = "client";
+    }
+    else if (isServer) {
+        moduleEnv = "worker";
+    }
+    return { moduleEnv, isClient, isServer };
+}
+export const runDirectivesScan = async ({ rootConfig, environments, clientFiles, serverFiles, entries: initialEntries, }) => {
+    deferredLog("\n… (rwsdk) Scanning for 'use client' and 'use server' directives...");
     // Set environment variable to indicate scanning is in progress
     process.env.RWSDK_DIRECTIVE_SCAN_ACTIVE = "true";
     try {
         const esbuild = await getViteEsbuild(rootConfig.root);
-        const input = environments.worker.config.build.rollupOptions?.input;
+        const input = initialEntries ?? environments.worker.config.build.rollupOptions?.input;
         let entries;
         if (Array.isArray(input)) {
             entries = input;
@@ -38,7 +74,9 @@ export const runDirectivesScan = async ({ rootConfig, environments, clientFiles,
             log("No entries found for directives scan in worker environment, skipping.");
             return;
         }
-        const absoluteEntries = entries.map((entry) => path.resolve(rootConfig.root, entry));
+        // Filter out virtual modules since they can't be scanned by esbuild
+        const realEntries = entries.filter((entry) => !entry.includes("virtual:"));
+        const absoluteEntries = realEntries.map((entry) => path.resolve(rootConfig.root, entry));
         log("Starting directives scan for worker environment with entries:", absoluteEntries);
         const workerResolver = createViteAwareResolver(rootConfig, environments.worker);
         const clientResolver = createViteAwareResolver(rootConfig, environments.client);
@@ -57,7 +95,7 @@ export const runDirectivesScan = async ({ rootConfig, environments, clientFiles,
             setup(build) {
                 // Match Vite's behavior by externalizing assets and special queries.
                 // This prevents esbuild from trying to bundle them, which would fail.
-                const scriptFilter = /\.(c|m)?[jt]sx?$/;
+                const scriptFilter = /\.(c|m)?[jt]sx?$|\.mdx$/;
                 const specialQueryFilter = /[?&](?:url|raw|worker|sharedworker|inline)\b/;
                 // This regex is used to identify if a path has any file extension.
                 const hasExtensionRegex = /\.[^/]+$/;
@@ -87,17 +125,12 @@ export const runDirectivesScan = async ({ rootConfig, environments, clientFiles,
                         /\.(m|c)?[jt]sx?$/.test(args.importer)) {
                         try {
                             const importerContents = await readFileWithCache(args.importer);
-                            if (hasDirective(importerContents, "use client")) {
-                                importerEnv = "client";
-                                log("Pre-detected importer 'use client' in:", args.importer);
-                            }
-                            else if (hasDirective(importerContents, "use server")) {
-                                importerEnv = "worker";
-                                log("Pre-detected importer 'use server' in:", args.importer);
-                            }
-                            else {
-                                importerEnv = "worker"; // Default for entry points
-                            }
+                            const classification = classifyModule({
+                                contents: importerContents,
+                                inheritedEnv: "worker", // Default for entry points
+                            });
+                            importerEnv = classification.moduleEnv;
+                            log("Pre-detected importer environment in:", args.importer, "as", importerEnv);
                             moduleEnvironments.set(args.importer, importerEnv);
                         }
                         catch (e) {
@@ -109,25 +142,12 @@ export const runDirectivesScan = async ({ rootConfig, environments, clientFiles,
                         importerEnv = "worker"; // Default for entry points or non-script files
                     }
                     log("Importer:", args.importer, "environment:", importerEnv);
-                    const resolver = importerEnv === "client" ? clientResolver : workerResolver;
-                    const resolved = await new Promise((resolve) => {
-                        resolver({}, args.importer || rootConfig.root, args.path, {}, (err, result) => {
-                            if (!err && result) {
-                                resolve({ id: result });
-                            }
-                            else {
-                                if (err) {
-                                    const errorMessage = err.message || String(err);
-                                    if (errorMessage.includes("Package path . is not exported")) {
-                                        log("Package exports error for %s, marking as external", args.path);
-                                    }
-                                    else {
-                                        log("Resolution failed for %s: %s", args.path, errorMessage);
-                                    }
-                                }
-                                resolve(null);
-                            }
-                        });
+                    const resolved = await resolveModuleWithEnvironment({
+                        path: args.path,
+                        importer: args.importer,
+                        importerEnv,
+                        clientResolver,
+                        workerResolver,
                     });
                     log("Resolution result:", resolved);
                     const resolvedPath = resolved?.id;
@@ -143,7 +163,7 @@ export const runDirectivesScan = async ({ rootConfig, environments, clientFiles,
                     log("Marking as external:", args.path, "resolved to:", resolvedPath);
                     return { external: true };
                 });
-                build.onLoad({ filter: /\.(m|c)?[jt]sx?$/ }, async (args) => {
+                build.onLoad({ filter: /\.(m|c)?[jt]sx?$|\.mdx$/ }, async (args) => {
                     log("onLoad called for:", args.path);
                     if (!args.path.startsWith("/") ||
                         args.path.includes("virtual:") ||
@@ -156,33 +176,52 @@ export const runDirectivesScan = async ({ rootConfig, environments, clientFiles,
                         return null;
                     }
                     try {
-                        const contents = await readFileWithCache(args.path);
+                        const originalContents = await readFileWithCache(args.path);
                         const inheritedEnv = args.pluginData?.inheritedEnv || "worker";
-                        let currentEnv = inheritedEnv;
-                        const isClient = hasDirective(contents, "use client");
-                        const isServer = hasDirective(contents, "use server");
-                        // A file's own directive takes precedence over the inherited environment.
-                        if (isClient) {
-                            currentEnv = "client";
-                        }
-                        else if (isServer) {
-                            // `else if` handles cases where a file might have both directives.
-                            // "use client" takes precedence.
-                            currentEnv = "worker";
-                        }
+                        const { moduleEnv, isClient, isServer } = classifyModule({
+                            contents: originalContents,
+                            inheritedEnv,
+                        });
                         // Store the definitive environment for this module, so it can be used when it becomes an importer.
-                        moduleEnvironments.set(args.path, currentEnv);
-                        log("Set environment for", args.path, "to", currentEnv);
+                        const realPath = await fsp.realpath(args.path);
+                        moduleEnvironments.set(realPath, moduleEnv);
+                        log("Set environment for", realPath, "to", moduleEnv);
                         // Finally, populate the output sets if the file has a directive.
                         if (isClient) {
-                            log("Discovered 'use client' in:", args.path);
-                            clientFiles.add(normalizeModulePath(args.path, rootConfig.root));
+                            log("Discovered 'use client' in:", realPath);
+                            clientFiles.add(normalizeModulePath(realPath, rootConfig.root));
                         }
                         if (isServer) {
-                            log("Discovered 'use server' in:", args.path);
-                            serverFiles.add(normalizeModulePath(args.path, rootConfig.root));
+                            log("Discovered 'use server' in:", realPath);
+                            serverFiles.add(normalizeModulePath(realPath, rootConfig.root));
+                        }
+                        let code;
+                        let loader;
+                        if (args.path.endsWith(".mdx")) {
+                            const result = await compile(originalContents, {
+                                jsx: true,
+                                jsxImportSource: "react",
+                            });
+                            code = String(result.value);
+                            loader = "tsx";
+                        }
+                        else if (/\.(m|c)?tsx$/.test(args.path)) {
+                            code = originalContents;
+                            loader = "tsx";
                         }
-                        return { contents, loader: "default" };
+                        else if (/\.(m|c)?ts$/.test(args.path)) {
+                            code = originalContents;
+                            loader = "ts";
+                        }
+                        else if (/\.(m|c)?jsx$/.test(args.path)) {
+                            code = originalContents;
+                            loader = "jsx";
+                        }
+                        else {
+                            code = originalContents;
+                            loader = "js";
+                        }
+                        return { contents: code, loader };
                     }
                     catch (e) {
                         log("Could not read file during scan, skipping:", args.path, e);
@@ -195,6 +234,7 @@ export const runDirectivesScan = async ({ rootConfig, environments, clientFiles,
             entryPoints: absoluteEntries,
             bundle: true,
             write: false,
+            outdir: path.join(INTERMEDIATES_OUTPUT_DIR, "directive-scan"),
             platform: "node",
             format: "esm",
             logLevel: "silent",
@@ -207,6 +247,13 @@ export const runDirectivesScan = async ({ rootConfig, environments, clientFiles,
     finally {
         // Always clear the scanning flag when done
         delete process.env.RWSDK_DIRECTIVE_SCAN_ACTIVE;
-        console.log("✅ Scan complete.");
+        deferredLog("✔ (rwsdk) Done scanning for 'use client' and 'use server' directives.");
+        process.env.VERBOSE &&
+            log("Client/server files after scanning: client=%O, server=%O", Array.from(clientFiles), Array.from(serverFiles));
     }
 };
+const deferredLog = (message) => {
+    setTimeout(() => {
+        console.log(message);
+    }, 500);
+};

package/dist/vite/runDirectivesScan.test.d.mts

@@ -0,0 +1 @@
+export {};

package/dist/vite/runDirectivesScan.test.mjs

@@ -0,0 +1,73 @@
+import { describe, it, expect, vi } from "vitest";
+import { resolveModuleWithEnvironment, classifyModule, } from "./runDirectivesScan.mjs";
+describe("runDirectivesScan helpers", () => {
+    describe("resolveModuleWithEnvironment", () => {
+        it("should use the client resolver when importerEnv is 'client'", async () => {
+            const clientResolver = vi.fn((_a, _b, _c, _d, cb) => cb(null, "/resolved/client"));
+            const workerResolver = vi.fn();
+            const result = await resolveModuleWithEnvironment({
+                path: "test-path",
+                importerEnv: "client",
+                clientResolver,
+                workerResolver,
+            });
+            expect(clientResolver).toHaveBeenCalled();
+            expect(workerResolver).not.toHaveBeenCalled();
+            expect(result).toEqual({ id: "/resolved/client" });
+        });
+        it("should use the worker resolver when importerEnv is 'worker'", async () => {
+            const clientResolver = vi.fn();
+            const workerResolver = vi.fn((_a, _b, _c, _d, cb) => cb(null, "/resolved/worker"));
+            const result = await resolveModuleWithEnvironment({
+                path: "test-path",
+                importerEnv: "worker",
+                clientResolver,
+                workerResolver,
+            });
+            expect(workerResolver).toHaveBeenCalled();
+            expect(clientResolver).not.toHaveBeenCalled();
+            expect(result).toEqual({ id: "/resolved/worker" });
+        });
+        it("should return null on resolution error", async () => {
+            const clientResolver = vi.fn((_a, _b, _c, _d, cb) => cb(new Error("Resolution failed")));
+            const workerResolver = vi.fn();
+            const result = await resolveModuleWithEnvironment({
+                path: "test-path",
+                importerEnv: "client",
+                clientResolver,
+                workerResolver,
+            });
+            expect(result).toBeNull();
+        });
+    });
+    describe("classifyModule", () => {
+        it("should return 'client' if 'use client' directive is present", () => {
+            const contents = `'use client';\nconsole.log('hello');`;
+            const result = classifyModule({ contents, inheritedEnv: "worker" });
+            expect(result.moduleEnv).toBe("client");
+            expect(result.isClient).toBe(true);
+            expect(result.isServer).toBe(false);
+        });
+        it("should return 'worker' if 'use server' directive is present", () => {
+            const contents = `"use server";\nexport default () => {};`;
+            const result = classifyModule({ contents, inheritedEnv: "client" });
+            expect(result.moduleEnv).toBe("worker");
+            expect(result.isClient).toBe(false);
+            expect(result.isServer).toBe(true);
+        });
+        it("should prioritize 'use client' over 'use server'", () => {
+            const contents = `'use client';\n'use server';\nconsole.log('hello');`;
+            const result = classifyModule({ contents, inheritedEnv: "worker" });
+            expect(result.moduleEnv).toBe("client");
+            expect(result.isClient).toBe(true);
+            expect(result.isServer).toBe(false);
+        });
+        it("should return the inherited environment if no directive is present", () => {
+            const contents = `console.log('no directive');`;
+            const result = classifyModule({ contents, inheritedEnv: "worker" });
+            expect(result.moduleEnv).toBe("worker");
+            expect(result.isClient).toBe(false);
+            expect(result.isServer).toBe(false);
+        });
+    });
+});

package/dist/vite/ssrBridgePlugin.mjs

@@ -34,7 +34,8 @@ export const ssrBridgePlugin = ({ clientFiles, serverFiles, }) => {
                         build.onResolve({ filter: /.*$/ }, (args) => {
                             process.env.VERBOSE &&
                                 log("Esbuild onResolve called for path=%s, args=%O", args.path, args);
-                            if (args.path === "rwsdk/__ssr_bridge") {
+                            if (args.path === "rwsdk/__ssr_bridge" ||
+                                args.path.startsWith(VIRTUAL_SSR_PREFIX)) {
                                 log("Marking as external: %s", args.path);
                                 return {
                                     path: args.path,
@@ -77,6 +78,12 @@ export const ssrBridgePlugin = ({ clientFiles, serverFiles, }) => {
             }
             else {
                 // In build mode, the behavior depends on the build pass
+                if (id.startsWith(VIRTUAL_SSR_PREFIX)) {
+                    if (this.environment.name === "worker") {
+                        log("Virtual SSR module case (build-worker pass): resolving to external");
+                        return { id, external: true };
+                    }
+                }
                 if (id === "rwsdk/__ssr_bridge" && this.environment.name === "worker") {
                     if (process.env.RWSDK_BUILD_PASS === "worker") {
                         // First pass: resolve to a temporary, external path

package/dist/vite/transformClientComponents.mjs

@@ -5,7 +5,8 @@ import { findExports } from "./findSpecifiers.mjs";
 const logVite = debug("rwsdk:vite:transform-client-components:vite");
 const logEsbuild = debug("rwsdk:vite:transform-client-components:esbuild");
 export async function transformClientComponents(code, normalizedId, ctx) {
-    if (!hasDirective(code, "use client")) {
+    if (!ctx.clientFiles?.has(normalizedId) &&
+        !hasDirective(code, "use client")) {
         return;
     }
     const log = ctx.isEsbuild ? logEsbuild : logVite;
@@ -76,13 +77,14 @@ export async function transformClientComponents(code, normalizedId, ctx) {
     // 4. Non-SSR files: replace all implementation with registerClientReference logic
     // Generate completely new code for worker/client environments
     const s = new MagicString("");
-    // Add import declaration
+    s.append('import { ssrLoadModule } from "rwsdk/__ssr_bridge";\n');
     s.append('import { registerClientReference } from "rwsdk/worker";\n');
+    s.append(`const SSRModule = await ssrLoadModule("${normalizedId}");\n`);
     // Compute unique computed local names first
     const computedLocalNames = new Map(processedExports.map((info) => [getComputedLocalName(info), info]));
     // Add registerClientReference assignments for unique names
     for (const [computedLocalName, correspondingInfo] of computedLocalNames) {
-        s.append(`const ${computedLocalName} = registerClientReference("${normalizedId}", "${correspondingInfo.exported}");\n`);
+        s.append(`const ${computedLocalName} = registerClientReference(SSRModule, "${normalizedId}", "${correspondingInfo.exported}");\n`);
     }
     // Add grouped export statement for named exports (preserving order and alias)
     if (processedExports.length > 0) {
@@ -94,7 +96,7 @@ export async function transformClientComponents(code, normalizedId, ctx) {
     // Add default export if present
     if (defaultExportInfo) {
         log(":isEsbuild=%s: Registering client reference for default export: %s", !!ctx.isEsbuild, defaultExportInfo.exported);
-        s.append(`export default registerClientReference("${normalizedId}", "${defaultExportInfo.exported}");\n`);
+        s.append(`export default registerClientReference(SSRModule, "${normalizedId}", "${defaultExportInfo.exported}");\n`);
     }
     const sourceMap = s.generateMap({
         source: normalizedId,