Apollo GraphQL over WebSocket Subscription Authentication

[KeithGillette]

From the AccountsJS GraphQL transport documentation, we have Queries and Mutations authenticating with our Apollo Client/Server setup over HTTP(S) but Subscriptions are not working. How do we configure AccountsJS with Apollo Client/Server to provide the same Context information on authenticated Subscriptions over graphql-ws WebSocket connections as we have over HTTP(S) connections?

[KeithGillette]

After much perusing of the source code, experimentation, and generally fumbling around, here's what I came up with:

Apollo Client

const webSocketClient = createClient({
	url: `ws${environment.serverSchemeSecure ? 's' : ''}://${environment.serverHost}:${environment.serverPort}${environment.graphQLpath}`,
	lazy: true,
	on: {
		closed: () => {
			this.apollo.client.defaultOptions.watchQuery.pollInterval = 0;
		},
		error: () => {
			this.apollo.client.defaultOptions.watchQuery.pollInterval = DefaultPollingInterval;
		},
	},
	connectionParams: () => {
		return this.authenticationService.accountsClient.getTokens(); // Add authToken to every connection
	},
});

Apollo Server

const webSocketServer = useServer({
		schema: schema,
		context: (webSocketContext, subscribeMessage, executionArguments) => {
			webSocketContext.extra.request.headers = {...webSocketContext.extra.request.headers, authorization: `Bearer ${webSocketContext.connectionParams.accessToken}`}; // Put authToken in shape ContextBuilder expects
			return accountsGraphQL.context({req: webSocketContext.extra.request});
		},
	},