Use virtual module to create rpc client

Author: jamesopstad

packages/plugin-rsc/src/plugin.ts

@@ -728,18 +728,7 @@ export default function vitePluginRsc(
             this.environment.mode === 'dev' &&
             rscPluginOptions.loadModuleDevProxy
           ) {
-            const origin = server.resolvedUrls?.local[0]
-            assert(origin, '[vite-rsc] no server for loadModueleDevProxy')
-            const endpoint =
-              origin +
-              '__vite_rsc_load_module_dev_proxy?' +
-              new URLSearchParams({ environmentName, entryName })
-            replacement = `__vite_rsc_rpc.createRpcClient(${JSON.stringify({
-              endpoint,
-            })})`
-            s.prepend(
-              `import * as __vite_rsc_rpc from "@vitejs/plugin-rsc/utils/rpc";`,
-            )
+            replacement = `import("virtual:vite-rsc/rpc-client").then((module) => module.createRpcClient(${JSON.stringify({ environmentName, entryName })}))`
           } else if (this.environment.mode === 'dev') {
             const environment = server.environments[environmentName]!
             const source = getEntrySource(environment.config, entryName)
@@ -851,6 +840,32 @@ export default function vitePluginRsc(
         })
       },
     },
+    {
+      name: 'rsc:virtual:vite-rsc/rpc-client',
+      resolveId(source) {
+        if (source === 'virtual:vite-rsc/rpc-client') {
+          return `\0${source}`
+        }
+      },
+      load(id) {
+        if (id === '\0virtual:vite-rsc/rpc-client') {
+          const { server } = manager
+          const origin = server.resolvedUrls?.local[0]
+          assert(origin, '[vite-rsc] no server for loadModuleDevProxy')
+
+          return `\
+import * as __vite_rsc_rpc from "@vitejs/plugin-rsc/utils/rpc";
+export function createRpcClient(params) {
+  const endpoint =
+    "${origin}" +
+    "__vite_rsc_load_module_dev_proxy?" +
+    new URLSearchParams(params);
+  return __vite_rsc_rpc.createRpcClient({ endpoint });
+}
+          `
+        }
+      },
+    },
     {
       name: 'rsc:virtual:vite-rsc/assets-manifest',
       resolveId(source) {