import type { SocketConnectRequest, SocketContext, SocketDisconnectRequest, SocketHandlerContext, SocketMessageRequest, SocketRequest, SocketResponse } from "./socket.js";
export interface SocketMiddlewareInput<Request extends SocketRequest, In extends SocketHandlerContext> {
    request: Request;
    context: In;
    next: <O extends SocketHandlerContext>(context: O) => Promise<SocketMiddlewareOutput<O>>;
}
export type SocketMiddlewareOutput<Context extends SocketHandlerContext> = SocketResponse & {
    context?: Context;
};
export type SocketMiddlewareFunction<Request extends SocketRequest, In extends SocketHandlerContext, Out extends SocketHandlerContext> = (input: SocketMiddlewareInput<Request, In>) => Promise<SocketMiddlewareOutput<Out>> | SocketMiddlewareOutput<Out>;
export interface SocketMiddleware<In extends SocketContext = any, Out extends SocketContext = any> {
    connect?: SocketMiddlewareFunction<SocketConnectRequest, In["connect"], Out["connect"]>;
    disconnect?: SocketMiddlewareFunction<SocketDisconnectRequest, In["disconnect"], Out["disconnect"]>;
    message?: SocketMiddlewareFunction<SocketMessageRequest, In["message"], Out["message"]>;
}
/**
 * Utility for creating a Socket Middleware function that combines its output context
 * with the input context.
 *
 * ```ts
 * const auth = socketMiddleware(({request, context, next}) => {
 *   return next({
 *     ...context,
 *     isAuthenticated: request.headers.Authorization !== undefined
 *   })
 * });
 *
 * socket.use(auth)("myAuthorizedCommand", async (request, { isAuthenticated }) => {
 *   if (isAuthenticated) {
 *     // do work
 *   }
 * })
 * ```
 */
export declare function socketMiddleware<PrevContext extends SocketContext = SocketContext, OutContext extends SocketContext = SocketContext>(fn: SocketMiddlewareFunction<SocketConnectRequest, PrevContext["connect"], OutContext["connect"]> | SocketMiddleware<PrevContext, OutContext>): {
    connect: (<In extends PrevContext["connect"]>(input: SocketMiddlewareInput<SocketConnectRequest, In>) => Promise<SocketMiddlewareOutput<OutContext["connect"]>>) | undefined;
    disconnect: (<In_1 extends PrevContext["disconnect"]>(input: SocketMiddlewareInput<SocketDisconnectRequest, In_1>) => Promise<SocketMiddlewareOutput<OutContext["disconnect"]>>) | undefined;
    message: (<In_2 extends PrevContext["message"]>(input: SocketMiddlewareInput<SocketMessageRequest, In_2>) => Promise<SocketMiddlewareOutput<OutContext["message"]>>) | undefined;
};
//# sourceMappingURL=middleware.d.ts.map