feat(plugin-rsc): add callback to be notified when client reference dependencies are loaded (#1065)

Co-authored-by: Hiroshi Ogawa <hi.ogawa.zz@gmail.com>

Author: schiller-manuel

packages/plugin-rsc/src/core/ssr.ts

@@ -6,6 +6,11 @@ let init = false
 
 export function setRequireModule(options: {
   load: (id: string) => unknown
+  /**
+   * Called EVERY time a module is requested
+   * @experimental
+   */
+  onLoad?: (id: string) => void
 }): void {
   if (init) return
   init = true
@@ -15,6 +20,8 @@ export function setRequireModule(options: {
   })
 
   const clientRequire = (id: string) => {
+    const cleanId = removeReferenceCacheTag(id)
+    options.onLoad?.(cleanId)
     return requireModule(id)
   }
   ;(globalThis as any).__vite_rsc_client_require__ = clientRequire

packages/plugin-rsc/src/ssr.tsx

@@ -9,6 +9,30 @@ export { createServerConsumerManifest } from './core/ssr'
 
 export * from './react/ssr'
 
+/**
+ * Callback type for client reference dependency notifications.
+ * Called during SSR when a client component's dependencies are loaded.
+ * @experimental
+ */
+export type OnClientReference = (reference: {
+  id: string
+  deps: ResolvedAssetDeps
+}) => void
+
+// Registered callback for client reference deps
+let onClientReference: OnClientReference | undefined
+
+/**
+ * Register a callback to be notified when client reference dependencies are loaded.
+ * Called during SSR when a client component is accessed.
+ * @experimental
+ */
+export function setOnClientReference(
+  callback: OnClientReference | undefined,
+): void {
+  onClientReference = callback
+}
+
 initialize()
 
 function initialize(): void {
@@ -38,6 +62,17 @@ function initialize(): void {
         return wrapResourceProxy(mod, deps)
       }
     },
+    // Called EVERY time a module is requested (not memoized).
+    // Notify framework callback for per-request asset collection.
+    onLoad: (id) => {
+      if (!import.meta.env.__vite_rsc_build__) return
+      if (onClientReference) {
+        const deps = assetsManifest.clientReferenceDeps[id]
+        if (deps) {
+          onClientReference({ id, deps: { js: deps.js, css: deps.css } })
+        }
+      }
+    },
   })
 }