only inject tail on started

Author: JounQin

package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-server-renderer",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "simple React SSR solution inspired by vue-server-render",
   "repository": "git@github.com:JounQin/react-server-render.git",
   "main": "lib/index.js",
@@ -27,12 +27,12 @@
   },
   "peerDependencies": {
     "react": "^16.2.0",
-    "react-dom": "^16.2.0",
-    "serialize-javascript": "^1.4.0"
+    "react-dom": "^16.2.0"
   },
   "dependencies": {
     "lodash": "^4.17.4",
     "resolve": "^1.5.0",
+    "serialize-javascript": "^1.4.0",
     "source-map": "^0.6.1"
   },
   "devDependencies": {
@@ -42,7 +42,6 @@
     "prettier": "^1.9.2",
     "react": "^16.2.0",
     "react-dom": "^16.2.0",
-    "serialize-javascript": "^1.4.0",
     "tslint": "^5.8.0",
     "tslint-config-prettier": "^1.6.0",
     "tslint-plugin-prettier": "^1.3.0",

src/bundle-renderer/create-bundle-renderer.ts

@@ -144,8 +144,8 @@ export function createBundleRendererCreator(
               if (rendererOptions && rendererOptions.template) {
                 renderStream
                   .on('beforeStart', () => res.emit('beforeStart'))
-                  .on('beforeEnd', () => res.emit('beforeEnd'))
-                  .on('redirect', url => res.emit('redirect', url))
+                  .on('beforeEnd', (...args) => res.emit('beforeEnd', ...args))
+                  .on('redirect', (...args) => res.emit('redirect', ...args))
               }
 
               renderStream.pipe(res)

src/template-renderer/template-stream.ts

@@ -58,15 +58,24 @@ export default class TemplateStream extends Transform {
       }
     }
 
-    this.push(this.template.neck(this.context) + '<div id="app">')
+    this.push(this.template.neck(this.context))
+
+    if (this.inject) {
+      this.push('<div id="app">')
+    }
   }
 
   _flush(done: () => void) {
-    this.emit('beforeEnd')
+    this.emit('beforeEnd', this.started)
 
-    this.push('</div>')
+    if (!this.started) {
+      done()
+      return
+    }
 
     if (this.inject) {
+      this.push('</div>')
+
       const { asyncContext } = this.context
 
       if (asyncContext) {