feat(vue-app): context.beforeSerialize method

packages/types/app/index.d.ts

@@ -72,6 +72,7 @@ export interface Context {
     redirected: boolean
     next: NextFunction
     beforeRenderFns: Array<() => any>
+    afterRenderFns: Array<() => any>
     fetchCounters: Record<string, number>
     nuxt: {
       layout: string

packages/vue-app/template/index.js

@@ -183,6 +183,7 @@ async function createApp(ssrContext, config = {}) {
     req: ssrContext ? ssrContext.req : undefined,
     res: ssrContext ? ssrContext.res : undefined,
     beforeRenderFns: ssrContext ? ssrContext.beforeRenderFns : undefined,
+    afterRenderFns: ssrContext ? ssrContext.afterRenderFns : undefined,
     ssrContext
   })
 

packages/vue-app/template/server.js

@@ -85,6 +85,8 @@ export default async (ssrContext) => {
   ssrContext.next = createNext(ssrContext)
   // Used for beforeNuxtRender({ Components, nuxtState })
   ssrContext.beforeRenderFns = []
+  // Used for afterNuxtRender({ Components, nuxtState })
+  ssrContext.afterRenderFns = []
   // Nuxt object (window.{{globals.context}}, defaults to window.__NUXT__)
   ssrContext.nuxt = { <% if (features.layouts) { %>layout: 'default', <% } %>data: [], <% if (features.fetch) { %>fetch: {}, <% } %>error: null<%= (store ? ', state: null' : '') %>, serverRendered: true, routePath: '' }
   <% if (features.fetch) { %>
@@ -120,16 +122,21 @@ export default async (ssrContext) => {
   const beforeRender = async () => {
     // Call beforeNuxtRender() methods
     await Promise.all(ssrContext.beforeRenderFns.map(fn => promisify(fn, { Components, nuxtState: ssrContext.nuxt })))
-    <% if (store) { %>
+
     ssrContext.rendered = () => {
+      // Call afterNuxtRender() methods
+      ssrContext.afterRenderFns.forEach(fn => fn({ Components, nuxtState: ssrContext.nuxt }))
+
+      <% if (store) { %>
       // Add the state from the vuex store
       ssrContext.nuxt.state = store.state
+      <% } %>
+
       <% if (isFullStatic && store) { %>
       // Stop recording store mutations
       ssrContext.unsetMutationObserver()
       <% } %>
     }
-    <% } %>
   }
 
   const renderErrorPage = async () => {

packages/vue-app/template/utils.js

@@ -259,6 +259,7 @@ export async function setContext (app, context) {
     }
     if (process.server) {
       app.context.beforeNuxtRender = fn => context.beforeRenderFns.push(fn)
+      app.context.afterNuxtRender = fn => context.afterRenderFns.push(fn)
     }
     if (process.client) {
       app.context.nuxtState = window.<%= globals.context %>

test/dev/basic.ssr.test.js

@@ -207,6 +207,13 @@ describe('basic ssr', () => {
     expect(window.__NUXT__.test).toBe(true)
   })
 
+  test('/special-state-after -> check window.__NUXT__.testAfter = true', async () => {
+    const window = await nuxt.server.renderAndGetWindow(url('/special-state-after'))
+    expect(window.document.title).toBe('Nuxt')
+    expect(window.__NUXT__.testBefore).toBeUndefined()
+    expect(window.__NUXT__.testAfter).toBe(true)
+  })
+
   test('/error', async () => {
     await expect(nuxt.server.renderRoute('/error', { req: {}, res: {} }))
       .rejects.toThrow('Error mouahahah')

test/fixtures/basic/pages/special-state-after.vue

@@ -0,0 +1,29 @@
+<template>
+  <div>
+    <h1>Special state in `window.__NUXT__`</h1>
+    <client-only><pre>{{ nuxtState }}</pre></client-only>
+  </div>
+</template>
+
+<script>
+export default {
+  data () {
+    return {
+      nuxtState: null
+    }
+  },
+  fetch () {
+    if (process.server) {
+      this.$root.context.beforeNuxtRender(({ nuxtState }) => {
+        nuxtState.testBefore = true
+      })
+      this.$root.context.afterNuxtRender(({ nuxtState }) => {
+        nuxtState.testAfter = true
+      })
+    }
+  },
+  beforeMount () {
+    this.nuxtState = window.__NUXT__
+  }
+}
+</script>