@tsed/graphql-ws 8.0.1 → 8.0.2

package/package.json

@@ -2,13 +2,14 @@
   "name": "@tsed/graphql-ws",
   "description": "An extension to allow GraphQL over WebSockets using graphql-ws protocol",
   "type": "module",
-  "version": "8.0.1",
+  "version": "8.0.2",
   "source": "./src/index.ts",
   "main": "./lib/esm/index.js",
   "module": "./lib/esm/index.js",
   "typings": "./lib/types/index.d.ts",
   "exports": {
     ".": {
+      "@tsed/source": "./src/index.ts",
       "types": "./lib/types/index.d.ts",
       "import": "./lib/esm/index.js",
       "default": "./lib/esm/index.js"
@@ -23,21 +24,21 @@
     "tslib": "2.7.0"
   },
   "devDependencies": {
-    "@tsed/barrels": "8.0.1",
-    "@tsed/core": "8.0.1",
-    "@tsed/di": "8.0.1",
+    "@tsed/barrels": "8.0.2",
+    "@tsed/core": "8.0.2",
+    "@tsed/di": "8.0.2",
     "@tsed/logger": "^6.7.8",
-    "@tsed/platform-http": "8.0.1",
-    "@tsed/typescript": "8.0.1",
+    "@tsed/platform-http": "8.0.2",
+    "@tsed/typescript": "8.0.2",
     "eslint": "9.12.0",
     "graphql-ws": "5.16.0",
     "typescript": "5.4.5"
   },
   "peerDependencies": {
-    "@tsed/core": "8.0.1",
-    "@tsed/di": "8.0.1",
+    "@tsed/core": "8.0.2",
+    "@tsed/di": "8.0.2",
     "@tsed/logger": ">=6.7.5",
-    "@tsed/platform-http": "8.0.1",
+    "@tsed/platform-http": "8.0.2",
     "graphql-ws": ">=5.14.2"
   },
   "repository": "https://github.com/tsedio/tsed",

package/src/GraphQLWSModule.ts

@@ -0,0 +1,59 @@
+import {Constant, Inject, InjectorService, Module} from "@tsed/di";
+import type {Disposable} from "graphql-ws";
+import {useServer} from "graphql-ws/lib/use/ws";
+import Http from "http";
+import Https from "https";
+
+import {GraphQLWSOptions} from "./GraphQLWSOptions.js";
+
+@Module()
+export class GraphQLWSModule {
+  @Constant("graphqlWs", {})
+  private settings: GraphQLWSOptions;
+
+  @Inject(Http.Server)
+  private httpServer: Http.Server | null;
+
+  @Inject(Https.Server)
+  private httpsServer: Https.Server | null;
+
+  @Inject()
+  private injector: InjectorService;
+
+  async createWSServer(settings: GraphQLWSOptions) {
+    // @ts-ignore
+    const {WebSocketServer} = await import("ws");
+
+    const wsServer = new WebSocketServer({
+      ...(this.settings.wsServerOptions || {}),
+      ...settings.wsServerOptions,
+      server: this.httpsServer || this.httpServer!,
+      path: settings.path
+    });
+
+    return useServer(
+      {
+        ...(this.settings.wsUseServerOptions || {}),
+        ...settings.wsUseServerOptions,
+        schema: settings.schema
+      },
+      wsServer
+    );
+  }
+
+  async $alterApolloServerPlugins(plugins: any[], settings: GraphQLWSOptions) {
+    const wsServer = await this.createWSServer(settings);
+
+    this.injector.logger.info(`Create GraphQL WS server on: ${settings.path}`);
+
+    return plugins.concat({
+      serverWillStart() {
+        return {
+          async drainServer() {
+            await wsServer.dispose();
+          }
+        };
+      }
+    } as any);
+  }
+}

package/src/GraphQLWSOptions.ts

@@ -0,0 +1,19 @@
+// @ts-ignore
+import {ServerOptions} from "graphql-ws/lib/server";
+import {ServerOptions as WSServerOptions} from "ws";
+
+export interface GraphQLWSOptions {
+  path: string;
+  schema: any;
+  wsUseServerOptions?: ServerOptions;
+  wsServerOptions?: WSServerOptions;
+}
+
+declare global {
+  namespace TsED {
+    interface ApolloSettings {
+      wsUseServerOptions?: ServerOptions;
+      wsServerOptions?: WSServerOptions;
+    }
+  }
+}

package/src/index.ts

@@ -0,0 +1,5 @@
+/**
+ * @file Automatically generated by @tsed/barrels.
+ */
+export * from "./GraphQLWSModule.js";
+export * from "./GraphQLWSOptions.js";