feature: pass route params to handlers (#29)

Author: Frando

README.md

@@ -78,6 +78,14 @@ fastify.listen(3000, err => {
 
 In this case there won't be any global handler, so it will respond with a 404 error on every unregistered route, closing the incoming upgrade connection requests.
 
+The route handler receives route params as a third argument:
+
+```js
+fastify.get('/ws/:id', { websocket: true }, (connection, req, params) => {
+  connection.write(`hi on stream ${params.id}`)
+})
+```
+
 However you can still pass a default global handler, that will be used as default handler.
 
 ```js

index.js

@@ -46,8 +46,8 @@ function fastifyWebsocket (fastify, opts, next) {
         throw new Error('invalid wsHandler function')
       }
 
-      router.on('GET', routeOptions.path, (req, _) => {
-        const result = wsHandler(req[kWs], req)
+      router.on('GET', routeOptions.path, (req, _, params) => {
+        const result = wsHandler(req[kWs], req, params)
 
         if (result && typeof result.catch === 'function') {
           result.catch((err) => req[kWs].destroy(err))

test/router.js

@@ -286,3 +286,48 @@ test(`Should return 404 on http request`, t => {
     t.end()
   })
 })
+
+test('Should pass route params to handlers', t => {
+  const fastify = Fastify()
+
+  t.tearDown(() => fastify.close())
+
+  fastify.register(fastifyWebsocket)
+  fastify.get('/ws', { websocket: true }, (conn, req, params) => {
+    t.equal(Object.keys(params).length, 0, 'params are empty')
+    conn.write('empty')
+    conn.end()
+  })
+  fastify.get('/ws/:id', { websocket: true }, (conn, req, params) => {
+    t.equal(params.id, 'foo', 'params are correct')
+    conn.write(params.id)
+    conn.end()
+  })
+
+  fastify.listen(0, err => {
+    let pending = 2
+    t.error(err)
+    const client = websocket(
+      'ws://localhost:' + (fastify.server.address()).port + '/ws/foo'
+    )
+    const client2 = websocket(
+      'ws://localhost:' + (fastify.server.address()).port + '/ws'
+    )
+    t.tearDown(client.destroy.bind(client))
+    t.tearDown(client2.destroy.bind(client))
+
+    client.setEncoding('utf8')
+    client2.setEncoding('utf8')
+
+    client.once('data', chunk => {
+      t.equal(chunk, 'foo')
+      client.end()
+      if (--pending === 0) t.end()
+    })
+    client2.once('data', chunk => {
+      t.equal(chunk, 'empty')
+      client2.end()
+      if (--pending === 0) t.end()
+    })
+  })
+})