/// <reference types="node" resolution-mode="require"/>
import type { FunctionRuntimeProps } from "../function-props.js";
import { type SocketSpec } from "../internal/service-spec.js";
import type { ServiceContext } from "../service.js";
import type { SocketMiddleware, SocketMiddlewareFunction } from "./middleware.js";
export interface SocketContext<ConnectContext extends SocketHandlerContext = SocketHandlerContext, DisconnectContext extends SocketHandlerContext = SocketHandlerContext, MessageContext extends SocketHandlerContext = SocketHandlerContext> {
    connect: ConnectContext;
    disconnect: DisconnectContext;
    message: MessageContext;
}
export interface SocketHandlerContext {
    socket: {
        socketName: string;
    };
    service: ServiceContext;
}
export type SocketHeaders = Record<string, string | undefined>;
export type SocketQuery = Record<string, string | undefined>;
export type SocketRequest = SocketConnectRequest | SocketDisconnectRequest | SocketMessageRequest;
export interface SocketConnectRequest {
    type: "connect";
    connectionId: string;
    query?: SocketQuery;
    headers: SocketHeaders;
}
export interface SocketDisconnectRequest {
    type: "disconnect";
    connectionId: string;
}
export interface SocketMessageRequest {
    type: "message";
    connectionId: string;
    body?: string | Buffer;
}
export interface SocketResponse {
    status: number;
    message?: string | Buffer | any;
}
export type SocketHandlers<Context extends SocketContext = SocketContext> = {
    $connect: (request: SocketConnectRequest, context: Context["connect"]) => Promise<SocketResponse | void> | SocketResponse | void;
    $disconnect: (request: SocketDisconnectRequest, context: Context["disconnect"]) => Promise<SocketResponse | void> | SocketResponse | void;
    $default: (request: SocketMessageRequest, context: Context["message"]) => Promise<SocketResponse | string | Buffer | any | void> | SocketResponse | string | Buffer | any | void;
};
export type Socket<Name extends string = string, Context extends SocketContext = SocketContext> = SocketSpec<Name> & {
    kind: "Socket";
    handlers: SocketHandlers<Context>;
    wssEndpoint: string;
    httpEndpoint: string;
    middlewares: SocketMiddleware[];
} & {
    send: (connectionId: string, input: Buffer | string) => Promise<void>;
    disconnect: (connectionId: string) => Promise<void>;
};
export type SocketOptions = FunctionRuntimeProps;
export interface SocketRouter<Context extends SocketContext = SocketContext> {
    middlewares: SocketMiddleware[];
    use<NextConnectContext extends SocketHandlerContext = Context["connect"], NextDisconnectContext extends SocketHandlerContext = Context["disconnect"], NextMessageContext extends SocketHandlerContext = Context["message"]>(socketMiddleware: SocketMiddleware<Context, SocketContext<NextConnectContext, NextDisconnectContext, NextMessageContext>> | SocketMiddlewareFunction<SocketConnectRequest, Context["connect"], NextConnectContext>): SocketRouter<SocketContext<NextConnectContext, NextDisconnectContext, NextMessageContext>>;
    socket<Name extends string>(name: Name, options: SocketOptions, handlers: SocketHandlers<Context>): Socket<Name, Context>;
    socket<Name extends string>(name: Name, handlers: SocketHandlers<Context>): Socket<Name, Context>;
}
export interface socket<Context extends SocketContext = SocketContext> extends SocketRouter<Context> {
    <Name extends string>(name: Name, options: SocketOptions, handlers: SocketHandlers<Context>): Socket<Name, Context>;
    <Name extends string>(name: Name, handlers: SocketHandlers<Context>): Socket<Name, Context>;
}
export declare const socket: socket<SocketContext<SocketHandlerContext, SocketHandlerContext, SocketHandlerContext>>;
export declare function parseSocketArgs<Name extends string>(args: any[]): {
    sourceLocation: import("../internal/service-spec.js").SourceLocation;
    name: Name;
    options: FunctionRuntimeProps;
    handlers: SocketHandlers<SocketContext<SocketHandlerContext, SocketHandlerContext, SocketHandlerContext>>;
};
//# sourceMappingURL=socket.d.ts.map