feat: call render:response hook for defineRenderHandler

unjs · Aug 8, 2022 · 8238f38 · 8238f38
1 parent e0ec2b2
commit 8238f38
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/runtime/app.ts b/src/runtime/app.ts
@@ -14,6 +14,7 @@ import { handlers } from '#internal/nitro/virtual/server-handlers'
 export interface NitroApp {
   h3App: H3App
   router: Router
+  // TODO: Type hooks and allow extending
   hooks: Hookable
   localCall: ReturnType<typeof createCall>
   localFetch: ReturnType<typeof createLocalFetch>

diff --git a/src/runtime/renderer.ts b/src/runtime/renderer.ts
@@ -1,4 +1,5 @@
 import { CompatibilityEvent, eventHandler } from 'h3'
+import { useNitroApp } from './app'
 
 export interface RenderResponse {
   body: string,
@@ -20,6 +21,10 @@ export function defineRenderHandler (handler: RenderHandler) {
 
     const response = await handler(event)
 
+    // Allow hooking and modifying response
+    const nitroApp = useNitroApp()
+    await nitroApp.hooks.callHook('render:response', response, { event })
+
     // TODO: Warn if response is already handled
 
     // TODO: Caching support