fix: missing register component in ssr (vuejs#1887)

mog422 · Jul 19, 2024 · 3fc2599 · 3fc2599
1 parent 6986365
commit 3fc2599
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -363,6 +363,18 @@ export default function loader(
         .join(`\n`) + `\n`
   }
 
+  if (isServer) {
+    code += `import { useSSRContext } from 'vue'\n`
+    code += `const _setup = script.setup\n`
+    ;(code += `script.setup = (props, ctx) => {`),
+      (code += `  const ssrContext = _useSSRContext()`),
+      (code += `  ;(ssrContext._registeredComponents || (ssrContext._registeredComponents = new Set())).add(${JSON.stringify(
+        hash(loaderContext.request)
+      )});`)
+    code += `  return _setup ? _setup(props, ctx) : undefined`
+    code += `}\n`
+  }
+
   // finalize
   if (!propsToAttach.length) {
     code += `\n\nconst __exports__ = script;`