@trigger.dev/core 3.0.0-beta.7 → 3.0.1

package/dist/commonjs/v3/zodMessageHandler.d.ts

@@ -0,0 +1,83 @@
+import { z } from "zod";
+import { StructuredLogger } from "./utils/structuredLogger.js";
+export declare class ZodSchemaParsedError extends Error {
+    error: z.ZodError;
+    payload: unknown;
+    constructor(error: z.ZodError, payload: unknown);
+}
+export type ZodMessageValueSchema<TDiscriminatedUnion extends z.ZodDiscriminatedUnion<any, any>> = z.ZodFirstPartySchemaTypes | TDiscriminatedUnion;
+export interface ZodMessageCatalogSchema {
+    [key: string]: ZodMessageValueSchema<any>;
+}
+export type ZodMessageHandlers<TCatalogSchema extends ZodMessageCatalogSchema> = Partial<{
+    [K in keyof TCatalogSchema]: (payload: z.infer<TCatalogSchema[K]>) => Promise<any>;
+}>;
+export type ZodMessageHandlerOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {
+    schema: TMessageCatalog;
+    messages?: ZodMessageHandlers<TMessageCatalog>;
+    logger?: StructuredLogger;
+};
+export type MessageFromSchema<K extends keyof TMessageCatalog, TMessageCatalog extends ZodMessageCatalogSchema> = {
+    type: K;
+    payload: z.input<TMessageCatalog[K]>;
+};
+export type MessagePayloadFromSchema<K extends keyof TMessageCatalog, TMessageCatalog extends ZodMessageCatalogSchema> = z.output<TMessageCatalog[K]>;
+export type MessageFromCatalog<TMessageCatalog extends ZodMessageCatalogSchema> = {
+    [K in keyof TMessageCatalog]: MessageFromSchema<K, TMessageCatalog>;
+}[keyof TMessageCatalog];
+export declare const ZodMessageSchema: z.ZodObject<{
+    version: z.ZodDefault<z.ZodLiteral<"v1">>;
+    type: z.ZodString;
+    payload: z.ZodUnknown;
+}, "strip", z.ZodTypeAny, {
+    type: string;
+    version: "v1";
+    payload?: unknown;
+}, {
+    type: string;
+    version?: "v1" | undefined;
+    payload?: unknown;
+}>;
+export interface EventEmitterLike {
+    on(eventName: string | symbol, listener: (...args: any[]) => void): this;
+}
+export declare class ZodMessageHandler<TMessageCatalog extends ZodMessageCatalogSchema> {
+    #private;
+    constructor(options: ZodMessageHandlerOptions<TMessageCatalog>);
+    handleMessage(message: unknown): Promise<{
+        success: true;
+        data: unknown;
+    } | {
+        success: false;
+        error: string;
+    }>;
+    parseMessage(message: unknown): {
+        success: true;
+        data: MessageFromCatalog<TMessageCatalog>;
+    } | {
+        success: false;
+        error: string;
+    };
+    registerHandlers(emitter: EventEmitterLike, logger?: StructuredLogger): void;
+}
+export declare function parseMessageFromCatalog<TMessageCatalog extends ZodMessageCatalogSchema>(message: unknown, schema: TMessageCatalog): MessageFromCatalog<TMessageCatalog>;
+type ZodMessageSenderCallback<TMessageCatalog extends ZodMessageCatalogSchema> = (message: {
+    type: keyof TMessageCatalog;
+    payload: z.infer<TMessageCatalog[keyof TMessageCatalog]>;
+    version: "v1";
+}) => Promise<void>;
+export type ZodMessageSenderOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {
+    schema: TMessageCatalog;
+    sender: ZodMessageSenderCallback<TMessageCatalog>;
+};
+export declare class ZodMessageSender<TMessageCatalog extends ZodMessageCatalogSchema> {
+    #private;
+    constructor(options: ZodMessageSenderOptions<TMessageCatalog>);
+    send<K extends keyof TMessageCatalog>(type: K, payload: z.input<TMessageCatalog[K]>): Promise<void>;
+    forwardMessage(message: unknown): Promise<void>;
+}
+export declare function sendMessageInCatalog<TMessageCatalog extends ZodMessageCatalogSchema>(catalog: TMessageCatalog, type: keyof TMessageCatalog, payload: z.input<TMessageCatalog[keyof TMessageCatalog]>, sender: ZodMessageSenderCallback<TMessageCatalog>): Promise<void>;
+export type MessageCatalogToSocketIoEvents<TCatalog extends ZodMessageCatalogSchema> = {
+    [K in keyof TCatalog]: (message: z.infer<TCatalog[K]>) => void;
+};
+export {};

package/dist/commonjs/v3/zodMessageHandler.js

@@ -0,0 +1,205 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ZodMessageSender = exports.ZodMessageHandler = exports.ZodMessageSchema = exports.ZodSchemaParsedError = void 0;
+exports.parseMessageFromCatalog = parseMessageFromCatalog;
+exports.sendMessageInCatalog = sendMessageInCatalog;
+const zod_1 = require("zod");
+class ZodSchemaParsedError extends Error {
+    error;
+    payload;
+    constructor(error, payload) {
+        super(error.message);
+        this.error = error;
+        this.payload = payload;
+    }
+}
+exports.ZodSchemaParsedError = ZodSchemaParsedError;
+exports.ZodMessageSchema = zod_1.z.object({
+    version: zod_1.z.literal("v1").default("v1"),
+    type: zod_1.z.string(),
+    payload: zod_1.z.unknown(),
+});
+class ZodMessageHandler {
+    #schema;
+    #handlers;
+    #logger;
+    constructor(options) {
+        this.#schema = options.schema;
+        this.#handlers = options.messages;
+        this.#logger = options.logger ?? console;
+    }
+    async handleMessage(message) {
+        const parsedMessage = this.parseMessage(message);
+        if (!parsedMessage.success) {
+            this.#logger.error(parsedMessage.error, { message });
+            return {
+                success: false,
+                error: parsedMessage.error,
+            };
+        }
+        if (!this.#handlers) {
+            this.#logger.error("No handlers provided", { message });
+            return {
+                success: false,
+                error: "No handlers provided",
+            };
+        }
+        const handler = this.#handlers[parsedMessage.data.type];
+        if (!handler) {
+            const error = `No handler for message type: ${String(parsedMessage.data.type)}`;
+            this.#logger.error(error, { message });
+            return {
+                success: false,
+                error,
+            };
+        }
+        const ack = await handler(parsedMessage.data.payload);
+        return {
+            success: true,
+            data: ack,
+        };
+    }
+    parseMessage(message) {
+        const parsedMessage = exports.ZodMessageSchema.safeParse(message);
+        if (!parsedMessage.success) {
+            return {
+                success: false,
+                error: `Failed to parse message: ${JSON.stringify(parsedMessage.error)}`,
+            };
+        }
+        const schema = this.#schema[parsedMessage.data.type];
+        if (!schema) {
+            return {
+                success: false,
+                error: `Unknown message type: ${parsedMessage.data.type}`,
+            };
+        }
+        const parsedPayload = schema.safeParse(parsedMessage.data.payload);
+        if (!parsedPayload.success) {
+            return {
+                success: false,
+                error: `Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`,
+            };
+        }
+        return {
+            success: true,
+            data: {
+                type: parsedMessage.data.type,
+                payload: parsedPayload.data,
+            },
+        };
+    }
+    registerHandlers(emitter, logger) {
+        const log = logger ?? console;
+        if (!this.#handlers) {
+            log.info("No handlers provided");
+            return;
+        }
+        for (const eventName of Object.keys(this.#schema)) {
+            emitter.on(eventName, async (message, callback) => {
+                log.info(`handling ${eventName}`, {
+                    payload: message,
+                    hasCallback: !!callback,
+                });
+                let ack;
+                // FIXME: this only works if the message doesn't have genuine payload prop
+                if ("payload" in message) {
+                    ack = await this.handleMessage({ type: eventName, ...message });
+                }
+                else {
+                    // Handle messages not sent by ZodMessageSender
+                    const { version, ...payload } = message;
+                    ack = await this.handleMessage({ type: eventName, version, payload });
+                }
+                if (callback && typeof callback === "function") {
+                    if (!ack.success) {
+                        // We don't know the callback type, so we can't do anything else - not all callbacks may accept a success prop
+                        log.error("Failed to handle message, skipping callback", { message, error: ack.error });
+                        return;
+                    }
+                    callback(ack.data);
+                }
+            });
+        }
+    }
+}
+exports.ZodMessageHandler = ZodMessageHandler;
+function parseMessageFromCatalog(message, schema) {
+    const parsedMessage = exports.ZodMessageSchema.safeParse(message);
+    if (!parsedMessage.success) {
+        throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);
+    }
+    const messageSchema = schema[parsedMessage.data.type];
+    if (!messageSchema) {
+        throw new Error(`Unknown message type: ${parsedMessage.data.type}`);
+    }
+    const parsedPayload = messageSchema.safeParse(parsedMessage.data.payload);
+    if (!parsedPayload.success) {
+        throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
+    }
+    return {
+        type: parsedMessage.data.type,
+        payload: parsedPayload.data,
+    };
+}
+class ZodMessageSender {
+    #schema;
+    #sender;
+    constructor(options) {
+        this.#schema = options.schema;
+        this.#sender = options.sender;
+    }
+    async send(type, payload) {
+        const schema = this.#schema[type];
+        if (!schema) {
+            throw new Error(`Unknown message type: ${type}`);
+        }
+        const parsedPayload = schema.safeParse(payload);
+        if (!parsedPayload.success) {
+            throw new ZodSchemaParsedError(parsedPayload.error, payload);
+        }
+        try {
+            await this.#sender({ type, payload, version: "v1" });
+        }
+        catch (error) {
+            console.error("[ZodMessageSender] Failed to send message", error);
+        }
+    }
+    async forwardMessage(message) {
+        const parsedMessage = exports.ZodMessageSchema.safeParse(message);
+        if (!parsedMessage.success) {
+            throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);
+        }
+        const schema = this.#schema[parsedMessage.data.type];
+        if (!schema) {
+            throw new Error(`Unknown message type: ${parsedMessage.data.type}`);
+        }
+        const parsedPayload = schema.safeParse(parsedMessage.data.payload);
+        if (!parsedPayload.success) {
+            throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
+        }
+        try {
+            await this.#sender({
+                type: parsedMessage.data.type,
+                payload: parsedPayload.data,
+                version: "v1",
+            });
+        }
+        catch (error) {
+            console.error("[ZodMessageSender] Failed to forward message", error);
+        }
+    }
+}
+exports.ZodMessageSender = ZodMessageSender;
+async function sendMessageInCatalog(catalog, type, payload, sender) {
+    const schema = catalog[type];
+    if (!schema) {
+        throw new Error(`Unknown message type: ${type}`);
+    }
+    const parsedPayload = schema.safeParse(payload);
+    if (!parsedPayload.success) {
+        throw new ZodSchemaParsedError(parsedPayload.error, payload);
+    }
+    await sender({ type, payload, version: "v1" });
+}
+//# sourceMappingURL=zodMessageHandler.js.map

package/dist/commonjs/v3/zodMessageHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zodMessageHandler.js","sourceRoot":"","sources":["../../../src/v3/zodMessageHandler.ts"],"names":[],"mappings":";;;AA4MA,0DA0BC;AA4ED,oDAmBC;AArUD,6BAAwB;AAGxB,MAAa,oBAAqB,SAAQ,KAAK;IAEpC;IACA;IAFT,YACS,KAAiB,EACjB,OAAgB;QAEvB,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAHd,UAAK,GAAL,KAAK,CAAY;QACjB,YAAO,GAAP,OAAO,CAAS;IAGzB,CAAC;CACF;AAPD,oDAOC;AAqCY,QAAA,gBAAgB,GAAG,OAAC,CAAC,MAAM,CAAC;IACvC,OAAO,EAAE,OAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;IACtC,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;IAChB,OAAO,EAAE,OAAC,CAAC,OAAO,EAAE;CACrB,CAAC,CAAC;AAMH,MAAa,iBAAiB;IAC5B,OAAO,CAAkB;IACzB,SAAS,CAAkD;IAC3D,OAAO,CAA6B;IAEpC,YAAY,OAAkD;QAC5D,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9B,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,QAAQ,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC;IAC3C,CAAC;IAEM,KAAK,CAAC,aAAa,CAAC,OAAgB;QAUzC,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAEjD,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YAC3B,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErD,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,aAAa,CAAC,KAAK;aAC3B,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAExD,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,sBAAsB;aAC9B,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAExD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,KAAK,GAAG,gCAAgC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAEhF,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAEvC,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK;aACN,CAAC;QACJ,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEtD,OAAO;YACL,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,GAAG;SACV,CAAC;IACJ,CAAC;IAEM,YAAY,CAAC,OAAgB;QASlC,MAAM,aAAa,GAAG,wBAAgB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAE1D,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YAC3B,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,4BAA4B,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE;aACzE,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAErD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,yBAAyB,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE;aAC1D,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEnE,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YAC3B,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,oCAAoC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE;aACjF,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,IAAI,EAAE;gBACJ,IAAI,EAAE,aAAa,CAAC,IAAI,CAAC,IAAI;gBAC7B,OAAO,EAAE,aAAa,CAAC,IAAI;aAC5B;SACF,CAAC;IACJ,CAAC;IAEM,gBAAgB,CAAC,OAAyB,EAAE,MAAyB;QAC1E,MAAM,GAAG,GAAG,MAAM,IAAI,OAAO,CAAC;QAE9B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,GAAG,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YACjC,OAAO;QACT,CAAC;QAED,KAAK,MAAM,SAAS,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;YAClD,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,OAAY,EAAE,QAAc,EAAiB,EAAE;gBAC1E,GAAG,CAAC,IAAI,CAAC,YAAY,SAAS,EAAE,EAAE;oBAChC,OAAO,EAAE,OAAO;oBAChB,WAAW,EAAE,CAAC,CAAC,QAAQ;iBACxB,CAAC,CAAC;gBAEH,IAAI,GAA6E,CAAC;gBAElF,0EAA0E;gBAC1E,IAAI,SAAS,IAAI,OAAO,EAAE,CAAC;oBACzB,GAAG,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;gBAClE,CAAC;qBAAM,CAAC;oBACN,+CAA+C;oBAC/C,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,GAAG,OAAO,CAAC;oBACxC,GAAG,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBACxE,CAAC;gBAED,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;oBAC/C,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;wBACjB,8GAA8G;wBAC9G,GAAG,CAAC,KAAK,CAAC,6CAA6C,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC;wBACxF,OAAO;oBACT,CAAC;oBAED,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBACrB,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;CACF;AAjJD,8CAiJC;AAED,SAAgB,uBAAuB,CACrC,OAAgB,EAChB,MAAuB;IAEvB,MAAM,aAAa,GAAG,wBAAgB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAE1D,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACrF,CAAC;IAED,MAAM,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEtD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,yBAAyB,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,aAAa,GAAG,aAAa,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAE1E,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,oCAAoC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC7F,CAAC;IAED,OAAO;QACL,IAAI,EAAE,aAAa,CAAC,IAAI,CAAC,IAAI;QAC7B,OAAO,EAAE,aAAa,CAAC,IAAI;KAC5B,CAAC;AACJ,CAAC;AAaD,MAAa,gBAAgB;IAC3B,OAAO,CAAkB;IACzB,OAAO,CAA4C;IAEnD,YAAY,OAAiD;QAC3D,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;IAChC,CAAC;IAEM,KAAK,CAAC,IAAI,CACf,IAAO,EACP,OAAoC;QAEpC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAElC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,yBAAyB,IAAc,EAAE,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAEhD,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YAC3B,MAAM,IAAI,oBAAoB,CAAC,aAAa,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/D,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2CAA2C,EAAE,KAAK,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,cAAc,CAAC,OAAgB;QAC1C,MAAM,aAAa,GAAG,wBAAgB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAE1D,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACrF,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAErD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,yBAAyB,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEnE,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,oCAAoC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC7F,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC;gBACjB,IAAI,EAAE,aAAa,CAAC,IAAI,CAAC,IAAI;gBAC7B,OAAO,EAAE,aAAa,CAAC,IAAI;gBAC3B,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,8CAA8C,EAAE,KAAK,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;CACF;AA7DD,4CA6DC;AAEM,KAAK,UAAU,oBAAoB,CACxC,OAAwB,EACxB,IAA2B,EAC3B,OAAwD,EACxD,MAAiD;IAEjD,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAE7B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,yBAAyB,IAAc,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,MAAM,aAAa,GAAG,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAEhD,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;QAC3B,MAAM,IAAI,oBAAoB,CAAC,aAAa,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAC/D,CAAC;IAED,MAAM,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;AACjD,CAAC"}

package/dist/commonjs/v3/zodNamespace.d.ts

@@ -0,0 +1,34 @@
+import type { DisconnectReason, Namespace, Server, Socket } from "socket.io";
+import { ZodMessageSender } from "./zodMessageHandler.js";
+import { ZodMessageCatalogToSocketIoEvents, ZodSocketMessageCatalogSchema, ZodSocketMessageHandler, ZodSocketMessageHandlers } from "./zodSocket.js";
+import type { DefaultEventsMap, EventsMap } from "socket.io/dist/typed-events";
+import { z } from "zod";
+import { StructuredLogger } from "./utils/structuredLogger.js";
+interface ExtendedError extends Error {
+    data?: any;
+}
+export type ZodNamespaceSocket<TClientMessages extends ZodSocketMessageCatalogSchema, TServerMessages extends ZodSocketMessageCatalogSchema, TServerSideEvents extends EventsMap = DefaultEventsMap, TSocketData extends z.ZodObject<any, any, any> = any> = Socket<ZodMessageCatalogToSocketIoEvents<TClientMessages>, ZodMessageCatalogToSocketIoEvents<TServerMessages>, TServerSideEvents, z.infer<TSocketData>>;
+interface ZodNamespaceOptions<TClientMessages extends ZodSocketMessageCatalogSchema, TServerMessages extends ZodSocketMessageCatalogSchema, TServerSideEvents extends EventsMap = DefaultEventsMap, TSocketData extends z.ZodObject<any, any, any> = any> {
+    io: Server;
+    name: string;
+    clientMessages: TClientMessages;
+    serverMessages: TServerMessages;
+    socketData?: TSocketData;
+    handlers?: ZodSocketMessageHandlers<TClientMessages>;
+    authToken?: string;
+    logger?: StructuredLogger;
+    preAuth?: (socket: ZodNamespaceSocket<TClientMessages, TServerMessages, TServerSideEvents, TSocketData>, next: (err?: ExtendedError) => void, logger: StructuredLogger) => Promise<void>;
+    postAuth?: (socket: ZodNamespaceSocket<TClientMessages, TServerMessages, TServerSideEvents, TSocketData>, next: (err?: ExtendedError) => void, logger: StructuredLogger) => Promise<void>;
+    onConnection?: (socket: ZodNamespaceSocket<TClientMessages, TServerMessages, TServerSideEvents, TSocketData>, handler: ZodSocketMessageHandler<TClientMessages>, sender: ZodMessageSender<TServerMessages>, logger: StructuredLogger) => Promise<void>;
+    onDisconnect?: (socket: ZodNamespaceSocket<TClientMessages, TServerMessages, TServerSideEvents, TSocketData>, reason: DisconnectReason, description: any, logger: StructuredLogger) => Promise<void>;
+    onError?: (socket: ZodNamespaceSocket<TClientMessages, TServerMessages, TServerSideEvents, TSocketData>, err: Error, logger: StructuredLogger) => Promise<void>;
+}
+export declare class ZodNamespace<TClientMessages extends ZodSocketMessageCatalogSchema, TServerMessages extends ZodSocketMessageCatalogSchema, TSocketData extends z.ZodObject<any, any, any> = any, TServerSideEvents extends EventsMap = DefaultEventsMap> {
+    #private;
+    sender: ZodMessageSender<TServerMessages>;
+    io: Server;
+    namespace: Namespace<ZodMessageCatalogToSocketIoEvents<TClientMessages>, ZodMessageCatalogToSocketIoEvents<TServerMessages>, TServerSideEvents, z.infer<TSocketData>>;
+    constructor(opts: ZodNamespaceOptions<TClientMessages, TServerMessages, TServerSideEvents, TSocketData>);
+    fetchSockets(): Promise<import("socket.io").RemoteSocket<ZodMessageCatalogToSocketIoEvents<TServerMessages>, z.TypeOf<TSocketData>>[]>;
+}
+export {};

package/dist/commonjs/v3/zodNamespace.js

@@ -0,0 +1,96 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ZodNamespace = void 0;
+const zodMessageHandler_js_1 = require("./zodMessageHandler.js");
+const zodSocket_js_1 = require("./zodSocket.js");
+const structuredLogger_js_1 = require("./utils/structuredLogger.js");
+class ZodNamespace {
+    #logger;
+    #handler;
+    sender;
+    io;
+    namespace;
+    constructor(opts) {
+        this.#logger = opts.logger ?? new structuredLogger_js_1.SimpleStructuredLogger(opts.name);
+        this.#handler = new zodSocket_js_1.ZodSocketMessageHandler({
+            schema: opts.clientMessages,
+            handlers: opts.handlers,
+        });
+        this.io = opts.io;
+        this.namespace = this.io.of(opts.name);
+        // FIXME: There's a bug here, this sender should not accept Socket schemas with callbacks
+        this.sender = new zodMessageHandler_js_1.ZodMessageSender({
+            schema: opts.serverMessages,
+            sender: async (message) => {
+                return new Promise((resolve, reject) => {
+                    try {
+                        // @ts-expect-error
+                        this.namespace.emit(message.type, message.payload);
+                        resolve();
+                    }
+                    catch (err) {
+                        reject(err);
+                    }
+                });
+            },
+        });
+        if (opts.preAuth) {
+            this.namespace.use(async (socket, next) => {
+                const logger = this.#logger.child({ socketId: socket.id, socketStage: "preAuth" });
+                if (typeof opts.preAuth === "function") {
+                    await opts.preAuth(socket, next, logger);
+                }
+            });
+        }
+        if (opts.authToken) {
+            this.namespace.use((socket, next) => {
+                const logger = this.#logger.child({ socketId: socket.id, socketStage: "auth" });
+                const { auth } = socket.handshake;
+                if (!("token" in auth)) {
+                    logger.error("no token");
+                    return socket.disconnect(true);
+                }
+                if (auth.token !== opts.authToken) {
+                    logger.error("invalid token");
+                    return socket.disconnect(true);
+                }
+                logger.info("success");
+                next();
+                return;
+            });
+        }
+        if (opts.postAuth) {
+            this.namespace.use(async (socket, next) => {
+                const logger = this.#logger.child({ socketId: socket.id, socketStage: "auth" });
+                if (typeof opts.postAuth === "function") {
+                    await opts.postAuth(socket, next, logger);
+                }
+            });
+        }
+        this.namespace.on("connection", async (socket) => {
+            const logger = this.#logger.child({ socketId: socket.id, socketStage: "connection" });
+            logger.info("connected");
+            this.#handler.registerHandlers(socket, logger);
+            socket.on("disconnect", async (reason, description) => {
+                logger.info("disconnect", { reason, description });
+                if (opts.onDisconnect) {
+                    await opts.onDisconnect(socket, reason, description, logger);
+                }
+            });
+            socket.on("error", async (error) => {
+                logger.error("error", { error });
+                if (opts.onError) {
+                    await opts.onError(socket, error, logger);
+                }
+            });
+            if (opts.onConnection) {
+                await opts.onConnection(socket, this.#handler, this.sender, logger);
+            }
+        });
+    }
+    fetchSockets() {
+        return this.namespace.fetchSockets();
+    }
+}
+exports.ZodNamespace = ZodNamespace;
+//# sourceMappingURL=zodNamespace.js.map

package/dist/commonjs/v3/zodNamespace.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zodNamespace.js","sourceRoot":"","sources":["../../../src/v3/zodNamespace.ts"],"names":[],"mappings":";;;AACA,iEAA0D;AAC1D,iDAKwB;AAIxB,qEAAuF;AA6DvF,MAAa,YAAY;IAMvB,OAAO,CAAmB;IAC1B,QAAQ,CAA2C;IACnD,MAAM,CAAoC;IAE1C,EAAE,CAAS;IACX,SAAS,CAKP;IAEF,YACE,IAA2F;QAE3F,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,4CAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEpE,IAAI,CAAC,QAAQ,GAAG,IAAI,sCAAuB,CAAC;YAC1C,MAAM,EAAE,IAAI,CAAC,cAAc;YAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC;QAElB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEvC,yFAAyF;QACzF,IAAI,CAAC,MAAM,GAAG,IAAI,uCAAgB,CAAC;YACjC,MAAM,EAAE,IAAI,CAAC,cAAc;YAC3B,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBACxB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBACrC,IAAI,CAAC;wBACH,mBAAmB;wBACnB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;wBACnD,OAAO,EAAE,CAAC;oBACZ,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,MAAM,CAAC,GAAG,CAAC,CAAC;oBACd,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE;gBACxC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAC;gBAEnF,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;oBACvC,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;gBAC3C,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE;gBAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC;gBAEhF,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC,SAAS,CAAC;gBAElC,IAAI,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,EAAE,CAAC;oBACvB,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;oBACzB,OAAO,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBACjC,CAAC;gBAED,IAAI,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,SAAS,EAAE,CAAC;oBAClC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;oBAC9B,OAAO,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBACjC,CAAC;gBAED,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAEvB,IAAI,EAAE,CAAC;gBAEP,OAAO;YACT,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE;gBACxC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC;gBAEhF,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;oBACxC,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC,CAAC;YACtF,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAEzB,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAE/C,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE;gBACpD,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;gBAEnD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtB,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;gBACjC,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBAEjC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;oBACjB,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACtE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC;IACvC,CAAC;CACF;AA3HD,oCA2HC"}

package/dist/commonjs/v3/zodSocket.d.ts

@@ -0,0 +1,91 @@
+import type { ManagerOptions, Socket, SocketOptions } from "socket.io-client";
+import { z } from "zod";
+import { EventEmitterLike, ZodMessageValueSchema } from "./zodMessageHandler.js";
+import { StructuredLogger } from "./utils/structuredLogger.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: z.infer<GetSocketMessageSchema<TCatalog, K>>, callback: (ack: z.infer<GetSocketCallbackSchema<TCatalog, K>>) => void) => void : (message: z.infer<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> = z.infer<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> = z.infer<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: z.infer<GetSocketMessageSchema<TCatalogSchema, K>>) => Promise<SocketMessageHasCallback<TCatalogSchema, K> extends true ? z.input<GetSocketCallbackSchema<TCatalogSchema, K>> : void>;
+}>;
+export type ZodSocketMessageHandlerOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
+    schema: TMessageCatalog;
+    handlers?: ZodSocketMessageHandlers<TMessageCatalog>;
+    logger?: StructuredLogger;
+};
+type MessageFromSocketSchema<K extends keyof TMessageCatalog, TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
+    type: K;
+    payload: z.input<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 ? z.input<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: z.input<GetSocketMessageSchema<TMessageCatalog, K>>): void;
+    sendWithAck<K extends GetSocketMessagesWithCallback<TMessageCatalog>>(type: K, payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>): Promise<z.infer<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>;
+    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: z.input<GetSocketMessageSchema<TClientMessages, K>>) => void;
+    get sendWithAck(): <K extends GetSocketMessagesWithCallback<TClientMessages>>(type: K, payload: z.input<GetSocketMessageSchema<TClientMessages, K>>) => Promise<z.TypeOf<GetSocketCallbackSchema<TClientMessages, K>>>;
+}
+export {};