fix(rsc): handle added/removed directive during dev

Author: hi-ogawa

packages/plugin-rsc/e2e/basic.test.ts

@@ -431,6 +431,22 @@ function defineTest(f: Fixture) {
       await page.reload()
       await expect(page.getByText('ok (test-shared)')).toBeVisible()
     })
+
+    test('hmr switch server to client', async ({ page }) => {
+      await page.goto(f.url())
+      await waitForHydration(page)
+
+      const editor = f.createEditor('src/routes/hmr-switch/client.tsx')
+      editor.edit((s) => `"use client";\n` + s)
+    })
+
+    test('hmr switch client to server', async ({ page }) => {
+      await page.goto(f.url())
+      await waitForHydration(page)
+
+      const editor = f.createEditor('src/routes/hmr-switch/client.tsx')
+      editor.edit((s) => s.replace('"use client";', ''))
+    })
   })
 
   test('css @js', async ({ page }) => {

packages/plugin-rsc/examples/basic/src/routes/hmr-switch/client.tsx

@@ -0,0 +1,5 @@
+'use client'
+
+export function TestHmrMixedClient() {
+  return <div>test-hmr-mixed-client</div>
+}

packages/plugin-rsc/examples/basic/src/routes/hmr-switch/server.tsx

@@ -0,0 +1,3 @@
+export function TestHmrMixedServer() {
+  return <div>test-hmr-mixed-server</div>
+}

packages/plugin-rsc/examples/basic/src/routes/root.tsx

@@ -38,6 +38,8 @@ import { TestHmrSharedAtomic } from './hmr-shared/atomic/server'
 import { TestCssQueries } from './css-queries/server'
 import { TestImportMetaGlob } from './import-meta-glob/server'
 import { TestAssetsServer } from './assets/server'
+import { TestHmrMixedServer } from './hmr-switch/server'
+import { TestHmrMixedClient } from './hmr-switch/client'
 
 export function Root(props: { url: URL }) {
   return (
@@ -65,6 +67,8 @@ export function Root(props: { url: URL }) {
         <TestHmrSharedServer />
         <TestHmrSharedClient />
         <TestHmrSharedAtomic />
+        <TestHmrMixedServer />
+        <TestHmrMixedClient />
         <TestTemporaryReference />
         <TestServerActionError />
         <TestReplayConsoleLogs url={props.url} />

packages/plugin-rsc/src/plugin.ts

@@ -1083,6 +1083,7 @@ function vitePluginUseClient(
         })
         if (!result) return
         const { output, exportNames } = result
+        // TODO
         manager.clientReferenceMetaMap[id] = {
           importId,
           referenceKey,