Support response schema update

index.js

@@ -30,9 +30,10 @@ function plugin (fastify, options, callback) {
       body: true,
       headers: false,
       params: false,
-      query: true
+      query: false,
+      response: false
     },
-    options
+    options || {}
   )
 
   fastify.addHook('onRoute', route => {
@@ -47,7 +48,15 @@ function plugin (fastify, options, callback) {
         route.schema.params = updateSchema(route.schema.params.valueOf())
       }
       if (route.schema.querystring && options.query) {
-        route.schema.querystring = updateSchema(route.schema.querystring.valueOf())
+        route.schema.querystring = updateSchema(
+          route.schema.querystring.valueOf()
+        )
+      }
+      if (route.schema.response) {
+        route.schema.response = mapValues(
+          route.schema.response,
+          schema => updateSchema(schema.valueOf())
+        )
       }
     }
   })

tests/default.js

@@ -14,14 +14,17 @@ function buildFastify () {
     body: true,
     headers: true,
     params: true,
-    query: true
+    query: true,
+    response: true
   })
 
   app.route({
     method: 'POST',
     url: '/foo/:a/:b',
     handler (request, reply) {
       reply.send({
+        a: '0',
+        b: '1',
         body: request.body,
         headers: request.headers,
         params: request.params,
@@ -32,15 +35,23 @@ function buildFastify () {
       body: schema,
       headers: schema,
       params: schema,
-      querystring: schema
+      querystring: schema,
+      response: {
+        200: S.object()
+          .prop('body')
+          .prop('headers')
+          .prop('params')
+          .prop('query')
+          .extend(schema)
+      }
     }
   })
 
   return app
 }
 
 test('default', t => {
-  t.plan(10)
+  t.plan(12)
 
   const app = buildFastify()
 
@@ -74,5 +85,8 @@ test('default', t => {
 
     t.strictEqual(data.query.a, 0)
     t.strictEqual(data.query.b, undefined)
+
+    t.strictEqual(data.a, 0)
+    t.strictEqual(data.b, undefined)
   })
 })