Update to @fastify/websocket v10 (#1089)

* Update to @fastify/websocket v10

Signed-off-by: Matteo Collina <[email protected]>

* drop node 16

Signed-off-by: Matteo Collina <[email protected]>

* fixup

Signed-off-by: Matteo Collina <[email protected]>

* Update index.d.ts

Co-authored-by: Jonny Green <[email protected]>

* double test timeout

Signed-off-by: Matteo Collina <[email protected]>

---------

Signed-off-by: Matteo Collina <[email protected]>
Co-authored-by: Jonny Green <[email protected]>

Author: mcollina

.taprc

@@ -2,4 +2,4 @@ ts: false
 jsx: false
 flow: false
 coverage: true
-timeout: 60
+timeout: 120

index.d.ts

@@ -16,7 +16,7 @@ import {
   FormattedExecutionResult,
   ParseOptions,
 } from "graphql";
-import { SocketStream } from "@fastify/websocket"
+import type { WebSocket } from "ws";
 import { IncomingMessage, OutgoingHttpHeaders } from "http";
 import { Readable } from "stream";
 
@@ -458,7 +458,7 @@ declare namespace mercurius {
             ) => void
           ) => void;
           context?: (
-            connection: SocketStream,
+            socket: WebSocket,
             request: FastifyRequest
           ) => Record<string, any> | Promise<Record<string, any>>;
           onConnect?: (data: {

lib/subscription.js

@@ -7,9 +7,7 @@ const SubscriptionConnection = require('./subscription-connection')
 const { getProtocolByName } = require('./subscription-protocol')
 
 function createConnectionHandler ({ subscriber, fastify, onConnect, onDisconnect, entityResolversFactory, subscriptionContextFn, keepAlive, fullWsTransport, errorFormatter }) {
-  return async (connection, request) => {
-    const { socket } = connection
-
+  return async (socket, request) => {
     if (socket.protocol === undefined || getProtocolByName(socket.protocol) === null) {
       request.log.warn('wrong websocket protocol')
       // Close the connection with an error code, ws v2 ensures that the
@@ -35,7 +33,7 @@ function createConnectionHandler ({ subscriber, fastify, onConnect, onDisconnect
     let resolveContext
 
     if (subscriptionContextFn) {
-      resolveContext = () => subscriptionContextFn(connection, request)
+      resolveContext = () => subscriptionContextFn(socket, request)
     }
 
     const subscriptionConnection = new SubscriptionConnection(socket, {
@@ -52,10 +50,10 @@ function createConnectionHandler ({ subscriber, fastify, onConnect, onDisconnect
     })
 
     /* istanbul ignore next */
-    connection.socket.on('error', () => {
+    socket.on('error', () => {
       subscriptionConnection.close()
     })
-    connection.socket.on('close', () => {
+    socket.on('close', () => {
       subscriptionConnection.close()
     })
   }

package.json

@@ -60,7 +60,7 @@
   "dependencies": {
     "@fastify/error": "^3.0.0",
     "@fastify/static": "^7.0.0",
-    "@fastify/websocket": "^8.0.0",
+    "@fastify/websocket": "^10.0.0",
     "fastify-plugin": "^4.2.0",
     "graphql-jit": "0.8.4",
     "mqemitter": "^5.0.0",

test/subscription.js

@@ -1968,9 +1968,9 @@ test('subscription server works with fastify websocket', t => {
   })
 
   app.register(async function (app) {
-    app.get('/fastify-websocket', { websocket: true }, (connection, req) => {
-      connection.socket.on('message', message => {
-        connection.socket.send('hi from server')
+    app.get('/fastify-websocket', { websocket: true }, (socket, req) => {
+      socket.on('message', message => {
+        socket.send('hi from server')
       })
     })
   })

test/types/index.ts

@@ -293,8 +293,8 @@ app.register(mercurius, {
       next(true)
       next(false)
     },
-    context: (connection, request) => {
-      connection.socket
+    context: (socket, request) => {
+      socket.send('hello')
       request.headers
       return {}
     },
@@ -319,8 +319,8 @@ app.register(mercurius, {
   schema,
   resolvers,
   subscription: {
-    context: async (connection, request) => {
-      connection.socket
+    context: async (socket, request) => {
+      socket.send('hello')
       request.headers
       return {}
     },