@trigger.dev/core 0.0.0-prerelease-20240724110801 → 0.0.0-prerelease-20240825150620

package/dist/v3/zodIpc.js

@@ -1,268 +0,0 @@
-'use strict';
-
-var crypto = require('crypto');
-var zod = require('zod');
-
-var __defProp = Object.defineProperty;
-var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-var __accessCheck = (obj, member, msg) => {
-  if (!member.has(obj))
-    throw TypeError("Cannot " + msg);
-};
-var __privateGet = (obj, member, getter) => {
-  __accessCheck(obj, member, "read from private field");
-  return getter ? getter.call(obj) : member.get(obj);
-};
-var __privateAdd = (obj, member, value) => {
-  if (member.has(obj))
-    throw TypeError("Cannot add the same private member more than once");
-  member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
-};
-var __privateSet = (obj, member, value, setter) => {
-  __accessCheck(obj, member, "write to private field");
-  setter ? setter.call(obj, value) : member.set(obj, value);
-  return value;
-};
-var __privateWrapper = (obj, member, setter, getter) => ({
-  set _(value) {
-    __privateSet(obj, member, value, setter);
-  },
-  get _() {
-    return __privateGet(obj, member, getter);
-  }
-});
-var __privateMethod = (obj, member, method) => {
-  __accessCheck(obj, member, "access private method");
-  return method;
-};
-var _ZodSchemaParsedError = class _ZodSchemaParsedError extends Error {
-  constructor(error, payload) {
-    super(error.message);
-    this.error = error;
-    this.payload = payload;
-  }
-};
-__name(_ZodSchemaParsedError, "ZodSchemaParsedError");
-var ZodSchemaParsedError = _ZodSchemaParsedError;
-zod.z.object({
-  version: zod.z.literal("v1").default("v1"),
-  type: zod.z.string(),
-  payload: zod.z.unknown()
-});
-
-// src/v3/zodIpc.ts
-var messageSchema = zod.z.object({
-  version: zod.z.literal("v1").default("v1"),
-  type: zod.z.string(),
-  payload: zod.z.unknown()
-});
-var _schema, _handlers, _sender, _a;
-var ZodIpcMessageHandler = (_a = class {
-  constructor(options) {
-    __privateAdd(this, _schema, void 0);
-    __privateAdd(this, _handlers, void 0);
-    __privateAdd(this, _sender, void 0);
-    __privateSet(this, _schema, options.schema);
-    __privateSet(this, _handlers, options.handlers);
-    __privateSet(this, _sender, options.sender);
-  }
-  async handleMessage(message) {
-    const parsedMessage = this.parseMessage(message);
-    if (!__privateGet(this, _handlers)) {
-      throw new Error("No handlers provided");
-    }
-    const handler = __privateGet(this, _handlers)[parsedMessage.type];
-    if (!handler) {
-      return;
-    }
-    const ack = await handler(parsedMessage.payload, __privateGet(this, _sender));
-    return ack;
-  }
-  parseMessage(message) {
-    const parsedMessage = messageSchema.safeParse(message);
-    if (!parsedMessage.success) {
-      throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);
-    }
-    const schema = __privateGet(this, _schema)[parsedMessage.data.type]["message"];
-    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)}`);
-    }
-    return {
-      type: parsedMessage.data.type,
-      payload: parsedPayload.data
-    };
-  }
-}, _schema = new WeakMap(), _handlers = new WeakMap(), _sender = new WeakMap(), __name(_a, "ZodIpcMessageHandler"), _a);
-var Packet = zod.z.discriminatedUnion("type", [
-  zod.z.object({
-    type: zod.z.literal("CONNECT"),
-    sessionId: zod.z.string().optional()
-  }),
-  zod.z.object({
-    type: zod.z.literal("ACK"),
-    message: zod.z.any(),
-    id: zod.z.number()
-  }),
-  zod.z.object({
-    type: zod.z.literal("EVENT"),
-    message: zod.z.any(),
-    id: zod.z.number().optional()
-  })
-]);
-var _sessionId, _messageCounter, _handler, _acks, _registerHandlers, registerHandlers_fn, _handlePacket, handlePacket_fn, _sendPacket, sendPacket_fn;
-var _ZodIpcConnection = class _ZodIpcConnection {
-  constructor(opts) {
-    __privateAdd(this, _registerHandlers);
-    __privateAdd(this, _handlePacket);
-    __privateAdd(this, _sendPacket);
-    __privateAdd(this, _sessionId, void 0);
-    __privateAdd(this, _messageCounter, void 0);
-    __privateAdd(this, _handler, void 0);
-    __privateAdd(this, _acks, void 0);
-    this.opts = opts;
-    __privateSet(this, _messageCounter, 0);
-    __privateSet(this, _acks, /* @__PURE__ */ new Map());
-    __privateSet(this, _handler, new ZodIpcMessageHandler({
-      schema: opts.listenSchema,
-      handlers: opts.handlers,
-      sender: {
-        send: this.send.bind(this),
-        sendWithAck: this.sendWithAck.bind(this)
-      }
-    }));
-    __privateMethod(this, _registerHandlers, registerHandlers_fn).call(this);
-  }
-  async connect() {
-    __privateMethod(this, _sendPacket, sendPacket_fn).call(this, {
-      type: "CONNECT"
-    });
-  }
-  async send(type, payload) {
-    const schema = this.opts.emitSchema[type]["message"];
-    if (!schema) {
-      throw new Error(`Unknown message type: ${type}`);
-    }
-    const parsedPayload = schema.safeParse(payload);
-    if (!parsedPayload.success) {
-      throw new ZodSchemaParsedError(parsedPayload.error, payload);
-    }
-    await __privateMethod(this, _sendPacket, sendPacket_fn).call(this, {
-      type: "EVENT",
-      message: {
-        type,
-        payload,
-        version: "v1"
-      }
-    });
-  }
-  async sendWithAck(type, payload, timeoutInMs) {
-    const currentId = __privateWrapper(this, _messageCounter)._++;
-    return new Promise(async (resolve, reject) => {
-      const defaultTimeoutInMs = 2e3;
-      const timeout = setTimeout(() => {
-        reject(JSON.stringify({
-          reason: "sendWithAck() timeout",
-          timeoutInMs: timeoutInMs ?? defaultTimeoutInMs,
-          type,
-          payload
-        }));
-      }, timeoutInMs ?? defaultTimeoutInMs);
-      __privateGet(this, _acks).set(currentId, {
-        resolve,
-        reject,
-        timeout
-      });
-      const schema = this.opts.emitSchema[type]["message"];
-      if (!schema) {
-        clearTimeout(timeout);
-        return reject(`Unknown message type: ${type}`);
-      }
-      const parsedPayload = schema.safeParse(payload);
-      if (!parsedPayload.success) {
-        clearTimeout(timeout);
-        return reject(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
-      }
-      await __privateMethod(this, _sendPacket, sendPacket_fn).call(this, {
-        type: "EVENT",
-        message: {
-          type,
-          payload,
-          version: "v1"
-        },
-        id: currentId
-      });
-    });
-  }
-};
-_sessionId = new WeakMap();
-_messageCounter = new WeakMap();
-_handler = new WeakMap();
-_acks = new WeakMap();
-_registerHandlers = new WeakSet();
-registerHandlers_fn = /* @__PURE__ */ __name(async function() {
-  if (!this.opts.process.on) {
-    return;
-  }
-  this.opts.process.on("message", async (message) => {
-    __privateMethod(this, _handlePacket, handlePacket_fn).call(this, message);
-  });
-}, "#registerHandlers");
-_handlePacket = new WeakSet();
-handlePacket_fn = /* @__PURE__ */ __name(async function(packet) {
-  const parsedPacket = Packet.safeParse(packet);
-  if (!parsedPacket.success) {
-    return;
-  }
-  switch (parsedPacket.data.type) {
-    case "ACK": {
-      const ack = __privateGet(this, _acks).get(parsedPacket.data.id);
-      if (!ack) {
-        return;
-      }
-      clearTimeout(ack.timeout);
-      ack.resolve(parsedPacket.data.message);
-      break;
-    }
-    case "CONNECT": {
-      if (!parsedPacket.data.sessionId) {
-        const id = crypto.randomUUID();
-        await __privateMethod(this, _sendPacket, sendPacket_fn).call(this, {
-          type: "CONNECT",
-          sessionId: id
-        });
-        return;
-      }
-      if (__privateGet(this, _sessionId)) {
-        return;
-      }
-      __privateSet(this, _sessionId, parsedPacket.data.sessionId);
-      break;
-    }
-    case "EVENT": {
-      const result = await __privateGet(this, _handler).handleMessage(parsedPacket.data.message);
-      if (typeof parsedPacket.data.id === "undefined") {
-        return;
-      }
-      await __privateMethod(this, _sendPacket, sendPacket_fn).call(this, {
-        type: "ACK",
-        id: parsedPacket.data.id,
-        message: result
-      });
-      break;
-    }
-  }
-}, "#handlePacket");
-_sendPacket = new WeakSet();
-sendPacket_fn = /* @__PURE__ */ __name(async function(packet1) {
-  await this.opts.process.send?.(packet1);
-}, "#sendPacket");
-__name(_ZodIpcConnection, "ZodIpcConnection");
-var ZodIpcConnection = _ZodIpcConnection;
-
-exports.ZodIpcConnection = ZodIpcConnection;
-//# sourceMappingURL=out.js.map
-//# sourceMappingURL=zodIpc.js.map

package/dist/v3/zodIpc.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/v3/zodIpc.ts","../../src/v3/zodMessageHandler.ts"],"names":["randomUUID","z","ZodSchemaParsedError","Error","constructor","error","payload","message","ZodMessageSchema","object","version","literal","default","type","string","unknown","messageSchema","ZodIpcMessageHandler","options","schema","handlers","sender","handleMessage","parsedMessage","parseMessage","handler","ack","safeParse","success","JSON","stringify","data","parsedPayload","Packet","discriminatedUnion","sessionId","optional","any","id","number","ZodIpcConnection","opts","Map","listenSchema","send","bind","sendWithAck","connect","emitSchema","timeoutInMs","currentId","Promise","resolve","reject","defaultTimeoutInMs","timeout","setTimeout","reason","set","clearTimeout","process","on","packet","parsedPacket","get","result"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,SAASA,kBAAkB;AAU3B,SAASC,KAAAA,UAAS;;;ACVlB,SAASA,SAAS;AAGX,IAAMC,wBAAN,MAAMA,8BAA6BC,MAAAA;EACxCC,YACSC,OACAC,SACP;AACA,UAAMD,MAAME,OAAO;iBAHZF;mBACAC;EAGT;AACF;AAP0CH;AAAnC,IAAMD,uBAAN;AAuCA,IAAMM,mBAAmBP,EAAEQ,OAAO;EACvCC,SAAST,EAAEU,QAAQ,IAAA,EAAMC,QAAQ,IAAA;EACjCC,MAAMZ,EAAEa,OAAM;EACdR,SAASL,EAAEc,QAAO;AACpB,CAAA;;;ADNA,IAAMC,gBAAgBf,GAAEQ,OAAO;EAC7BC,SAAST,GAAEU,QAAQ,IAAA,EAAMC,QAAQ,IAAA;EACjCC,MAAMZ,GAAEa,OAAM;EACdR,SAASL,GAAEc,QAAO;AACpB,CAAA;AA5CA;AAuDA,IAAME,wBAAN,WAAMA;EAQJb,YAAYc,SAAoE;AAJhF;AACA;AACA;AAGE,uBAAK,SAAUA,QAAQC;AACvB,uBAAK,WAAYD,QAAQE;AACzB,uBAAK,SAAUF,QAAQG;EACzB;EAEA,MAAaC,cAAcf,SAAkB;AAC3C,UAAMgB,gBAAgB,KAAKC,aAAajB,OAAAA;AAExC,QAAI,CAAC,mBAAK,YAAW;AACnB,YAAM,IAAIJ,MAAM,sBAAA;IAClB;AAEA,UAAMsB,UAAU,mBAAK,WAAUF,cAAcV,IAAI;AAEjD,QAAI,CAACY,SAAS;AAEZ;IACF;AAEA,UAAMC,MAAM,MAAMD,QAAQF,cAAcjB,SAAS,mBAAK,QAAO;AAE7D,WAAOoB;EACT;EAEOF,aAAajB,SAA6D;AAC/E,UAAMgB,gBAAgBP,cAAcW,UAAUpB,OAAAA;AAE9C,QAAI,CAACgB,cAAcK,SAAS;AAC1B,YAAM,IAAIzB,MAAM,4BAA4B0B,KAAKC,UAAUP,cAAclB,KAAK,CAAA,EAAG;IACnF;AAEA,UAAMc,SAAS,mBAAK,SAAQI,cAAcQ,KAAKlB,IAAI,EAAE,SAAA;AAErD,QAAI,CAACM,QAAQ;AACX,YAAM,IAAIhB,MAAM,yBAAyBoB,cAAcQ,KAAKlB,IAAI,EAAE;IACpE;AAEA,UAAMmB,gBAAgBb,OAAOQ,UAAUJ,cAAcQ,KAAKzB,OAAO;AAEjE,QAAI,CAAC0B,cAAcJ,SAAS;AAC1B,YAAM,IAAIzB,MAAM,oCAAoC0B,KAAKC,UAAUE,cAAc3B,KAAK,CAAA,EAAG;IAC3F;AAEA,WAAO;MACLQ,MAAMU,cAAcQ,KAAKlB;MACzBP,SAAS0B,cAAcD;IACzB;EACF;AACF,GArDE,yBACA,2BACA,yBANId,oCAAN;AA2DA,IAAMgB,SAAShC,GAAEiC,mBAAmB,QAAQ;EAC1CjC,GAAEQ,OAAO;IACPI,MAAMZ,GAAEU,QAAQ,SAAA;IAChBwB,WAAWlC,GAAEa,OAAM,EAAGsB,SAAQ;EAChC,CAAA;EACAnC,GAAEQ,OAAO;IACPI,MAAMZ,GAAEU,QAAQ,KAAA;IAChBJ,SAASN,GAAEoC,IAAG;IACdC,IAAIrC,GAAEsC,OAAM;EACd,CAAA;EACAtC,GAAEQ,OAAO;IACPI,MAAMZ,GAAEU,QAAQ,OAAA;IAChBJ,SAASN,GAAEoC,IAAG;IACdC,IAAIrC,GAAEsC,OAAM,EAAGH,SAAQ;EACzB,CAAA;CACD;AAjID;AAkJO,IAAMI,oBAAN,MAAMA,kBAAAA;EAkBXpC,YAAoBqC,MAA6D;AAcjF,uBAAM;AAcN,uBAAM;AA+DN,uBAAM;AAzGN;AACA;AAEA;AAEA;gBASoBA;uBAbpB,iBAA0B;uBAI1B,OAOI,oBAAIC,IAAAA;AAGN,uBAAK,UAAW,IAAIzB,qBAAqB;MACvCE,QAAQsB,KAAKE;MACbvB,UAAUqB,KAAKrB;MACfC,QAAQ;QACNuB,MAAM,KAAKA,KAAKC,KAAK,IAAI;QACzBC,aAAa,KAAKA,YAAYD,KAAK,IAAI;MACzC;IACF,CAAA;AAEA,0BAAK,wCAAL;EAEF;EAYA,MAAME,UAAU;AACd,0BAAK,4BAAL,WAAiB;MAAElC,MAAM;IAAU;EACrC;EAqEA,MAAM+B,KACJ/B,MACAP,SACe;AACf,UAAMa,SAAS,KAAKsB,KAAKO,WAAWnC,IAAAA,EAAM,SAAA;AAE1C,QAAI,CAACM,QAAQ;AACX,YAAM,IAAIhB,MAAM,yBAAyBU,IAAAA,EAAgB;IAC3D;AAEA,UAAMmB,gBAAgBb,OAAOQ,UAAUrB,OAAAA;AAEvC,QAAI,CAAC0B,cAAcJ,SAAS;AAC1B,YAAM,IAAI1B,qBAAqB8B,cAAc3B,OAAOC,OAAAA;IACtD;AAEA,UAAM,sBAAK,4BAAL,WAAiB;MACrBO,MAAM;MACNN,SAAS;QACPM;QACAP;QACAI,SAAS;MACX;IACF;EACF;EAEA,MAAaoC,YACXjC,MACAP,SACA2C,aAC4D;AAC5D,UAAMC,YAAY,uBAAK,iBAAL;AAElB,WAAO,IAAIC,QAAQ,OAAOC,SAASC,WAAW;AAC5C,YAAMC,qBAAqB;AAG3B,YAAMC,UAAUC,WAAW,MAAM;AAC/BH,eACExB,KAAKC,UAAU;UACb2B,QAAQ;UACRR,aAAaA,eAAeK;UAC5BzC;UACAP;QACF,CAAA,CAAA;MAEJ,GAAG2C,eAAeK,kBAAAA;AAElB,yBAAK,OAAMI,IAAIR,WAAW;QAAEE;QAASC;QAAQE;MAAQ,CAAA;AAErD,YAAMpC,SAAS,KAAKsB,KAAKO,WAAWnC,IAAAA,EAAM,SAAA;AAE1C,UAAI,CAACM,QAAQ;AACXwC,qBAAaJ,OAAAA;AACb,eAAOF,OAAO,yBAAyBxC,IAAAA,EAAgB;MACzD;AAEA,YAAMmB,gBAAgBb,OAAOQ,UAAUrB,OAAAA;AAEvC,UAAI,CAAC0B,cAAcJ,SAAS;AAC1B+B,qBAAaJ,OAAAA;AACb,eAAOF,OAAO,oCAAoCxB,KAAKC,UAAUE,cAAc3B,KAAK,CAAA,EAAG;MACzF;AAEA,YAAM,sBAAK,4BAAL,WAAiB;QACrBQ,MAAM;QACNN,SAAS;UACPM;UACAP;UACAI,SAAS;QACX;QACA4B,IAAIY;MACN;IACF,CAAA;EACF;AACF;AAxLE;AACA;AAEA;AAEA;AAuBM;sBAAiB,wCAAG;AACxB,MAAI,CAAC,KAAKT,KAAKmB,QAAQC,IAAI;AACzB;EACF;AAEA,OAAKpB,KAAKmB,QAAQC,GAAG,WAAW,OAAOtD,YAAY;AACjD,0BAAK,gCAAL,WAAmBA;EACrB,CAAA;AACF,GARuB;AAcjB;kBAAa,sCAACuD,QAA+B;AACjD,QAAMC,eAAe9B,OAAON,UAAUmC,MAAAA;AAEtC,MAAI,CAACC,aAAanC,SAAS;AACzB;EACF;AAEA,UAAQmC,aAAahC,KAAKlB,MAAI;IAC5B,KAAK,OAAO;AAEV,YAAMa,MAAM,mBAAK,OAAMsC,IAAID,aAAahC,KAAKO,EAAE;AAE/C,UAAI,CAACZ,KAAK;AACR;MACF;AAEAiC,mBAAajC,IAAI6B,OAAO;AACxB7B,UAAI0B,QAAQW,aAAahC,KAAKxB,OAAO;AAErC;IACF;IACA,KAAK,WAAW;AACd,UAAI,CAACwD,aAAahC,KAAKI,WAAW;AAEhC,cAAMG,KAAKtC,WAAAA;AAEX,cAAM,sBAAK,4BAAL,WAAiB;UAAEa,MAAM;UAAWsB,WAAWG;QAAG;AAExD;MACF;AAGA,UAAI,mBAAK,aAAY;AAEnB;MACF;AAEA,yBAAK,YAAayB,aAAahC,KAAKI;AAEpC;IACF;IACA,KAAK,SAAS;AACZ,YAAM8B,SAAS,MAAM,mBAAK,UAAS3C,cAAcyC,aAAahC,KAAKxB,OAAO;AAE1E,UAAI,OAAOwD,aAAahC,KAAKO,OAAO,aAAa;AAC/C;MACF;AAGA,YAAM,sBAAK,4BAAL,WAAiB;QACrBzB,MAAM;QACNyB,IAAIyB,aAAahC,KAAKO;QACtB/B,SAAS0D;MACX;AAEA;IACF;IACA,SAAS;AACP;IACF;EACF;AACF,GA7DmB;AA+Db;gBAAW,sCAACH,SAAgB;AAChC,QAAM,KAAKrB,KAAKmB,QAAQhB,OAAOkB,OAAAA;AACjC,GAFiB;AA7GNtB;AAAN,IAAMA,mBAAN","sourcesContent":["import { randomUUID } from \"crypto\";\nimport {\n  GetSocketCallbackSchema,\n  GetSocketMessageSchema,\n  GetSocketMessagesWithCallback,\n  GetSocketMessagesWithoutCallback,\n  MessagesFromSocketCatalog,\n  SocketMessageHasCallback,\n  ZodSocketMessageCatalogSchema,\n} from \"./zodSocket\";\nimport { z } from \"zod\";\nimport { ZodSchemaParsedError } from \"./zodMessageHandler\";\nimport { inspect } from \"node:util\";\n\ninterface ZodIpcMessageSender<TEmitCatalog extends ZodSocketMessageCatalogSchema> {\n  send<K extends GetSocketMessagesWithoutCallback<TEmitCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TEmitCatalog, K>>\n  ): Promise<void>;\n\n  sendWithAck<K extends GetSocketMessagesWithCallback<TEmitCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TEmitCatalog, K>>\n  ): Promise<z.infer<GetSocketCallbackSchema<TEmitCatalog, K>>>;\n}\n\ntype ZodIpcMessageHandlers<\n  TListenCatalog extends ZodSocketMessageCatalogSchema,\n  TEmitCatalog extends ZodSocketMessageCatalogSchema,\n> = Partial<{\n  [K in keyof TListenCatalog]: (\n    payload: z.infer<GetSocketMessageSchema<TListenCatalog, K>>,\n    sender: ZodIpcMessageSender<TEmitCatalog>\n  ) => Promise<\n    SocketMessageHasCallback<TListenCatalog, K> extends true\n      ? z.input<GetSocketCallbackSchema<TListenCatalog, K>>\n      : void\n  >;\n}>;\n\nconst messageSchema = z.object({\n  version: z.literal(\"v1\").default(\"v1\"),\n  type: z.string(),\n  payload: z.unknown(),\n});\n\ntype ZodIpcMessageHandlerOptions<\n  TListenCatalog extends ZodSocketMessageCatalogSchema,\n  TEmitCatalog extends ZodSocketMessageCatalogSchema,\n> = {\n  schema: TListenCatalog;\n  handlers?: ZodIpcMessageHandlers<TListenCatalog, TEmitCatalog>;\n  sender: ZodIpcMessageSender<TEmitCatalog>;\n};\n\nclass ZodIpcMessageHandler<\n  TListenCatalog extends ZodSocketMessageCatalogSchema,\n  TEmitCatalog extends ZodSocketMessageCatalogSchema,\n> {\n  #schema: TListenCatalog;\n  #handlers: ZodIpcMessageHandlers<TListenCatalog, TEmitCatalog> | undefined;\n  #sender: ZodIpcMessageSender<TEmitCatalog>;\n\n  constructor(options: ZodIpcMessageHandlerOptions<TListenCatalog, TEmitCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.handlers;\n    this.#sender = options.sender;\n  }\n\n  public async handleMessage(message: unknown) {\n    const parsedMessage = this.parseMessage(message);\n\n    if (!this.#handlers) {\n      throw new Error(\"No handlers provided\");\n    }\n\n    const handler = this.#handlers[parsedMessage.type];\n\n    if (!handler) {\n      // console.error(`No handler for message type: ${String(parsedMessage.type)}`);\n      return;\n    }\n\n    const ack = await handler(parsedMessage.payload, this.#sender);\n\n    return ack;\n  }\n\n  public parseMessage(message: unknown): MessagesFromSocketCatalog<TListenCatalog> {\n    const parsedMessage = messageSchema.safeParse(message);\n\n    if (!parsedMessage.success) {\n      throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);\n    }\n\n    const schema = this.#schema[parsedMessage.data.type][\"message\"];\n\n    if (!schema) {\n      throw new Error(`Unknown message type: ${parsedMessage.data.type}`);\n    }\n\n    const parsedPayload = schema.safeParse(parsedMessage.data.payload);\n\n    if (!parsedPayload.success) {\n      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    return {\n      type: parsedMessage.data.type,\n      payload: parsedPayload.data,\n    };\n  }\n}\n\nconst Packet = z.discriminatedUnion(\"type\", [\n  z.object({\n    type: z.literal(\"CONNECT\"),\n    sessionId: z.string().optional(),\n  }),\n  z.object({\n    type: z.literal(\"ACK\"),\n    message: z.any(),\n    id: z.number(),\n  }),\n  z.object({\n    type: z.literal(\"EVENT\"),\n    message: z.any(),\n    id: z.number().optional(),\n  }),\n]);\n\ntype Packet = z.infer<typeof Packet>;\n\ninterface ZodIpcConnectionOptions<\n  TListenCatalog extends ZodSocketMessageCatalogSchema,\n  TEmitCatalog extends ZodSocketMessageCatalogSchema,\n> {\n  listenSchema: TListenCatalog;\n  emitSchema: TEmitCatalog;\n  process: {\n    send?: (message: any) => any;\n    on?: (event: \"message\", listener: (message: any) => void) => void;\n  };\n  handlers?: ZodIpcMessageHandlers<TListenCatalog, TEmitCatalog>;\n}\n\nexport class ZodIpcConnection<\n  TListenCatalog extends ZodSocketMessageCatalogSchema,\n  TEmitCatalog extends ZodSocketMessageCatalogSchema,\n> {\n  #sessionId?: string;\n  #messageCounter: number = 0;\n\n  #handler: ZodIpcMessageHandler<TListenCatalog, TEmitCatalog>;\n\n  #acks: Map<\n    number,\n    {\n      resolve: (value: unknown) => void;\n      reject: (reason?: any) => void;\n      timeout: NodeJS.Timeout;\n    }\n  > = new Map();\n\n  constructor(private opts: ZodIpcConnectionOptions<TListenCatalog, TEmitCatalog>) {\n    this.#handler = new ZodIpcMessageHandler({\n      schema: opts.listenSchema,\n      handlers: opts.handlers,\n      sender: {\n        send: this.send.bind(this),\n        sendWithAck: this.sendWithAck.bind(this),\n      },\n    });\n\n    this.#registerHandlers();\n    // this.connect();\n  }\n\n  async #registerHandlers() {\n    if (!this.opts.process.on) {\n      return;\n    }\n\n    this.opts.process.on(\"message\", async (message) => {\n      this.#handlePacket(message);\n    });\n  }\n\n  async connect() {\n    this.#sendPacket({ type: \"CONNECT\" });\n  }\n\n  async #handlePacket(packet: Packet): Promise<void> {\n    const parsedPacket = Packet.safeParse(packet);\n\n    if (!parsedPacket.success) {\n      return;\n    }\n\n    switch (parsedPacket.data.type) {\n      case \"ACK\": {\n        // Check our list of ACKs and resolve with the message\n        const ack = this.#acks.get(parsedPacket.data.id);\n\n        if (!ack) {\n          return;\n        }\n\n        clearTimeout(ack.timeout);\n        ack.resolve(parsedPacket.data.message);\n\n        break;\n      }\n      case \"CONNECT\": {\n        if (!parsedPacket.data.sessionId) {\n          // This is a client trying to connect, so we generate and send back a session ID\n          const id = randomUUID();\n\n          await this.#sendPacket({ type: \"CONNECT\", sessionId: id });\n\n          return;\n        }\n\n        // This is a server replying to our connect message\n        if (this.#sessionId) {\n          // We're already connected\n          return;\n        }\n\n        this.#sessionId = parsedPacket.data.sessionId;\n\n        break;\n      }\n      case \"EVENT\": {\n        const result = await this.#handler.handleMessage(parsedPacket.data.message);\n\n        if (typeof parsedPacket.data.id === \"undefined\") {\n          return;\n        }\n\n        // There's an ID so we should ACK\n        await this.#sendPacket({\n          type: \"ACK\",\n          id: parsedPacket.data.id,\n          message: result,\n        });\n\n        break;\n      }\n      default: {\n        break;\n      }\n    }\n  }\n\n  async #sendPacket(packet: Packet) {\n    await this.opts.process.send?.(packet);\n  }\n\n  async send<K extends GetSocketMessagesWithoutCallback<TEmitCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TEmitCatalog, K>>\n  ): Promise<void> {\n    const schema = this.opts.emitSchema[type][\"message\"];\n\n    if (!schema) {\n      throw new Error(`Unknown message type: ${type as string}`);\n    }\n\n    const parsedPayload = schema.safeParse(payload);\n\n    if (!parsedPayload.success) {\n      throw new ZodSchemaParsedError(parsedPayload.error, payload);\n    }\n\n    await this.#sendPacket({\n      type: \"EVENT\",\n      message: {\n        type,\n        payload,\n        version: \"v1\",\n      },\n    });\n  }\n\n  public async sendWithAck<K extends GetSocketMessagesWithCallback<TEmitCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TEmitCatalog, K>>,\n    timeoutInMs?: number\n  ): Promise<z.infer<GetSocketCallbackSchema<TEmitCatalog, K>>> {\n    const currentId = this.#messageCounter++;\n\n    return new Promise(async (resolve, reject) => {\n      const defaultTimeoutInMs = 2000;\n\n      // Timeout if the ACK takes too long to get back to us\n      const timeout = setTimeout(() => {\n        reject(\n          JSON.stringify({\n            reason: \"sendWithAck() timeout\",\n            timeoutInMs: timeoutInMs ?? defaultTimeoutInMs,\n            type,\n            payload,\n          })\n        );\n      }, timeoutInMs ?? defaultTimeoutInMs);\n\n      this.#acks.set(currentId, { resolve, reject, timeout });\n\n      const schema = this.opts.emitSchema[type][\"message\"];\n\n      if (!schema) {\n        clearTimeout(timeout);\n        return reject(`Unknown message type: ${type as string}`);\n      }\n\n      const parsedPayload = schema.safeParse(payload);\n\n      if (!parsedPayload.success) {\n        clearTimeout(timeout);\n        return reject(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n      }\n\n      await this.#sendPacket({\n        type: \"EVENT\",\n        message: {\n          type,\n          payload,\n          version: \"v1\",\n        },\n        id: currentId,\n      });\n    });\n  }\n}\n","import { z } from \"zod\";\nimport { StructuredLogger } from \"./utils/structuredLogger\";\n\nexport class ZodSchemaParsedError extends Error {\n  constructor(\n    public error: z.ZodError,\n    public payload: unknown\n  ) {\n    super(error.message);\n  }\n}\n\nexport type ZodMessageValueSchema<TDiscriminatedUnion extends z.ZodDiscriminatedUnion<any, any>> =\n  | z.ZodFirstPartySchemaTypes\n  | TDiscriminatedUnion;\n\nexport interface ZodMessageCatalogSchema {\n  [key: string]: ZodMessageValueSchema<any>;\n}\n\nexport type ZodMessageHandlers<TCatalogSchema extends ZodMessageCatalogSchema> = Partial<{\n  [K in keyof TCatalogSchema]: (payload: z.infer<TCatalogSchema[K]>) => Promise<any>;\n}>;\n\nexport type ZodMessageHandlerOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {\n  schema: TMessageCatalog;\n  messages?: ZodMessageHandlers<TMessageCatalog>;\n  logger?: StructuredLogger;\n};\n\nexport type MessageFromSchema<\n  K extends keyof TMessageCatalog,\n  TMessageCatalog extends ZodMessageCatalogSchema,\n> = {\n  type: K;\n  payload: z.input<TMessageCatalog[K]>;\n};\n\nexport type MessageFromCatalog<TMessageCatalog extends ZodMessageCatalogSchema> = {\n  [K in keyof TMessageCatalog]: MessageFromSchema<K, TMessageCatalog>;\n}[keyof TMessageCatalog];\n\nexport const ZodMessageSchema = z.object({\n  version: z.literal(\"v1\").default(\"v1\"),\n  type: z.string(),\n  payload: z.unknown(),\n});\n\nexport interface EventEmitterLike {\n  on(eventName: string | symbol, listener: (...args: any[]) => void): this;\n}\n\nexport class ZodMessageHandler<TMessageCatalog extends ZodMessageCatalogSchema> {\n  #schema: TMessageCatalog;\n  #handlers: ZodMessageHandlers<TMessageCatalog> | undefined;\n  #logger: StructuredLogger | Console;\n\n  constructor(options: ZodMessageHandlerOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.messages;\n    this.#logger = options.logger ?? console;\n  }\n\n  public async handleMessage(message: unknown): Promise<\n    | {\n        success: true;\n        data: unknown;\n      }\n    | {\n        success: false;\n        error: string;\n      }\n  > {\n    const parsedMessage = this.parseMessage(message);\n\n    if (!parsedMessage.success) {\n      this.#logger.error(parsedMessage.error, { message });\n\n      return {\n        success: false,\n        error: parsedMessage.error,\n      };\n    }\n\n    if (!this.#handlers) {\n      this.#logger.error(\"No handlers provided\", { message });\n\n      return {\n        success: false,\n        error: \"No handlers provided\",\n      };\n    }\n\n    const handler = this.#handlers[parsedMessage.data.type];\n\n    if (!handler) {\n      const error = `No handler for message type: ${String(parsedMessage.data.type)}`;\n\n      this.#logger.error(error, { message });\n\n      return {\n        success: false,\n        error,\n      };\n    }\n\n    const ack = await handler(parsedMessage.data.payload);\n\n    return {\n      success: true,\n      data: ack,\n    };\n  }\n\n  public parseMessage(message: unknown):\n    | {\n        success: true;\n        data: MessageFromCatalog<TMessageCatalog>;\n      }\n    | {\n        success: false;\n        error: string;\n      } {\n    const parsedMessage = ZodMessageSchema.safeParse(message);\n\n    if (!parsedMessage.success) {\n      return {\n        success: false,\n        error: `Failed to parse message: ${JSON.stringify(parsedMessage.error)}`,\n      };\n    }\n\n    const schema = this.#schema[parsedMessage.data.type];\n\n    if (!schema) {\n      return {\n        success: false,\n        error: `Unknown message type: ${parsedMessage.data.type}`,\n      };\n    }\n\n    const parsedPayload = schema.safeParse(parsedMessage.data.payload);\n\n    if (!parsedPayload.success) {\n      return {\n        success: false,\n        error: `Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`,\n      };\n    }\n\n    return {\n      success: true,\n      data: {\n        type: parsedMessage.data.type,\n        payload: parsedPayload.data,\n      },\n    };\n  }\n\n  public registerHandlers(emitter: EventEmitterLike, logger?: StructuredLogger) {\n    const log = logger ?? console;\n\n    if (!this.#handlers) {\n      log.info(\"No handlers provided\");\n      return;\n    }\n\n    for (const eventName of Object.keys(this.#schema)) {\n      emitter.on(eventName, async (message: any, callback?: any): Promise<void> => {\n        log.info(`handling ${eventName}`, {\n          payload: message,\n          hasCallback: !!callback,\n        });\n\n        let ack: Awaited<ReturnType<ZodMessageHandler<TMessageCatalog>[\"handleMessage\"]>>;\n\n        // FIXME: this only works if the message doesn't have genuine payload prop\n        if (\"payload\" in message) {\n          ack = await this.handleMessage({ type: eventName, ...message });\n        } else {\n          // Handle messages not sent by ZodMessageSender\n          const { version, ...payload } = message;\n          ack = await this.handleMessage({ type: eventName, version, payload });\n        }\n\n        if (callback && typeof callback === \"function\") {\n          if (!ack.success) {\n            // We don't know the callback type, so we can't do anything else - not all callbacks may accept a success prop\n            log.error(\"Failed to handle message, skipping callback\", { message, error: ack.error });\n            return;\n          }\n\n          callback(ack.data);\n        }\n      });\n    }\n  }\n}\n\ntype ZodMessageSenderCallback<TMessageCatalog extends ZodMessageCatalogSchema> = (message: {\n  type: keyof TMessageCatalog;\n  payload: z.infer<TMessageCatalog[keyof TMessageCatalog]>;\n  version: \"v1\";\n}) => Promise<void>;\n\nexport type ZodMessageSenderOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {\n  schema: TMessageCatalog;\n  sender: ZodMessageSenderCallback<TMessageCatalog>;\n};\n\nexport class ZodMessageSender<TMessageCatalog extends ZodMessageCatalogSchema> {\n  #schema: TMessageCatalog;\n  #sender: ZodMessageSenderCallback<TMessageCatalog>;\n\n  constructor(options: ZodMessageSenderOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#sender = options.sender;\n  }\n\n  public async send<K extends keyof TMessageCatalog>(\n    type: K,\n    payload: z.input<TMessageCatalog[K]>\n  ) {\n    const schema = this.#schema[type];\n\n    if (!schema) {\n      throw new Error(`Unknown message type: ${type as string}`);\n    }\n\n    const parsedPayload = schema.safeParse(payload);\n\n    if (!parsedPayload.success) {\n      throw new ZodSchemaParsedError(parsedPayload.error, payload);\n    }\n\n    try {\n      await this.#sender({ type, payload, version: \"v1\" });\n    } catch (error) {\n      console.error(\"[ZodMessageSender] Failed to send message\", error);\n    }\n  }\n\n  public async forwardMessage(message: unknown) {\n    const parsedMessage = ZodMessageSchema.safeParse(message);\n\n    if (!parsedMessage.success) {\n      throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);\n    }\n\n    const schema = this.#schema[parsedMessage.data.type];\n\n    if (!schema) {\n      throw new Error(`Unknown message type: ${parsedMessage.data.type}`);\n    }\n\n    const parsedPayload = schema.safeParse(parsedMessage.data.payload);\n\n    if (!parsedPayload.success) {\n      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    try {\n      await this.#sender({\n        type: parsedMessage.data.type,\n        payload: parsedPayload.data,\n        version: \"v1\",\n      });\n    } catch (error) {\n      console.error(\"[ZodMessageSender] Failed to forward message\", error);\n    }\n  }\n}\n\nexport type MessageCatalogToSocketIoEvents<TCatalog extends ZodMessageCatalogSchema> = {\n  [K in keyof TCatalog]: (message: z.infer<TCatalog[K]>) => void;\n};\n"]}

package/dist/v3/zodIpc.mjs

@@ -1,266 +0,0 @@
-import { randomUUID } from 'crypto';
-import { z } from 'zod';
-
-var __defProp = Object.defineProperty;
-var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-var __accessCheck = (obj, member, msg) => {
-  if (!member.has(obj))
-    throw TypeError("Cannot " + msg);
-};
-var __privateGet = (obj, member, getter) => {
-  __accessCheck(obj, member, "read from private field");
-  return getter ? getter.call(obj) : member.get(obj);
-};
-var __privateAdd = (obj, member, value) => {
-  if (member.has(obj))
-    throw TypeError("Cannot add the same private member more than once");
-  member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
-};
-var __privateSet = (obj, member, value, setter) => {
-  __accessCheck(obj, member, "write to private field");
-  setter ? setter.call(obj, value) : member.set(obj, value);
-  return value;
-};
-var __privateWrapper = (obj, member, setter, getter) => ({
-  set _(value) {
-    __privateSet(obj, member, value, setter);
-  },
-  get _() {
-    return __privateGet(obj, member, getter);
-  }
-});
-var __privateMethod = (obj, member, method) => {
-  __accessCheck(obj, member, "access private method");
-  return method;
-};
-var _ZodSchemaParsedError = class _ZodSchemaParsedError extends Error {
-  constructor(error, payload) {
-    super(error.message);
-    this.error = error;
-    this.payload = payload;
-  }
-};
-__name(_ZodSchemaParsedError, "ZodSchemaParsedError");
-var ZodSchemaParsedError = _ZodSchemaParsedError;
-z.object({
-  version: z.literal("v1").default("v1"),
-  type: z.string(),
-  payload: z.unknown()
-});
-
-// src/v3/zodIpc.ts
-var messageSchema = z.object({
-  version: z.literal("v1").default("v1"),
-  type: z.string(),
-  payload: z.unknown()
-});
-var _schema, _handlers, _sender, _a;
-var ZodIpcMessageHandler = (_a = class {
-  constructor(options) {
-    __privateAdd(this, _schema, void 0);
-    __privateAdd(this, _handlers, void 0);
-    __privateAdd(this, _sender, void 0);
-    __privateSet(this, _schema, options.schema);
-    __privateSet(this, _handlers, options.handlers);
-    __privateSet(this, _sender, options.sender);
-  }
-  async handleMessage(message) {
-    const parsedMessage = this.parseMessage(message);
-    if (!__privateGet(this, _handlers)) {
-      throw new Error("No handlers provided");
-    }
-    const handler = __privateGet(this, _handlers)[parsedMessage.type];
-    if (!handler) {
-      return;
-    }
-    const ack = await handler(parsedMessage.payload, __privateGet(this, _sender));
-    return ack;
-  }
-  parseMessage(message) {
-    const parsedMessage = messageSchema.safeParse(message);
-    if (!parsedMessage.success) {
-      throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);
-    }
-    const schema = __privateGet(this, _schema)[parsedMessage.data.type]["message"];
-    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)}`);
-    }
-    return {
-      type: parsedMessage.data.type,
-      payload: parsedPayload.data
-    };
-  }
-}, _schema = new WeakMap(), _handlers = new WeakMap(), _sender = new WeakMap(), __name(_a, "ZodIpcMessageHandler"), _a);
-var Packet = z.discriminatedUnion("type", [
-  z.object({
-    type: z.literal("CONNECT"),
-    sessionId: z.string().optional()
-  }),
-  z.object({
-    type: z.literal("ACK"),
-    message: z.any(),
-    id: z.number()
-  }),
-  z.object({
-    type: z.literal("EVENT"),
-    message: z.any(),
-    id: z.number().optional()
-  })
-]);
-var _sessionId, _messageCounter, _handler, _acks, _registerHandlers, registerHandlers_fn, _handlePacket, handlePacket_fn, _sendPacket, sendPacket_fn;
-var _ZodIpcConnection = class _ZodIpcConnection {
-  constructor(opts) {
-    __privateAdd(this, _registerHandlers);
-    __privateAdd(this, _handlePacket);
-    __privateAdd(this, _sendPacket);
-    __privateAdd(this, _sessionId, void 0);
-    __privateAdd(this, _messageCounter, void 0);
-    __privateAdd(this, _handler, void 0);
-    __privateAdd(this, _acks, void 0);
-    this.opts = opts;
-    __privateSet(this, _messageCounter, 0);
-    __privateSet(this, _acks, /* @__PURE__ */ new Map());
-    __privateSet(this, _handler, new ZodIpcMessageHandler({
-      schema: opts.listenSchema,
-      handlers: opts.handlers,
-      sender: {
-        send: this.send.bind(this),
-        sendWithAck: this.sendWithAck.bind(this)
-      }
-    }));
-    __privateMethod(this, _registerHandlers, registerHandlers_fn).call(this);
-  }
-  async connect() {
-    __privateMethod(this, _sendPacket, sendPacket_fn).call(this, {
-      type: "CONNECT"
-    });
-  }
-  async send(type, payload) {
-    const schema = this.opts.emitSchema[type]["message"];
-    if (!schema) {
-      throw new Error(`Unknown message type: ${type}`);
-    }
-    const parsedPayload = schema.safeParse(payload);
-    if (!parsedPayload.success) {
-      throw new ZodSchemaParsedError(parsedPayload.error, payload);
-    }
-    await __privateMethod(this, _sendPacket, sendPacket_fn).call(this, {
-      type: "EVENT",
-      message: {
-        type,
-        payload,
-        version: "v1"
-      }
-    });
-  }
-  async sendWithAck(type, payload, timeoutInMs) {
-    const currentId = __privateWrapper(this, _messageCounter)._++;
-    return new Promise(async (resolve, reject) => {
-      const defaultTimeoutInMs = 2e3;
-      const timeout = setTimeout(() => {
-        reject(JSON.stringify({
-          reason: "sendWithAck() timeout",
-          timeoutInMs: timeoutInMs ?? defaultTimeoutInMs,
-          type,
-          payload
-        }));
-      }, timeoutInMs ?? defaultTimeoutInMs);
-      __privateGet(this, _acks).set(currentId, {
-        resolve,
-        reject,
-        timeout
-      });
-      const schema = this.opts.emitSchema[type]["message"];
-      if (!schema) {
-        clearTimeout(timeout);
-        return reject(`Unknown message type: ${type}`);
-      }
-      const parsedPayload = schema.safeParse(payload);
-      if (!parsedPayload.success) {
-        clearTimeout(timeout);
-        return reject(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
-      }
-      await __privateMethod(this, _sendPacket, sendPacket_fn).call(this, {
-        type: "EVENT",
-        message: {
-          type,
-          payload,
-          version: "v1"
-        },
-        id: currentId
-      });
-    });
-  }
-};
-_sessionId = new WeakMap();
-_messageCounter = new WeakMap();
-_handler = new WeakMap();
-_acks = new WeakMap();
-_registerHandlers = new WeakSet();
-registerHandlers_fn = /* @__PURE__ */ __name(async function() {
-  if (!this.opts.process.on) {
-    return;
-  }
-  this.opts.process.on("message", async (message) => {
-    __privateMethod(this, _handlePacket, handlePacket_fn).call(this, message);
-  });
-}, "#registerHandlers");
-_handlePacket = new WeakSet();
-handlePacket_fn = /* @__PURE__ */ __name(async function(packet) {
-  const parsedPacket = Packet.safeParse(packet);
-  if (!parsedPacket.success) {
-    return;
-  }
-  switch (parsedPacket.data.type) {
-    case "ACK": {
-      const ack = __privateGet(this, _acks).get(parsedPacket.data.id);
-      if (!ack) {
-        return;
-      }
-      clearTimeout(ack.timeout);
-      ack.resolve(parsedPacket.data.message);
-      break;
-    }
-    case "CONNECT": {
-      if (!parsedPacket.data.sessionId) {
-        const id = randomUUID();
-        await __privateMethod(this, _sendPacket, sendPacket_fn).call(this, {
-          type: "CONNECT",
-          sessionId: id
-        });
-        return;
-      }
-      if (__privateGet(this, _sessionId)) {
-        return;
-      }
-      __privateSet(this, _sessionId, parsedPacket.data.sessionId);
-      break;
-    }
-    case "EVENT": {
-      const result = await __privateGet(this, _handler).handleMessage(parsedPacket.data.message);
-      if (typeof parsedPacket.data.id === "undefined") {
-        return;
-      }
-      await __privateMethod(this, _sendPacket, sendPacket_fn).call(this, {
-        type: "ACK",
-        id: parsedPacket.data.id,
-        message: result
-      });
-      break;
-    }
-  }
-}, "#handlePacket");
-_sendPacket = new WeakSet();
-sendPacket_fn = /* @__PURE__ */ __name(async function(packet1) {
-  await this.opts.process.send?.(packet1);
-}, "#sendPacket");
-__name(_ZodIpcConnection, "ZodIpcConnection");
-var ZodIpcConnection = _ZodIpcConnection;
-
-export { ZodIpcConnection };
-//# sourceMappingURL=out.js.map
-//# sourceMappingURL=zodIpc.mjs.map

package/dist/v3/zodIpc.mjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/v3/zodIpc.ts","../../src/v3/zodMessageHandler.ts"],"names":["randomUUID","z","ZodSchemaParsedError","Error","constructor","error","payload","message","ZodMessageSchema","object","version","literal","default","type","string","unknown","messageSchema","ZodIpcMessageHandler","options","schema","handlers","sender","handleMessage","parsedMessage","parseMessage","handler","ack","safeParse","success","JSON","stringify","data","parsedPayload","Packet","discriminatedUnion","sessionId","optional","any","id","number","ZodIpcConnection","opts","Map","listenSchema","send","bind","sendWithAck","connect","emitSchema","timeoutInMs","currentId","Promise","resolve","reject","defaultTimeoutInMs","timeout","setTimeout","reason","set","clearTimeout","process","on","packet","parsedPacket","get","result"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,SAASA,kBAAkB;AAU3B,SAASC,KAAAA,UAAS;;;ACVlB,SAASA,SAAS;AAGX,IAAMC,wBAAN,MAAMA,8BAA6BC,MAAAA;EACxCC,YACSC,OACAC,SACP;AACA,UAAMD,MAAME,OAAO;iBAHZF;mBACAC;EAGT;AACF;AAP0CH;AAAnC,IAAMD,uBAAN;AAuCA,IAAMM,mBAAmBP,EAAEQ,OAAO;EACvCC,SAAST,EAAEU,QAAQ,IAAA,EAAMC,QAAQ,IAAA;EACjCC,MAAMZ,EAAEa,OAAM;EACdR,SAASL,EAAEc,QAAO;AACpB,CAAA;;;ADNA,IAAMC,gBAAgBf,GAAEQ,OAAO;EAC7BC,SAAST,GAAEU,QAAQ,IAAA,EAAMC,QAAQ,IAAA;EACjCC,MAAMZ,GAAEa,OAAM;EACdR,SAASL,GAAEc,QAAO;AACpB,CAAA;AA5CA;AAuDA,IAAME,wBAAN,WAAMA;EAQJb,YAAYc,SAAoE;AAJhF;AACA;AACA;AAGE,uBAAK,SAAUA,QAAQC;AACvB,uBAAK,WAAYD,QAAQE;AACzB,uBAAK,SAAUF,QAAQG;EACzB;EAEA,MAAaC,cAAcf,SAAkB;AAC3C,UAAMgB,gBAAgB,KAAKC,aAAajB,OAAAA;AAExC,QAAI,CAAC,mBAAK,YAAW;AACnB,YAAM,IAAIJ,MAAM,sBAAA;IAClB;AAEA,UAAMsB,UAAU,mBAAK,WAAUF,cAAcV,IAAI;AAEjD,QAAI,CAACY,SAAS;AAEZ;IACF;AAEA,UAAMC,MAAM,MAAMD,QAAQF,cAAcjB,SAAS,mBAAK,QAAO;AAE7D,WAAOoB;EACT;EAEOF,aAAajB,SAA6D;AAC/E,UAAMgB,gBAAgBP,cAAcW,UAAUpB,OAAAA;AAE9C,QAAI,CAACgB,cAAcK,SAAS;AAC1B,YAAM,IAAIzB,MAAM,4BAA4B0B,KAAKC,UAAUP,cAAclB,KAAK,CAAA,EAAG;IACnF;AAEA,UAAMc,SAAS,mBAAK,SAAQI,cAAcQ,KAAKlB,IAAI,EAAE,SAAA;AAErD,QAAI,CAACM,QAAQ;AACX,YAAM,IAAIhB,MAAM,yBAAyBoB,cAAcQ,KAAKlB,IAAI,EAAE;IACpE;AAEA,UAAMmB,gBAAgBb,OAAOQ,UAAUJ,cAAcQ,KAAKzB,OAAO;AAEjE,QAAI,CAAC0B,cAAcJ,SAAS;AAC1B,YAAM,IAAIzB,MAAM,oCAAoC0B,KAAKC,UAAUE,cAAc3B,KAAK,CAAA,EAAG;IAC3F;AAEA,WAAO;MACLQ,MAAMU,cAAcQ,KAAKlB;MACzBP,SAAS0B,cAAcD;IACzB;EACF;AACF,GArDE,yBACA,2BACA,yBANId,oCAAN;AA2DA,IAAMgB,SAAShC,GAAEiC,mBAAmB,QAAQ;EAC1CjC,GAAEQ,OAAO;IACPI,MAAMZ,GAAEU,QAAQ,SAAA;IAChBwB,WAAWlC,GAAEa,OAAM,EAAGsB,SAAQ;EAChC,CAAA;EACAnC,GAAEQ,OAAO;IACPI,MAAMZ,GAAEU,QAAQ,KAAA;IAChBJ,SAASN,GAAEoC,IAAG;IACdC,IAAIrC,GAAEsC,OAAM;EACd,CAAA;EACAtC,GAAEQ,OAAO;IACPI,MAAMZ,GAAEU,QAAQ,OAAA;IAChBJ,SAASN,GAAEoC,IAAG;IACdC,IAAIrC,GAAEsC,OAAM,EAAGH,SAAQ;EACzB,CAAA;CACD;AAjID;AAkJO,IAAMI,oBAAN,MAAMA,kBAAAA;EAkBXpC,YAAoBqC,MAA6D;AAcjF,uBAAM;AAcN,uBAAM;AA+DN,uBAAM;AAzGN;AACA;AAEA;AAEA;gBASoBA;uBAbpB,iBAA0B;uBAI1B,OAOI,oBAAIC,IAAAA;AAGN,uBAAK,UAAW,IAAIzB,qBAAqB;MACvCE,QAAQsB,KAAKE;MACbvB,UAAUqB,KAAKrB;MACfC,QAAQ;QACNuB,MAAM,KAAKA,KAAKC,KAAK,IAAI;QACzBC,aAAa,KAAKA,YAAYD,KAAK,IAAI;MACzC;IACF,CAAA;AAEA,0BAAK,wCAAL;EAEF;EAYA,MAAME,UAAU;AACd,0BAAK,4BAAL,WAAiB;MAAElC,MAAM;IAAU;EACrC;EAqEA,MAAM+B,KACJ/B,MACAP,SACe;AACf,UAAMa,SAAS,KAAKsB,KAAKO,WAAWnC,IAAAA,EAAM,SAAA;AAE1C,QAAI,CAACM,QAAQ;AACX,YAAM,IAAIhB,MAAM,yBAAyBU,IAAAA,EAAgB;IAC3D;AAEA,UAAMmB,gBAAgBb,OAAOQ,UAAUrB,OAAAA;AAEvC,QAAI,CAAC0B,cAAcJ,SAAS;AAC1B,YAAM,IAAI1B,qBAAqB8B,cAAc3B,OAAOC,OAAAA;IACtD;AAEA,UAAM,sBAAK,4BAAL,WAAiB;MACrBO,MAAM;MACNN,SAAS;QACPM;QACAP;QACAI,SAAS;MACX;IACF;EACF;EAEA,MAAaoC,YACXjC,MACAP,SACA2C,aAC4D;AAC5D,UAAMC,YAAY,uBAAK,iBAAL;AAElB,WAAO,IAAIC,QAAQ,OAAOC,SAASC,WAAW;AAC5C,YAAMC,qBAAqB;AAG3B,YAAMC,UAAUC,WAAW,MAAM;AAC/BH,eACExB,KAAKC,UAAU;UACb2B,QAAQ;UACRR,aAAaA,eAAeK;UAC5BzC;UACAP;QACF,CAAA,CAAA;MAEJ,GAAG2C,eAAeK,kBAAAA;AAElB,yBAAK,OAAMI,IAAIR,WAAW;QAAEE;QAASC;QAAQE;MAAQ,CAAA;AAErD,YAAMpC,SAAS,KAAKsB,KAAKO,WAAWnC,IAAAA,EAAM,SAAA;AAE1C,UAAI,CAACM,QAAQ;AACXwC,qBAAaJ,OAAAA;AACb,eAAOF,OAAO,yBAAyBxC,IAAAA,EAAgB;MACzD;AAEA,YAAMmB,gBAAgBb,OAAOQ,UAAUrB,OAAAA;AAEvC,UAAI,CAAC0B,cAAcJ,SAAS;AAC1B+B,qBAAaJ,OAAAA;AACb,eAAOF,OAAO,oCAAoCxB,KAAKC,UAAUE,cAAc3B,KAAK,CAAA,EAAG;MACzF;AAEA,YAAM,sBAAK,4BAAL,WAAiB;QACrBQ,MAAM;QACNN,SAAS;UACPM;UACAP;UACAI,SAAS;QACX;QACA4B,IAAIY;MACN;IACF,CAAA;EACF;AACF;AAxLE;AACA;AAEA;AAEA;AAuBM;sBAAiB,wCAAG;AACxB,MAAI,CAAC,KAAKT,KAAKmB,QAAQC,IAAI;AACzB;EACF;AAEA,OAAKpB,KAAKmB,QAAQC,GAAG,WAAW,OAAOtD,YAAY;AACjD,0BAAK,gCAAL,WAAmBA;EACrB,CAAA;AACF,GARuB;AAcjB;kBAAa,sCAACuD,QAA+B;AACjD,QAAMC,eAAe9B,OAAON,UAAUmC,MAAAA;AAEtC,MAAI,CAACC,aAAanC,SAAS;AACzB;EACF;AAEA,UAAQmC,aAAahC,KAAKlB,MAAI;IAC5B,KAAK,OAAO;AAEV,YAAMa,MAAM,mBAAK,OAAMsC,IAAID,aAAahC,KAAKO,EAAE;AAE/C,UAAI,CAACZ,KAAK;AACR;MACF;AAEAiC,mBAAajC,IAAI6B,OAAO;AACxB7B,UAAI0B,QAAQW,aAAahC,KAAKxB,OAAO;AAErC;IACF;IACA,KAAK,WAAW;AACd,UAAI,CAACwD,aAAahC,KAAKI,WAAW;AAEhC,cAAMG,KAAKtC,WAAAA;AAEX,cAAM,sBAAK,4BAAL,WAAiB;UAAEa,MAAM;UAAWsB,WAAWG;QAAG;AAExD;MACF;AAGA,UAAI,mBAAK,aAAY;AAEnB;MACF;AAEA,yBAAK,YAAayB,aAAahC,KAAKI;AAEpC;IACF;IACA,KAAK,SAAS;AACZ,YAAM8B,SAAS,MAAM,mBAAK,UAAS3C,cAAcyC,aAAahC,KAAKxB,OAAO;AAE1E,UAAI,OAAOwD,aAAahC,KAAKO,OAAO,aAAa;AAC/C;MACF;AAGA,YAAM,sBAAK,4BAAL,WAAiB;QACrBzB,MAAM;QACNyB,IAAIyB,aAAahC,KAAKO;QACtB/B,SAAS0D;MACX;AAEA;IACF;IACA,SAAS;AACP;IACF;EACF;AACF,GA7DmB;AA+Db;gBAAW,sCAACH,SAAgB;AAChC,QAAM,KAAKrB,KAAKmB,QAAQhB,OAAOkB,OAAAA;AACjC,GAFiB;AA7GNtB;AAAN,IAAMA,mBAAN","sourcesContent":["import { randomUUID } from \"crypto\";\nimport {\n  GetSocketCallbackSchema,\n  GetSocketMessageSchema,\n  GetSocketMessagesWithCallback,\n  GetSocketMessagesWithoutCallback,\n  MessagesFromSocketCatalog,\n  SocketMessageHasCallback,\n  ZodSocketMessageCatalogSchema,\n} from \"./zodSocket\";\nimport { z } from \"zod\";\nimport { ZodSchemaParsedError } from \"./zodMessageHandler\";\nimport { inspect } from \"node:util\";\n\ninterface ZodIpcMessageSender<TEmitCatalog extends ZodSocketMessageCatalogSchema> {\n  send<K extends GetSocketMessagesWithoutCallback<TEmitCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TEmitCatalog, K>>\n  ): Promise<void>;\n\n  sendWithAck<K extends GetSocketMessagesWithCallback<TEmitCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TEmitCatalog, K>>\n  ): Promise<z.infer<GetSocketCallbackSchema<TEmitCatalog, K>>>;\n}\n\ntype ZodIpcMessageHandlers<\n  TListenCatalog extends ZodSocketMessageCatalogSchema,\n  TEmitCatalog extends ZodSocketMessageCatalogSchema,\n> = Partial<{\n  [K in keyof TListenCatalog]: (\n    payload: z.infer<GetSocketMessageSchema<TListenCatalog, K>>,\n    sender: ZodIpcMessageSender<TEmitCatalog>\n  ) => Promise<\n    SocketMessageHasCallback<TListenCatalog, K> extends true\n      ? z.input<GetSocketCallbackSchema<TListenCatalog, K>>\n      : void\n  >;\n}>;\n\nconst messageSchema = z.object({\n  version: z.literal(\"v1\").default(\"v1\"),\n  type: z.string(),\n  payload: z.unknown(),\n});\n\ntype ZodIpcMessageHandlerOptions<\n  TListenCatalog extends ZodSocketMessageCatalogSchema,\n  TEmitCatalog extends ZodSocketMessageCatalogSchema,\n> = {\n  schema: TListenCatalog;\n  handlers?: ZodIpcMessageHandlers<TListenCatalog, TEmitCatalog>;\n  sender: ZodIpcMessageSender<TEmitCatalog>;\n};\n\nclass ZodIpcMessageHandler<\n  TListenCatalog extends ZodSocketMessageCatalogSchema,\n  TEmitCatalog extends ZodSocketMessageCatalogSchema,\n> {\n  #schema: TListenCatalog;\n  #handlers: ZodIpcMessageHandlers<TListenCatalog, TEmitCatalog> | undefined;\n  #sender: ZodIpcMessageSender<TEmitCatalog>;\n\n  constructor(options: ZodIpcMessageHandlerOptions<TListenCatalog, TEmitCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.handlers;\n    this.#sender = options.sender;\n  }\n\n  public async handleMessage(message: unknown) {\n    const parsedMessage = this.parseMessage(message);\n\n    if (!this.#handlers) {\n      throw new Error(\"No handlers provided\");\n    }\n\n    const handler = this.#handlers[parsedMessage.type];\n\n    if (!handler) {\n      // console.error(`No handler for message type: ${String(parsedMessage.type)}`);\n      return;\n    }\n\n    const ack = await handler(parsedMessage.payload, this.#sender);\n\n    return ack;\n  }\n\n  public parseMessage(message: unknown): MessagesFromSocketCatalog<TListenCatalog> {\n    const parsedMessage = messageSchema.safeParse(message);\n\n    if (!parsedMessage.success) {\n      throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);\n    }\n\n    const schema = this.#schema[parsedMessage.data.type][\"message\"];\n\n    if (!schema) {\n      throw new Error(`Unknown message type: ${parsedMessage.data.type}`);\n    }\n\n    const parsedPayload = schema.safeParse(parsedMessage.data.payload);\n\n    if (!parsedPayload.success) {\n      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    return {\n      type: parsedMessage.data.type,\n      payload: parsedPayload.data,\n    };\n  }\n}\n\nconst Packet = z.discriminatedUnion(\"type\", [\n  z.object({\n    type: z.literal(\"CONNECT\"),\n    sessionId: z.string().optional(),\n  }),\n  z.object({\n    type: z.literal(\"ACK\"),\n    message: z.any(),\n    id: z.number(),\n  }),\n  z.object({\n    type: z.literal(\"EVENT\"),\n    message: z.any(),\n    id: z.number().optional(),\n  }),\n]);\n\ntype Packet = z.infer<typeof Packet>;\n\ninterface ZodIpcConnectionOptions<\n  TListenCatalog extends ZodSocketMessageCatalogSchema,\n  TEmitCatalog extends ZodSocketMessageCatalogSchema,\n> {\n  listenSchema: TListenCatalog;\n  emitSchema: TEmitCatalog;\n  process: {\n    send?: (message: any) => any;\n    on?: (event: \"message\", listener: (message: any) => void) => void;\n  };\n  handlers?: ZodIpcMessageHandlers<TListenCatalog, TEmitCatalog>;\n}\n\nexport class ZodIpcConnection<\n  TListenCatalog extends ZodSocketMessageCatalogSchema,\n  TEmitCatalog extends ZodSocketMessageCatalogSchema,\n> {\n  #sessionId?: string;\n  #messageCounter: number = 0;\n\n  #handler: ZodIpcMessageHandler<TListenCatalog, TEmitCatalog>;\n\n  #acks: Map<\n    number,\n    {\n      resolve: (value: unknown) => void;\n      reject: (reason?: any) => void;\n      timeout: NodeJS.Timeout;\n    }\n  > = new Map();\n\n  constructor(private opts: ZodIpcConnectionOptions<TListenCatalog, TEmitCatalog>) {\n    this.#handler = new ZodIpcMessageHandler({\n      schema: opts.listenSchema,\n      handlers: opts.handlers,\n      sender: {\n        send: this.send.bind(this),\n        sendWithAck: this.sendWithAck.bind(this),\n      },\n    });\n\n    this.#registerHandlers();\n    // this.connect();\n  }\n\n  async #registerHandlers() {\n    if (!this.opts.process.on) {\n      return;\n    }\n\n    this.opts.process.on(\"message\", async (message) => {\n      this.#handlePacket(message);\n    });\n  }\n\n  async connect() {\n    this.#sendPacket({ type: \"CONNECT\" });\n  }\n\n  async #handlePacket(packet: Packet): Promise<void> {\n    const parsedPacket = Packet.safeParse(packet);\n\n    if (!parsedPacket.success) {\n      return;\n    }\n\n    switch (parsedPacket.data.type) {\n      case \"ACK\": {\n        // Check our list of ACKs and resolve with the message\n        const ack = this.#acks.get(parsedPacket.data.id);\n\n        if (!ack) {\n          return;\n        }\n\n        clearTimeout(ack.timeout);\n        ack.resolve(parsedPacket.data.message);\n\n        break;\n      }\n      case \"CONNECT\": {\n        if (!parsedPacket.data.sessionId) {\n          // This is a client trying to connect, so we generate and send back a session ID\n          const id = randomUUID();\n\n          await this.#sendPacket({ type: \"CONNECT\", sessionId: id });\n\n          return;\n        }\n\n        // This is a server replying to our connect message\n        if (this.#sessionId) {\n          // We're already connected\n          return;\n        }\n\n        this.#sessionId = parsedPacket.data.sessionId;\n\n        break;\n      }\n      case \"EVENT\": {\n        const result = await this.#handler.handleMessage(parsedPacket.data.message);\n\n        if (typeof parsedPacket.data.id === \"undefined\") {\n          return;\n        }\n\n        // There's an ID so we should ACK\n        await this.#sendPacket({\n          type: \"ACK\",\n          id: parsedPacket.data.id,\n          message: result,\n        });\n\n        break;\n      }\n      default: {\n        break;\n      }\n    }\n  }\n\n  async #sendPacket(packet: Packet) {\n    await this.opts.process.send?.(packet);\n  }\n\n  async send<K extends GetSocketMessagesWithoutCallback<TEmitCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TEmitCatalog, K>>\n  ): Promise<void> {\n    const schema = this.opts.emitSchema[type][\"message\"];\n\n    if (!schema) {\n      throw new Error(`Unknown message type: ${type as string}`);\n    }\n\n    const parsedPayload = schema.safeParse(payload);\n\n    if (!parsedPayload.success) {\n      throw new ZodSchemaParsedError(parsedPayload.error, payload);\n    }\n\n    await this.#sendPacket({\n      type: \"EVENT\",\n      message: {\n        type,\n        payload,\n        version: \"v1\",\n      },\n    });\n  }\n\n  public async sendWithAck<K extends GetSocketMessagesWithCallback<TEmitCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TEmitCatalog, K>>,\n    timeoutInMs?: number\n  ): Promise<z.infer<GetSocketCallbackSchema<TEmitCatalog, K>>> {\n    const currentId = this.#messageCounter++;\n\n    return new Promise(async (resolve, reject) => {\n      const defaultTimeoutInMs = 2000;\n\n      // Timeout if the ACK takes too long to get back to us\n      const timeout = setTimeout(() => {\n        reject(\n          JSON.stringify({\n            reason: \"sendWithAck() timeout\",\n            timeoutInMs: timeoutInMs ?? defaultTimeoutInMs,\n            type,\n            payload,\n          })\n        );\n      }, timeoutInMs ?? defaultTimeoutInMs);\n\n      this.#acks.set(currentId, { resolve, reject, timeout });\n\n      const schema = this.opts.emitSchema[type][\"message\"];\n\n      if (!schema) {\n        clearTimeout(timeout);\n        return reject(`Unknown message type: ${type as string}`);\n      }\n\n      const parsedPayload = schema.safeParse(payload);\n\n      if (!parsedPayload.success) {\n        clearTimeout(timeout);\n        return reject(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n      }\n\n      await this.#sendPacket({\n        type: \"EVENT\",\n        message: {\n          type,\n          payload,\n          version: \"v1\",\n        },\n        id: currentId,\n      });\n    });\n  }\n}\n","import { z } from \"zod\";\nimport { StructuredLogger } from \"./utils/structuredLogger\";\n\nexport class ZodSchemaParsedError extends Error {\n  constructor(\n    public error: z.ZodError,\n    public payload: unknown\n  ) {\n    super(error.message);\n  }\n}\n\nexport type ZodMessageValueSchema<TDiscriminatedUnion extends z.ZodDiscriminatedUnion<any, any>> =\n  | z.ZodFirstPartySchemaTypes\n  | TDiscriminatedUnion;\n\nexport interface ZodMessageCatalogSchema {\n  [key: string]: ZodMessageValueSchema<any>;\n}\n\nexport type ZodMessageHandlers<TCatalogSchema extends ZodMessageCatalogSchema> = Partial<{\n  [K in keyof TCatalogSchema]: (payload: z.infer<TCatalogSchema[K]>) => Promise<any>;\n}>;\n\nexport type ZodMessageHandlerOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {\n  schema: TMessageCatalog;\n  messages?: ZodMessageHandlers<TMessageCatalog>;\n  logger?: StructuredLogger;\n};\n\nexport type MessageFromSchema<\n  K extends keyof TMessageCatalog,\n  TMessageCatalog extends ZodMessageCatalogSchema,\n> = {\n  type: K;\n  payload: z.input<TMessageCatalog[K]>;\n};\n\nexport type MessageFromCatalog<TMessageCatalog extends ZodMessageCatalogSchema> = {\n  [K in keyof TMessageCatalog]: MessageFromSchema<K, TMessageCatalog>;\n}[keyof TMessageCatalog];\n\nexport const ZodMessageSchema = z.object({\n  version: z.literal(\"v1\").default(\"v1\"),\n  type: z.string(),\n  payload: z.unknown(),\n});\n\nexport interface EventEmitterLike {\n  on(eventName: string | symbol, listener: (...args: any[]) => void): this;\n}\n\nexport class ZodMessageHandler<TMessageCatalog extends ZodMessageCatalogSchema> {\n  #schema: TMessageCatalog;\n  #handlers: ZodMessageHandlers<TMessageCatalog> | undefined;\n  #logger: StructuredLogger | Console;\n\n  constructor(options: ZodMessageHandlerOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.messages;\n    this.#logger = options.logger ?? console;\n  }\n\n  public async handleMessage(message: unknown): Promise<\n    | {\n        success: true;\n        data: unknown;\n      }\n    | {\n        success: false;\n        error: string;\n      }\n  > {\n    const parsedMessage = this.parseMessage(message);\n\n    if (!parsedMessage.success) {\n      this.#logger.error(parsedMessage.error, { message });\n\n      return {\n        success: false,\n        error: parsedMessage.error,\n      };\n    }\n\n    if (!this.#handlers) {\n      this.#logger.error(\"No handlers provided\", { message });\n\n      return {\n        success: false,\n        error: \"No handlers provided\",\n      };\n    }\n\n    const handler = this.#handlers[parsedMessage.data.type];\n\n    if (!handler) {\n      const error = `No handler for message type: ${String(parsedMessage.data.type)}`;\n\n      this.#logger.error(error, { message });\n\n      return {\n        success: false,\n        error,\n      };\n    }\n\n    const ack = await handler(parsedMessage.data.payload);\n\n    return {\n      success: true,\n      data: ack,\n    };\n  }\n\n  public parseMessage(message: unknown):\n    | {\n        success: true;\n        data: MessageFromCatalog<TMessageCatalog>;\n      }\n    | {\n        success: false;\n        error: string;\n      } {\n    const parsedMessage = ZodMessageSchema.safeParse(message);\n\n    if (!parsedMessage.success) {\n      return {\n        success: false,\n        error: `Failed to parse message: ${JSON.stringify(parsedMessage.error)}`,\n      };\n    }\n\n    const schema = this.#schema[parsedMessage.data.type];\n\n    if (!schema) {\n      return {\n        success: false,\n        error: `Unknown message type: ${parsedMessage.data.type}`,\n      };\n    }\n\n    const parsedPayload = schema.safeParse(parsedMessage.data.payload);\n\n    if (!parsedPayload.success) {\n      return {\n        success: false,\n        error: `Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`,\n      };\n    }\n\n    return {\n      success: true,\n      data: {\n        type: parsedMessage.data.type,\n        payload: parsedPayload.data,\n      },\n    };\n  }\n\n  public registerHandlers(emitter: EventEmitterLike, logger?: StructuredLogger) {\n    const log = logger ?? console;\n\n    if (!this.#handlers) {\n      log.info(\"No handlers provided\");\n      return;\n    }\n\n    for (const eventName of Object.keys(this.#schema)) {\n      emitter.on(eventName, async (message: any, callback?: any): Promise<void> => {\n        log.info(`handling ${eventName}`, {\n          payload: message,\n          hasCallback: !!callback,\n        });\n\n        let ack: Awaited<ReturnType<ZodMessageHandler<TMessageCatalog>[\"handleMessage\"]>>;\n\n        // FIXME: this only works if the message doesn't have genuine payload prop\n        if (\"payload\" in message) {\n          ack = await this.handleMessage({ type: eventName, ...message });\n        } else {\n          // Handle messages not sent by ZodMessageSender\n          const { version, ...payload } = message;\n          ack = await this.handleMessage({ type: eventName, version, payload });\n        }\n\n        if (callback && typeof callback === \"function\") {\n          if (!ack.success) {\n            // We don't know the callback type, so we can't do anything else - not all callbacks may accept a success prop\n            log.error(\"Failed to handle message, skipping callback\", { message, error: ack.error });\n            return;\n          }\n\n          callback(ack.data);\n        }\n      });\n    }\n  }\n}\n\ntype ZodMessageSenderCallback<TMessageCatalog extends ZodMessageCatalogSchema> = (message: {\n  type: keyof TMessageCatalog;\n  payload: z.infer<TMessageCatalog[keyof TMessageCatalog]>;\n  version: \"v1\";\n}) => Promise<void>;\n\nexport type ZodMessageSenderOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {\n  schema: TMessageCatalog;\n  sender: ZodMessageSenderCallback<TMessageCatalog>;\n};\n\nexport class ZodMessageSender<TMessageCatalog extends ZodMessageCatalogSchema> {\n  #schema: TMessageCatalog;\n  #sender: ZodMessageSenderCallback<TMessageCatalog>;\n\n  constructor(options: ZodMessageSenderOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#sender = options.sender;\n  }\n\n  public async send<K extends keyof TMessageCatalog>(\n    type: K,\n    payload: z.input<TMessageCatalog[K]>\n  ) {\n    const schema = this.#schema[type];\n\n    if (!schema) {\n      throw new Error(`Unknown message type: ${type as string}`);\n    }\n\n    const parsedPayload = schema.safeParse(payload);\n\n    if (!parsedPayload.success) {\n      throw new ZodSchemaParsedError(parsedPayload.error, payload);\n    }\n\n    try {\n      await this.#sender({ type, payload, version: \"v1\" });\n    } catch (error) {\n      console.error(\"[ZodMessageSender] Failed to send message\", error);\n    }\n  }\n\n  public async forwardMessage(message: unknown) {\n    const parsedMessage = ZodMessageSchema.safeParse(message);\n\n    if (!parsedMessage.success) {\n      throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);\n    }\n\n    const schema = this.#schema[parsedMessage.data.type];\n\n    if (!schema) {\n      throw new Error(`Unknown message type: ${parsedMessage.data.type}`);\n    }\n\n    const parsedPayload = schema.safeParse(parsedMessage.data.payload);\n\n    if (!parsedPayload.success) {\n      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    try {\n      await this.#sender({\n        type: parsedMessage.data.type,\n        payload: parsedPayload.data,\n        version: \"v1\",\n      });\n    } catch (error) {\n      console.error(\"[ZodMessageSender] Failed to forward message\", error);\n    }\n  }\n}\n\nexport type MessageCatalogToSocketIoEvents<TCatalog extends ZodMessageCatalogSchema> = {\n  [K in keyof TCatalog]: (message: z.infer<TCatalog[K]>) => void;\n};\n"]}