refactor(plugin-rsc): move onClientReference notification into wrapResourceProxy

Consolidate client reference notification by moving it into the resource
proxy getter, which is called on every export access (not memoized by React).
This eliminates the separate `onLoad` callback mechanism.

- Remove `onLoad` option from `setRequireModule`
- Add `id` parameter to `wrapResourceProxy` and make `deps` required
- Notify `onClientReference` early in `load` (production) alongside preload
- Notify `onClientReference` in proxy getter for per-request asset collection

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: hi-ogawa

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

@@ -6,25 +6,14 @@ 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
 
   const requireModule = memoize((id: string) => {
     return options.load(removeReferenceCacheTag(id))
   })
-
-  const clientRequire = (id: string) => {
-    const cleanId = removeReferenceCacheTag(id)
-    options.onLoad?.(cleanId)
-    return requireModule(id)
-  }
-  ;(globalThis as any).__vite_rsc_client_require__ = clientRequire
+  ;(globalThis as any).__vite_rsc_client_require__ = requireModule
 
   setInternalRequire()
 }

packages/plugin-rsc/src/ssr.tsx

@@ -47,43 +47,33 @@ function initialize(): void {
         const modCss = await import(
           /* @vite-ignore */ '/@id/__x00__' + toCssVirtual({ id, type: 'ssr' })
         )
-        return wrapResourceProxy(mod, { js: [], css: modCss.default })
+        return wrapResourceProxy(mod, id, { js: [], css: modCss.default })
       } else {
         const import_ = clientReferences.default[id]
         if (!import_) {
           throw new Error(`client reference not found '${id}'`)
         }
-        const deps = assetsManifest.clientReferenceDeps[id]
-        // kick off preload before initial async import, which is not sync-cached
-        if (deps) {
-          preloadDeps(deps)
-        }
+        const deps =
+          assetsManifest.clientReferenceDeps[id] ??
+          ({ js: [], css: [] } as const)
+        // kick off preload/notify before initial async import, which is not sync-cached
+        preloadDeps(deps)
+        onClientReference?.({ id, deps })
         const mod: any = await import_()
-        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 } })
-        }
+        return wrapResourceProxy(mod, id, deps)
       }
     },
   })
 }
 
-// preload/preinit during getter access since `load` is cached on production
-function wrapResourceProxy(mod: any, deps?: ResolvedAssetDeps) {
+// preload/preinit during getter access since `load` is cached on production.
+// also notify `onClientReference` callback here since module export access is not memoized by React.
+function wrapResourceProxy(mod: any, id: string, deps: ResolvedAssetDeps) {
   return new Proxy(mod, {
     get(target, p, receiver) {
       if (p in mod) {
-        if (deps) {
-          preloadDeps(deps)
-        }
+        preloadDeps(deps)
+        onClientReference?.({ id, deps })
       }
       return Reflect.get(target, p, receiver)
     },