@roj-ai/transport 0.0.2

package/dist/server/connection-manager.js

@@ -0,0 +1,234 @@
+/**
+ * Connection Manager
+ *
+ * Manages multiple WebSocket connections and session subscriptions.
+ */
+// ============================================================================
+// Connection Manager
+// ============================================================================
+/**
+ * Manages multiple server connections and session subscriptions.
+ *
+ * Features:
+ * - Track all active connections
+ * - Manage session subscriptions
+ * - Broadcast to subscribed connections
+ * - Connection statistics
+ */
+export class ConnectionManager {
+    connections = new Map();
+    sessionSubscriptions = new Map(); // sessionId -> connectionIds
+    // ============================================================================
+    // Connection Management
+    // ============================================================================
+    /**
+     * Add a connection.
+     */
+    add(connection) {
+        this.connections.set(connection.getConnectionId(), connection);
+    }
+    /**
+     * Remove a connection.
+     */
+    remove(connectionId) {
+        const connection = this.connections.get(connectionId);
+        if (!connection)
+            return;
+        // Clean up subscriptions
+        for (const sessionId of connection.getSubscribedSessions()) {
+            this.unsubscribeInternal(connectionId, sessionId);
+        }
+        this.connections.delete(connectionId);
+    }
+    /**
+     * Get a connection by ID.
+     */
+    get(connectionId) {
+        return this.connections.get(connectionId);
+    }
+    /**
+     * Check if a connection exists.
+     */
+    has(connectionId) {
+        return this.connections.has(connectionId);
+    }
+    /**
+     * Get all connections.
+     */
+    getAll() {
+        return this.connections.values();
+    }
+    /**
+     * Get all connection IDs.
+     */
+    getAllIds() {
+        return this.connections.keys();
+    }
+    /**
+     * Get connection count.
+     */
+    get size() {
+        return this.connections.size;
+    }
+    // ============================================================================
+    // Session Subscriptions
+    // ============================================================================
+    /**
+     * Subscribe a connection to a session.
+     */
+    subscribe(connectionId, sessionId) {
+        const connection = this.connections.get(connectionId);
+        if (!connection)
+            return false;
+        connection.subscribe(sessionId);
+        let subscribers = this.sessionSubscriptions.get(sessionId);
+        if (!subscribers) {
+            subscribers = new Set();
+            this.sessionSubscriptions.set(sessionId, subscribers);
+        }
+        subscribers.add(connectionId);
+        return true;
+    }
+    /**
+     * Unsubscribe a connection from a session.
+     */
+    unsubscribe(connectionId, sessionId) {
+        const connection = this.connections.get(connectionId);
+        if (!connection)
+            return false;
+        connection.unsubscribe(sessionId);
+        this.unsubscribeInternal(connectionId, sessionId);
+        return true;
+    }
+    /**
+     * Get all subscribers for a session.
+     */
+    getSubscribers(sessionId) {
+        const subscriberIds = this.sessionSubscriptions.get(sessionId);
+        if (!subscriberIds)
+            return [];
+        const subscribers = [];
+        const deadIds = [];
+        for (const connectionId of subscriberIds) {
+            const connection = this.connections.get(connectionId);
+            if (connection?.isConnected()) {
+                subscribers.push(connection);
+            }
+            else {
+                deadIds.push(connectionId);
+            }
+        }
+        // Clean up dead subscriptions
+        for (const id of deadIds) {
+            subscriberIds.delete(id);
+        }
+        if (subscriberIds.size === 0) {
+            this.sessionSubscriptions.delete(sessionId);
+        }
+        return subscribers;
+    }
+    /**
+     * Get subscriber count for a session.
+     */
+    getSubscriberCount(sessionId) {
+        return this.sessionSubscriptions.get(sessionId)?.size ?? 0;
+    }
+    /**
+     * Check if a session has any subscribers.
+     */
+    hasSubscribers(sessionId) {
+        return this.getSubscriberCount(sessionId) > 0;
+    }
+    // ============================================================================
+    // Broadcasting
+    // ============================================================================
+    /**
+     * Broadcast a raw message to all subscribers of a session.
+     */
+    broadcast(sessionId, message) {
+        const subscribers = this.getSubscribers(sessionId);
+        let sent = 0;
+        for (const connection of subscribers) {
+            if (connection.send(message)) {
+                sent++;
+            }
+        }
+        return sent;
+    }
+    /**
+     * Broadcast to all connections.
+     */
+    broadcastToAll(message) {
+        let sent = 0;
+        for (const connection of this.connections.values()) {
+            if (connection.isConnected() && connection.send(message)) {
+                sent++;
+            }
+        }
+        return sent;
+    }
+    // ============================================================================
+    // Statistics
+    // ============================================================================
+    /**
+     * Get connection statistics.
+     */
+    getStats() {
+        const byClientType = {};
+        let activeConnections = 0;
+        for (const connection of this.connections.values()) {
+            const clientType = connection.getClientType();
+            byClientType[clientType] = (byClientType[clientType] ?? 0) + 1;
+            if (connection.isConnected()) {
+                activeConnections++;
+            }
+        }
+        return {
+            totalConnections: this.connections.size,
+            activeConnections,
+            activeSessions: this.sessionSubscriptions.size,
+            byClientType,
+        };
+    }
+    // ============================================================================
+    // Cleanup
+    // ============================================================================
+    /**
+     * Clear all connections.
+     */
+    clear() {
+        for (const connection of this.connections.values()) {
+            connection.disconnect();
+        }
+        this.connections.clear();
+        this.sessionSubscriptions.clear();
+    }
+    /**
+     * Remove disconnected connections.
+     */
+    cleanup() {
+        let removed = 0;
+        for (const [connectionId, connection] of this.connections) {
+            if (!connection.isConnected()) {
+                this.remove(connectionId);
+                removed++;
+            }
+        }
+        return removed;
+    }
+    // ============================================================================
+    // Private Methods
+    // ============================================================================
+    /**
+     * Internal unsubscribe (updates subscription map only).
+     */
+    unsubscribeInternal(connectionId, sessionId) {
+        const subscribers = this.sessionSubscriptions.get(sessionId);
+        if (subscribers) {
+            subscribers.delete(connectionId);
+            if (subscribers.size === 0) {
+                this.sessionSubscriptions.delete(sessionId);
+            }
+        }
+    }
+}

package/dist/server/index.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Server Connection Module
+ *
+ * WebSocket server for handling incoming client connections.
+ */
+export { createServerConnection, ServerConnection } from './server-connection.js';
+export type { ServerConnectionConfig, ServerHandlerContext } from './server-connection.js';
+export { ConnectionManager } from './connection-manager.js';
+export type { ConnectionStats } from './connection-manager.js';
+export type { ConnectionState, HandlerContext, Protocol, ProtocolCaller, ProtocolDef, ProtocolHandlers, ProtocolNotifier, TransportEvent, TransportEventListener, } from '../core/index.js';
+export { defineProtocol, method, notification } from '../core/index.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/server/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/server/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,sBAAsB,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAA;AACjF,YAAY,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAA;AAE1F,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAA;AAC3D,YAAY,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAA;AAG9D,YAAY,EACX,eAAe,EACf,cAAc,EACd,QAAQ,EACR,cAAc,EACd,WAAW,EACX,gBAAgB,EAChB,gBAAgB,EAChB,cAAc,EACd,sBAAsB,GACtB,MAAM,kBAAkB,CAAA;AAEzB,OAAO,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAA"}

package/dist/server/index.js

@@ -0,0 +1,8 @@
+/**
+ * Server Connection Module
+ *
+ * WebSocket server for handling incoming client connections.
+ */
+export { createServerConnection, ServerConnection } from './server-connection.js';
+export { ConnectionManager } from './connection-manager.js';
+export { defineProtocol, method, notification } from '../core/index.js';

package/dist/server/server-connection.d.ts

@@ -0,0 +1,50 @@
+/**
+ * Server Connection
+ *
+ * Represents a single client connection on the server side.
+ * Simplified: no handshake - connection is ready immediately when attached.
+ */
+import { Connection } from '../core/connection.js';
+import type { ConnectionConfig } from '../core/connection.js';
+import type { HandlerContext, ProtocolDef } from '../core/protocol.js';
+import type { IServerWebSocket } from '../platform/types.js';
+/**
+ * Server connection configuration.
+ */
+export interface ServerConnectionConfig<TReceive extends ProtocolDef, TSend extends ProtocolDef> extends ConnectionConfig<TReceive, TSend> {
+    connectionId?: string;
+}
+/**
+ * Extended handler context for server connections.
+ */
+export interface ServerHandlerContext extends HandlerContext {
+    clientType?: string;
+    subscribedSessions: Set<string>;
+}
+/**
+ * Server-side WebSocket connection representing a single client.
+ */
+export declare class ServerConnection<TReceive extends ProtocolDef, TSend extends ProtocolDef> extends Connection<TReceive, TSend> {
+    private serverWs;
+    private clientType;
+    private subscribedSessions;
+    constructor(config: ServerConnectionConfig<TReceive, TSend>);
+    /**
+     * Attach an already-accepted WebSocket connection.
+     * Connection is immediately ready - no handshake needed.
+     */
+    attach(ws: IServerWebSocket): void;
+    connect(): Promise<void>;
+    disconnect(): Promise<void>;
+    getClientType(): string;
+    setClientType(clientType: string): void;
+    getSubscribedSessions(): ReadonlySet<string>;
+    subscribe(sessionId: string): void;
+    unsubscribe(sessionId: string): void;
+    isSubscribedTo(sessionId: string): boolean;
+    publishToSession(sessionId: string, message: string): void;
+    getHandlerContext(): ServerHandlerContext;
+    private setupServerWebSocket;
+}
+export declare function createServerConnection<TReceive extends ProtocolDef, TSend extends ProtocolDef>(config: ServerConnectionConfig<TReceive, TSend>): ServerConnection<TReceive, TSend>;
+//# sourceMappingURL=server-connection.d.ts.map

package/dist/server/server-connection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server-connection.d.ts","sourceRoot":"","sources":["../../src/server/server-connection.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAA;AAClD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAA;AAC7D,OAAO,KAAK,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAA;AACtE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAA;AAG5D;;GAEG;AACH,MAAM,WAAW,sBAAsB,CAAC,QAAQ,SAAS,WAAW,EAAE,KAAK,SAAS,WAAW,CAAE,SAAQ,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC;IACzI,YAAY,CAAC,EAAE,MAAM,CAAA;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAqB,SAAQ,cAAc;IAC3D,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,kBAAkB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAA;CAC/B;AAED;;GAEG;AACH,qBAAa,gBAAgB,CAAC,QAAQ,SAAS,WAAW,EAAE,KAAK,SAAS,WAAW,CAAE,SAAQ,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC;IACzH,OAAO,CAAC,QAAQ,CAAgC;IAChD,OAAO,CAAC,UAAU,CAAoB;IACtC,OAAO,CAAC,kBAAkB,CAAoB;gBAElC,MAAM,EAAE,sBAAsB,CAAC,QAAQ,EAAE,KAAK,CAAC;IAO3D;;;OAGG;IACH,MAAM,CAAC,EAAE,EAAE,gBAAgB,GAAG,IAAI;IAM5B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAIxB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAYjC,aAAa,IAAI,MAAM;IAIvB,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAIvC,qBAAqB,IAAI,WAAW,CAAC,MAAM,CAAC;IAI5C,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAKlC,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAKpC,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAI1C,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAI1D,iBAAiB,IAAI,oBAAoB;IASzC,OAAO,CAAC,oBAAoB;CAO5B;AAED,wBAAgB,sBAAsB,CAAC,QAAQ,SAAS,WAAW,EAAE,KAAK,SAAS,WAAW,EAC7F,MAAM,EAAE,sBAAsB,CAAC,QAAQ,EAAE,KAAK,CAAC,GAC7C,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAEnC"}

package/dist/server/server-connection.js

@@ -0,0 +1,84 @@
+/**
+ * Server Connection
+ *
+ * Represents a single client connection on the server side.
+ * Simplified: no handshake - connection is ready immediately when attached.
+ */
+import { Connection } from '../core/connection.js';
+import { WebSocketReadyState } from '../platform/types.js';
+/**
+ * Server-side WebSocket connection representing a single client.
+ */
+export class ServerConnection extends Connection {
+    serverWs = null;
+    clientType = 'unknown';
+    subscribedSessions = new Set();
+    constructor(config) {
+        super(config);
+        if (config.connectionId) {
+            this.connectionId = config.connectionId;
+        }
+    }
+    /**
+     * Attach an already-accepted WebSocket connection.
+     * Connection is immediately ready - no handshake needed.
+     */
+    attach(ws) {
+        this.serverWs = ws;
+        this.setupServerWebSocket(ws);
+        this.setState('connected');
+    }
+    async connect() {
+        throw new Error('Server connections use attach() instead of connect()');
+    }
+    async disconnect() {
+        this.cleanup();
+        if (this.serverWs && this.serverWs.readyState !== WebSocketReadyState.CLOSED) {
+            this.serverWs.close(1000, 'Server disconnect');
+        }
+        this.serverWs = null;
+        this.ws = null;
+        this.setState('disconnected');
+    }
+    getClientType() {
+        return this.clientType;
+    }
+    setClientType(clientType) {
+        this.clientType = clientType;
+    }
+    getSubscribedSessions() {
+        return this.subscribedSessions;
+    }
+    subscribe(sessionId) {
+        this.subscribedSessions.add(sessionId);
+        this.serverWs?.subscribe(`session:${sessionId}`);
+    }
+    unsubscribe(sessionId) {
+        this.subscribedSessions.delete(sessionId);
+        this.serverWs?.unsubscribe(`session:${sessionId}`);
+    }
+    isSubscribedTo(sessionId) {
+        return this.subscribedSessions.has(sessionId);
+    }
+    publishToSession(sessionId, message) {
+        this.serverWs?.publish(`session:${sessionId}`, message);
+    }
+    getHandlerContext() {
+        return {
+            connectionId: this.connectionId,
+            clientType: this.clientType,
+            subscribedSessions: this.subscribedSessions,
+        };
+    }
+    // Use server context for handlers
+    setupServerWebSocket(ws) {
+        this.ws = ws;
+        this.router.setSender((message) => this.send(message));
+        ws.onmessage = (data) => this.router.handleMessage(data, this.getHandlerContext());
+        ws.onerror = (error) => this.emit('error', error);
+        ws.onclose = (event) => this.handleClose(event.code, event.reason);
+    }
+}
+export function createServerConnection(config) {
+    return new ServerConnection(config);
+}

package/package.json

@@ -0,0 +1,52 @@
+{
+	"name": "@roj-ai/transport",
+	"version": "0.0.2",
+	"type": "module",
+	"main": "./src/index.ts",
+	"exports": {
+		".": {
+			"types": "./dist/index.d.ts",
+			"import": "./src/index.ts",
+			"default": "./src/index.ts"
+		},
+		"./client": {
+			"types": "./dist/client/index.d.ts",
+			"import": "./src/client/index.ts",
+			"default": "./src/client/index.ts"
+		},
+		"./server": {
+			"types": "./dist/server/index.d.ts",
+			"import": "./src/server/index.ts",
+			"default": "./src/server/index.ts"
+		},
+		"./browser": {
+			"types": "./dist/platform/browser.d.ts",
+			"import": "./src/platform/browser.ts",
+			"default": "./src/platform/browser.ts"
+		},
+		"./bun": {
+			"types": "./dist/platform/bun.d.ts",
+			"import": "./src/platform/bun.ts",
+			"default": "./src/platform/bun.ts"
+		}
+	},
+	"files": [
+		"src",
+		"dist"
+	],
+	"publishConfig": {
+		"access": "public"
+	},
+	"scripts": {
+		"build": "tsc",
+		"type-check": "tsc --noEmit"
+	},
+	"dependencies": {
+		"zod": "4.3.6"
+	},
+	"devDependencies": {
+		"@types/bun": "latest",
+		"typescript": "^5.7.2"
+	},
+	"types": "./dist/index.d.ts"
+}