@trigger.dev/core 0.0.0-v3-prerelease-20240813130521 → 0.0.0-v3-prerelease-20240913125036

package/dist/{v3/zodMessageHandler.d.mts → esm/v3/zodMessageHandler.d.ts}

@@ -1,31 +1,31 @@
-import { z } from 'zod';
-import { StructuredLogger } from './utils/structuredLogger.mjs';
-
-declare class ZodSchemaParsedError extends Error {
+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);
 }
-type ZodMessageValueSchema<TDiscriminatedUnion extends z.ZodDiscriminatedUnion<any, any>> = z.ZodFirstPartySchemaTypes | TDiscriminatedUnion;
-interface ZodMessageCatalogSchema {
+export type ZodMessageValueSchema<TDiscriminatedUnion extends z.ZodDiscriminatedUnion<any, any>> = z.ZodFirstPartySchemaTypes | TDiscriminatedUnion;
+export interface ZodMessageCatalogSchema {
     [key: string]: ZodMessageValueSchema<any>;
 }
-type ZodMessageHandlers<TCatalogSchema extends ZodMessageCatalogSchema> = Partial<{
+export type ZodMessageHandlers<TCatalogSchema extends ZodMessageCatalogSchema> = Partial<{
     [K in keyof TCatalogSchema]: (payload: z.infer<TCatalogSchema[K]>) => Promise<any>;
 }>;
-type ZodMessageHandlerOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {
+export type ZodMessageHandlerOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {
     schema: TMessageCatalog;
     messages?: ZodMessageHandlers<TMessageCatalog>;
     logger?: StructuredLogger;
 };
-type MessageFromSchema<K extends keyof TMessageCatalog, TMessageCatalog extends ZodMessageCatalogSchema> = {
+export type MessageFromSchema<K extends keyof TMessageCatalog, TMessageCatalog extends ZodMessageCatalogSchema> = {
     type: K;
     payload: z.input<TMessageCatalog[K]>;
 };
-type MessageFromCatalog<TMessageCatalog extends ZodMessageCatalogSchema> = {
+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];
-declare const ZodMessageSchema: z.ZodObject<{
+export declare const ZodMessageSchema: z.ZodObject<{
     version: z.ZodDefault<z.ZodLiteral<"v1">>;
     type: z.ZodString;
     payload: z.ZodUnknown;
@@ -38,10 +38,10 @@ declare const ZodMessageSchema: z.ZodObject<{
     version?: "v1" | undefined;
     payload?: unknown;
 }>;
-interface EventEmitterLike {
+export interface EventEmitterLike {
     on(eventName: string | symbol, listener: (...args: any[]) => void): this;
 }
-declare class ZodMessageHandler<TMessageCatalog extends ZodMessageCatalogSchema> {
+export declare class ZodMessageHandler<TMessageCatalog extends ZodMessageCatalogSchema> {
     #private;
     constructor(options: ZodMessageHandlerOptions<TMessageCatalog>);
     handleMessage(message: unknown): Promise<{
@@ -60,23 +60,24 @@ declare class ZodMessageHandler<TMessageCatalog extends ZodMessageCatalogSchema>
     };
     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>;
-type ZodMessageSenderOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {
+export type ZodMessageSenderOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {
     schema: TMessageCatalog;
     sender: ZodMessageSenderCallback<TMessageCatalog>;
 };
-declare class ZodMessageSender<TMessageCatalog extends ZodMessageCatalogSchema> {
+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>;
 }
-type MessageCatalogToSocketIoEvents<TCatalog extends ZodMessageCatalogSchema> = {
+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 { type EventEmitterLike, type MessageCatalogToSocketIoEvents, type MessageFromCatalog, type MessageFromSchema, type ZodMessageCatalogSchema, ZodMessageHandler, type ZodMessageHandlerOptions, type ZodMessageHandlers, ZodMessageSchema, ZodMessageSender, type ZodMessageSenderOptions, type ZodMessageValueSchema, ZodSchemaParsedError };
+export {};

package/dist/esm/v3/zodMessageHandler.js

@@ -0,0 +1,197 @@
+import { z } from "zod";
+export class ZodSchemaParsedError extends Error {
+    error;
+    payload;
+    constructor(error, payload) {
+        super(error.message);
+        this.error = error;
+        this.payload = payload;
+    }
+}
+export const ZodMessageSchema = z.object({
+    version: z.literal("v1").default("v1"),
+    type: z.string(),
+    payload: z.unknown(),
+});
+export 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 = 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);
+                }
+            });
+        }
+    }
+}
+export function parseMessageFromCatalog(message, schema) {
+    const parsedMessage = 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,
+    };
+}
+export 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 = 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);
+        }
+    }
+}
+export 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/esm/v3/zodMessageHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zodMessageHandler.js","sourceRoot":"","sources":["../../../src/v3/zodMessageHandler.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,MAAM,OAAO,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;AAqCD,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;IACtC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE;CACrB,CAAC,CAAC;AAMH,MAAM,OAAO,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,gBAAgB,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;AAED,MAAM,UAAU,uBAAuB,CACrC,OAAgB,EAChB,MAAuB;IAEvB,MAAM,aAAa,GAAG,gBAAgB,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,MAAM,OAAO,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,gBAAgB,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;AAED,MAAM,CAAC,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/esm/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/esm/v3/zodNamespace.js

@@ -0,0 +1,92 @@
+import { ZodMessageSender } from "./zodMessageHandler.js";
+import { ZodSocketMessageHandler, } from "./zodSocket.js";
+import { SimpleStructuredLogger } from "./utils/structuredLogger.js";
+export class ZodNamespace {
+    #logger;
+    #handler;
+    sender;
+    io;
+    namespace;
+    constructor(opts) {
+        this.#logger = opts.logger ?? new SimpleStructuredLogger(opts.name);
+        this.#handler = new 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 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();
+    }
+}
+//# sourceMappingURL=zodNamespace.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"zodNamespace.js","sourceRoot":"","sources":["../../../src/v3/zodNamespace.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,EAGL,uBAAuB,GAExB,MAAM,gBAAgB,CAAC;AAIxB,OAAO,EAAE,sBAAsB,EAAoB,MAAM,6BAA6B,CAAC;AA6DvF,MAAM,OAAO,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,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEpE,IAAI,CAAC,QAAQ,GAAG,IAAI,uBAAuB,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,gBAAgB,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"}

package/dist/{v3/zodSocket.d.mts → esm/v3/zodSocket.d.ts}

@@ -1,9 +1,8 @@
-import { Socket, ManagerOptions, SocketOptions } from 'socket.io-client';
-import { z } from 'zod';
-import { ZodMessageValueSchema, EventEmitterLike } from './zodMessageHandler.mjs';
-import { StructuredLogger } from './utils/structuredLogger.mjs';
-
-interface ZodSocketMessageCatalogSchema {
+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>;
     } | {
@@ -11,20 +10,20 @@ interface ZodSocketMessageCatalogSchema {
         callback?: ZodMessageValueSchema<any>;
     };
 }
-type ZodMessageCatalogToSocketIoEvents<TCatalog extends ZodSocketMessageCatalogSchema> = {
+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;
 };
-type GetSocketMessageSchema<TRPCCatalog extends ZodSocketMessageCatalogSchema, TMessageType extends keyof TRPCCatalog> = TRPCCatalog[TMessageType]["message"];
-type InferSocketMessageSchema<TRPCCatalog extends ZodSocketMessageCatalogSchema, TMessageType extends keyof TRPCCatalog> = z.infer<GetSocketMessageSchema<TRPCCatalog, TMessageType>>;
-type GetSocketCallbackSchema<TRPCCatalog extends ZodSocketMessageCatalogSchema, TMessageType extends keyof TRPCCatalog> = TRPCCatalog[TMessageType] extends {
+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;
-type InferSocketCallbackSchema<TRPCCatalog extends ZodSocketMessageCatalogSchema, TMessageType extends keyof TRPCCatalog> = z.infer<GetSocketCallbackSchema<TRPCCatalog, TMessageType>>;
-type SocketMessageHasCallback<TRPCCatalog extends ZodSocketMessageCatalogSchema, TMessageType extends keyof TRPCCatalog> = GetSocketCallbackSchema<TRPCCatalog, TMessageType> extends never ? false : true;
-type ZodSocketMessageHandlers<TCatalogSchema extends ZodSocketMessageCatalogSchema> = Partial<{
+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>;
 }>;
-type ZodSocketMessageHandlerOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
+export type ZodSocketMessageHandlerOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
     schema: TMessageCatalog;
     handlers?: ZodSocketMessageHandlers<TMessageCatalog>;
     logger?: StructuredLogger;
@@ -33,34 +32,34 @@ type MessageFromSocketSchema<K extends keyof TMessageCatalog, TMessageCatalog ex
     type: K;
     payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>;
 };
-type MessagesFromSocketCatalog<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
+export type MessagesFromSocketCatalog<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
     [K in keyof TMessageCatalog]: MessageFromSocketSchema<K, TMessageCatalog>;
 }[keyof TMessageCatalog];
-declare class ZodSocketMessageHandler<TRPCCatalog extends ZodSocketMessageCatalogSchema> {
+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;
 }
-type ZodSocketMessageSenderOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
+export type ZodSocketMessageSenderOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
     schema: TMessageCatalog;
     socket: ZodSocket<any, TMessageCatalog>;
     logger?: StructuredLogger;
 };
-type GetSocketMessagesWithCallback<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
+export type GetSocketMessagesWithCallback<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
     [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true ? K : never;
 }[keyof TMessageCatalog];
-type GetSocketMessagesWithoutCallback<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
+export type GetSocketMessagesWithoutCallback<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {
     [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true ? never : K;
 }[keyof TMessageCatalog];
-declare class ZodSocketMessageSender<TMessageCatalog extends ZodSocketMessageCatalogSchema> {
+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>>>;
 }
-type ZodSocket<TListenEvents extends ZodSocketMessageCatalogSchema, TEmitEvents extends ZodSocketMessageCatalogSchema> = Omit<Socket<ZodMessageCatalogToSocketIoEvents<TListenEvents>, ZodMessageCatalogToSocketIoEvents<TEmitEvents>>, "timeout"> & {
+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> {
@@ -80,7 +79,7 @@ interface ZodSocketConnectionOptions<TClientMessages extends ZodSocketMessageCat
     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>;
 }
-declare class ZodSocketConnection<TClientMessages extends ZodSocketMessageCatalogSchema, TServerMessages extends ZodSocketMessageCatalogSchema> {
+export declare class ZodSocketConnection<TClientMessages extends ZodSocketMessageCatalogSchema, TServerMessages extends ZodSocketMessageCatalogSchema> {
     #private;
     socket: ZodSocket<TServerMessages, TClientMessages>;
     constructor(opts: ZodSocketConnectionOptions<TClientMessages, TServerMessages>);
@@ -89,5 +88,4 @@ declare class ZodSocketConnection<TClientMessages extends ZodSocketMessageCatalo
     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 { type GetSocketCallbackSchema, type GetSocketMessageSchema, type GetSocketMessagesWithCallback, type GetSocketMessagesWithoutCallback, type InferSocketCallbackSchema, type InferSocketMessageSchema, type MessagesFromSocketCatalog, type SocketMessageHasCallback, type ZodMessageCatalogToSocketIoEvents, type ZodSocket, ZodSocketConnection, type ZodSocketMessageCatalogSchema, ZodSocketMessageHandler, type ZodSocketMessageHandlerOptions, type ZodSocketMessageHandlers, ZodSocketMessageSender, type ZodSocketMessageSenderOptions };
+export {};