feat: support dva ssr (#199)

* feat: support dva ssr * feat: dva ssr * chore: umi beta 6 * chore: yarn.lock * fix: chore * chore: yarn.lock * chore: yarn.lock
umijs · May 19, 2020 · 9930e41 · 9930e41
1 parent 70ffc80
commit 9930e41
Show file tree

Hide file tree

Showing 5 changed files with 151 additions and 100 deletions.
diff --git a/package.json b/package.json
@@ -44,7 +44,7 @@
     "react": "^16.12.0",
     "react-dom": "^16.12.0",
     "react-test-renderer": "^16.9.0",
-    "umi": "^3.2.0-beta.6",
+    "umi": "^3.2.0",
     "yorkie": "^2.0.0"
   },
   "gitHooks": {

diff --git a/packages/plugin-dva/src/dva.tpl b/packages/plugin-dva/src/dva.tpl
@@ -7,7 +7,7 @@ import { plugin, history } from '../core/umiExports';
 
 let app:any = null;
 
-function _onCreate() {
+export function _onCreate(options = {}) {
   const runtimeDva = plugin.applyPlugins({
     key: 'dva',
     type: ApplyPluginsType.modify,
@@ -18,7 +18,8 @@ function _onCreate() {
     {{{ ExtendDvaConfig }}}
     ...(runtimeDva.config || {}),
     // @ts-ignore
-    ...(window.g_useSSR ? { initialState: window.g_initialData } : {}),
+    ...(typeof window !== 'undefined' && window.g_useSSR ? { initialState: window.g_initialProps } : {}),
+    ...(options || {}),
   });
   {{{ EnhanceApp }}}
   app.use(createLoading());
@@ -37,7 +38,10 @@ export function getApp() {
 export class _DvaContainer extends Component {
   constructor(props: any) {
     super(props);
-    _onCreate();
+    // run only in client, avoid override server _onCreate()
+    if (typeof window !== 'undefined') {
+      _onCreate();
+    }
   }
 
   componentWillUnmount() {

diff --git a/packages/plugin-dva/src/index.ts b/packages/plugin-dva/src/index.ts
@@ -112,7 +112,9 @@ app.model({ namespace: '${basename(path, extname(path))}', ...(require('${path}'
       const runtimeTpl = readFileSync(join(__dirname, 'runtime.tpl'), 'utf-8');
       api.writeTmpFile({
         path: 'plugin-dva/runtime.tsx',
-        content: Mustache.render(runtimeTpl, {}),
+        content: Mustache.render(runtimeTpl, {
+          SSR: !!api.config?.ssr,
+        }),
       });
 
       // exports.ts

diff --git a/packages/plugin-dva/src/runtime.tpl b/packages/plugin-dva/src/runtime.tpl
@@ -1,6 +1,25 @@
 import React from 'react';
-import { _DvaContainer } from './dva';
+import { _DvaContainer, getApp, _onCreate } from './dva';
 
 export function rootContainer(container) {
   return React.createElement(_DvaContainer, null, container);
 }
+
+{{#SSR}}
+export const ssr = {
+  modifyGetInitialPropsCtx: async (ctx) => {
+    // 服务端执行早于 constructor 中的 onCreate
+    if (process.env.__IS_SERVER && ctx.history) {
+      const tmpApp = _onCreate({
+        // server
+        history: ctx.history,
+      })
+      tmpApp.router(() => {})
+      tmpApp.start();
+    }
+    // 一定有 app
+    const { _store } = getApp();
+    ctx.store = _store;
+  },
+}
+{{/SSR}}