@trigger.dev/core 3.0.0-beta.47 → 3.0.0-beta.49

package/dist/v3/zodIpc.js.map

@@ -1 +1 @@
-{"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;AAsCA,IAAMM,mBAAmBP,EAAEQ,OAAO;EACvCC,SAAST,EAAEU,QAAQ,IAAA,EAAMC,QAAQ,IAAA;EACjCC,MAAMZ,EAAEa,OAAM;EACdR,SAASL,EAAEc,QAAO;AACpB,CAAA;;;ADLA,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};\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\n  constructor(options: ZodMessageHandlerOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.messages;\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);\n\n    return ack;\n  }\n\n  public parseMessage(message: unknown): MessageFromCatalog<TMessageCatalog> {\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    return {\n      type: parsedMessage.data.type,\n      payload: parsedPayload.data,\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;\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          callback(ack);\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"]}
+{"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.map

@@ -1 +1 @@
-{"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;AAsCA,IAAMM,mBAAmBP,EAAEQ,OAAO;EACvCC,SAAST,EAAEU,QAAQ,IAAA,EAAMC,QAAQ,IAAA;EACjCC,MAAMZ,EAAEa,OAAM;EACdR,SAASL,EAAEc,QAAO;AACpB,CAAA;;;ADLA,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};\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\n  constructor(options: ZodMessageHandlerOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.messages;\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);\n\n    return ack;\n  }\n\n  public parseMessage(message: unknown): MessageFromCatalog<TMessageCatalog> {\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    return {\n      type: parsedMessage.data.type,\n      payload: parsedPayload.data,\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;\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          callback(ack);\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"]}
+{"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/zodMessageHandler.d.mts

@@ -16,6 +16,7 @@ type ZodMessageHandlers<TCatalogSchema extends ZodMessageCatalogSchema> = Partia
 type ZodMessageHandlerOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {
     schema: TMessageCatalog;
     messages?: ZodMessageHandlers<TMessageCatalog>;
+    logger?: StructuredLogger;
 };
 type MessageFromSchema<K extends keyof TMessageCatalog, TMessageCatalog extends ZodMessageCatalogSchema> = {
     type: K;
@@ -29,8 +30,8 @@ declare const ZodMessageSchema: z.ZodObject<{
     type: z.ZodString;
     payload: z.ZodUnknown;
 }, "strip", z.ZodTypeAny, {
-    version: "v1";
     type: string;
+    version: "v1";
     payload?: unknown;
 }, {
     type: string;
@@ -43,8 +44,20 @@ interface EventEmitterLike {
 declare class ZodMessageHandler<TMessageCatalog extends ZodMessageCatalogSchema> {
     #private;
     constructor(options: ZodMessageHandlerOptions<TMessageCatalog>);
-    handleMessage(message: unknown): Promise<any>;
-    parseMessage(message: unknown): MessageFromCatalog<TMessageCatalog>;
+    handleMessage(message: unknown): Promise<{
+        success: true;
+        data: unknown;
+    } | {
+        success: false;
+        error: string;
+    }>;
+    parseMessage(message: unknown): {
+        success: true;
+        data: MessageFromCatalog<TMessageCatalog>;
+    } | {
+        success: false;
+        error: string;
+    };
     registerHandlers(emitter: EventEmitterLike, logger?: StructuredLogger): void;
 }
 type ZodMessageSenderCallback<TMessageCatalog extends ZodMessageCatalogSchema> = (message: {

package/dist/v3/zodMessageHandler.d.ts

@@ -16,6 +16,7 @@ type ZodMessageHandlers<TCatalogSchema extends ZodMessageCatalogSchema> = Partia
 type ZodMessageHandlerOptions<TMessageCatalog extends ZodMessageCatalogSchema> = {
     schema: TMessageCatalog;
     messages?: ZodMessageHandlers<TMessageCatalog>;
+    logger?: StructuredLogger;
 };
 type MessageFromSchema<K extends keyof TMessageCatalog, TMessageCatalog extends ZodMessageCatalogSchema> = {
     type: K;
@@ -29,8 +30,8 @@ declare const ZodMessageSchema: z.ZodObject<{
     type: z.ZodString;
     payload: z.ZodUnknown;
 }, "strip", z.ZodTypeAny, {
-    version: "v1";
     type: string;
+    version: "v1";
     payload?: unknown;
 }, {
     type: string;
@@ -43,8 +44,20 @@ interface EventEmitterLike {
 declare class ZodMessageHandler<TMessageCatalog extends ZodMessageCatalogSchema> {
     #private;
     constructor(options: ZodMessageHandlerOptions<TMessageCatalog>);
-    handleMessage(message: unknown): Promise<any>;
-    parseMessage(message: unknown): MessageFromCatalog<TMessageCatalog>;
+    handleMessage(message: unknown): Promise<{
+        success: true;
+        data: unknown;
+    } | {
+        success: false;
+        error: string;
+    }>;
+    parseMessage(message: unknown): {
+        success: true;
+        data: MessageFromCatalog<TMessageCatalog>;
+    } | {
+        success: false;
+        error: string;
+    };
     registerHandlers(emitter: EventEmitterLike, logger?: StructuredLogger): void;
 }
 type ZodMessageSenderCallback<TMessageCatalog extends ZodMessageCatalogSchema> = (message: {

package/dist/v3/zodMessageHandler.js

@@ -36,43 +36,81 @@ var ZodMessageSchema = zod.z.object({
   type: zod.z.string(),
   payload: zod.z.unknown()
 });
-var _schema, _handlers;
+var _schema, _handlers, _logger;
 var _ZodMessageHandler = class _ZodMessageHandler {
   constructor(options) {
     __privateAdd(this, _schema, void 0);
     __privateAdd(this, _handlers, void 0);
+    __privateAdd(this, _logger, void 0);
     __privateSet(this, _schema, options.schema);
     __privateSet(this, _handlers, options.messages);
+    __privateSet(this, _logger, options.logger ?? console);
   }
   async handleMessage(message) {
     const parsedMessage = this.parseMessage(message);
+    if (!parsedMessage.success) {
+      __privateGet(this, _logger).error(parsedMessage.error, {
+        message
+      });
+      return {
+        success: false,
+        error: parsedMessage.error
+      };
+    }
     if (!__privateGet(this, _handlers)) {
-      throw new Error("No handlers provided");
+      __privateGet(this, _logger).error("No handlers provided", {
+        message
+      });
+      return {
+        success: false,
+        error: "No handlers provided"
+      };
     }
-    const handler = __privateGet(this, _handlers)[parsedMessage.type];
+    const handler = __privateGet(this, _handlers)[parsedMessage.data.type];
     if (!handler) {
-      console.error(`No handler for message type: ${String(parsedMessage.type)}`);
-      return;
+      const error = `No handler for message type: ${String(parsedMessage.data.type)}`;
+      __privateGet(this, _logger).error(error, {
+        message
+      });
+      return {
+        success: false,
+        error
+      };
     }
-    const ack = await handler(parsedMessage.payload);
-    return ack;
+    const ack = await handler(parsedMessage.data.payload);
+    return {
+      success: true,
+      data: ack
+    };
   }
   parseMessage(message) {
     const parsedMessage = ZodMessageSchema.safeParse(message);
     if (!parsedMessage.success) {
-      throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);
+      return {
+        success: false,
+        error: `Failed to parse message: ${JSON.stringify(parsedMessage.error)}`
+      };
     }
     const schema = __privateGet(this, _schema)[parsedMessage.data.type];
     if (!schema) {
-      throw new Error(`Unknown message type: ${parsedMessage.data.type}`);
+      return {
+        success: false,
+        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 {
+        success: false,
+        error: `Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`
+      };
     }
     return {
-      type: parsedMessage.data.type,
-      payload: parsedPayload.data
+      success: true,
+      data: {
+        type: parsedMessage.data.type,
+        payload: parsedPayload.data
+      }
     };
   }
   registerHandlers(emitter, logger) {
@@ -102,7 +140,14 @@ var _ZodMessageHandler = class _ZodMessageHandler {
           });
         }
         if (callback && typeof callback === "function") {
-          callback(ack);
+          if (!ack.success) {
+            log.error("Failed to handle message, skipping callback", {
+              message,
+              error: ack.error
+            });
+            return;
+          }
+          callback(ack.data);
         }
       });
     }
@@ -110,6 +155,7 @@ var _ZodMessageHandler = class _ZodMessageHandler {
 };
 _schema = new WeakMap();
 _handlers = new WeakMap();
+_logger = new WeakMap();
 __name(_ZodMessageHandler, "ZodMessageHandler");
 var ZodMessageHandler = _ZodMessageHandler;
 var _schema2, _sender;

package/dist/v3/zodMessageHandler.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/v3/zodMessageHandler.ts"],"names":["z","ZodSchemaParsedError","Error","constructor","error","payload","message","ZodMessageSchema","object","version","literal","default","type","string","unknown","ZodMessageHandler","options","schema","messages","handleMessage","parsedMessage","parseMessage","handler","console","String","ack","safeParse","success","JSON","stringify","data","parsedPayload","registerHandlers","emitter","logger","log","info","eventName","Object","keys","on","callback","hasCallback","_schema","ZodMessageSender","sender","send","forwardMessage"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAAA,SAASA,SAAS;AAGX,IAAMC,wBAAN,MAAMA,8BAA6BC,MAAAA;EACxCC,YACSC,OACAC,SACP;AACA,UAAMD,MAAME,OAAO;iBAHZF;mBACAC;EAGT;AACF;AAP0CH;AAAnC,IAAMD,uBAAN;AAsCA,IAAMM,mBAAmBP,EAAEQ,OAAO;EACvCC,SAAST,EAAEU,QAAQ,IAAA,EAAMC,QAAQ,IAAA;EACjCC,MAAMZ,EAAEa,OAAM;EACdR,SAASL,EAAEc,QAAO;AACpB,CAAA;AA7CA;AAmDO,IAAMC,qBAAN,MAAMA,mBAAAA;EAIXZ,YAAYa,SAAoD;AAHhE;AACA;AAGE,uBAAK,SAAUA,QAAQC;AACvB,uBAAK,WAAYD,QAAQE;EAC3B;EAEA,MAAaC,cAAcb,SAAkB;AAC3C,UAAMc,gBAAgB,KAAKC,aAAaf,OAAAA;AAExC,QAAI,CAAC,mBAAK,YAAW;AACnB,YAAM,IAAIJ,MAAM,sBAAA;IAClB;AAEA,UAAMoB,UAAU,mBAAK,WAAUF,cAAcR,IAAI;AAEjD,QAAI,CAACU,SAAS;AACZC,cAAQnB,MAAM,gCAAgCoB,OAAOJ,cAAcR,IAAI,CAAA,EAAG;AAC1E;IACF;AAEA,UAAMa,MAAM,MAAMH,QAAQF,cAAcf,OAAO;AAE/C,WAAOoB;EACT;EAEOJ,aAAaf,SAAuD;AACzE,UAAMc,gBAAgBb,iBAAiBmB,UAAUpB,OAAAA;AAEjD,QAAI,CAACc,cAAcO,SAAS;AAC1B,YAAM,IAAIzB,MAAM,4BAA4B0B,KAAKC,UAAUT,cAAchB,KAAK,CAAA,EAAG;IACnF;AAEA,UAAMa,SAAS,mBAAK,SAAQG,cAAcU,KAAKlB,IAAI;AAEnD,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIf,MAAM,yBAAyBkB,cAAcU,KAAKlB,IAAI,EAAE;IACpE;AAEA,UAAMmB,gBAAgBd,OAAOS,UAAUN,cAAcU,KAAKzB,OAAO;AAEjE,QAAI,CAAC0B,cAAcJ,SAAS;AAC1B,YAAM,IAAIzB,MAAM,oCAAoC0B,KAAKC,UAAUE,cAAc3B,KAAK,CAAA,EAAG;IAC3F;AAEA,WAAO;MACLQ,MAAMQ,cAAcU,KAAKlB;MACzBP,SAAS0B,cAAcD;IACzB;EACF;EAEOE,iBAAiBC,SAA2BC,QAA2B;AAC5E,UAAMC,MAAMD,UAAUX;AAEtB,QAAI,CAAC,mBAAK,YAAW;AACnBY,UAAIC,KAAK,sBAAA;AACT;IACF;AAEA,eAAWC,aAAaC,OAAOC,KAAK,mBAAK,QAAO,GAAG;AACjDN,cAAQO,GAAGH,WAAW,OAAO/B,SAAcmC,aAAkC;AAC3EN,YAAIC,KAAK,YAAYC,SAAAA,IAAa;UAChChC,SAASC;UACToC,aAAa,CAAC,CAACD;QACjB,CAAA;AAEA,YAAIhB;AAGJ,YAAI,aAAanB,SAAS;AACxBmB,gBAAM,MAAM,KAAKN,cAAc;YAAEP,MAAMyB;YAAW,GAAG/B;UAAQ,CAAA;QAC/D,OAAO;AAEL,gBAAM,EAAEG,SAAS,GAAGJ,QAAAA,IAAYC;AAChCmB,gBAAM,MAAM,KAAKN,cAAc;YAAEP,MAAMyB;YAAW5B;YAASJ;UAAQ,CAAA;QACrE;AAEA,YAAIoC,YAAY,OAAOA,aAAa,YAAY;AAC9CA,mBAAShB,GAAAA;QACX;MACF,CAAA;IACF;EACF;AACF;AApFE;AACA;AAFWV;AAAN,IAAMA,oBAAN;AAnDP,IAAA4B,UAAA;AAqJO,IAAMC,oBAAN,MAAMA,kBAAAA;EAIXzC,YAAYa,SAAmD;AAH/D,uBAAA2B,UAAA;AACA;AAGE,uBAAKA,UAAU3B,QAAQC;AACvB,uBAAK,SAAUD,QAAQ6B;EACzB;EAEA,MAAaC,KACXlC,MACAP,SACA;AACA,UAAMY,SAAS,mBAAK0B,UAAQ/B,IAAAA;AAE5B,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIf,MAAM,yBAAyBU,IAAAA,EAAgB;IAC3D;AAEA,UAAMmB,gBAAgBd,OAAOS,UAAUrB,OAAAA;AAEvC,QAAI,CAAC0B,cAAcJ,SAAS;AAC1B,YAAM,IAAI1B,qBAAqB8B,cAAc3B,OAAOC,OAAAA;IACtD;AAEA,QAAI;AACF,YAAM,mBAAK,SAAL,WAAa;QAAEO;QAAMP;QAASI,SAAS;MAAK;IACpD,SAASL,OAAO;AACdmB,cAAQnB,MAAM,6CAA6CA,KAAAA;IAC7D;EACF;EAEA,MAAa2C,eAAezC,SAAkB;AAC5C,UAAMc,gBAAgBb,iBAAiBmB,UAAUpB,OAAAA;AAEjD,QAAI,CAACc,cAAcO,SAAS;AAC1B,YAAM,IAAIzB,MAAM,4BAA4B0B,KAAKC,UAAUT,cAAchB,KAAK,CAAA,EAAG;IACnF;AAEA,UAAMa,SAAS,mBAAK0B,UAAQvB,cAAcU,KAAKlB,IAAI;AAEnD,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIf,MAAM,yBAAyBkB,cAAcU,KAAKlB,IAAI,EAAE;IACpE;AAEA,UAAMmB,gBAAgBd,OAAOS,UAAUN,cAAcU,KAAKzB,OAAO;AAEjE,QAAI,CAAC0B,cAAcJ,SAAS;AAC1B,YAAM,IAAIzB,MAAM,oCAAoC0B,KAAKC,UAAUE,cAAc3B,KAAK,CAAA,EAAG;IAC3F;AAEA,QAAI;AACF,YAAM,mBAAK,SAAL,WAAa;QACjBQ,MAAMQ,cAAcU,KAAKlB;QACzBP,SAAS0B,cAAcD;QACvBrB,SAAS;MACX;IACF,SAASL,OAAO;AACdmB,cAAQnB,MAAM,gDAAgDA,KAAAA;IAChE;EACF;AACF;AA5DEuC,WAAA;AACA;AAFWC;AAAN,IAAMA,mBAAN","sourcesContent":["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};\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\n  constructor(options: ZodMessageHandlerOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.messages;\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);\n\n    return ack;\n  }\n\n  public parseMessage(message: unknown): MessageFromCatalog<TMessageCatalog> {\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    return {\n      type: parsedMessage.data.type,\n      payload: parsedPayload.data,\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;\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          callback(ack);\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"]}
+{"version":3,"sources":["../../src/v3/zodMessageHandler.ts"],"names":["z","ZodSchemaParsedError","Error","constructor","error","payload","message","ZodMessageSchema","object","version","literal","default","type","string","unknown","ZodMessageHandler","options","schema","messages","logger","console","handleMessage","parsedMessage","parseMessage","success","handler","data","String","ack","safeParse","JSON","stringify","parsedPayload","registerHandlers","emitter","log","info","eventName","Object","keys","on","callback","hasCallback","_schema","ZodMessageSender","sender","send","forwardMessage"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAAA,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;AA9CA;AAoDO,IAAMC,qBAAN,MAAMA,mBAAAA;EAKXZ,YAAYa,SAAoD;AAJhE;AACA;AACA;AAGE,uBAAK,SAAUA,QAAQC;AACvB,uBAAK,WAAYD,QAAQE;AACzB,uBAAK,SAAUF,QAAQG,UAAUC;EACnC;EAEA,MAAaC,cAAcf,SASzB;AACA,UAAMgB,gBAAgB,KAAKC,aAAajB,OAAAA;AAExC,QAAI,CAACgB,cAAcE,SAAS;AAC1B,yBAAK,SAAQpB,MAAMkB,cAAclB,OAAO;QAAEE;MAAQ,CAAA;AAElD,aAAO;QACLkB,SAAS;QACTpB,OAAOkB,cAAclB;MACvB;IACF;AAEA,QAAI,CAAC,mBAAK,YAAW;AACnB,yBAAK,SAAQA,MAAM,wBAAwB;QAAEE;MAAQ,CAAA;AAErD,aAAO;QACLkB,SAAS;QACTpB,OAAO;MACT;IACF;AAEA,UAAMqB,UAAU,mBAAK,WAAUH,cAAcI,KAAKd,IAAI;AAEtD,QAAI,CAACa,SAAS;AACZ,YAAMrB,QAAQ,gCAAgCuB,OAAOL,cAAcI,KAAKd,IAAI,CAAA;AAE5E,yBAAK,SAAQR,MAAMA,OAAO;QAAEE;MAAQ,CAAA;AAEpC,aAAO;QACLkB,SAAS;QACTpB;MACF;IACF;AAEA,UAAMwB,MAAM,MAAMH,QAAQH,cAAcI,KAAKrB,OAAO;AAEpD,WAAO;MACLmB,SAAS;MACTE,MAAME;IACR;EACF;EAEOL,aAAajB,SAQd;AACJ,UAAMgB,gBAAgBf,iBAAiBsB,UAAUvB,OAAAA;AAEjD,QAAI,CAACgB,cAAcE,SAAS;AAC1B,aAAO;QACLA,SAAS;QACTpB,OAAO,4BAA4B0B,KAAKC,UAAUT,cAAclB,KAAK,CAAA;MACvE;IACF;AAEA,UAAMa,SAAS,mBAAK,SAAQK,cAAcI,KAAKd,IAAI;AAEnD,QAAI,CAACK,QAAQ;AACX,aAAO;QACLO,SAAS;QACTpB,OAAO,yBAAyBkB,cAAcI,KAAKd,IAAI;MACzD;IACF;AAEA,UAAMoB,gBAAgBf,OAAOY,UAAUP,cAAcI,KAAKrB,OAAO;AAEjE,QAAI,CAAC2B,cAAcR,SAAS;AAC1B,aAAO;QACLA,SAAS;QACTpB,OAAO,oCAAoC0B,KAAKC,UAAUC,cAAc5B,KAAK,CAAA;MAC/E;IACF;AAEA,WAAO;MACLoB,SAAS;MACTE,MAAM;QACJd,MAAMU,cAAcI,KAAKd;QACzBP,SAAS2B,cAAcN;MACzB;IACF;EACF;EAEOO,iBAAiBC,SAA2Bf,QAA2B;AAC5E,UAAMgB,MAAMhB,UAAUC;AAEtB,QAAI,CAAC,mBAAK,YAAW;AACnBe,UAAIC,KAAK,sBAAA;AACT;IACF;AAEA,eAAWC,aAAaC,OAAOC,KAAK,mBAAK,QAAO,GAAG;AACjDL,cAAQM,GAAGH,WAAW,OAAO/B,SAAcmC,aAAkC;AAC3EN,YAAIC,KAAK,YAAYC,SAAAA,IAAa;UAChChC,SAASC;UACToC,aAAa,CAAC,CAACD;QACjB,CAAA;AAEA,YAAIb;AAGJ,YAAI,aAAatB,SAAS;AACxBsB,gBAAM,MAAM,KAAKP,cAAc;YAAET,MAAMyB;YAAW,GAAG/B;UAAQ,CAAA;QAC/D,OAAO;AAEL,gBAAM,EAAEG,SAAS,GAAGJ,QAAAA,IAAYC;AAChCsB,gBAAM,MAAM,KAAKP,cAAc;YAAET,MAAMyB;YAAW5B;YAASJ;UAAQ,CAAA;QACrE;AAEA,YAAIoC,YAAY,OAAOA,aAAa,YAAY;AAC9C,cAAI,CAACb,IAAIJ,SAAS;AAEhBW,gBAAI/B,MAAM,+CAA+C;cAAEE;cAASF,OAAOwB,IAAIxB;YAAM,CAAA;AACrF;UACF;AAEAqC,mBAASb,IAAIF,IAAI;QACnB;MACF,CAAA;IACF;EACF;AACF;AAhJE;AACA;AACA;AAHWX;AAAN,IAAMA,oBAAN;AApDP,IAAA4B,UAAA;AAkNO,IAAMC,oBAAN,MAAMA,kBAAAA;EAIXzC,YAAYa,SAAmD;AAH/D,uBAAA2B,UAAA;AACA;AAGE,uBAAKA,UAAU3B,QAAQC;AACvB,uBAAK,SAAUD,QAAQ6B;EACzB;EAEA,MAAaC,KACXlC,MACAP,SACA;AACA,UAAMY,SAAS,mBAAK0B,UAAQ/B,IAAAA;AAE5B,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIf,MAAM,yBAAyBU,IAAAA,EAAgB;IAC3D;AAEA,UAAMoB,gBAAgBf,OAAOY,UAAUxB,OAAAA;AAEvC,QAAI,CAAC2B,cAAcR,SAAS;AAC1B,YAAM,IAAIvB,qBAAqB+B,cAAc5B,OAAOC,OAAAA;IACtD;AAEA,QAAI;AACF,YAAM,mBAAK,SAAL,WAAa;QAAEO;QAAMP;QAASI,SAAS;MAAK;IACpD,SAASL,OAAO;AACdgB,cAAQhB,MAAM,6CAA6CA,KAAAA;IAC7D;EACF;EAEA,MAAa2C,eAAezC,SAAkB;AAC5C,UAAMgB,gBAAgBf,iBAAiBsB,UAAUvB,OAAAA;AAEjD,QAAI,CAACgB,cAAcE,SAAS;AAC1B,YAAM,IAAItB,MAAM,4BAA4B4B,KAAKC,UAAUT,cAAclB,KAAK,CAAA,EAAG;IACnF;AAEA,UAAMa,SAAS,mBAAK0B,UAAQrB,cAAcI,KAAKd,IAAI;AAEnD,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIf,MAAM,yBAAyBoB,cAAcI,KAAKd,IAAI,EAAE;IACpE;AAEA,UAAMoB,gBAAgBf,OAAOY,UAAUP,cAAcI,KAAKrB,OAAO;AAEjE,QAAI,CAAC2B,cAAcR,SAAS;AAC1B,YAAM,IAAItB,MAAM,oCAAoC4B,KAAKC,UAAUC,cAAc5B,KAAK,CAAA,EAAG;IAC3F;AAEA,QAAI;AACF,YAAM,mBAAK,SAAL,WAAa;QACjBQ,MAAMU,cAAcI,KAAKd;QACzBP,SAAS2B,cAAcN;QACvBjB,SAAS;MACX;IACF,SAASL,OAAO;AACdgB,cAAQhB,MAAM,gDAAgDA,KAAAA;IAChE;EACF;AACF;AA5DEuC,WAAA;AACA;AAFWC;AAAN,IAAMA,mBAAN","sourcesContent":["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/zodMessageHandler.mjs

@@ -34,43 +34,81 @@ var ZodMessageSchema = z.object({
   type: z.string(),
   payload: z.unknown()
 });
-var _schema, _handlers;
+var _schema, _handlers, _logger;
 var _ZodMessageHandler = class _ZodMessageHandler {
   constructor(options) {
     __privateAdd(this, _schema, void 0);
     __privateAdd(this, _handlers, void 0);
+    __privateAdd(this, _logger, void 0);
     __privateSet(this, _schema, options.schema);
     __privateSet(this, _handlers, options.messages);
+    __privateSet(this, _logger, options.logger ?? console);
   }
   async handleMessage(message) {
     const parsedMessage = this.parseMessage(message);
+    if (!parsedMessage.success) {
+      __privateGet(this, _logger).error(parsedMessage.error, {
+        message
+      });
+      return {
+        success: false,
+        error: parsedMessage.error
+      };
+    }
     if (!__privateGet(this, _handlers)) {
-      throw new Error("No handlers provided");
+      __privateGet(this, _logger).error("No handlers provided", {
+        message
+      });
+      return {
+        success: false,
+        error: "No handlers provided"
+      };
     }
-    const handler = __privateGet(this, _handlers)[parsedMessage.type];
+    const handler = __privateGet(this, _handlers)[parsedMessage.data.type];
     if (!handler) {
-      console.error(`No handler for message type: ${String(parsedMessage.type)}`);
-      return;
+      const error = `No handler for message type: ${String(parsedMessage.data.type)}`;
+      __privateGet(this, _logger).error(error, {
+        message
+      });
+      return {
+        success: false,
+        error
+      };
     }
-    const ack = await handler(parsedMessage.payload);
-    return ack;
+    const ack = await handler(parsedMessage.data.payload);
+    return {
+      success: true,
+      data: ack
+    };
   }
   parseMessage(message) {
     const parsedMessage = ZodMessageSchema.safeParse(message);
     if (!parsedMessage.success) {
-      throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);
+      return {
+        success: false,
+        error: `Failed to parse message: ${JSON.stringify(parsedMessage.error)}`
+      };
     }
     const schema = __privateGet(this, _schema)[parsedMessage.data.type];
     if (!schema) {
-      throw new Error(`Unknown message type: ${parsedMessage.data.type}`);
+      return {
+        success: false,
+        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 {
+        success: false,
+        error: `Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`
+      };
     }
     return {
-      type: parsedMessage.data.type,
-      payload: parsedPayload.data
+      success: true,
+      data: {
+        type: parsedMessage.data.type,
+        payload: parsedPayload.data
+      }
     };
   }
   registerHandlers(emitter, logger) {
@@ -100,7 +138,14 @@ var _ZodMessageHandler = class _ZodMessageHandler {
           });
         }
         if (callback && typeof callback === "function") {
-          callback(ack);
+          if (!ack.success) {
+            log.error("Failed to handle message, skipping callback", {
+              message,
+              error: ack.error
+            });
+            return;
+          }
+          callback(ack.data);
         }
       });
     }
@@ -108,6 +153,7 @@ var _ZodMessageHandler = class _ZodMessageHandler {
 };
 _schema = new WeakMap();
 _handlers = new WeakMap();
+_logger = new WeakMap();
 __name(_ZodMessageHandler, "ZodMessageHandler");
 var ZodMessageHandler = _ZodMessageHandler;
 var _schema2, _sender;