@forklaunch/ws 0.1.0 → 0.1.2

package/lib/index.d.mts

@@ -554,11 +554,13 @@ declare class ForklaunchWebSocket<SV extends AnySchemaValidator, const ES extend
  * @see {@link EventSchema} for schema structure
  */
 declare class ForklaunchWebSocketServer<SV extends AnySchemaValidator, const ES extends EventSchema<SV>> extends WebSocketServer {
+    private readonly schemaValidator;
+    private readonly eventSchemas;
     /**
      * Creates a new ForklaunchWebSocketServer with schema validation.
      *
-     * @param _schemaValidator - The schema validator instance (e.g., ZodSchemaValidator)
-     * @param _eventSchemas - Schema definitions for all WebSocket events
+     * @param schemaValidator - The schema validator instance (e.g., ZodSchemaValidator)
+     * @param eventSchemas - Schema definitions for all WebSocket events
      * @param options - Standard WebSocketServer options (port, host, server, etc.)
      * @param callback - Optional callback invoked when the server starts listening
      *
@@ -601,7 +603,18 @@ declare class ForklaunchWebSocketServer<SV extends AnySchemaValidator, const ES
      * });
      * ```
      */
-    constructor(_schemaValidator: SV, _eventSchemas: ES, options?: ConstructorParameters<typeof WebSocketServer>[0], callback?: () => void);
+    constructor(schemaValidator: SV, eventSchemas: ES, options?: ConstructorParameters<typeof WebSocketServer>[0], callback?: () => void);
+    /**
+     * Enhances a plain WebSocket instance with ForklaunchWebSocket functionality.
+     *
+     * This method:
+     * 1. Changes the prototype to ForklaunchWebSocket.prototype to inject all methods
+     * 2. Initializes all required properties (schemaValidator, eventSchemas, schemas)
+     * 3. Ensures methods like on(), send(), close() etc. work with validation
+     *
+     * @internal
+     */
+    private enhanceWebSocket;
     /**
      * Registers an event listener for server events.
      *

package/lib/index.d.ts

@@ -554,11 +554,13 @@ declare class ForklaunchWebSocket<SV extends AnySchemaValidator, const ES extend
  * @see {@link EventSchema} for schema structure
  */
 declare class ForklaunchWebSocketServer<SV extends AnySchemaValidator, const ES extends EventSchema<SV>> extends WebSocketServer {
+    private readonly schemaValidator;
+    private readonly eventSchemas;
     /**
      * Creates a new ForklaunchWebSocketServer with schema validation.
      *
-     * @param _schemaValidator - The schema validator instance (e.g., ZodSchemaValidator)
-     * @param _eventSchemas - Schema definitions for all WebSocket events
+     * @param schemaValidator - The schema validator instance (e.g., ZodSchemaValidator)
+     * @param eventSchemas - Schema definitions for all WebSocket events
      * @param options - Standard WebSocketServer options (port, host, server, etc.)
      * @param callback - Optional callback invoked when the server starts listening
      *
@@ -601,7 +603,18 @@ declare class ForklaunchWebSocketServer<SV extends AnySchemaValidator, const ES
      * });
      * ```
      */
-    constructor(_schemaValidator: SV, _eventSchemas: ES, options?: ConstructorParameters<typeof WebSocketServer>[0], callback?: () => void);
+    constructor(schemaValidator: SV, eventSchemas: ES, options?: ConstructorParameters<typeof WebSocketServer>[0], callback?: () => void);
+    /**
+     * Enhances a plain WebSocket instance with ForklaunchWebSocket functionality.
+     *
+     * This method:
+     * 1. Changes the prototype to ForklaunchWebSocket.prototype to inject all methods
+     * 2. Initializes all required properties (schemaValidator, eventSchemas, schemas)
+     * 3. Ensures methods like on(), send(), close() etc. work with validation
+     *
+     * @internal
+     */
+    private enhanceWebSocket;
     /**
      * Registers an event listener for server events.
      *

package/lib/index.js

@@ -27,8 +27,8 @@ __export(index_exports, {
   ForklaunchWebSocket: () => ForklaunchWebSocket,
   ForklaunchWebSocketServer: () => ForklaunchWebSocketServer,
   OPEN: () => OPEN,
-  WebSocket: () => import_ws4.WebSocket,
-  WebSocketServer: () => import_ws4.WebSocketServer,
+  WebSocket: () => import_ws5.WebSocket,
+  WebSocketServer: () => import_ws5.WebSocketServer,
   default: () => ForklaunchWebSocket
 });
 module.exports = __toCommonJS(index_exports);
@@ -427,13 +427,16 @@ var ForklaunchWebSocket = class extends import_ws2.WebSocket {
 };
 
 // src/webSocketServer.ts
-var import_ws3 = require("ws");
-var ForklaunchWebSocketServer = class extends import_ws3.WebSocketServer {
+var import_ws3 = require("@forklaunch/core/ws");
+var import_ws4 = require("ws");
+var ForklaunchWebSocketServer = class extends import_ws4.WebSocketServer {
+  schemaValidator;
+  eventSchemas;
   /**
    * Creates a new ForklaunchWebSocketServer with schema validation.
    *
-   * @param _schemaValidator - The schema validator instance (e.g., ZodSchemaValidator)
-   * @param _eventSchemas - Schema definitions for all WebSocket events
+   * @param schemaValidator - The schema validator instance (e.g., ZodSchemaValidator)
+   * @param eventSchemas - Schema definitions for all WebSocket events
    * @param options - Standard WebSocketServer options (port, host, server, etc.)
    * @param callback - Optional callback invoked when the server starts listening
    *
@@ -476,8 +479,40 @@ var ForklaunchWebSocketServer = class extends import_ws3.WebSocketServer {
    * });
    * ```
    */
-  constructor(_schemaValidator, _eventSchemas, options, callback) {
+  constructor(schemaValidator, eventSchemas, options, callback) {
     super(options, callback);
+    this.schemaValidator = schemaValidator;
+    this.eventSchemas = eventSchemas;
+    super.on("connection", (ws, request) => {
+      const forklaunchWs = this.enhanceWebSocket(ws);
+      this.emit("connection", forklaunchWs, request);
+    });
+  }
+  /**
+   * Enhances a plain WebSocket instance with ForklaunchWebSocket functionality.
+   *
+   * This method:
+   * 1. Changes the prototype to ForklaunchWebSocket.prototype to inject all methods
+   * 2. Initializes all required properties (schemaValidator, eventSchemas, schemas)
+   * 3. Ensures methods like on(), send(), close() etc. work with validation
+   *
+   * @internal
+   */
+  enhanceWebSocket(ws) {
+    Object.setPrototypeOf(ws, ForklaunchWebSocket.prototype);
+    const enhancedWs = ws;
+    enhancedWs.schemaValidator = this.schemaValidator;
+    enhancedWs.eventSchemas = this.eventSchemas;
+    const serverEventSchemas = {
+      ...this.eventSchemas,
+      clientMessages: this.eventSchemas.serverMessages,
+      serverMessages: this.eventSchemas.clientMessages
+    };
+    enhancedWs.schemas = (0, import_ws3.createWebSocketSchemas)(
+      this.schemaValidator,
+      serverEventSchemas
+    );
+    return enhancedWs;
   }
   on(event, listener) {
     return super.on(event, listener);
@@ -500,13 +535,13 @@ var ForklaunchWebSocketServer = class extends import_ws3.WebSocketServer {
 };
 
 // index.ts
-var import_ws4 = require("ws");
 var import_ws5 = require("ws");
+var import_ws6 = require("ws");
 var import_events = require("events");
-var CLOSED = import_ws5.WebSocket.CLOSED;
-var CLOSING = import_ws5.WebSocket.CLOSING;
-var CONNECTING = import_ws5.WebSocket.CONNECTING;
-var OPEN = import_ws5.WebSocket.OPEN;
+var CLOSED = import_ws6.WebSocket.CLOSED;
+var CLOSING = import_ws6.WebSocket.CLOSING;
+var CONNECTING = import_ws6.WebSocket.CONNECTING;
+var OPEN = import_ws6.WebSocket.OPEN;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   CLOSED,

package/lib/index.mjs

@@ -397,13 +397,18 @@ var ForklaunchWebSocket = class extends WebSocket {
 };
 
 // src/webSocketServer.ts
+import {
+  createWebSocketSchemas as createWebSocketSchemas2
+} from "@forklaunch/core/ws";
 import { WebSocketServer } from "ws";
 var ForklaunchWebSocketServer = class extends WebSocketServer {
+  schemaValidator;
+  eventSchemas;
   /**
    * Creates a new ForklaunchWebSocketServer with schema validation.
    *
-   * @param _schemaValidator - The schema validator instance (e.g., ZodSchemaValidator)
-   * @param _eventSchemas - Schema definitions for all WebSocket events
+   * @param schemaValidator - The schema validator instance (e.g., ZodSchemaValidator)
+   * @param eventSchemas - Schema definitions for all WebSocket events
    * @param options - Standard WebSocketServer options (port, host, server, etc.)
    * @param callback - Optional callback invoked when the server starts listening
    *
@@ -446,8 +451,40 @@ var ForklaunchWebSocketServer = class extends WebSocketServer {
    * });
    * ```
    */
-  constructor(_schemaValidator, _eventSchemas, options, callback) {
+  constructor(schemaValidator, eventSchemas, options, callback) {
     super(options, callback);
+    this.schemaValidator = schemaValidator;
+    this.eventSchemas = eventSchemas;
+    super.on("connection", (ws, request) => {
+      const forklaunchWs = this.enhanceWebSocket(ws);
+      this.emit("connection", forklaunchWs, request);
+    });
+  }
+  /**
+   * Enhances a plain WebSocket instance with ForklaunchWebSocket functionality.
+   *
+   * This method:
+   * 1. Changes the prototype to ForklaunchWebSocket.prototype to inject all methods
+   * 2. Initializes all required properties (schemaValidator, eventSchemas, schemas)
+   * 3. Ensures methods like on(), send(), close() etc. work with validation
+   *
+   * @internal
+   */
+  enhanceWebSocket(ws) {
+    Object.setPrototypeOf(ws, ForklaunchWebSocket.prototype);
+    const enhancedWs = ws;
+    enhancedWs.schemaValidator = this.schemaValidator;
+    enhancedWs.eventSchemas = this.eventSchemas;
+    const serverEventSchemas = {
+      ...this.eventSchemas,
+      clientMessages: this.eventSchemas.serverMessages,
+      serverMessages: this.eventSchemas.clientMessages
+    };
+    enhancedWs.schemas = createWebSocketSchemas2(
+      this.schemaValidator,
+      serverEventSchemas
+    );
+    return enhancedWs;
   }
   on(event, listener) {
     return super.on(event, listener);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@forklaunch/ws",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Typed framework for ws, by ForkLaunch.",
   "homepage": "https://github.com/forklaunch/forklaunch-js#readme",
   "bugs": {
@@ -29,21 +29,21 @@
     "@forklaunch/fastmcp-fork": "^1.0.5",
     "@types/ws": "^8.18.1",
     "ws": "^8.18.3",
-    "@forklaunch/common": "0.6.22",
-    "@forklaunch/core": "0.16.0",
-    "@forklaunch/validator": "0.10.22"
+    "@forklaunch/common": "0.6.23",
+    "@forklaunch/validator": "0.10.23",
+    "@forklaunch/core": "0.16.1"
   },
   "devDependencies": {
-    "@eslint/js": "^9.38.0",
-    "@typescript/native-preview": "7.0.0-dev.20251027.1",
+    "@eslint/js": "^9.39.1",
+    "@typescript/native-preview": "7.0.0-dev.20251117.1",
     "jest": "^30.2.0",
     "prettier": "^3.6.2",
     "ts-jest": "^29.4.5",
     "ts-node": "^10.9.2",
-    "tsup": "^8.5.0",
+    "tsup": "^8.5.1",
     "typedoc": "^0.28.14",
     "typescript": "^5.9.3",
-    "typescript-eslint": "^8.46.2",
+    "typescript-eslint": "^8.46.4",
     "zod": "^4.1.12"
   },
   "scripts": {