@agentick/gateway 0.0.1

package/dist/types.d.ts

@@ -0,0 +1,314 @@
+/**
+ * Gateway Types
+ */
+import type { App } from "@agentick/core";
+import type { KernelContext, UserContext } from "@agentick/kernel";
+import type { AuthConfig } from "@agentick/server";
+export type { AuthConfig, AuthResult } from "@agentick/server";
+/**
+ * Schema type that works with both Zod 3 and Zod 4.
+ * We only need parse() and type inference (_output).
+ */
+export interface ZodLikeSchema<T = unknown> {
+    parse(data: unknown): T;
+    _output: T;
+}
+export type { UserContext } from "@agentick/kernel";
+export interface GatewayConfig {
+    /**
+     * Port to listen on (ignored in embedded mode)
+     * @default 18789
+     */
+    port?: number;
+    /**
+     * Host to bind to (ignored in embedded mode)
+     * @default "127.0.0.1"
+     */
+    host?: string;
+    /**
+     * Gateway ID (auto-generated if not provided)
+     */
+    id?: string;
+    /**
+     * App definitions
+     */
+    apps: Record<string, App>;
+    /**
+     * Default app to use when session key doesn't specify one
+     */
+    defaultApp: string;
+    /**
+     * Authentication configuration
+     */
+    auth?: AuthConfig;
+    /**
+     * Run in embedded mode (no standalone server).
+     * Use handleRequest() to process requests from your framework.
+     * @default false
+     */
+    embedded?: boolean;
+    /**
+     * Persistence configuration
+     */
+    storage?: StorageConfig;
+    /**
+     * Channel adapters (WhatsApp, Slack, etc.)
+     */
+    channels?: ChannelAdapter[];
+    /**
+     * Message routing configuration
+     */
+    routing?: RoutingConfig;
+    /**
+     * Transport mode (ignored in embedded mode)
+     * - "websocket": WebSocket only (default, good for CLI/native clients)
+     * - "http": HTTP/SSE only (good for web browsers)
+     * - "both": Both transports on different ports
+     * @default "websocket"
+     */
+    transport?: "websocket" | "http" | "both";
+    /**
+     * HTTP path prefix (e.g., "/api")
+     * @default ""
+     */
+    httpPathPrefix?: string;
+    /**
+     * CORS origin for HTTP transport
+     * @default "*"
+     */
+    httpCorsOrigin?: string;
+    /**
+     * HTTP port when using "both" mode
+     * @default port + 1
+     */
+    httpPort?: number;
+    /**
+     * Custom methods - runs within Agentick ALS context.
+     *
+     * Supports:
+     * - Simple handlers: `async (params) => result`
+     * - Streaming: `async function* (params) { yield value }`
+     * - With config: `method({ schema, handler, roles, guard })`
+     * - Namespaces: `{ tasks: { list, create, admin: { ... } } }` (recursive)
+     *
+     * Use method() wrapper for schema validation, roles, guards, etc.
+     * ctx param is optional - use Context.get() for idiomatic access.
+     */
+    methods?: MethodsConfig;
+}
+export interface StorageConfig {
+    /**
+     * Base directory for storage
+     * @default "~/.agentick"
+     */
+    directory?: string;
+    /**
+     * Enable session persistence
+     * @default true
+     */
+    sessions?: boolean;
+    /**
+     * Enable memory persistence
+     * @default true
+     */
+    memory?: boolean;
+}
+export interface ChannelAdapter {
+    /**
+     * Channel identifier
+     */
+    id: string;
+    /**
+     * Human-readable name
+     */
+    name: string;
+    /**
+     * Initialize the channel
+     */
+    initialize(gateway: GatewayContext): Promise<void>;
+    /**
+     * Clean up resources
+     */
+    destroy(): Promise<void>;
+}
+export interface GatewayContext {
+    /**
+     * Send a message to a session
+     */
+    sendToSession(sessionId: string, message: string): Promise<void>;
+    /**
+     * Get available apps
+     */
+    getApps(): string[];
+    /**
+     * Get or create a session
+     */
+    getSession(sessionId: string): SessionContext;
+}
+export interface SessionContext {
+    id: string;
+    appId: string;
+    send(message: string): AsyncGenerator<SessionEvent>;
+}
+export interface SessionEvent {
+    type: string;
+    data: unknown;
+}
+export interface RoutingConfig {
+    /**
+     * Map channels to agents
+     */
+    channels?: Record<string, string>;
+    /**
+     * Custom routing function
+     */
+    custom?: (message: IncomingMessage, context: RoutingContext) => string | null;
+}
+export interface IncomingMessage {
+    text: string;
+    channel?: string;
+    from?: string;
+    metadata?: Record<string, unknown>;
+}
+export interface RoutingContext {
+    availableApps: string[];
+    defaultApp: string;
+    sessionHistory?: Array<{
+        role: string;
+        content: string;
+    }>;
+}
+export interface ClientState {
+    id: string;
+    connectedAt: Date;
+    authenticated: boolean;
+    /** Full user context from auth */
+    user?: UserContext;
+    subscriptions: Set<string>;
+    metadata?: Record<string, unknown>;
+}
+export interface SessionState {
+    id: string;
+    appId: string;
+    createdAt: Date;
+    lastActivityAt: Date;
+    messageCount: number;
+    isActive: boolean;
+    subscribers: Set<string>;
+}
+export interface GatewayEvents {
+    started: {
+        port: number;
+        host: string;
+    };
+    stopped: Record<string, never>;
+    "client:connected": {
+        clientId: string;
+        ip?: string;
+    };
+    "client:disconnected": {
+        clientId: string;
+        reason?: string;
+    };
+    "client:authenticated": {
+        clientId: string;
+        user?: UserContext;
+    };
+    "session:created": {
+        sessionId: string;
+        appId: string;
+    };
+    "session:closed": {
+        sessionId: string;
+    };
+    "session:message": {
+        sessionId: string;
+        role: "user" | "assistant";
+        content: string;
+    };
+    "app:message": {
+        appId: string;
+        sessionId: string;
+        message: string;
+    };
+    "channel:message": {
+        channel: string;
+        from: string;
+        message: string;
+    };
+    "channel:error": {
+        channel: string;
+        error: Error;
+    };
+    error: Error;
+}
+/** Symbol for detecting method definitions vs namespaces */
+export declare const METHOD_DEFINITION: unique symbol;
+/**
+ * Simple method handler - ctx param is optional since Context.get() works
+ */
+export type SimpleMethodHandler<TParams = Record<string, unknown>, TResult = unknown> = (params: TParams, ctx?: KernelContext) => Promise<TResult> | TResult;
+/**
+ * Streaming method handler - yields values to client
+ */
+export type StreamingMethodHandler<TParams = Record<string, unknown>, TYield = unknown> = (params: TParams, ctx?: KernelContext) => AsyncGenerator<TYield>;
+/**
+ * Method definition input (what you pass to method())
+ */
+export interface MethodDefinitionInput<TSchema extends ZodLikeSchema = ZodLikeSchema> {
+    /** Zod schema for params validation + TypeScript inference */
+    schema?: TSchema;
+    /** Handler function - receives validated & typed params */
+    handler: SimpleMethodHandler<TSchema["_output"]> | StreamingMethodHandler<TSchema["_output"]>;
+    /** Required roles - checked before handler */
+    roles?: string[];
+    /** Custom guard function */
+    guard?: (ctx: KernelContext) => boolean | Promise<boolean>;
+    /** Method description for discovery */
+    description?: string;
+}
+/**
+ * Method definition with symbol marker (returned by method())
+ */
+export interface MethodDefinition<TSchema extends ZodLikeSchema = ZodLikeSchema> extends MethodDefinitionInput<TSchema> {
+    [METHOD_DEFINITION]: true;
+}
+/**
+ * Factory function to create a method definition.
+ * Stores config (schema, roles, guards) - the gateway creates the
+ * actual procedure during initialization with the full inferred path name.
+ *
+ * @example
+ * methods: {
+ *   tasks: {
+ *     list: async (params) => { ... },  // Simple - auto-wrapped
+ *     create: method({                   // With config
+ *       schema: z.object({ title: z.string() }),
+ *       handler: async (params) => { ... }
+ *     }),
+ *   }
+ * }
+ */
+export declare function method<TSchema extends ZodLikeSchema>(definition: MethodDefinitionInput<TSchema>): MethodDefinition<TSchema>;
+/**
+ * Check if a value is a method definition (vs a namespace)
+ */
+export declare function isMethodDefinition(value: unknown): value is MethodDefinition;
+/**
+ * Method can be:
+ * - Simple function: async (params) => result
+ * - Streaming function: async function* (params) { yield }
+ * - Method definition: method({ schema, handler, roles, ... })
+ */
+export type Method = SimpleMethodHandler | StreamingMethodHandler | MethodDefinition<any>;
+/**
+ * Method namespace - recursively nested, arbitrary depth
+ */
+export type MethodNamespace = {
+    [key: string]: Method | MethodNamespace;
+};
+/**
+ * Methods config - supports flat or nested namespaces
+ */
+export type MethodsConfig = MethodNamespace;
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACnE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAGnD,YAAY,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE/D;;;GAGG;AACH,MAAM,WAAW,aAAa,CAAC,CAAC,GAAG,OAAO;IACxC,KAAK,CAAC,IAAI,EAAE,OAAO,GAAG,CAAC,CAAC;IACxB,OAAO,EAAE,CAAC,CAAC;CACZ;AAED,YAAY,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAMpD,MAAM,WAAW,aAAa;IAC5B;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,EAAE,CAAC,EAAE,MAAM,CAAC;IAEZ;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAE1B;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IAEnB;;OAEG;IACH,IAAI,CAAC,EAAE,UAAU,CAAC;IAElB;;;;OAIG;IACH,QAAQ,CAAC,EAAE,OAAO,CAAC;IAEnB;;OAEG;IACH,OAAO,CAAC,EAAE,aAAa,CAAC;IAExB;;OAEG;IACH,QAAQ,CAAC,EAAE,cAAc,EAAE,CAAC;IAE5B;;OAEG;IACH,OAAO,CAAC,EAAE,aAAa,CAAC;IAExB;;;;;;OAMG;IACH,SAAS,CAAC,EAAE,WAAW,GAAG,MAAM,GAAG,MAAM,CAAC;IAE1C;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB;;;OAGG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB;;;;;;;;;;;OAWG;IACH,OAAO,CAAC,EAAE,aAAa,CAAC;CACzB;AAMD,MAAM,WAAW,aAAa;IAC5B;;;OAGG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;;OAGG;IACH,QAAQ,CAAC,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAMD,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IAEX;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,UAAU,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEnD;;OAEG;IACH,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1B;AAED,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjE;;OAEG;IACH,OAAO,IAAI,MAAM,EAAE,CAAC;IAEpB;;OAEG;IACH,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,cAAc,CAAC;CAC/C;AAED,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;CACrD;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,OAAO,CAAC;CACf;AAMD,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAElC;;OAEG;IACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,eAAe,EAAE,OAAO,EAAE,cAAc,KAAK,MAAM,GAAG,IAAI,CAAC;CAC/E;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,cAAc;IAC7B,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAC3D;AAMD,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,IAAI,CAAC;IAClB,aAAa,EAAE,OAAO,CAAC;IACvB,kCAAkC;IAClC,IAAI,CAAC,EAAE,WAAW,CAAC;IACnB,aAAa,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAMD,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,IAAI,CAAC;IAChB,cAAc,EAAE,IAAI,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,OAAO,CAAC;IAClB,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;CAC1B;AAMD,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IACxC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC/B,kBAAkB,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IACtD,qBAAqB,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAC7D,sBAAsB,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,WAAW,CAAA;KAAE,CAAC;IACjE,iBAAiB,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IACxD,gBAAgB,EAAE;QAAE,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IACxC,iBAAiB,EAAE;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;QAC3B,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;IACF,aAAa,EAAE;QACb,KAAK,EAAE,MAAM,CAAC;QACd,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;IACF,iBAAiB,EAAE;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;IACF,eAAe,EAAE;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,KAAK,CAAC;KACd,CAAC;IACF,KAAK,EAAE,KAAK,CAAC;CACd;AAMD,4DAA4D;AAC5D,eAAO,MAAM,iBAAiB,eAA2C,CAAC;AAE1E;;GAEG;AACH,MAAM,MAAM,mBAAmB,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,IAAI,CACtF,MAAM,EAAE,OAAO,EACf,GAAG,CAAC,EAAE,aAAa,KAChB,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC;AAEhC;;GAEG;AACH,MAAM,MAAM,sBAAsB,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,IAAI,CACxF,MAAM,EAAE,OAAO,EACf,GAAG,CAAC,EAAE,aAAa,KAChB,cAAc,CAAC,MAAM,CAAC,CAAC;AAE5B;;GAEG;AACH,MAAM,WAAW,qBAAqB,CAAC,OAAO,SAAS,aAAa,GAAG,aAAa;IAClF,8DAA8D;IAC9D,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,2DAA2D;IAC3D,OAAO,EAAE,mBAAmB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,GAAG,sBAAsB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;IAC9F,8CAA8C;IAC9C,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,4BAA4B;IAC5B,KAAK,CAAC,EAAE,CAAC,GAAG,EAAE,aAAa,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3D,uCAAuC;IACvC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB,CAC/B,OAAO,SAAS,aAAa,GAAG,aAAa,CAC7C,SAAQ,qBAAqB,CAAC,OAAO,CAAC;IACtC,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC;CAC3B;AAED;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,MAAM,CAAC,OAAO,SAAS,aAAa,EAClD,UAAU,EAAE,qBAAqB,CAAC,OAAO,CAAC,GACzC,gBAAgB,CAAC,OAAO,CAAC,CAK3B;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,gBAAgB,CAE5E;AAED;;;;;GAKG;AAEH,MAAM,MAAM,MAAM,GAAG,mBAAmB,GAAG,sBAAsB,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;AAE1F;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG;IAC5B,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,eAAe,CAAC;CACzC,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,aAAa,GAAG,eAAe,CAAC"}

package/dist/types.js

@@ -0,0 +1,37 @@
+/**
+ * Gateway Types
+ */
+// ============================================================================
+// Custom Methods
+// ============================================================================
+/** Symbol for detecting method definitions vs namespaces */
+export const METHOD_DEFINITION = Symbol.for("agentick:method-definition");
+/**
+ * Factory function to create a method definition.
+ * Stores config (schema, roles, guards) - the gateway creates the
+ * actual procedure during initialization with the full inferred path name.
+ *
+ * @example
+ * methods: {
+ *   tasks: {
+ *     list: async (params) => { ... },  // Simple - auto-wrapped
+ *     create: method({                   // With config
+ *       schema: z.object({ title: z.string() }),
+ *       handler: async (params) => { ... }
+ *     }),
+ *   }
+ * }
+ */
+export function method(definition) {
+    return {
+        [METHOD_DEFINITION]: true,
+        ...definition,
+    };
+}
+/**
+ * Check if a value is a method definition (vs a namespace)
+ */
+export function isMethodDefinition(value) {
+    return typeof value === "object" && value !== null && METHOD_DEFINITION in value;
+}
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAkSH,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E,4DAA4D;AAC5D,MAAM,CAAC,MAAM,iBAAiB,GAAG,MAAM,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;AA2C1E;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,MAAM,CACpB,UAA0C;IAE1C,OAAO;QACL,CAAC,iBAAiB,CAAC,EAAE,IAAI;QACzB,GAAG,UAAU;KACd,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAc;IAC/C,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,iBAAiB,IAAI,KAAK,CAAC;AACnF,CAAC"}

package/dist/websocket-server.d.ts

@@ -0,0 +1,87 @@
+/**
+ * WebSocket Server
+ *
+ * Handles WebSocket connections and message routing.
+ */
+import { WebSocket } from "ws";
+import type { ClientMessage, GatewayMessage } from "./transport-protocol.js";
+import type { ClientState, AuthConfig } from "./types.js";
+export interface WSServerConfig {
+    port: number;
+    host: string;
+    auth?: AuthConfig;
+}
+export interface WSServerEvents {
+    connection: (client: WSClient) => void;
+    disconnect: (clientId: string, reason?: string) => void;
+    message: (clientId: string, message: ClientMessage) => void;
+    error: (error: Error) => void;
+}
+export declare class WSClient {
+    readonly id: string;
+    readonly socket: WebSocket;
+    readonly state: ClientState;
+    private server;
+    constructor(id: string, socket: WebSocket, server: WSServer);
+    /**
+     * Send a message to this client
+     */
+    send(message: GatewayMessage): void;
+    /**
+     * Close the connection
+     */
+    close(code?: number, reason?: string): void;
+    /**
+     * Check if connected
+     */
+    get isConnected(): boolean;
+}
+export declare class WSServer {
+    private wss;
+    private clients;
+    private config;
+    private handlers;
+    private clientIdCounter;
+    constructor(config: WSServerConfig);
+    /**
+     * Start the WebSocket server
+     */
+    start(): Promise<void>;
+    /**
+     * Stop the server
+     */
+    stop(): Promise<void>;
+    /**
+     * Register event handlers
+     */
+    on<K extends keyof WSServerEvents>(event: K, handler: WSServerEvents[K]): void;
+    /**
+     * Get a client by ID
+     */
+    getClient(id: string): WSClient | undefined;
+    /**
+     * Get all clients
+     */
+    getClients(): WSClient[];
+    /**
+     * Get authenticated clients
+     */
+    getAuthenticatedClients(): WSClient[];
+    /**
+     * Broadcast a message to all authenticated clients
+     */
+    broadcast(message: GatewayMessage): void;
+    /**
+     * Send a message to clients subscribed to a session
+     */
+    sendToSubscribers(sessionId: string, message: GatewayMessage): void;
+    /**
+     * Get connected client count
+     */
+    get clientCount(): number;
+    private handleConnection;
+    private handleMessage;
+    private handleConnect;
+    private validateAuth;
+}
+//# sourceMappingURL=websocket-server.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"websocket-server.d.ts","sourceRoot":"","sources":["../src/websocket-server.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAmB,SAAS,EAAE,MAAM,IAAI,CAAC;AAEhD,OAAO,KAAK,EACV,aAAa,EACb,cAAc,EAGf,MAAM,yBAAyB,CAAC;AACjC,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAc,MAAM,YAAY,CAAC;AAEtE,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,UAAU,CAAC;CACnB;AAED,MAAM,WAAW,cAAc;IAC7B,UAAU,EAAE,CAAC,MAAM,EAAE,QAAQ,KAAK,IAAI,CAAC;IACvC,UAAU,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;IACxD,OAAO,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,KAAK,IAAI,CAAC;IAC5D,KAAK,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CAC/B;AAED,qBAAa,QAAQ;IACnB,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC;IAC3B,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC;IAC5B,OAAO,CAAC,MAAM,CAAW;gBAEb,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ;IAY3D;;OAEG;IACH,IAAI,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAMnC;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI;IAI3C;;OAEG;IACH,IAAI,WAAW,IAAI,OAAO,CAEzB;CACF;AAED,qBAAa,QAAQ;IACnB,OAAO,CAAC,GAAG,CAAgC;IAC3C,OAAO,CAAC,OAAO,CAA+B;IAC9C,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,QAAQ,CAA+B;IAC/C,OAAO,CAAC,eAAe,CAAK;gBAEhB,MAAM,EAAE,cAAc;IAIlC;;OAEG;IACH,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAuBtB;;OAEG;IACH,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAqBrB;;OAEG;IACH,EAAE,CAAC,CAAC,SAAS,MAAM,cAAc,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,IAAI;IAI9E;;OAEG;IACH,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS;IAI3C;;OAEG;IACH,UAAU,IAAI,QAAQ,EAAE;IAIxB;;OAEG;IACH,uBAAuB,IAAI,QAAQ,EAAE;IAIrC;;OAEG;IACH,SAAS,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAMxC;;OAEG;IACH,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,IAAI;IAQnE;;OAEG;IACH,IAAI,WAAW,IAAI,MAAM,CAExB;IAED,OAAO,CAAC,gBAAgB;YA+BV,aAAa;YA2Bb,aAAa;YAgCb,YAAY;CAyB3B"}

package/dist/websocket-server.js

@@ -0,0 +1,245 @@
+/**
+ * WebSocket Server
+ *
+ * Handles WebSocket connections and message routing.
+ */
+import { WebSocketServer, WebSocket } from "ws";
+export class WSClient {
+    id;
+    socket;
+    state;
+    server;
+    constructor(id, socket, server) {
+        this.id = id;
+        this.socket = socket;
+        this.server = server;
+        this.state = {
+            id,
+            connectedAt: new Date(),
+            authenticated: false,
+            subscriptions: new Set(),
+        };
+    }
+    /**
+     * Send a message to this client
+     */
+    send(message) {
+        if (this.socket.readyState === WebSocket.OPEN) {
+            this.socket.send(JSON.stringify(message));
+        }
+    }
+    /**
+     * Close the connection
+     */
+    close(code, reason) {
+        this.socket.close(code, reason);
+    }
+    /**
+     * Check if connected
+     */
+    get isConnected() {
+        return this.socket.readyState === WebSocket.OPEN;
+    }
+}
+export class WSServer {
+    wss = null;
+    clients = new Map();
+    config;
+    handlers = {};
+    clientIdCounter = 0;
+    constructor(config) {
+        this.config = config;
+    }
+    /**
+     * Start the WebSocket server
+     */
+    start() {
+        return new Promise((resolve, reject) => {
+            try {
+                this.wss = new WebSocketServer({
+                    port: this.config.port,
+                    host: this.config.host,
+                });
+                this.wss.on("connection", this.handleConnection.bind(this));
+                this.wss.on("error", (error) => {
+                    this.handlers.error?.(error);
+                    reject(error);
+                });
+                this.wss.on("listening", () => {
+                    resolve();
+                });
+            }
+            catch (error) {
+                reject(error);
+            }
+        });
+    }
+    /**
+     * Stop the server
+     */
+    stop() {
+        return new Promise((resolve) => {
+            if (!this.wss) {
+                resolve();
+                return;
+            }
+            // Close all client connections
+            for (const client of this.clients.values()) {
+                client.close(1001, "Server shutting down");
+            }
+            this.clients.clear();
+            // Close the server
+            this.wss.close(() => {
+                this.wss = null;
+                resolve();
+            });
+        });
+    }
+    /**
+     * Register event handlers
+     */
+    on(event, handler) {
+        this.handlers[event] = handler;
+    }
+    /**
+     * Get a client by ID
+     */
+    getClient(id) {
+        return this.clients.get(id);
+    }
+    /**
+     * Get all clients
+     */
+    getClients() {
+        return Array.from(this.clients.values());
+    }
+    /**
+     * Get authenticated clients
+     */
+    getAuthenticatedClients() {
+        return this.getClients().filter((c) => c.state.authenticated);
+    }
+    /**
+     * Broadcast a message to all authenticated clients
+     */
+    broadcast(message) {
+        for (const client of this.getAuthenticatedClients()) {
+            client.send(message);
+        }
+    }
+    /**
+     * Send a message to clients subscribed to a session
+     */
+    sendToSubscribers(sessionId, message) {
+        for (const client of this.getAuthenticatedClients()) {
+            if (client.state.subscriptions.has(sessionId)) {
+                client.send(message);
+            }
+        }
+    }
+    /**
+     * Get connected client count
+     */
+    get clientCount() {
+        return this.clients.size;
+    }
+    handleConnection(socket, _request) {
+        const clientId = `client-${++this.clientIdCounter}`;
+        const client = new WSClient(clientId, socket, this);
+        this.clients.set(clientId, client);
+        socket.on("message", (data) => {
+            try {
+                const message = JSON.parse(data.toString());
+                this.handleMessage(client, message);
+            }
+            catch (error) {
+                client.send({
+                    type: "error",
+                    code: "INVALID_MESSAGE",
+                    message: "Failed to parse message",
+                });
+            }
+        });
+        socket.on("close", () => {
+            this.clients.delete(clientId);
+            this.handlers.disconnect?.(clientId);
+        });
+        socket.on("error", (error) => {
+            this.handlers.error?.(error);
+        });
+        // Notify handler of new connection (before auth)
+        this.handlers.connection?.(client);
+    }
+    async handleMessage(client, message) {
+        // Handle connect message (authentication)
+        if (message.type === "connect") {
+            await this.handleConnect(client, message);
+            return;
+        }
+        // Handle ping
+        if (message.type === "ping") {
+            client.send({ type: "pong", timestamp: message.timestamp });
+            return;
+        }
+        // All other messages require authentication
+        if (!client.state.authenticated) {
+            client.send({
+                type: "error",
+                code: "UNAUTHORIZED",
+                message: "Authentication required. Send connect message first.",
+            });
+            return;
+        }
+        // Forward to message handler
+        this.handlers.message?.(client.id, message);
+    }
+    async handleConnect(client, message) {
+        // Validate authentication
+        const authResult = await this.validateAuth(message.token);
+        if (!authResult.valid) {
+            client.send({
+                type: "error",
+                code: "AUTH_FAILED",
+                message: "Authentication failed",
+            });
+            client.close(4001, "Authentication failed");
+            return;
+        }
+        // Update client state
+        client.state.authenticated = true;
+        client.state.user = authResult.user;
+        client.state.metadata = {
+            ...client.state.metadata,
+            ...authResult.metadata,
+            ...message.metadata,
+        };
+        // Client ID from message takes precedence
+        if (message.clientId) {
+            // Update internal tracking if client provides their own ID
+            this.clients.delete(client.id);
+            client.id = message.clientId;
+            this.clients.set(message.clientId, client);
+        }
+    }
+    async validateAuth(token) {
+        const auth = this.config.auth;
+        // No auth configured
+        if (!auth || auth.type === "none") {
+            return { valid: true };
+        }
+        // Token auth
+        if (auth.type === "token") {
+            return { valid: token === auth.token };
+        }
+        // JWT auth
+        if (auth.type === "jwt") {
+            return { valid: false };
+        }
+        // Custom auth
+        if (auth.type === "custom") {
+            return await auth.validate(token ?? "");
+        }
+        return { valid: false };
+    }
+}
+//# sourceMappingURL=websocket-server.js.map