fix: missing register component in ssr (#1887)

Author: mog422

package.json

@@ -38,6 +38,7 @@
   "packageManager": "[email protected]",
   "dependencies": {
     "chalk": "^4.1.0",
+    "hash-sum": "^2.0.0",
     "watchpack": "^2.4.0"
   },
   "peerDependencies": {
@@ -57,6 +58,7 @@
     "@intlify/vue-i18n-loader": "^3.0.0",
     "@types/cssesc": "^3.0.2",
     "@types/estree": "^0.0.45",
+    "@types/hash-sum": "^1.0.2",
     "@types/jest": "^26.0.13",
     "@types/jsdom": "^16.2.13",
     "@types/mini-css-extract-plugin": "^0.9.1",

src/index.ts

@@ -11,6 +11,9 @@ import type {
   SFCTemplateCompileOptions,
   SFCScriptCompileOptions,
 } from 'vue/compiler-sfc'
+
+import hashSum from 'hash-sum'
+
 import { selectBlock } from './select'
 import { genHotReloadCode } from './hotReload'
 import { genCSSModulesCode } from './cssModules'
@@ -363,6 +366,18 @@ export default function loader(
         .join(`\n`) + `\n`
   }
 
+  if (isServer) {
+    code += `\nimport { 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(
+        hashSum(loaderContext.request)
+      )});`)
+    code += `  return _setup ? _setup(props, ctx) : undefined`
+    code += `}\n`
+  }
+
   // finalize
   if (!propsToAttach.length) {
     code += `\n\nconst __exports__ = script;`