Use decorateReply (fixes #1)

fastify · Sep 10, 2017 · 4baddb2 · 4baddb2
1 parent c868c89
commit 4baddb2
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/plugin.js b/plugin.js
@@ -4,6 +4,12 @@ const fp = require('fastify-plugin')
 const cookie = require('cookie')
 
 function plugin (fastify, options, next) {
+  fastify.decorateReply('setCookie', function (name, value, options) {
+    const seriaized = cookie.serialize(name, value, options || {})
+    this.header('Set-Cookie', seriaized)
+    return this
+  })
+
   fastify.addHook('preHandler', (fastifyReq, fastifyRes, done) => {
     if (fastifyReq.cookies) done()
     if (!fastifyReq.req.headers) fastifyReq.req.headers = {}
@@ -12,12 +18,6 @@ function plugin (fastify, options, next) {
     const cookies = (cookieHeader) ? cookie.parse(cookieHeader) : []
     fastifyReq.cookies = cookies
 
-    fastifyRes.setCookie = function (name, value, options) {
-      const seriaized = cookie.serialize(name, value, options || {})
-      this.header('Set-Cookie', seriaized)
-      return this
-    }
-
     done()
   })