import type { ManagerOptions, Socket, SocketOptions } from "socket.io-client";
import type { inferZodSchemaInput, inferZodSchemaOutput } from "./types/schemas.js";
import type { StructuredLogger } from "./utils/structuredLogger.js";
import type { EventEmitterLike, ZodMessageValueSchema } from "./zodMessageHandler.js";
export interface ZodSocketMessageCatalogSchema {
    [key: string]: {
        message: ZodMessageValueSchema<any>;
    } | {
        message: ZodMessageValueSchema<any>;
        callback?: ZodMessageValueSchema<any>;
    };
}
export type ZodMessageCatalogToSocketIoEvents<TCatalog extends ZodSocketMessageCatalogSchema> = {
    [K in keyof TCatalog]: SocketMessageHasCallback<TCatalog, K> extends true ? (message: inferZodSchemaOutput<GetSocketMessageSchema<TCatalog, K>>, callback: (ack: inferZodSchemaOutput<GetSocketCallbackSchema<TCatalog, K>>) => void) => void : (message: inferZodSchemaOutput<GetSocketMessageSchema<TCatalog, K>>) => void;
};
export type GetSocketMessageSchema<TRPCCatalog extends ZodSocketMessageCatalogSchema, TMessageType extends keyof TRPCCatalog> = TRPCCatalog[TMessageType]["message"];
export type InferSocketMessageSchema<TRPCCatalog extends ZodSocketMessageCatalogSchema, TMessageType extends keyof TRPCCatalog> = inferZodSchemaOutput<GetSocketMessageSchema<TRPCCatalog, TMessageType>>;
export type GetSocketCallbackSchema<TRPCCatalog extends ZodSocketMessageCatalogSchema, TMessageType extends keyof TRPCCatalog> = TRPCCatalog[TMessageType] extends {
    callback: any;
} ? TRPCCatalog[TMessageType]["callback"] : never;
export type InferSocketCallbackSchema<TRPCCatalog extends ZodSocketMessageCatalogSchema, TMessageType extends keyof TRPCCatalog> = inferZodSchemaOutput<GetSocketCallbackSchema<TRPCCatalog, TMessageType>>;
export type SocketMessageHasCallback<TRPCCatalog extends ZodSocketMessageCatalogSchema, TMessageType extends keyof TRPCCatalog> = GetSocketCallbackSchema<TRPCCatalog, TMessageType> extends never ? false : true;
export type ZodSocketMessageHandlers<TCatalogSchema extends ZodSocketMessageCatalogSchema> = Partial<{
    [K in keyof TCatalogSchema]: (payload: inferZodSchemaOutput<GetSocketMessageSchema<TCatalogSchema, K>>) => Promise<SocketMessageHasCallback<TCatalogSchema, K> extends true ? inferZodSchemaInput<GetSocketCallbackSchema<TCatalogSchema, K>> : void>;
}>;
export type ZodSocketMessageHandlerOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
    schema: TMessageCatalog;
    handlers?: ZodSocketMessageHandlers<TMessageCatalog>;
    logger?: StructuredLogger;
    logPayloads?: boolean;
};
type MessageFromSocketSchema<K extends keyof TMessageCatalog, TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
    type: K;
    payload: inferZodSchemaInput<GetSocketMessageSchema<TMessageCatalog, K>>;
};
export type MessagesFromSocketCatalog<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
    [K in keyof TMessageCatalog]: MessageFromSocketSchema<K, TMessageCatalog>;
}[keyof TMessageCatalog];
export declare class ZodSocketMessageHandler<TRPCCatalog extends ZodSocketMessageCatalogSchema> {
    #private;
    constructor(options: ZodSocketMessageHandlerOptions<TRPCCatalog>);
    handleMessage(message: unknown): Promise<(SocketMessageHasCallback<TRPCCatalog, keyof TRPCCatalog> extends true ? inferZodSchemaInput<GetSocketCallbackSchema<TRPCCatalog, keyof TRPCCatalog>> : void) | undefined>;
    private parseMessage;
    registerHandlers(emitter: EventEmitterLike, logger?: StructuredLogger): void;
}
export type ZodSocketMessageSenderOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
    schema: TMessageCatalog;
    socket: ZodSocket<any, TMessageCatalog>;
    logger?: StructuredLogger;
};
export type GetSocketMessagesWithCallback<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
    [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true ? K : never;
}[keyof TMessageCatalog];
export type GetSocketMessagesWithoutCallback<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
    [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true ? never : K;
}[keyof TMessageCatalog];
export declare class ZodSocketMessageSender<TMessageCatalog extends ZodSocketMessageCatalogSchema> {
    #private;
    constructor(options: ZodSocketMessageSenderOptions<TMessageCatalog>);
    send<K extends GetSocketMessagesWithoutCallback<TMessageCatalog>>(type: K, payload: inferZodSchemaInput<GetSocketMessageSchema<TMessageCatalog, K>>): void;
    sendWithAck<K extends GetSocketMessagesWithCallback<TMessageCatalog>>(type: K, payload: inferZodSchemaInput<GetSocketMessageSchema<TMessageCatalog, K>>, timeout?: number): Promise<inferZodSchemaOutput<GetSocketCallbackSchema<TMessageCatalog, K>>>;
}
export type ZodSocket<TListenEvents extends ZodSocketMessageCatalogSchema, TEmitEvents extends ZodSocketMessageCatalogSchema> = Omit<Socket<ZodMessageCatalogToSocketIoEvents<TListenEvents>, ZodMessageCatalogToSocketIoEvents<TEmitEvents>>, "timeout"> & {
    timeout: (timeout: number) => Socket<ZodMessageCatalogToSocketIoEvents<TListenEvents>, ZodMessageCatalogToSocketIoEvents<TEmitEvents>>;
};
interface ZodSocketConnectionOptions<TClientMessages extends ZodSocketMessageCatalogSchema, TServerMessages extends ZodSocketMessageCatalogSchema> {
    host: string;
    port?: number;
    secure?: boolean;
    namespace: string;
    clientMessages: TClientMessages;
    serverMessages: TServerMessages;
    extraHeaders?: {
        [header: string]: string;
    };
    handlers?: ZodSocketMessageHandlers<TServerMessages>;
    authToken?: string;
    ioOptions?: Partial<ManagerOptions & SocketOptions>;
    logHandlerPayloads?: boolean;
    onConnection?: (socket: ZodSocket<TServerMessages, TClientMessages>, handler: ZodSocketMessageHandler<TServerMessages>, sender: ZodSocketMessageSender<TClientMessages>, logger: StructuredLogger) => Promise<void>;
    onDisconnect?: (socket: ZodSocket<TServerMessages, TClientMessages>, reason: Socket.DisconnectReason, description: any, logger: StructuredLogger) => Promise<void>;
    onError?: (socket: ZodSocket<TServerMessages, TClientMessages>, err: Error, logger: StructuredLogger) => Promise<void>;
}
export declare class ZodSocketConnection<TClientMessages extends ZodSocketMessageCatalogSchema, TServerMessages extends ZodSocketMessageCatalogSchema> {
    #private;
    socket: ZodSocket<TServerMessages, TClientMessages>;
    constructor(opts: ZodSocketConnectionOptions<TClientMessages, TServerMessages>);
    close(): void;
    connect(): void;
    get send(): <K extends GetSocketMessagesWithoutCallback<TClientMessages>>(type: K, payload: inferZodSchemaInput<GetSocketMessageSchema<TClientMessages, K>>) => void;
    get sendWithAck(): <K extends GetSocketMessagesWithCallback<TClientMessages>>(type: K, payload: inferZodSchemaInput<GetSocketMessageSchema<TClientMessages, K>>, timeout?: number) => Promise<inferZodSchemaOutput<GetSocketCallbackSchema<TClientMessages, K>>>;
}
export {};