import { type WebSocket } from "ws";
import { z } from "zod";
import { type WebSocketAudience } from "./ws-audience";
import { type WebSocketClusterBus } from "./ws-cluster-bus";
import { type WebSocketPresenceStore } from "./ws-presence-store";
import { type ManagedWebSocketConnection, WebSocketRegistry, type WebSocketRoomId, type WebSocketUserId } from "./ws-registry";
type MessageHandler<T> = (data: T) => void | Promise<void>;
type CloseHandler = () => void | Promise<void>;
export type WebSocketEventMap = Record<string, unknown>;
type InferWebSocketEventMap<TSchema extends z.ZodRawShape> = z.infer<z.ZodObject<TSchema>>;
export type WebSocketOutEvents<TOut extends WebSocketEventMap = WebSocketEventMap> = TOut;
export type WebSocketInEvents<TIn extends WebSocketEventMap = WebSocketEventMap> = TIn;
export interface WebSocketConnection<TOut extends WebSocketEventMap = WebSocketEventMap, TIn extends WebSocketEventMap = WebSocketEventMap> extends ManagedWebSocketConnection {
    transport: "ws";
    onClose(callback: CloseHandler): void;
    onMessage<K extends keyof WebSocketInEvents<TIn>>(event: K, handler: MessageHandler<WebSocketInEvents<TIn>[K]>): void;
    publish<K extends keyof WebSocketOutEvents<TOut>>(event: K, data: WebSocketOutEvents<TOut>[K]): void;
    waitForClose(): Promise<void>;
    join(roomId: WebSocketRoomId): void;
    leave(roomId: WebSocketRoomId): void;
    setUserId(userId: WebSocketUserId): void;
    clearUserId(): void;
}
export type AnyWebSocketConnection = WebSocketConnection<WebSocketEventMap, WebSocketEventMap>;
type WebSocketConnectionOptions<TOut extends z.ZodRawShape, TIn extends z.ZodRawShape> = {
    namespace?: string;
    heartbeat?: number;
    maxPayload?: number;
    active?: boolean;
    outEvents: z.ZodObject<TOut>;
    inEvents: z.ZodObject<TIn>;
    registry: WebSocketRegistry;
};
export type WebSocketRuntimeOptions = {
    nodeId?: string;
    presenceStore?: WebSocketPresenceStore;
    clusterBus?: WebSocketClusterBus;
};
export declare class WebSocketRuntime {
    readonly registry: WebSocketRegistry;
    constructor(options?: WebSocketRuntimeOptions);
    registerConnection<TOutSchema extends z.ZodRawShape, TInSchema extends z.ZodRawShape>(socket: WebSocket, options: Omit<WebSocketConnectionOptions<TOutSchema, TInSchema>, "registry">): WebSocketConnection<InferWebSocketEventMap<TOutSchema>, InferWebSocketEventMap<TInSchema>>;
    activateConnection(connectionId: string): void;
    broadcast(event: string, data: unknown, namespace?: string): void;
    publishToRoom(roomId: WebSocketRoomId, event: string, data: unknown, namespace?: string): void;
    publishToUser(userId: WebSocketUserId, event: string, data: unknown, namespace?: string): void;
    publishToAudience(audience: WebSocketAudience, event: string, data: unknown): void;
    shutdown(code?: number, reason?: string): Promise<void>;
}
export declare function createWebSocketRuntime(options?: WebSocketRuntimeOptions): WebSocketRuntime;
export {};
//# sourceMappingURL=ws.d.ts.map