@trigger.dev/core 3.0.0-beta.34 → 3.0.0-beta.36

package/dist/v3/zodSocket.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/v3/zodSocket.ts","../../src/v3/utils/structuredLogger.ts"],"names":["io","z","LogLevel","SimpleStructuredLogger","constructor","name","level","includes","process","env","DEBUG","debug","info","fields","child","log","message","args","console","error","warn","loggerFunction","structuredLog","length","timestamp","Date","JSON","stringify","messageSchema","object","version","literal","default","type","string","payload","unknown","ZodSocketMessageHandler","options","schema","handlers","handleMessage","parsedMessage","parseMessage","Error","handler","String","ack","safeParse","success","data","parsedPayload","registerHandlers","emitter","logger","eventName","Object","keys","on","callback","hasCallback","_schema","ZodSocketMessageSender","socket","send","emit","sendWithAck","callbackResult","emitWithAck","ZodSocketConnection","opts","uri","secure","host","port","namespace","transports","auth","token","authToken","extraHeaders","reconnectionDelay","reconnectionDelayMax","socketId","id","serverMessages","clientMessages","onError","onConnection","reason","description","onDisconnect","close","connect","bind"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AACA,SAASA,UAAU;AACnB,SAASC,SAAS;;;ICSX;UAAKC,WAAQ;AAARA,EAAAA,UAAAA,UACV,KAAA,IAAA,CAAA,IAAA;AADUA,EAAAA,UAAAA,UAEV,OAAA,IAAA,CAAA,IAAA;AAFUA,EAAAA,UAAAA,UAGV,MAAA,IAAA,CAAA,IAAA;AAHUA,EAAAA,UAAAA,UAIV,MAAA,IAAA,CAAA,IAAA;AAJUA,EAAAA,UAAAA,UAKV,OAAA,IAAA,CAAA,IAAA;GALUA,aAAAA,WAAAA,CAAAA,EAAAA;;AAQL,IAAMC,0BAAN,MAAMA,wBAAAA;EACXC,YACUC,MACAC,QAAkB;IAAC;IAAK;IAAQC,SAASC,QAAQC,IAAIC,SAAS,EAAA,IAClER,SAASS,QACTT,SAASU,MACLC,QACR;AAoCF;gBAzCUR;iBACAC;kBAGAO;EACP;EAEHC,MAAMD,QAAiCP,OAAkB;AACvD,WAAO,IAAIH,wBAAuB,KAAKE,MAAMC,OAAO;MAAE,GAAG,KAAKO;MAAQ,GAAGA;IAAO,CAAA;EAClF;EAEAE,IAAIC,YAAoBC,MAAsB;AAC5C,QAAI,KAAKX,QAAQJ,SAASa;AAAK;AAE/B,0BAAK,kCAAL,WAAoBG,QAAQH,KAAKC,SAAS,OAAA,GAAUC;EACtD;EAEAE,MAAMH,YAAoBC,MAAsB;AAC9C,QAAI,KAAKX,QAAQJ,SAASiB;AAAO;AAEjC,0BAAK,kCAAL,WAAoBD,QAAQC,OAAOH,SAAS,SAAA,GAAYC;EAC1D;EAEAG,KAAKJ,YAAoBC,MAAsB;AAC7C,QAAI,KAAKX,QAAQJ,SAASkB;AAAM;AAEhC,0BAAK,kCAAL,WAAoBF,QAAQE,MAAMJ,SAAS,QAAA,GAAWC;EACxD;EAEAL,KAAKI,YAAoBC,MAAsB;AAC7C,QAAI,KAAKX,QAAQJ,SAASU;AAAM;AAEhC,0BAAK,kCAAL,WAAoBM,QAAQN,MAAMI,SAAS,QAAA,GAAWC;EACxD;EAEAN,MAAMK,YAAoBC,MAAsB;AAC9C,QAAI,KAAKX,QAAQJ,SAASS;AAAO;AAEjC,0BAAK,kCAAL,WAAoBO,QAAQP,OAAOK,SAAS,SAAA,GAAYC;EAC1D;AAmBF;AAjBE;mBAAc,gCACZI,gBACAL,SACAV,UACGW,MACH;AACA,QAAMK,gBAAgB;IACpB,GAAIL,KAAKM,WAAW,IAAIN,KAAK,CAAA,IAAKA;IAClC,GAAG,KAAKJ;IACRW,WAAW,oBAAIC,KAAAA;IACfpB,MAAM,KAAKA;IACXW;IACAV;EACF;AAEAe,iBAAeK,KAAKC,UAAUL,aAAAA,CAAAA;AAChC,GAhBc;AA3CHnB;AAAN,IAAMA,yBAAN;;;AD+DP,IAAMyB,gBAAgB3B,EAAE4B,OAAO;EAC7BC,SAAS7B,EAAE8B,QAAQ,IAAA,EAAMC,QAAQ,IAAA;EACjCC,MAAMhC,EAAEiC,OAAM;EACdC,SAASlC,EAAEmC,QAAO;AACpB,CAAA;AArFA;AAuFO,IAAMC,2BAAN,MAAMA,yBAAAA;EAIXjC,YAAYkC,SAAsD;AAHlE;AACA;AAGE,uBAAK,SAAUA,QAAQC;AACvB,uBAAK,WAAYD,QAAQE;EAC3B;EAEA,MAAaC,cAAczB,SAAkB;AAC3C,UAAM0B,gBAAgB,KAAKC,aAAa3B,OAAAA;AAExC,QAAI,CAAC,mBAAK,YAAW;AACnB,YAAM,IAAI4B,MAAM,sBAAA;IAClB;AAEA,UAAMC,UAAU,mBAAK,WAAUH,cAAcT,IAAI;AAEjD,QAAI,CAACY,SAAS;AACZ3B,cAAQC,MAAM,gCAAgC2B,OAAOJ,cAAcT,IAAI,CAAA,EAAG;AAC1E;IACF;AAEA,UAAMc,MAAM,MAAMF,QAAQH,cAAcP,OAAO;AAE/C,WAAOY;EACT;EAEOJ,aAAa3B,SAA0D;AAC5E,UAAM0B,gBAAgBd,cAAcoB,UAAUhC,OAAAA;AAE9C,QAAI,CAAC0B,cAAcO,SAAS;AAC1B,YAAM,IAAIL,MAAM,4BAA4BlB,KAAKC,UAAUe,cAAcvB,KAAK,CAAA,EAAG;IACnF;AAEA,UAAMoB,SAAS,mBAAK,SAAQG,cAAcQ,KAAKjB,IAAI,EAAE,SAAA;AAErD,QAAI,CAACM,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBF,cAAcQ,KAAKjB,IAAI,EAAE;IACpE;AAEA,UAAMkB,gBAAgBZ,OAAOS,UAAUN,cAAcQ,KAAKf,OAAO;AAEjE,QAAI,CAACgB,cAAcF,SAAS;AAC1B,YAAM,IAAIL,MAAM,oCAAoClB,KAAKC,UAAUwB,cAAchC,KAAK,CAAA,EAAG;IAC3F;AAEA,WAAO;MACLc,MAAMS,cAAcQ,KAAKjB;MACzBE,SAASgB,cAAcD;IACzB;EACF;EAEOE,iBAAiBC,SAA2BC,QAA2B;AAC5E,UAAMvC,MAAMuC,UAAUpC;AAEtB,QAAI,CAAC,mBAAK,YAAW;AACnBH,UAAIH,KAAK,sBAAA;AACT;IACF;AAEA,eAAW2C,aAAaC,OAAOC,KAAK,mBAAK,UAAS,GAAG;AACnDJ,cAAQK,GAAGH,WAAW,OAAOvC,SAAc2C,aAAkC;AAC3E5C,YAAIH,KAAK,YAAY2C,SAAAA,IAAa;UAChCpB,SAASnB;UACT4C,aAAa,CAAC,CAACD;QACjB,CAAA;AAEA,YAAIZ;AAEJ,YAAI;AAEF,cAAI,aAAa/B,SAAS;AACxB+B,kBAAM,MAAM,KAAKN,cAAc;cAAER,MAAMsB;cAAW,GAAGvC;YAAQ,CAAA;UAC/D,OAAO;AAEL,kBAAM,EAAEc,SAAS,GAAGK,QAAAA,IAAYnB;AAChC+B,kBAAM,MAAM,KAAKN,cAAc;cAAER,MAAMsB;cAAWzB;cAASK;YAAQ,CAAA;UACrE;QACF,SAAShB,OAAO;AACdJ,cAAII,MAAM,gCAAgC;YAAEA;UAAM,CAAA;AAClD;QACF;AAEA,YAAIwC,YAAY,OAAOA,aAAa,YAAY;AAC9CA,mBAASZ,GAAAA;QACX;MACF,CAAA;IACF;EACF;AACF;AAzFE;AACA;AAFWV;AAAN,IAAMA,0BAAN;AAvFP,IAAAwB,UAAA;AAsMO,IAAMC,0BAAN,MAAMA,wBAAAA;EAIX1D,YAAYkC,SAAyD;AAHrE,uBAAAuB,UAAA;AACA;AAGE,uBAAKA,UAAUvB,QAAQC;AACvB,uBAAK,SAAUD,QAAQyB;EACzB;EAEOC,KACL/B,MACAE,SACM;AACN,UAAMI,SAAS,mBAAKsB,UAAQ5B,IAAAA,EAAM,SAAA;AAElC,QAAI,CAACM,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBX,IAAAA,EAAgB;IAC3D;AAEA,UAAMkB,gBAAgBZ,OAAOS,UAAUb,OAAAA;AAEvC,QAAI,CAACgB,cAAcF,SAAS;AAC1B,YAAM,IAAIL,MAAM,oCAAoClB,KAAKC,UAAUwB,cAAchC,KAAK,CAAA,EAAG;IAC3F;AAGA,uBAAK,SAAQ8C,KAAKhC,MAAM;MAAEE;MAASL,SAAS;IAAK,CAAA;AAEjD;EACF;EAEA,MAAaoC,YACXjC,MACAE,SAC+D;AAC/D,UAAMI,SAAS,mBAAKsB,UAAQ5B,IAAAA,EAAM,SAAA;AAElC,QAAI,CAACM,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBX,IAAAA,EAAgB;IAC3D;AAEA,UAAMkB,gBAAgBZ,OAAOS,UAAUb,OAAAA;AAEvC,QAAI,CAACgB,cAAcF,SAAS;AAC1B,YAAM,IAAIL,MAAM,oCAAoClB,KAAKC,UAAUwB,cAAchC,KAAK,CAAA,EAAG;IAC3F;AAGA,UAAMgD,iBAAiB,MAAM,mBAAK,SAAQC,YAAYnC,MAAM;MAAEE;MAASL,SAAS;IAAK,CAAA;AAErF,WAAOqC;EACT;AACF;AAnDEN,WAAA;AACA;AAFWC;AAAN,IAAMA,yBAAN;AAtMP;AAsSO,IAAMO,uBAAN,MAAMA,qBAAAA;EAUXjE,YAAYkE,MAAoE;AANhF;AAGA;AACA;AAGE,UAAMC,MAAM,GAAGD,KAAKE,SAAS,QAAQ,IAAI,MAAMF,KAAKG,IAAI,IACtDH,KAAKI,SAASJ,KAAKE,SAAS,QAAQ,KAAG,IACrCF,KAAKK,SAAS;AAElB,UAAMrB,SAAS,IAAInD,uBAAuBmE,KAAKK,WAAWzE,SAASU,IAAI;AACvE0C,WAAOvC,IAAI,kBAAkB;MAAEwD;IAAI,CAAA;AAEnC,SAAKR,SAAS/D,GAAGuE,KAAK;MACpBK,YAAY;QAAC;;MACbC,MAAM;QACJC,OAAOR,KAAKS;MACd;MACAC,cAAcV,KAAKU;MACnBC,mBAAmB;MACnBC,sBAAsB;IACxB,CAAA;AAEA,uBAAK,SAAU5B,OAAOxC,MAAM;MAC1BqE,UAAU,KAAKpB,OAAOqB;IACxB,CAAA;AAEA,uBAAK,UAAW,IAAI/C,wBAAwB;MAC1CE,QAAQ+B,KAAKe;MACb7C,UAAU8B,KAAK9B;IACjB,CAAA;AACA,uBAAK,UAASY,iBAAiB,KAAKW,QAAQ,mBAAK,QAAO;AAExD,uBAAK,SAAU,IAAID,uBAAuB;MACxCvB,QAAQ+B,KAAKgB;MACbvB,QAAQ,KAAKA;IACf,CAAA;AAEA,SAAKA,OAAOL,GAAG,iBAAiB,OAAOvC,UAAU;AAC/C,yBAAK,SAAQA,MAAM,kBAAkBA,KAAAA,EAAO;AAE5C,UAAImD,KAAKiB,SAAS;AAChB,cAAMjB,KAAKiB,QAAQ,KAAKxB,QAAQ5C,OAAO,mBAAK,QAAO;MACrD;IACF,CAAA;AAEA,SAAK4C,OAAOL,GAAG,WAAW,YAAY;AACpC,yBAAK,SAAQ9C,KAAK,SAAA;AAElB,UAAI0D,KAAKkB,cAAc;AACrB,cAAMlB,KAAKkB,aAAa,KAAKzB,QAAQ,mBAAK,WAAU,mBAAK,UAAS,mBAAK,QAAO;MAChF;IACF,CAAA;AAEA,SAAKA,OAAOL,GAAG,cAAc,OAAO+B,QAAQC,gBAAgB;AAC1D,yBAAK,SAAQ9E,KAAK,cAAc;QAAE6E;QAAQC;MAAY,CAAA;AAEtD,UAAIpB,KAAKqB,cAAc;AACrB,cAAMrB,KAAKqB,aAAa,KAAK5B,QAAQ0B,QAAQC,aAAa,mBAAK,QAAO;MACxE;IACF,CAAA;EACF;EAEAE,QAAQ;AACN,SAAK7B,OAAO6B,MAAK;EACnB;EAEAC,UAAU;AACR,SAAK9B,OAAO8B,QAAO;EACrB;EAEA,IAAI7B,OAAO;AACT,WAAO,mBAAK,SAAQA,KAAK8B,KAAK,mBAAK,QAAO;EAC5C;EAEA,IAAI5B,cAAc;AAChB,WAAO,mBAAK,SAAQA,YAAY4B,KAAK,mBAAK,QAAO;EACnD;AACF;AA/EE;AAGA;AACA;AARWzB;AAAN,IAAMA,sBAAN","sourcesContent":["import type { Socket } from \"socket.io-client\";\nimport { io } from \"socket.io-client\";\nimport { z } from \"zod\";\nimport { EventEmitterLike, ZodMessageValueSchema } from \"./zodMessageHandler\";\nimport { LogLevel, SimpleStructuredLogger, StructuredLogger } from \"./utils/structuredLogger\";\n\nexport interface ZodSocketMessageCatalogSchema {\n  [key: string]:\n    | {\n        message: ZodMessageValueSchema<any>;\n      }\n    | {\n        message: ZodMessageValueSchema<any>;\n        callback?: ZodMessageValueSchema<any>;\n      };\n}\n\nexport type ZodMessageCatalogToSocketIoEvents<TCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TCatalog]: SocketMessageHasCallback<TCatalog, K> extends true\n    ? (\n        message: z.infer<GetSocketMessageSchema<TCatalog, K>>,\n        callback: (ack: z.infer<GetSocketCallbackSchema<TCatalog, K>>) => void\n      ) => void\n    : (message: z.infer<GetSocketMessageSchema<TCatalog, K>>) => void;\n};\n\nexport type GetSocketMessageSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = TRPCCatalog[TMessageType][\"message\"];\n\nexport type InferSocketMessageSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = z.infer<GetSocketMessageSchema<TRPCCatalog, TMessageType>>;\n\nexport type GetSocketCallbackSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = TRPCCatalog[TMessageType] extends { callback: any }\n  ? TRPCCatalog[TMessageType][\"callback\"]\n  : never;\n\nexport type InferSocketCallbackSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = z.infer<GetSocketCallbackSchema<TRPCCatalog, TMessageType>>;\n\nexport type SocketMessageHasCallback<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = GetSocketCallbackSchema<TRPCCatalog, TMessageType> extends never ? false : true;\n\nexport type ZodSocketMessageHandlers<TCatalogSchema extends ZodSocketMessageCatalogSchema> =\n  Partial<{\n    [K in keyof TCatalogSchema]: (\n      payload: z.infer<GetSocketMessageSchema<TCatalogSchema, K>>\n    ) => Promise<\n      SocketMessageHasCallback<TCatalogSchema, K> extends true\n        ? z.input<GetSocketCallbackSchema<TCatalogSchema, K>>\n        : void\n    >;\n  }>;\n\nexport type ZodSocketMessageHandlerOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> =\n  {\n    schema: TMessageCatalog;\n    handlers?: ZodSocketMessageHandlers<TMessageCatalog>;\n  };\n\ntype MessageFromSocketSchema<\n  K extends keyof TMessageCatalog,\n  TMessageCatalog extends ZodSocketMessageCatalogSchema,\n> = {\n  type: K;\n  payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>;\n};\n\nexport type MessagesFromSocketCatalog<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TMessageCatalog]: MessageFromSocketSchema<K, TMessageCatalog>;\n}[keyof TMessageCatalog];\n\nconst messageSchema = z.object({\n  version: z.literal(\"v1\").default(\"v1\"),\n  type: z.string(),\n  payload: z.unknown(),\n});\n\nexport class ZodSocketMessageHandler<TRPCCatalog extends ZodSocketMessageCatalogSchema> {\n  #schema: TRPCCatalog;\n  #handlers: ZodSocketMessageHandlers<TRPCCatalog> | undefined;\n\n  constructor(options: ZodSocketMessageHandlerOptions<TRPCCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.handlers;\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): MessagesFromSocketCatalog<TRPCCatalog> {\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  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.#handlers)) {\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        try {\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        } catch (error) {\n          log.error(\"Error while handling message\", { error });\n          return;\n        }\n\n        if (callback && typeof callback === \"function\") {\n          callback(ack);\n        }\n      });\n    }\n  }\n}\n\nexport type ZodSocketMessageSenderOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  schema: TMessageCatalog;\n  socket: ZodSocket<any, TMessageCatalog>;\n};\n\nexport type GetSocketMessagesWithCallback<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true\n    ? K\n    : never;\n}[keyof TMessageCatalog];\n\nexport type GetSocketMessagesWithoutCallback<\n  TMessageCatalog extends ZodSocketMessageCatalogSchema,\n> = {\n  [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true\n    ? never\n    : K;\n}[keyof TMessageCatalog];\n\nexport class ZodSocketMessageSender<TMessageCatalog extends ZodSocketMessageCatalogSchema> {\n  #schema: TMessageCatalog;\n  #socket: ZodSocket<any, TMessageCatalog>;\n\n  constructor(options: ZodSocketMessageSenderOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#socket = options.socket;\n  }\n\n  public send<K extends GetSocketMessagesWithoutCallback<TMessageCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>\n  ): void {\n    const schema = this.#schema[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 Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    // @ts-expect-error\n    this.#socket.emit(type, { payload, version: \"v1\" });\n\n    return;\n  }\n\n  public async sendWithAck<K extends GetSocketMessagesWithCallback<TMessageCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>\n  ): Promise<z.infer<GetSocketCallbackSchema<TMessageCatalog, K>>> {\n    const schema = this.#schema[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 Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    // @ts-expect-error\n    const callbackResult = await this.#socket.emitWithAck(type, { payload, version: \"v1\" });\n\n    return callbackResult;\n  }\n}\n\nexport type ZodSocket<\n  TListenEvents extends ZodSocketMessageCatalogSchema,\n  TEmitEvents extends ZodSocketMessageCatalogSchema,\n> = Socket<\n  ZodMessageCatalogToSocketIoEvents<TListenEvents>,\n  ZodMessageCatalogToSocketIoEvents<TEmitEvents>\n>;\n\ninterface ZodSocketConnectionOptions<\n  TClientMessages extends ZodSocketMessageCatalogSchema,\n  TServerMessages extends ZodSocketMessageCatalogSchema,\n> {\n  host: string;\n  port?: number;\n  secure?: boolean;\n  namespace: string;\n  clientMessages: TClientMessages;\n  serverMessages: TServerMessages;\n  extraHeaders?: {\n    [header: string]: string;\n  };\n  handlers?: ZodSocketMessageHandlers<TServerMessages>;\n  authToken?: string;\n  onConnection?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    handler: ZodSocketMessageHandler<TServerMessages>,\n    sender: ZodSocketMessageSender<TClientMessages>,\n    logger: StructuredLogger\n  ) => Promise<void>;\n  onDisconnect?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    reason: Socket.DisconnectReason,\n    description: any,\n    logger: StructuredLogger\n  ) => Promise<void>;\n  onError?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    err: Error,\n    logger: StructuredLogger\n  ) => Promise<void>;\n}\n\nexport class ZodSocketConnection<\n  TClientMessages extends ZodSocketMessageCatalogSchema,\n  TServerMessages extends ZodSocketMessageCatalogSchema,\n> {\n  #sender: ZodSocketMessageSender<TClientMessages>;\n  socket: ZodSocket<TServerMessages, TClientMessages>;\n\n  #handler: ZodSocketMessageHandler<TServerMessages>;\n  #logger: StructuredLogger;\n\n  constructor(opts: ZodSocketConnectionOptions<TClientMessages, TServerMessages>) {\n    const uri = `${opts.secure ? \"wss\" : \"ws\"}://${opts.host}:${\n      opts.port ?? (opts.secure ? \"443\" : \"80\")\n    }/${opts.namespace}`;\n\n    const logger = new SimpleStructuredLogger(opts.namespace, LogLevel.info);\n    logger.log(\"new zod socket\", { uri });\n\n    this.socket = io(uri, {\n      transports: [\"websocket\"],\n      auth: {\n        token: opts.authToken,\n      },\n      extraHeaders: opts.extraHeaders,\n      reconnectionDelay: 500,\n      reconnectionDelayMax: 1000,\n    });\n\n    this.#logger = logger.child({\n      socketId: this.socket.id,\n    });\n\n    this.#handler = new ZodSocketMessageHandler({\n      schema: opts.serverMessages,\n      handlers: opts.handlers,\n    });\n    this.#handler.registerHandlers(this.socket, this.#logger);\n\n    this.#sender = new ZodSocketMessageSender({\n      schema: opts.clientMessages,\n      socket: this.socket,\n    });\n\n    this.socket.on(\"connect_error\", async (error) => {\n      this.#logger.error(`connect_error: ${error}`);\n\n      if (opts.onError) {\n        await opts.onError(this.socket, error, this.#logger);\n      }\n    });\n\n    this.socket.on(\"connect\", async () => {\n      this.#logger.info(\"connect\");\n\n      if (opts.onConnection) {\n        await opts.onConnection(this.socket, this.#handler, this.#sender, this.#logger);\n      }\n    });\n\n    this.socket.on(\"disconnect\", async (reason, description) => {\n      this.#logger.info(\"disconnect\", { reason, description });\n\n      if (opts.onDisconnect) {\n        await opts.onDisconnect(this.socket, reason, description, this.#logger);\n      }\n    });\n  }\n\n  close() {\n    this.socket.close();\n  }\n\n  connect() {\n    this.socket.connect();\n  }\n\n  get send() {\n    return this.#sender.send.bind(this.#sender);\n  }\n\n  get sendWithAck() {\n    return this.#sender.sendWithAck.bind(this.#sender);\n  }\n}\n\nfunction createLogger(prefix: string) {\n  return (...args: any[]) => console.log(prefix, ...args);\n}\n","type StructuredArgs = (Record<string, unknown> | undefined)[];\n\nexport interface StructuredLogger {\n  log: (message: string, ...args: StructuredArgs) => any;\n  error: (message: string, ...args: StructuredArgs) => any;\n  warn: (message: string, ...args: StructuredArgs) => any;\n  info: (message: string, ...args: StructuredArgs) => any;\n  debug: (message: string, ...args: StructuredArgs) => any;\n  child: (fields: Record<string, unknown>) => StructuredLogger;\n}\n\nexport enum LogLevel {\n  \"log\",\n  \"error\",\n  \"warn\",\n  \"info\",\n  \"debug\",\n}\n\nexport class SimpleStructuredLogger implements StructuredLogger {\n  constructor(\n    private name: string,\n    private level: LogLevel = [\"1\", \"true\"].includes(process.env.DEBUG ?? \"\")\n      ? LogLevel.debug\n      : LogLevel.info,\n    private fields?: Record<string, unknown>\n  ) {}\n\n  child(fields: Record<string, unknown>, level?: LogLevel) {\n    return new SimpleStructuredLogger(this.name, level, { ...this.fields, ...fields });\n  }\n\n  log(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.log) return;\n\n    this.#structuredLog(console.log, message, \"log\", ...args);\n  }\n\n  error(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.error) return;\n\n    this.#structuredLog(console.error, message, \"error\", ...args);\n  }\n\n  warn(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.warn) return;\n\n    this.#structuredLog(console.warn, message, \"warn\", ...args);\n  }\n\n  info(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.info) return;\n\n    this.#structuredLog(console.info, message, \"info\", ...args);\n  }\n\n  debug(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.debug) return;\n\n    this.#structuredLog(console.debug, message, \"debug\", ...args);\n  }\n\n  #structuredLog(\n    loggerFunction: (message: string, ...args: any[]) => void,\n    message: string,\n    level: string,\n    ...args: StructuredArgs\n  ) {\n    const structuredLog = {\n      ...(args.length === 1 ? args[0] : args),\n      ...this.fields,\n      timestamp: new Date(),\n      name: this.name,\n      message,\n      level,\n    };\n\n    loggerFunction(JSON.stringify(structuredLog));\n  }\n}\n"]}
+{"version":3,"sources":["../../src/v3/zodSocket.ts","../../src/v3/utils/structuredLogger.ts"],"names":["io","ZodError","z","LogLevel","SimpleStructuredLogger","constructor","name","level","includes","process","env","DEBUG","debug","info","fields","child","log","message","args","console","error","warn","loggerFunction","structuredLog","length","timestamp","Date","JSON","stringify","fromZodError","messageSchema","object","version","string","type","payload","unknown","ZodSocketMessageHandler","options","schema","handlers","handleMessage","parsedMessage","parseMessage","Error","handler","String","ack","safeParse","success","data","messageWithVersion","parsedPayload","registerHandlers","emitter","logger","eventName","Object","keys","on","callback","hasCallback","stack","_schema","ZodSocketMessageSender","socket","send","emit","sendWithAck","callbackResult","emitWithAck","ZodSocketConnection","opts","uri","secure","host","port","namespace","transports","auth","token","authToken","extraHeaders","reconnectionDelay","reconnectionDelayMax","socketId","id","serverMessages","clientMessages","onError","onConnection","reason","description","onDisconnect","close","connect","bind"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AACA,SAASA,UAAU;AACnB,SAASC,UAAUC,SAAS;;;ICSrB;UAAKC,WAAQ;AAARA,EAAAA,UAAAA,UACV,KAAA,IAAA,CAAA,IAAA;AADUA,EAAAA,UAAAA,UAEV,OAAA,IAAA,CAAA,IAAA;AAFUA,EAAAA,UAAAA,UAGV,MAAA,IAAA,CAAA,IAAA;AAHUA,EAAAA,UAAAA,UAIV,MAAA,IAAA,CAAA,IAAA;AAJUA,EAAAA,UAAAA,UAKV,OAAA,IAAA,CAAA,IAAA;GALUA,aAAAA,WAAAA,CAAAA,EAAAA;;AAQL,IAAMC,0BAAN,MAAMA,wBAAAA;EACXC,YACUC,MACAC,QAAkB;IAAC;IAAK;IAAQC,SAASC,QAAQC,IAAIC,SAAS,EAAA,IAClER,SAASS,QACTT,SAASU,MACLC,QACR;AAoCF;gBAzCUR;iBACAC;kBAGAO;EACP;EAEHC,MAAMD,QAAiCP,OAAkB;AACvD,WAAO,IAAIH,wBAAuB,KAAKE,MAAMC,OAAO;MAAE,GAAG,KAAKO;MAAQ,GAAGA;IAAO,CAAA;EAClF;EAEAE,IAAIC,YAAoBC,MAAsB;AAC5C,QAAI,KAAKX,QAAQJ,SAASa;AAAK;AAE/B,0BAAK,kCAAL,WAAoBG,QAAQH,KAAKC,SAAS,OAAA,GAAUC;EACtD;EAEAE,MAAMH,YAAoBC,MAAsB;AAC9C,QAAI,KAAKX,QAAQJ,SAASiB;AAAO;AAEjC,0BAAK,kCAAL,WAAoBD,QAAQC,OAAOH,SAAS,SAAA,GAAYC;EAC1D;EAEAG,KAAKJ,YAAoBC,MAAsB;AAC7C,QAAI,KAAKX,QAAQJ,SAASkB;AAAM;AAEhC,0BAAK,kCAAL,WAAoBF,QAAQE,MAAMJ,SAAS,QAAA,GAAWC;EACxD;EAEAL,KAAKI,YAAoBC,MAAsB;AAC7C,QAAI,KAAKX,QAAQJ,SAASU;AAAM;AAEhC,0BAAK,kCAAL,WAAoBM,QAAQN,MAAMI,SAAS,QAAA,GAAWC;EACxD;EAEAN,MAAMK,YAAoBC,MAAsB;AAC9C,QAAI,KAAKX,QAAQJ,SAASS;AAAO;AAEjC,0BAAK,kCAAL,WAAoBO,QAAQP,OAAOK,SAAS,SAAA,GAAYC;EAC1D;AAmBF;AAjBE;mBAAc,gCACZI,gBACAL,SACAV,UACGW,MACH;AACA,QAAMK,gBAAgB;IACpB,GAAIL,KAAKM,WAAW,IAAIN,KAAK,CAAA,IAAKA;IAClC,GAAG,KAAKJ;IACRW,WAAW,oBAAIC,KAAAA;IACfpB,MAAM,KAAKA;IACXW;IACAV;EACF;AAEAe,iBAAeK,KAAKC,UAAUL,aAAAA,CAAAA;AAChC,GAhBc;AA3CHnB;AAAN,IAAMA,yBAAN;;;ADdP,SAASyB,oBAAoB;AA8E7B,IAAMC,gBAAgB5B,EAAE6B,OAAO;EAC7BC,SAAS9B,EAAE+B,OAAM;EACjBC,MAAMhC,EAAE+B,OAAM;EACdE,SAASjC,EAAEkC,QAAO;AACpB,CAAA;AAtFA;AAwFO,IAAMC,2BAAN,MAAMA,yBAAAA;EAIXhC,YAAYiC,SAAsD;AAHlE;AACA;AAGE,uBAAK,SAAUA,QAAQC;AACvB,uBAAK,WAAYD,QAAQE;EAC3B;EAEA,MAAaC,cAAcxB,SAAkB;AAC3C,UAAMyB,gBAAgB,KAAKC,aAAa1B,OAAAA;AAExC,QAAI,CAAC,mBAAK,YAAW;AACnB,YAAM,IAAI2B,MAAM,sBAAA;IAClB;AAEA,UAAMC,UAAU,mBAAK,WAAUH,cAAcR,IAAI;AAEjD,QAAI,CAACW,SAAS;AACZ1B,cAAQC,MAAM,gCAAgC0B,OAAOJ,cAAcR,IAAI,CAAA,EAAG;AAC1E;IACF;AAEA,UAAMa,MAAM,MAAMF,QAAQH,cAAcP,OAAO;AAE/C,WAAOY;EACT;EAEOJ,aAAa1B,SAA0D;AAC5E,UAAMyB,gBAAgBZ,cAAckB,UAAU/B,OAAAA;AAE9C,QAAI,CAACyB,cAAcO,SAAS;AAC1B,YAAM,IAAIL,MAAM,4BAA4BjB,KAAKC,UAAUc,cAActB,KAAK,CAAA,EAAG;IACnF;AAEA,UAAMmB,SAAS,mBAAK,SAAQG,cAAcQ,KAAKhB,IAAI,EAAE,SAAA;AAErD,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBF,cAAcQ,KAAKhB,IAAI,EAAE;IACpE;AAEA,UAAMiB,qBAAqB;MACzBnB,SAASU,cAAcQ,KAAKlB;MAC5B,GAAI,OAAOU,cAAcQ,KAAKf,YAAY,WAAWO,cAAcQ,KAAKf,UAAU,CAAC;IACrF;AAEA,UAAMiB,gBAAgBb,OAAOS,UAAUG,kBAAAA;AAEvC,QAAI,CAACC,cAAcH,SAAS;AAC1B9B,cAAQC,MAAM,mCAAmC;QAC/CH;QACAkB,SAASgB;MACX,CAAA;AAEA,YAAMC,cAAchC,iBAAiBnB,WACjC4B,aAAauB,cAAchC,KAAK,IAChCgC,cAAchC;IACpB;AAEA,WAAO;MACLc,MAAMQ,cAAcQ,KAAKhB;MACzBC,SAASiB,cAAcF;IACzB;EACF;EAEOG,iBAAiBC,SAA2BC,QAA2B;AAC5E,UAAMvC,MAAMuC,UAAUpC;AAEtB,QAAI,CAAC,mBAAK,YAAW;AACnBH,UAAIH,KAAK,sBAAA;AACT;IACF;AAEA,eAAW2C,aAAaC,OAAOC,KAAK,mBAAK,UAAS,GAAG;AACnDJ,cAAQK,GAAGH,WAAW,OAAOvC,SAAc2C,aAAkC;AAC3E5C,YAAIH,KAAK,YAAY2C,SAAAA,IAAa;UAChCrB,SAASlB;UACT4C,aAAa,CAAC,CAACD;QACjB,CAAA;AAEA,YAAIb;AAEJ,YAAI;AAEF,cAAI,aAAa9B,SAAS;AACxB8B,kBAAM,MAAM,KAAKN,cAAc;cAAEP,MAAMsB;cAAW,GAAGvC;YAAQ,CAAA;UAC/D,OAAO;AAEL,kBAAM,EAAEe,SAAS,GAAGG,QAAAA,IAAYlB;AAChC8B,kBAAM,MAAM,KAAKN,cAAc;cAAEP,MAAMsB;cAAWxB;cAASG;YAAQ,CAAA;UACrE;QACF,SAASf,OAAO;AACdJ,cAAII,MAAM,gCAAgC;YACxCA,OACEA,iBAAiBwB,QACb;cACE3B,SAASG,MAAMH;cACf6C,OAAO1C,MAAM0C;YACf,IACA1C;UACR,CAAA;AACA;QACF;AAEA,YAAIwC,YAAY,OAAOA,aAAa,YAAY;AAC9CA,mBAASb,GAAAA;QACX;MACF,CAAA;IACF;EACF;AACF;AA7GE;AACA;AAFWV;AAAN,IAAMA,0BAAN;AAxFP,IAAA0B,UAAA;AA2NO,IAAMC,0BAAN,MAAMA,wBAAAA;EAIX3D,YAAYiC,SAAyD;AAHrE,uBAAAyB,UAAA;AACA;AAGE,uBAAKA,UAAUzB,QAAQC;AACvB,uBAAK,SAAUD,QAAQ2B;EACzB;EAEOC,KACLhC,MACAC,SACM;AACN,UAAMI,SAAS,mBAAKwB,UAAQ7B,IAAAA,EAAM,SAAA;AAElC,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBV,IAAAA,EAAgB;IAC3D;AAEA,UAAMkB,gBAAgBb,OAAOS,UAAUb,OAAAA;AAEvC,QAAI,CAACiB,cAAcH,SAAS;AAC1B,YAAM,IAAIL,MAAM,oCAAoCjB,KAAKC,UAAUwB,cAAchC,KAAK,CAAA,EAAG;IAC3F;AAGA,uBAAK,SAAQ+C,KAAKjC,MAAM;MAAEC;MAASH,SAAS;IAAK,CAAA;AAEjD;EACF;EAEA,MAAaoC,YACXlC,MACAC,SAC+D;AAC/D,UAAMI,SAAS,mBAAKwB,UAAQ7B,IAAAA,EAAM,SAAA;AAElC,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBV,IAAAA,EAAgB;IAC3D;AAEA,UAAMkB,gBAAgBb,OAAOS,UAAUb,OAAAA;AAEvC,QAAI,CAACiB,cAAcH,SAAS;AAC1B,YAAM,IAAIL,MAAM,oCAAoCjB,KAAKC,UAAUwB,cAAchC,KAAK,CAAA,EAAG;IAC3F;AAGA,UAAMiD,iBAAiB,MAAM,mBAAK,SAAQC,YAAYpC,MAAM;MAAEC;MAASH,SAAS;IAAK,CAAA;AAErF,WAAOqC;EACT;AACF;AAnDEN,WAAA;AACA;AAFWC;AAAN,IAAMA,yBAAN;AA3NP;AA2TO,IAAMO,uBAAN,MAAMA,qBAAAA;EAUXlE,YAAYmE,MAAoE;AANhF;AAGA;AACA;AAGE,UAAMC,MAAM,GAAGD,KAAKE,SAAS,QAAQ,IAAI,MAAMF,KAAKG,IAAI,IACtDH,KAAKI,SAASJ,KAAKE,SAAS,QAAQ,KAAG,IACrCF,KAAKK,SAAS;AAElB,UAAMtB,SAAS,IAAInD,uBAAuBoE,KAAKK,WAAW1E,SAASU,IAAI;AACvE0C,WAAOvC,IAAI,kBAAkB;MAAEyD;IAAI,CAAA;AAEnC,SAAKR,SAASjE,GAAGyE,KAAK;MACpBK,YAAY;QAAC;;MACbC,MAAM;QACJC,OAAOR,KAAKS;MACd;MACAC,cAAcV,KAAKU;MACnBC,mBAAmB;MACnBC,sBAAsB;IACxB,CAAA;AAEA,uBAAK,SAAU7B,OAAOxC,MAAM;MAC1BsE,UAAU,KAAKpB,OAAOqB;IACxB,CAAA;AAEA,uBAAK,UAAW,IAAIjD,wBAAwB;MAC1CE,QAAQiC,KAAKe;MACb/C,UAAUgC,KAAKhC;IACjB,CAAA;AACA,uBAAK,UAASa,iBAAiB,KAAKY,QAAQ,mBAAK,QAAO;AAExD,uBAAK,SAAU,IAAID,uBAAuB;MACxCzB,QAAQiC,KAAKgB;MACbvB,QAAQ,KAAKA;IACf,CAAA;AAEA,SAAKA,OAAON,GAAG,iBAAiB,OAAOvC,UAAU;AAC/C,yBAAK,SAAQA,MAAM,kBAAkBA,KAAAA,EAAO;AAE5C,UAAIoD,KAAKiB,SAAS;AAChB,cAAMjB,KAAKiB,QAAQ,KAAKxB,QAAQ7C,OAAO,mBAAK,QAAO;MACrD;IACF,CAAA;AAEA,SAAK6C,OAAON,GAAG,WAAW,YAAY;AACpC,yBAAK,SAAQ9C,KAAK,SAAA;AAElB,UAAI2D,KAAKkB,cAAc;AACrB,cAAMlB,KAAKkB,aAAa,KAAKzB,QAAQ,mBAAK,WAAU,mBAAK,UAAS,mBAAK,QAAO;MAChF;IACF,CAAA;AAEA,SAAKA,OAAON,GAAG,cAAc,OAAOgC,QAAQC,gBAAgB;AAC1D,yBAAK,SAAQ/E,KAAK,cAAc;QAAE8E;QAAQC;MAAY,CAAA;AAEtD,UAAIpB,KAAKqB,cAAc;AACrB,cAAMrB,KAAKqB,aAAa,KAAK5B,QAAQ0B,QAAQC,aAAa,mBAAK,QAAO;MACxE;IACF,CAAA;EACF;EAEAE,QAAQ;AACN,SAAK7B,OAAO6B,MAAK;EACnB;EAEAC,UAAU;AACR,SAAK9B,OAAO8B,QAAO;EACrB;EAEA,IAAI7B,OAAO;AACT,WAAO,mBAAK,SAAQA,KAAK8B,KAAK,mBAAK,QAAO;EAC5C;EAEA,IAAI5B,cAAc;AAChB,WAAO,mBAAK,SAAQA,YAAY4B,KAAK,mBAAK,QAAO;EACnD;AACF;AA/EE;AAGA;AACA;AARWzB;AAAN,IAAMA,sBAAN","sourcesContent":["import type { Socket } from \"socket.io-client\";\nimport { io } from \"socket.io-client\";\nimport { ZodError, z } from \"zod\";\nimport { EventEmitterLike, ZodMessageValueSchema } from \"./zodMessageHandler\";\nimport { LogLevel, SimpleStructuredLogger, StructuredLogger } from \"./utils/structuredLogger\";\nimport { fromZodError } from \"zod-validation-error\";\n\nexport interface ZodSocketMessageCatalogSchema {\n  [key: string]:\n    | {\n        message: ZodMessageValueSchema<any>;\n      }\n    | {\n        message: ZodMessageValueSchema<any>;\n        callback?: ZodMessageValueSchema<any>;\n      };\n}\n\nexport type ZodMessageCatalogToSocketIoEvents<TCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TCatalog]: SocketMessageHasCallback<TCatalog, K> extends true\n    ? (\n        message: z.infer<GetSocketMessageSchema<TCatalog, K>>,\n        callback: (ack: z.infer<GetSocketCallbackSchema<TCatalog, K>>) => void\n      ) => void\n    : (message: z.infer<GetSocketMessageSchema<TCatalog, K>>) => void;\n};\n\nexport type GetSocketMessageSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = TRPCCatalog[TMessageType][\"message\"];\n\nexport type InferSocketMessageSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = z.infer<GetSocketMessageSchema<TRPCCatalog, TMessageType>>;\n\nexport type GetSocketCallbackSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = TRPCCatalog[TMessageType] extends { callback: any }\n  ? TRPCCatalog[TMessageType][\"callback\"]\n  : never;\n\nexport type InferSocketCallbackSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = z.infer<GetSocketCallbackSchema<TRPCCatalog, TMessageType>>;\n\nexport type SocketMessageHasCallback<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = GetSocketCallbackSchema<TRPCCatalog, TMessageType> extends never ? false : true;\n\nexport type ZodSocketMessageHandlers<TCatalogSchema extends ZodSocketMessageCatalogSchema> =\n  Partial<{\n    [K in keyof TCatalogSchema]: (\n      payload: z.infer<GetSocketMessageSchema<TCatalogSchema, K>>\n    ) => Promise<\n      SocketMessageHasCallback<TCatalogSchema, K> extends true\n        ? z.input<GetSocketCallbackSchema<TCatalogSchema, K>>\n        : void\n    >;\n  }>;\n\nexport type ZodSocketMessageHandlerOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> =\n  {\n    schema: TMessageCatalog;\n    handlers?: ZodSocketMessageHandlers<TMessageCatalog>;\n  };\n\ntype MessageFromSocketSchema<\n  K extends keyof TMessageCatalog,\n  TMessageCatalog extends ZodSocketMessageCatalogSchema,\n> = {\n  type: K;\n  payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>;\n};\n\nexport type MessagesFromSocketCatalog<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TMessageCatalog]: MessageFromSocketSchema<K, TMessageCatalog>;\n}[keyof TMessageCatalog];\n\nconst messageSchema = z.object({\n  version: z.string(),\n  type: z.string(),\n  payload: z.unknown(),\n});\n\nexport class ZodSocketMessageHandler<TRPCCatalog extends ZodSocketMessageCatalogSchema> {\n  #schema: TRPCCatalog;\n  #handlers: ZodSocketMessageHandlers<TRPCCatalog> | undefined;\n\n  constructor(options: ZodSocketMessageHandlerOptions<TRPCCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.handlers;\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): MessagesFromSocketCatalog<TRPCCatalog> {\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 messageWithVersion = {\n      version: parsedMessage.data.version,\n      ...(typeof parsedMessage.data.payload === \"object\" ? parsedMessage.data.payload : {}),\n    };\n\n    const parsedPayload = schema.safeParse(messageWithVersion);\n\n    if (!parsedPayload.success) {\n      console.error(\"Failed to parse message payload\", {\n        message,\n        payload: messageWithVersion,\n      });\n\n      throw parsedPayload.error instanceof ZodError\n        ? fromZodError(parsedPayload.error)\n        : 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.#handlers)) {\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        try {\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        } catch (error) {\n          log.error(\"Error while handling message\", {\n            error:\n              error instanceof Error\n                ? {\n                    message: error.message,\n                    stack: error.stack,\n                  }\n                : error,\n          });\n          return;\n        }\n\n        if (callback && typeof callback === \"function\") {\n          callback(ack);\n        }\n      });\n    }\n  }\n}\n\nexport type ZodSocketMessageSenderOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  schema: TMessageCatalog;\n  socket: ZodSocket<any, TMessageCatalog>;\n};\n\nexport type GetSocketMessagesWithCallback<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true\n    ? K\n    : never;\n}[keyof TMessageCatalog];\n\nexport type GetSocketMessagesWithoutCallback<\n  TMessageCatalog extends ZodSocketMessageCatalogSchema,\n> = {\n  [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true\n    ? never\n    : K;\n}[keyof TMessageCatalog];\n\nexport class ZodSocketMessageSender<TMessageCatalog extends ZodSocketMessageCatalogSchema> {\n  #schema: TMessageCatalog;\n  #socket: ZodSocket<any, TMessageCatalog>;\n\n  constructor(options: ZodSocketMessageSenderOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#socket = options.socket;\n  }\n\n  public send<K extends GetSocketMessagesWithoutCallback<TMessageCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>\n  ): void {\n    const schema = this.#schema[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 Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    // @ts-expect-error\n    this.#socket.emit(type, { payload, version: \"v1\" });\n\n    return;\n  }\n\n  public async sendWithAck<K extends GetSocketMessagesWithCallback<TMessageCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>\n  ): Promise<z.infer<GetSocketCallbackSchema<TMessageCatalog, K>>> {\n    const schema = this.#schema[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 Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    // @ts-expect-error\n    const callbackResult = await this.#socket.emitWithAck(type, { payload, version: \"v1\" });\n\n    return callbackResult;\n  }\n}\n\nexport type ZodSocket<\n  TListenEvents extends ZodSocketMessageCatalogSchema,\n  TEmitEvents extends ZodSocketMessageCatalogSchema,\n> = Socket<\n  ZodMessageCatalogToSocketIoEvents<TListenEvents>,\n  ZodMessageCatalogToSocketIoEvents<TEmitEvents>\n>;\n\ninterface ZodSocketConnectionOptions<\n  TClientMessages extends ZodSocketMessageCatalogSchema,\n  TServerMessages extends ZodSocketMessageCatalogSchema,\n> {\n  host: string;\n  port?: number;\n  secure?: boolean;\n  namespace: string;\n  clientMessages: TClientMessages;\n  serverMessages: TServerMessages;\n  extraHeaders?: {\n    [header: string]: string;\n  };\n  handlers?: ZodSocketMessageHandlers<TServerMessages>;\n  authToken?: string;\n  onConnection?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    handler: ZodSocketMessageHandler<TServerMessages>,\n    sender: ZodSocketMessageSender<TClientMessages>,\n    logger: StructuredLogger\n  ) => Promise<void>;\n  onDisconnect?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    reason: Socket.DisconnectReason,\n    description: any,\n    logger: StructuredLogger\n  ) => Promise<void>;\n  onError?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    err: Error,\n    logger: StructuredLogger\n  ) => Promise<void>;\n}\n\nexport class ZodSocketConnection<\n  TClientMessages extends ZodSocketMessageCatalogSchema,\n  TServerMessages extends ZodSocketMessageCatalogSchema,\n> {\n  #sender: ZodSocketMessageSender<TClientMessages>;\n  socket: ZodSocket<TServerMessages, TClientMessages>;\n\n  #handler: ZodSocketMessageHandler<TServerMessages>;\n  #logger: StructuredLogger;\n\n  constructor(opts: ZodSocketConnectionOptions<TClientMessages, TServerMessages>) {\n    const uri = `${opts.secure ? \"wss\" : \"ws\"}://${opts.host}:${\n      opts.port ?? (opts.secure ? \"443\" : \"80\")\n    }/${opts.namespace}`;\n\n    const logger = new SimpleStructuredLogger(opts.namespace, LogLevel.info);\n    logger.log(\"new zod socket\", { uri });\n\n    this.socket = io(uri, {\n      transports: [\"websocket\"],\n      auth: {\n        token: opts.authToken,\n      },\n      extraHeaders: opts.extraHeaders,\n      reconnectionDelay: 500,\n      reconnectionDelayMax: 1000,\n    });\n\n    this.#logger = logger.child({\n      socketId: this.socket.id,\n    });\n\n    this.#handler = new ZodSocketMessageHandler({\n      schema: opts.serverMessages,\n      handlers: opts.handlers,\n    });\n    this.#handler.registerHandlers(this.socket, this.#logger);\n\n    this.#sender = new ZodSocketMessageSender({\n      schema: opts.clientMessages,\n      socket: this.socket,\n    });\n\n    this.socket.on(\"connect_error\", async (error) => {\n      this.#logger.error(`connect_error: ${error}`);\n\n      if (opts.onError) {\n        await opts.onError(this.socket, error, this.#logger);\n      }\n    });\n\n    this.socket.on(\"connect\", async () => {\n      this.#logger.info(\"connect\");\n\n      if (opts.onConnection) {\n        await opts.onConnection(this.socket, this.#handler, this.#sender, this.#logger);\n      }\n    });\n\n    this.socket.on(\"disconnect\", async (reason, description) => {\n      this.#logger.info(\"disconnect\", { reason, description });\n\n      if (opts.onDisconnect) {\n        await opts.onDisconnect(this.socket, reason, description, this.#logger);\n      }\n    });\n  }\n\n  close() {\n    this.socket.close();\n  }\n\n  connect() {\n    this.socket.connect();\n  }\n\n  get send() {\n    return this.#sender.send.bind(this.#sender);\n  }\n\n  get sendWithAck() {\n    return this.#sender.sendWithAck.bind(this.#sender);\n  }\n}\n\nfunction createLogger(prefix: string) {\n  return (...args: any[]) => console.log(prefix, ...args);\n}\n","type StructuredArgs = (Record<string, unknown> | undefined)[];\n\nexport interface StructuredLogger {\n  log: (message: string, ...args: StructuredArgs) => any;\n  error: (message: string, ...args: StructuredArgs) => any;\n  warn: (message: string, ...args: StructuredArgs) => any;\n  info: (message: string, ...args: StructuredArgs) => any;\n  debug: (message: string, ...args: StructuredArgs) => any;\n  child: (fields: Record<string, unknown>) => StructuredLogger;\n}\n\nexport enum LogLevel {\n  \"log\",\n  \"error\",\n  \"warn\",\n  \"info\",\n  \"debug\",\n}\n\nexport class SimpleStructuredLogger implements StructuredLogger {\n  constructor(\n    private name: string,\n    private level: LogLevel = [\"1\", \"true\"].includes(process.env.DEBUG ?? \"\")\n      ? LogLevel.debug\n      : LogLevel.info,\n    private fields?: Record<string, unknown>\n  ) {}\n\n  child(fields: Record<string, unknown>, level?: LogLevel) {\n    return new SimpleStructuredLogger(this.name, level, { ...this.fields, ...fields });\n  }\n\n  log(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.log) return;\n\n    this.#structuredLog(console.log, message, \"log\", ...args);\n  }\n\n  error(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.error) return;\n\n    this.#structuredLog(console.error, message, \"error\", ...args);\n  }\n\n  warn(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.warn) return;\n\n    this.#structuredLog(console.warn, message, \"warn\", ...args);\n  }\n\n  info(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.info) return;\n\n    this.#structuredLog(console.info, message, \"info\", ...args);\n  }\n\n  debug(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.debug) return;\n\n    this.#structuredLog(console.debug, message, \"debug\", ...args);\n  }\n\n  #structuredLog(\n    loggerFunction: (message: string, ...args: any[]) => void,\n    message: string,\n    level: string,\n    ...args: StructuredArgs\n  ) {\n    const structuredLog = {\n      ...(args.length === 1 ? args[0] : args),\n      ...this.fields,\n      timestamp: new Date(),\n      name: this.name,\n      message,\n      level,\n    };\n\n    loggerFunction(JSON.stringify(structuredLog));\n  }\n}\n"]}

package/dist/v3/zodSocket.mjs

@@ -1,5 +1,6 @@
 import { io } from 'socket.io-client';
-import { z } from 'zod';
+import { z, ZodError } from 'zod';
+import { fromZodError } from 'zod-validation-error';
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -92,10 +93,8 @@ structuredLog_fn = /* @__PURE__ */ __name(function(loggerFunction, message, leve
 }, "#structuredLog");
 __name(_SimpleStructuredLogger, "SimpleStructuredLogger");
 var SimpleStructuredLogger = _SimpleStructuredLogger;
-
-// src/v3/zodSocket.ts
 var messageSchema = z.object({
-  version: z.literal("v1").default("v1"),
+  version: z.string(),
   type: z.string(),
   payload: z.unknown()
 });
@@ -129,9 +128,17 @@ var _ZodSocketMessageHandler = class _ZodSocketMessageHandler {
     if (!schema) {
       throw new Error(`Unknown message type: ${parsedMessage.data.type}`);
     }
-    const parsedPayload = schema.safeParse(parsedMessage.data.payload);
+    const messageWithVersion = {
+      version: parsedMessage.data.version,
+      ...typeof parsedMessage.data.payload === "object" ? parsedMessage.data.payload : {}
+    };
+    const parsedPayload = schema.safeParse(messageWithVersion);
     if (!parsedPayload.success) {
-      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
+      console.error("Failed to parse message payload", {
+        message,
+        payload: messageWithVersion
+      });
+      throw parsedPayload.error instanceof ZodError ? fromZodError(parsedPayload.error) : parsedPayload.error;
     }
     return {
       type: parsedMessage.data.type,
@@ -167,7 +174,10 @@ var _ZodSocketMessageHandler = class _ZodSocketMessageHandler {
           }
         } catch (error) {
           log.error("Error while handling message", {
-            error
+            error: error instanceof Error ? {
+              message: error.message,
+              stack: error.stack
+            } : error
           });
           return;
         }

package/dist/v3/zodSocket.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/v3/zodSocket.ts","../../src/v3/utils/structuredLogger.ts"],"names":["io","z","LogLevel","SimpleStructuredLogger","constructor","name","level","includes","process","env","DEBUG","debug","info","fields","child","log","message","args","console","error","warn","loggerFunction","structuredLog","length","timestamp","Date","JSON","stringify","messageSchema","object","version","literal","default","type","string","payload","unknown","ZodSocketMessageHandler","options","schema","handlers","handleMessage","parsedMessage","parseMessage","Error","handler","String","ack","safeParse","success","data","parsedPayload","registerHandlers","emitter","logger","eventName","Object","keys","on","callback","hasCallback","_schema","ZodSocketMessageSender","socket","send","emit","sendWithAck","callbackResult","emitWithAck","ZodSocketConnection","opts","uri","secure","host","port","namespace","transports","auth","token","authToken","extraHeaders","reconnectionDelay","reconnectionDelayMax","socketId","id","serverMessages","clientMessages","onError","onConnection","reason","description","onDisconnect","close","connect","bind"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AACA,SAASA,UAAU;AACnB,SAASC,SAAS;;;ICSX;UAAKC,WAAQ;AAARA,EAAAA,UAAAA,UACV,KAAA,IAAA,CAAA,IAAA;AADUA,EAAAA,UAAAA,UAEV,OAAA,IAAA,CAAA,IAAA;AAFUA,EAAAA,UAAAA,UAGV,MAAA,IAAA,CAAA,IAAA;AAHUA,EAAAA,UAAAA,UAIV,MAAA,IAAA,CAAA,IAAA;AAJUA,EAAAA,UAAAA,UAKV,OAAA,IAAA,CAAA,IAAA;GALUA,aAAAA,WAAAA,CAAAA,EAAAA;;AAQL,IAAMC,0BAAN,MAAMA,wBAAAA;EACXC,YACUC,MACAC,QAAkB;IAAC;IAAK;IAAQC,SAASC,QAAQC,IAAIC,SAAS,EAAA,IAClER,SAASS,QACTT,SAASU,MACLC,QACR;AAoCF;gBAzCUR;iBACAC;kBAGAO;EACP;EAEHC,MAAMD,QAAiCP,OAAkB;AACvD,WAAO,IAAIH,wBAAuB,KAAKE,MAAMC,OAAO;MAAE,GAAG,KAAKO;MAAQ,GAAGA;IAAO,CAAA;EAClF;EAEAE,IAAIC,YAAoBC,MAAsB;AAC5C,QAAI,KAAKX,QAAQJ,SAASa;AAAK;AAE/B,0BAAK,kCAAL,WAAoBG,QAAQH,KAAKC,SAAS,OAAA,GAAUC;EACtD;EAEAE,MAAMH,YAAoBC,MAAsB;AAC9C,QAAI,KAAKX,QAAQJ,SAASiB;AAAO;AAEjC,0BAAK,kCAAL,WAAoBD,QAAQC,OAAOH,SAAS,SAAA,GAAYC;EAC1D;EAEAG,KAAKJ,YAAoBC,MAAsB;AAC7C,QAAI,KAAKX,QAAQJ,SAASkB;AAAM;AAEhC,0BAAK,kCAAL,WAAoBF,QAAQE,MAAMJ,SAAS,QAAA,GAAWC;EACxD;EAEAL,KAAKI,YAAoBC,MAAsB;AAC7C,QAAI,KAAKX,QAAQJ,SAASU;AAAM;AAEhC,0BAAK,kCAAL,WAAoBM,QAAQN,MAAMI,SAAS,QAAA,GAAWC;EACxD;EAEAN,MAAMK,YAAoBC,MAAsB;AAC9C,QAAI,KAAKX,QAAQJ,SAASS;AAAO;AAEjC,0BAAK,kCAAL,WAAoBO,QAAQP,OAAOK,SAAS,SAAA,GAAYC;EAC1D;AAmBF;AAjBE;mBAAc,gCACZI,gBACAL,SACAV,UACGW,MACH;AACA,QAAMK,gBAAgB;IACpB,GAAIL,KAAKM,WAAW,IAAIN,KAAK,CAAA,IAAKA;IAClC,GAAG,KAAKJ;IACRW,WAAW,oBAAIC,KAAAA;IACfpB,MAAM,KAAKA;IACXW;IACAV;EACF;AAEAe,iBAAeK,KAAKC,UAAUL,aAAAA,CAAAA;AAChC,GAhBc;AA3CHnB;AAAN,IAAMA,yBAAN;;;AD+DP,IAAMyB,gBAAgB3B,EAAE4B,OAAO;EAC7BC,SAAS7B,EAAE8B,QAAQ,IAAA,EAAMC,QAAQ,IAAA;EACjCC,MAAMhC,EAAEiC,OAAM;EACdC,SAASlC,EAAEmC,QAAO;AACpB,CAAA;AArFA;AAuFO,IAAMC,2BAAN,MAAMA,yBAAAA;EAIXjC,YAAYkC,SAAsD;AAHlE;AACA;AAGE,uBAAK,SAAUA,QAAQC;AACvB,uBAAK,WAAYD,QAAQE;EAC3B;EAEA,MAAaC,cAAczB,SAAkB;AAC3C,UAAM0B,gBAAgB,KAAKC,aAAa3B,OAAAA;AAExC,QAAI,CAAC,mBAAK,YAAW;AACnB,YAAM,IAAI4B,MAAM,sBAAA;IAClB;AAEA,UAAMC,UAAU,mBAAK,WAAUH,cAAcT,IAAI;AAEjD,QAAI,CAACY,SAAS;AACZ3B,cAAQC,MAAM,gCAAgC2B,OAAOJ,cAAcT,IAAI,CAAA,EAAG;AAC1E;IACF;AAEA,UAAMc,MAAM,MAAMF,QAAQH,cAAcP,OAAO;AAE/C,WAAOY;EACT;EAEOJ,aAAa3B,SAA0D;AAC5E,UAAM0B,gBAAgBd,cAAcoB,UAAUhC,OAAAA;AAE9C,QAAI,CAAC0B,cAAcO,SAAS;AAC1B,YAAM,IAAIL,MAAM,4BAA4BlB,KAAKC,UAAUe,cAAcvB,KAAK,CAAA,EAAG;IACnF;AAEA,UAAMoB,SAAS,mBAAK,SAAQG,cAAcQ,KAAKjB,IAAI,EAAE,SAAA;AAErD,QAAI,CAACM,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBF,cAAcQ,KAAKjB,IAAI,EAAE;IACpE;AAEA,UAAMkB,gBAAgBZ,OAAOS,UAAUN,cAAcQ,KAAKf,OAAO;AAEjE,QAAI,CAACgB,cAAcF,SAAS;AAC1B,YAAM,IAAIL,MAAM,oCAAoClB,KAAKC,UAAUwB,cAAchC,KAAK,CAAA,EAAG;IAC3F;AAEA,WAAO;MACLc,MAAMS,cAAcQ,KAAKjB;MACzBE,SAASgB,cAAcD;IACzB;EACF;EAEOE,iBAAiBC,SAA2BC,QAA2B;AAC5E,UAAMvC,MAAMuC,UAAUpC;AAEtB,QAAI,CAAC,mBAAK,YAAW;AACnBH,UAAIH,KAAK,sBAAA;AACT;IACF;AAEA,eAAW2C,aAAaC,OAAOC,KAAK,mBAAK,UAAS,GAAG;AACnDJ,cAAQK,GAAGH,WAAW,OAAOvC,SAAc2C,aAAkC;AAC3E5C,YAAIH,KAAK,YAAY2C,SAAAA,IAAa;UAChCpB,SAASnB;UACT4C,aAAa,CAAC,CAACD;QACjB,CAAA;AAEA,YAAIZ;AAEJ,YAAI;AAEF,cAAI,aAAa/B,SAAS;AACxB+B,kBAAM,MAAM,KAAKN,cAAc;cAAER,MAAMsB;cAAW,GAAGvC;YAAQ,CAAA;UAC/D,OAAO;AAEL,kBAAM,EAAEc,SAAS,GAAGK,QAAAA,IAAYnB;AAChC+B,kBAAM,MAAM,KAAKN,cAAc;cAAER,MAAMsB;cAAWzB;cAASK;YAAQ,CAAA;UACrE;QACF,SAAShB,OAAO;AACdJ,cAAII,MAAM,gCAAgC;YAAEA;UAAM,CAAA;AAClD;QACF;AAEA,YAAIwC,YAAY,OAAOA,aAAa,YAAY;AAC9CA,mBAASZ,GAAAA;QACX;MACF,CAAA;IACF;EACF;AACF;AAzFE;AACA;AAFWV;AAAN,IAAMA,0BAAN;AAvFP,IAAAwB,UAAA;AAsMO,IAAMC,0BAAN,MAAMA,wBAAAA;EAIX1D,YAAYkC,SAAyD;AAHrE,uBAAAuB,UAAA;AACA;AAGE,uBAAKA,UAAUvB,QAAQC;AACvB,uBAAK,SAAUD,QAAQyB;EACzB;EAEOC,KACL/B,MACAE,SACM;AACN,UAAMI,SAAS,mBAAKsB,UAAQ5B,IAAAA,EAAM,SAAA;AAElC,QAAI,CAACM,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBX,IAAAA,EAAgB;IAC3D;AAEA,UAAMkB,gBAAgBZ,OAAOS,UAAUb,OAAAA;AAEvC,QAAI,CAACgB,cAAcF,SAAS;AAC1B,YAAM,IAAIL,MAAM,oCAAoClB,KAAKC,UAAUwB,cAAchC,KAAK,CAAA,EAAG;IAC3F;AAGA,uBAAK,SAAQ8C,KAAKhC,MAAM;MAAEE;MAASL,SAAS;IAAK,CAAA;AAEjD;EACF;EAEA,MAAaoC,YACXjC,MACAE,SAC+D;AAC/D,UAAMI,SAAS,mBAAKsB,UAAQ5B,IAAAA,EAAM,SAAA;AAElC,QAAI,CAACM,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBX,IAAAA,EAAgB;IAC3D;AAEA,UAAMkB,gBAAgBZ,OAAOS,UAAUb,OAAAA;AAEvC,QAAI,CAACgB,cAAcF,SAAS;AAC1B,YAAM,IAAIL,MAAM,oCAAoClB,KAAKC,UAAUwB,cAAchC,KAAK,CAAA,EAAG;IAC3F;AAGA,UAAMgD,iBAAiB,MAAM,mBAAK,SAAQC,YAAYnC,MAAM;MAAEE;MAASL,SAAS;IAAK,CAAA;AAErF,WAAOqC;EACT;AACF;AAnDEN,WAAA;AACA;AAFWC;AAAN,IAAMA,yBAAN;AAtMP;AAsSO,IAAMO,uBAAN,MAAMA,qBAAAA;EAUXjE,YAAYkE,MAAoE;AANhF;AAGA;AACA;AAGE,UAAMC,MAAM,GAAGD,KAAKE,SAAS,QAAQ,IAAI,MAAMF,KAAKG,IAAI,IACtDH,KAAKI,SAASJ,KAAKE,SAAS,QAAQ,KAAG,IACrCF,KAAKK,SAAS;AAElB,UAAMrB,SAAS,IAAInD,uBAAuBmE,KAAKK,WAAWzE,SAASU,IAAI;AACvE0C,WAAOvC,IAAI,kBAAkB;MAAEwD;IAAI,CAAA;AAEnC,SAAKR,SAAS/D,GAAGuE,KAAK;MACpBK,YAAY;QAAC;;MACbC,MAAM;QACJC,OAAOR,KAAKS;MACd;MACAC,cAAcV,KAAKU;MACnBC,mBAAmB;MACnBC,sBAAsB;IACxB,CAAA;AAEA,uBAAK,SAAU5B,OAAOxC,MAAM;MAC1BqE,UAAU,KAAKpB,OAAOqB;IACxB,CAAA;AAEA,uBAAK,UAAW,IAAI/C,wBAAwB;MAC1CE,QAAQ+B,KAAKe;MACb7C,UAAU8B,KAAK9B;IACjB,CAAA;AACA,uBAAK,UAASY,iBAAiB,KAAKW,QAAQ,mBAAK,QAAO;AAExD,uBAAK,SAAU,IAAID,uBAAuB;MACxCvB,QAAQ+B,KAAKgB;MACbvB,QAAQ,KAAKA;IACf,CAAA;AAEA,SAAKA,OAAOL,GAAG,iBAAiB,OAAOvC,UAAU;AAC/C,yBAAK,SAAQA,MAAM,kBAAkBA,KAAAA,EAAO;AAE5C,UAAImD,KAAKiB,SAAS;AAChB,cAAMjB,KAAKiB,QAAQ,KAAKxB,QAAQ5C,OAAO,mBAAK,QAAO;MACrD;IACF,CAAA;AAEA,SAAK4C,OAAOL,GAAG,WAAW,YAAY;AACpC,yBAAK,SAAQ9C,KAAK,SAAA;AAElB,UAAI0D,KAAKkB,cAAc;AACrB,cAAMlB,KAAKkB,aAAa,KAAKzB,QAAQ,mBAAK,WAAU,mBAAK,UAAS,mBAAK,QAAO;MAChF;IACF,CAAA;AAEA,SAAKA,OAAOL,GAAG,cAAc,OAAO+B,QAAQC,gBAAgB;AAC1D,yBAAK,SAAQ9E,KAAK,cAAc;QAAE6E;QAAQC;MAAY,CAAA;AAEtD,UAAIpB,KAAKqB,cAAc;AACrB,cAAMrB,KAAKqB,aAAa,KAAK5B,QAAQ0B,QAAQC,aAAa,mBAAK,QAAO;MACxE;IACF,CAAA;EACF;EAEAE,QAAQ;AACN,SAAK7B,OAAO6B,MAAK;EACnB;EAEAC,UAAU;AACR,SAAK9B,OAAO8B,QAAO;EACrB;EAEA,IAAI7B,OAAO;AACT,WAAO,mBAAK,SAAQA,KAAK8B,KAAK,mBAAK,QAAO;EAC5C;EAEA,IAAI5B,cAAc;AAChB,WAAO,mBAAK,SAAQA,YAAY4B,KAAK,mBAAK,QAAO;EACnD;AACF;AA/EE;AAGA;AACA;AARWzB;AAAN,IAAMA,sBAAN","sourcesContent":["import type { Socket } from \"socket.io-client\";\nimport { io } from \"socket.io-client\";\nimport { z } from \"zod\";\nimport { EventEmitterLike, ZodMessageValueSchema } from \"./zodMessageHandler\";\nimport { LogLevel, SimpleStructuredLogger, StructuredLogger } from \"./utils/structuredLogger\";\n\nexport interface ZodSocketMessageCatalogSchema {\n  [key: string]:\n    | {\n        message: ZodMessageValueSchema<any>;\n      }\n    | {\n        message: ZodMessageValueSchema<any>;\n        callback?: ZodMessageValueSchema<any>;\n      };\n}\n\nexport type ZodMessageCatalogToSocketIoEvents<TCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TCatalog]: SocketMessageHasCallback<TCatalog, K> extends true\n    ? (\n        message: z.infer<GetSocketMessageSchema<TCatalog, K>>,\n        callback: (ack: z.infer<GetSocketCallbackSchema<TCatalog, K>>) => void\n      ) => void\n    : (message: z.infer<GetSocketMessageSchema<TCatalog, K>>) => void;\n};\n\nexport type GetSocketMessageSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = TRPCCatalog[TMessageType][\"message\"];\n\nexport type InferSocketMessageSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = z.infer<GetSocketMessageSchema<TRPCCatalog, TMessageType>>;\n\nexport type GetSocketCallbackSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = TRPCCatalog[TMessageType] extends { callback: any }\n  ? TRPCCatalog[TMessageType][\"callback\"]\n  : never;\n\nexport type InferSocketCallbackSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = z.infer<GetSocketCallbackSchema<TRPCCatalog, TMessageType>>;\n\nexport type SocketMessageHasCallback<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = GetSocketCallbackSchema<TRPCCatalog, TMessageType> extends never ? false : true;\n\nexport type ZodSocketMessageHandlers<TCatalogSchema extends ZodSocketMessageCatalogSchema> =\n  Partial<{\n    [K in keyof TCatalogSchema]: (\n      payload: z.infer<GetSocketMessageSchema<TCatalogSchema, K>>\n    ) => Promise<\n      SocketMessageHasCallback<TCatalogSchema, K> extends true\n        ? z.input<GetSocketCallbackSchema<TCatalogSchema, K>>\n        : void\n    >;\n  }>;\n\nexport type ZodSocketMessageHandlerOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> =\n  {\n    schema: TMessageCatalog;\n    handlers?: ZodSocketMessageHandlers<TMessageCatalog>;\n  };\n\ntype MessageFromSocketSchema<\n  K extends keyof TMessageCatalog,\n  TMessageCatalog extends ZodSocketMessageCatalogSchema,\n> = {\n  type: K;\n  payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>;\n};\n\nexport type MessagesFromSocketCatalog<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TMessageCatalog]: MessageFromSocketSchema<K, TMessageCatalog>;\n}[keyof TMessageCatalog];\n\nconst messageSchema = z.object({\n  version: z.literal(\"v1\").default(\"v1\"),\n  type: z.string(),\n  payload: z.unknown(),\n});\n\nexport class ZodSocketMessageHandler<TRPCCatalog extends ZodSocketMessageCatalogSchema> {\n  #schema: TRPCCatalog;\n  #handlers: ZodSocketMessageHandlers<TRPCCatalog> | undefined;\n\n  constructor(options: ZodSocketMessageHandlerOptions<TRPCCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.handlers;\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): MessagesFromSocketCatalog<TRPCCatalog> {\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  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.#handlers)) {\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        try {\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        } catch (error) {\n          log.error(\"Error while handling message\", { error });\n          return;\n        }\n\n        if (callback && typeof callback === \"function\") {\n          callback(ack);\n        }\n      });\n    }\n  }\n}\n\nexport type ZodSocketMessageSenderOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  schema: TMessageCatalog;\n  socket: ZodSocket<any, TMessageCatalog>;\n};\n\nexport type GetSocketMessagesWithCallback<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true\n    ? K\n    : never;\n}[keyof TMessageCatalog];\n\nexport type GetSocketMessagesWithoutCallback<\n  TMessageCatalog extends ZodSocketMessageCatalogSchema,\n> = {\n  [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true\n    ? never\n    : K;\n}[keyof TMessageCatalog];\n\nexport class ZodSocketMessageSender<TMessageCatalog extends ZodSocketMessageCatalogSchema> {\n  #schema: TMessageCatalog;\n  #socket: ZodSocket<any, TMessageCatalog>;\n\n  constructor(options: ZodSocketMessageSenderOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#socket = options.socket;\n  }\n\n  public send<K extends GetSocketMessagesWithoutCallback<TMessageCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>\n  ): void {\n    const schema = this.#schema[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 Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    // @ts-expect-error\n    this.#socket.emit(type, { payload, version: \"v1\" });\n\n    return;\n  }\n\n  public async sendWithAck<K extends GetSocketMessagesWithCallback<TMessageCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>\n  ): Promise<z.infer<GetSocketCallbackSchema<TMessageCatalog, K>>> {\n    const schema = this.#schema[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 Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    // @ts-expect-error\n    const callbackResult = await this.#socket.emitWithAck(type, { payload, version: \"v1\" });\n\n    return callbackResult;\n  }\n}\n\nexport type ZodSocket<\n  TListenEvents extends ZodSocketMessageCatalogSchema,\n  TEmitEvents extends ZodSocketMessageCatalogSchema,\n> = Socket<\n  ZodMessageCatalogToSocketIoEvents<TListenEvents>,\n  ZodMessageCatalogToSocketIoEvents<TEmitEvents>\n>;\n\ninterface ZodSocketConnectionOptions<\n  TClientMessages extends ZodSocketMessageCatalogSchema,\n  TServerMessages extends ZodSocketMessageCatalogSchema,\n> {\n  host: string;\n  port?: number;\n  secure?: boolean;\n  namespace: string;\n  clientMessages: TClientMessages;\n  serverMessages: TServerMessages;\n  extraHeaders?: {\n    [header: string]: string;\n  };\n  handlers?: ZodSocketMessageHandlers<TServerMessages>;\n  authToken?: string;\n  onConnection?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    handler: ZodSocketMessageHandler<TServerMessages>,\n    sender: ZodSocketMessageSender<TClientMessages>,\n    logger: StructuredLogger\n  ) => Promise<void>;\n  onDisconnect?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    reason: Socket.DisconnectReason,\n    description: any,\n    logger: StructuredLogger\n  ) => Promise<void>;\n  onError?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    err: Error,\n    logger: StructuredLogger\n  ) => Promise<void>;\n}\n\nexport class ZodSocketConnection<\n  TClientMessages extends ZodSocketMessageCatalogSchema,\n  TServerMessages extends ZodSocketMessageCatalogSchema,\n> {\n  #sender: ZodSocketMessageSender<TClientMessages>;\n  socket: ZodSocket<TServerMessages, TClientMessages>;\n\n  #handler: ZodSocketMessageHandler<TServerMessages>;\n  #logger: StructuredLogger;\n\n  constructor(opts: ZodSocketConnectionOptions<TClientMessages, TServerMessages>) {\n    const uri = `${opts.secure ? \"wss\" : \"ws\"}://${opts.host}:${\n      opts.port ?? (opts.secure ? \"443\" : \"80\")\n    }/${opts.namespace}`;\n\n    const logger = new SimpleStructuredLogger(opts.namespace, LogLevel.info);\n    logger.log(\"new zod socket\", { uri });\n\n    this.socket = io(uri, {\n      transports: [\"websocket\"],\n      auth: {\n        token: opts.authToken,\n      },\n      extraHeaders: opts.extraHeaders,\n      reconnectionDelay: 500,\n      reconnectionDelayMax: 1000,\n    });\n\n    this.#logger = logger.child({\n      socketId: this.socket.id,\n    });\n\n    this.#handler = new ZodSocketMessageHandler({\n      schema: opts.serverMessages,\n      handlers: opts.handlers,\n    });\n    this.#handler.registerHandlers(this.socket, this.#logger);\n\n    this.#sender = new ZodSocketMessageSender({\n      schema: opts.clientMessages,\n      socket: this.socket,\n    });\n\n    this.socket.on(\"connect_error\", async (error) => {\n      this.#logger.error(`connect_error: ${error}`);\n\n      if (opts.onError) {\n        await opts.onError(this.socket, error, this.#logger);\n      }\n    });\n\n    this.socket.on(\"connect\", async () => {\n      this.#logger.info(\"connect\");\n\n      if (opts.onConnection) {\n        await opts.onConnection(this.socket, this.#handler, this.#sender, this.#logger);\n      }\n    });\n\n    this.socket.on(\"disconnect\", async (reason, description) => {\n      this.#logger.info(\"disconnect\", { reason, description });\n\n      if (opts.onDisconnect) {\n        await opts.onDisconnect(this.socket, reason, description, this.#logger);\n      }\n    });\n  }\n\n  close() {\n    this.socket.close();\n  }\n\n  connect() {\n    this.socket.connect();\n  }\n\n  get send() {\n    return this.#sender.send.bind(this.#sender);\n  }\n\n  get sendWithAck() {\n    return this.#sender.sendWithAck.bind(this.#sender);\n  }\n}\n\nfunction createLogger(prefix: string) {\n  return (...args: any[]) => console.log(prefix, ...args);\n}\n","type StructuredArgs = (Record<string, unknown> | undefined)[];\n\nexport interface StructuredLogger {\n  log: (message: string, ...args: StructuredArgs) => any;\n  error: (message: string, ...args: StructuredArgs) => any;\n  warn: (message: string, ...args: StructuredArgs) => any;\n  info: (message: string, ...args: StructuredArgs) => any;\n  debug: (message: string, ...args: StructuredArgs) => any;\n  child: (fields: Record<string, unknown>) => StructuredLogger;\n}\n\nexport enum LogLevel {\n  \"log\",\n  \"error\",\n  \"warn\",\n  \"info\",\n  \"debug\",\n}\n\nexport class SimpleStructuredLogger implements StructuredLogger {\n  constructor(\n    private name: string,\n    private level: LogLevel = [\"1\", \"true\"].includes(process.env.DEBUG ?? \"\")\n      ? LogLevel.debug\n      : LogLevel.info,\n    private fields?: Record<string, unknown>\n  ) {}\n\n  child(fields: Record<string, unknown>, level?: LogLevel) {\n    return new SimpleStructuredLogger(this.name, level, { ...this.fields, ...fields });\n  }\n\n  log(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.log) return;\n\n    this.#structuredLog(console.log, message, \"log\", ...args);\n  }\n\n  error(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.error) return;\n\n    this.#structuredLog(console.error, message, \"error\", ...args);\n  }\n\n  warn(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.warn) return;\n\n    this.#structuredLog(console.warn, message, \"warn\", ...args);\n  }\n\n  info(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.info) return;\n\n    this.#structuredLog(console.info, message, \"info\", ...args);\n  }\n\n  debug(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.debug) return;\n\n    this.#structuredLog(console.debug, message, \"debug\", ...args);\n  }\n\n  #structuredLog(\n    loggerFunction: (message: string, ...args: any[]) => void,\n    message: string,\n    level: string,\n    ...args: StructuredArgs\n  ) {\n    const structuredLog = {\n      ...(args.length === 1 ? args[0] : args),\n      ...this.fields,\n      timestamp: new Date(),\n      name: this.name,\n      message,\n      level,\n    };\n\n    loggerFunction(JSON.stringify(structuredLog));\n  }\n}\n"]}
+{"version":3,"sources":["../../src/v3/zodSocket.ts","../../src/v3/utils/structuredLogger.ts"],"names":["io","ZodError","z","LogLevel","SimpleStructuredLogger","constructor","name","level","includes","process","env","DEBUG","debug","info","fields","child","log","message","args","console","error","warn","loggerFunction","structuredLog","length","timestamp","Date","JSON","stringify","fromZodError","messageSchema","object","version","string","type","payload","unknown","ZodSocketMessageHandler","options","schema","handlers","handleMessage","parsedMessage","parseMessage","Error","handler","String","ack","safeParse","success","data","messageWithVersion","parsedPayload","registerHandlers","emitter","logger","eventName","Object","keys","on","callback","hasCallback","stack","_schema","ZodSocketMessageSender","socket","send","emit","sendWithAck","callbackResult","emitWithAck","ZodSocketConnection","opts","uri","secure","host","port","namespace","transports","auth","token","authToken","extraHeaders","reconnectionDelay","reconnectionDelayMax","socketId","id","serverMessages","clientMessages","onError","onConnection","reason","description","onDisconnect","close","connect","bind"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AACA,SAASA,UAAU;AACnB,SAASC,UAAUC,SAAS;;;ICSrB;UAAKC,WAAQ;AAARA,EAAAA,UAAAA,UACV,KAAA,IAAA,CAAA,IAAA;AADUA,EAAAA,UAAAA,UAEV,OAAA,IAAA,CAAA,IAAA;AAFUA,EAAAA,UAAAA,UAGV,MAAA,IAAA,CAAA,IAAA;AAHUA,EAAAA,UAAAA,UAIV,MAAA,IAAA,CAAA,IAAA;AAJUA,EAAAA,UAAAA,UAKV,OAAA,IAAA,CAAA,IAAA;GALUA,aAAAA,WAAAA,CAAAA,EAAAA;;AAQL,IAAMC,0BAAN,MAAMA,wBAAAA;EACXC,YACUC,MACAC,QAAkB;IAAC;IAAK;IAAQC,SAASC,QAAQC,IAAIC,SAAS,EAAA,IAClER,SAASS,QACTT,SAASU,MACLC,QACR;AAoCF;gBAzCUR;iBACAC;kBAGAO;EACP;EAEHC,MAAMD,QAAiCP,OAAkB;AACvD,WAAO,IAAIH,wBAAuB,KAAKE,MAAMC,OAAO;MAAE,GAAG,KAAKO;MAAQ,GAAGA;IAAO,CAAA;EAClF;EAEAE,IAAIC,YAAoBC,MAAsB;AAC5C,QAAI,KAAKX,QAAQJ,SAASa;AAAK;AAE/B,0BAAK,kCAAL,WAAoBG,QAAQH,KAAKC,SAAS,OAAA,GAAUC;EACtD;EAEAE,MAAMH,YAAoBC,MAAsB;AAC9C,QAAI,KAAKX,QAAQJ,SAASiB;AAAO;AAEjC,0BAAK,kCAAL,WAAoBD,QAAQC,OAAOH,SAAS,SAAA,GAAYC;EAC1D;EAEAG,KAAKJ,YAAoBC,MAAsB;AAC7C,QAAI,KAAKX,QAAQJ,SAASkB;AAAM;AAEhC,0BAAK,kCAAL,WAAoBF,QAAQE,MAAMJ,SAAS,QAAA,GAAWC;EACxD;EAEAL,KAAKI,YAAoBC,MAAsB;AAC7C,QAAI,KAAKX,QAAQJ,SAASU;AAAM;AAEhC,0BAAK,kCAAL,WAAoBM,QAAQN,MAAMI,SAAS,QAAA,GAAWC;EACxD;EAEAN,MAAMK,YAAoBC,MAAsB;AAC9C,QAAI,KAAKX,QAAQJ,SAASS;AAAO;AAEjC,0BAAK,kCAAL,WAAoBO,QAAQP,OAAOK,SAAS,SAAA,GAAYC;EAC1D;AAmBF;AAjBE;mBAAc,gCACZI,gBACAL,SACAV,UACGW,MACH;AACA,QAAMK,gBAAgB;IACpB,GAAIL,KAAKM,WAAW,IAAIN,KAAK,CAAA,IAAKA;IAClC,GAAG,KAAKJ;IACRW,WAAW,oBAAIC,KAAAA;IACfpB,MAAM,KAAKA;IACXW;IACAV;EACF;AAEAe,iBAAeK,KAAKC,UAAUL,aAAAA,CAAAA;AAChC,GAhBc;AA3CHnB;AAAN,IAAMA,yBAAN;;;ADdP,SAASyB,oBAAoB;AA8E7B,IAAMC,gBAAgB5B,EAAE6B,OAAO;EAC7BC,SAAS9B,EAAE+B,OAAM;EACjBC,MAAMhC,EAAE+B,OAAM;EACdE,SAASjC,EAAEkC,QAAO;AACpB,CAAA;AAtFA;AAwFO,IAAMC,2BAAN,MAAMA,yBAAAA;EAIXhC,YAAYiC,SAAsD;AAHlE;AACA;AAGE,uBAAK,SAAUA,QAAQC;AACvB,uBAAK,WAAYD,QAAQE;EAC3B;EAEA,MAAaC,cAAcxB,SAAkB;AAC3C,UAAMyB,gBAAgB,KAAKC,aAAa1B,OAAAA;AAExC,QAAI,CAAC,mBAAK,YAAW;AACnB,YAAM,IAAI2B,MAAM,sBAAA;IAClB;AAEA,UAAMC,UAAU,mBAAK,WAAUH,cAAcR,IAAI;AAEjD,QAAI,CAACW,SAAS;AACZ1B,cAAQC,MAAM,gCAAgC0B,OAAOJ,cAAcR,IAAI,CAAA,EAAG;AAC1E;IACF;AAEA,UAAMa,MAAM,MAAMF,QAAQH,cAAcP,OAAO;AAE/C,WAAOY;EACT;EAEOJ,aAAa1B,SAA0D;AAC5E,UAAMyB,gBAAgBZ,cAAckB,UAAU/B,OAAAA;AAE9C,QAAI,CAACyB,cAAcO,SAAS;AAC1B,YAAM,IAAIL,MAAM,4BAA4BjB,KAAKC,UAAUc,cAActB,KAAK,CAAA,EAAG;IACnF;AAEA,UAAMmB,SAAS,mBAAK,SAAQG,cAAcQ,KAAKhB,IAAI,EAAE,SAAA;AAErD,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBF,cAAcQ,KAAKhB,IAAI,EAAE;IACpE;AAEA,UAAMiB,qBAAqB;MACzBnB,SAASU,cAAcQ,KAAKlB;MAC5B,GAAI,OAAOU,cAAcQ,KAAKf,YAAY,WAAWO,cAAcQ,KAAKf,UAAU,CAAC;IACrF;AAEA,UAAMiB,gBAAgBb,OAAOS,UAAUG,kBAAAA;AAEvC,QAAI,CAACC,cAAcH,SAAS;AAC1B9B,cAAQC,MAAM,mCAAmC;QAC/CH;QACAkB,SAASgB;MACX,CAAA;AAEA,YAAMC,cAAchC,iBAAiBnB,WACjC4B,aAAauB,cAAchC,KAAK,IAChCgC,cAAchC;IACpB;AAEA,WAAO;MACLc,MAAMQ,cAAcQ,KAAKhB;MACzBC,SAASiB,cAAcF;IACzB;EACF;EAEOG,iBAAiBC,SAA2BC,QAA2B;AAC5E,UAAMvC,MAAMuC,UAAUpC;AAEtB,QAAI,CAAC,mBAAK,YAAW;AACnBH,UAAIH,KAAK,sBAAA;AACT;IACF;AAEA,eAAW2C,aAAaC,OAAOC,KAAK,mBAAK,UAAS,GAAG;AACnDJ,cAAQK,GAAGH,WAAW,OAAOvC,SAAc2C,aAAkC;AAC3E5C,YAAIH,KAAK,YAAY2C,SAAAA,IAAa;UAChCrB,SAASlB;UACT4C,aAAa,CAAC,CAACD;QACjB,CAAA;AAEA,YAAIb;AAEJ,YAAI;AAEF,cAAI,aAAa9B,SAAS;AACxB8B,kBAAM,MAAM,KAAKN,cAAc;cAAEP,MAAMsB;cAAW,GAAGvC;YAAQ,CAAA;UAC/D,OAAO;AAEL,kBAAM,EAAEe,SAAS,GAAGG,QAAAA,IAAYlB;AAChC8B,kBAAM,MAAM,KAAKN,cAAc;cAAEP,MAAMsB;cAAWxB;cAASG;YAAQ,CAAA;UACrE;QACF,SAASf,OAAO;AACdJ,cAAII,MAAM,gCAAgC;YACxCA,OACEA,iBAAiBwB,QACb;cACE3B,SAASG,MAAMH;cACf6C,OAAO1C,MAAM0C;YACf,IACA1C;UACR,CAAA;AACA;QACF;AAEA,YAAIwC,YAAY,OAAOA,aAAa,YAAY;AAC9CA,mBAASb,GAAAA;QACX;MACF,CAAA;IACF;EACF;AACF;AA7GE;AACA;AAFWV;AAAN,IAAMA,0BAAN;AAxFP,IAAA0B,UAAA;AA2NO,IAAMC,0BAAN,MAAMA,wBAAAA;EAIX3D,YAAYiC,SAAyD;AAHrE,uBAAAyB,UAAA;AACA;AAGE,uBAAKA,UAAUzB,QAAQC;AACvB,uBAAK,SAAUD,QAAQ2B;EACzB;EAEOC,KACLhC,MACAC,SACM;AACN,UAAMI,SAAS,mBAAKwB,UAAQ7B,IAAAA,EAAM,SAAA;AAElC,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBV,IAAAA,EAAgB;IAC3D;AAEA,UAAMkB,gBAAgBb,OAAOS,UAAUb,OAAAA;AAEvC,QAAI,CAACiB,cAAcH,SAAS;AAC1B,YAAM,IAAIL,MAAM,oCAAoCjB,KAAKC,UAAUwB,cAAchC,KAAK,CAAA,EAAG;IAC3F;AAGA,uBAAK,SAAQ+C,KAAKjC,MAAM;MAAEC;MAASH,SAAS;IAAK,CAAA;AAEjD;EACF;EAEA,MAAaoC,YACXlC,MACAC,SAC+D;AAC/D,UAAMI,SAAS,mBAAKwB,UAAQ7B,IAAAA,EAAM,SAAA;AAElC,QAAI,CAACK,QAAQ;AACX,YAAM,IAAIK,MAAM,yBAAyBV,IAAAA,EAAgB;IAC3D;AAEA,UAAMkB,gBAAgBb,OAAOS,UAAUb,OAAAA;AAEvC,QAAI,CAACiB,cAAcH,SAAS;AAC1B,YAAM,IAAIL,MAAM,oCAAoCjB,KAAKC,UAAUwB,cAAchC,KAAK,CAAA,EAAG;IAC3F;AAGA,UAAMiD,iBAAiB,MAAM,mBAAK,SAAQC,YAAYpC,MAAM;MAAEC;MAASH,SAAS;IAAK,CAAA;AAErF,WAAOqC;EACT;AACF;AAnDEN,WAAA;AACA;AAFWC;AAAN,IAAMA,yBAAN;AA3NP;AA2TO,IAAMO,uBAAN,MAAMA,qBAAAA;EAUXlE,YAAYmE,MAAoE;AANhF;AAGA;AACA;AAGE,UAAMC,MAAM,GAAGD,KAAKE,SAAS,QAAQ,IAAI,MAAMF,KAAKG,IAAI,IACtDH,KAAKI,SAASJ,KAAKE,SAAS,QAAQ,KAAG,IACrCF,KAAKK,SAAS;AAElB,UAAMtB,SAAS,IAAInD,uBAAuBoE,KAAKK,WAAW1E,SAASU,IAAI;AACvE0C,WAAOvC,IAAI,kBAAkB;MAAEyD;IAAI,CAAA;AAEnC,SAAKR,SAASjE,GAAGyE,KAAK;MACpBK,YAAY;QAAC;;MACbC,MAAM;QACJC,OAAOR,KAAKS;MACd;MACAC,cAAcV,KAAKU;MACnBC,mBAAmB;MACnBC,sBAAsB;IACxB,CAAA;AAEA,uBAAK,SAAU7B,OAAOxC,MAAM;MAC1BsE,UAAU,KAAKpB,OAAOqB;IACxB,CAAA;AAEA,uBAAK,UAAW,IAAIjD,wBAAwB;MAC1CE,QAAQiC,KAAKe;MACb/C,UAAUgC,KAAKhC;IACjB,CAAA;AACA,uBAAK,UAASa,iBAAiB,KAAKY,QAAQ,mBAAK,QAAO;AAExD,uBAAK,SAAU,IAAID,uBAAuB;MACxCzB,QAAQiC,KAAKgB;MACbvB,QAAQ,KAAKA;IACf,CAAA;AAEA,SAAKA,OAAON,GAAG,iBAAiB,OAAOvC,UAAU;AAC/C,yBAAK,SAAQA,MAAM,kBAAkBA,KAAAA,EAAO;AAE5C,UAAIoD,KAAKiB,SAAS;AAChB,cAAMjB,KAAKiB,QAAQ,KAAKxB,QAAQ7C,OAAO,mBAAK,QAAO;MACrD;IACF,CAAA;AAEA,SAAK6C,OAAON,GAAG,WAAW,YAAY;AACpC,yBAAK,SAAQ9C,KAAK,SAAA;AAElB,UAAI2D,KAAKkB,cAAc;AACrB,cAAMlB,KAAKkB,aAAa,KAAKzB,QAAQ,mBAAK,WAAU,mBAAK,UAAS,mBAAK,QAAO;MAChF;IACF,CAAA;AAEA,SAAKA,OAAON,GAAG,cAAc,OAAOgC,QAAQC,gBAAgB;AAC1D,yBAAK,SAAQ/E,KAAK,cAAc;QAAE8E;QAAQC;MAAY,CAAA;AAEtD,UAAIpB,KAAKqB,cAAc;AACrB,cAAMrB,KAAKqB,aAAa,KAAK5B,QAAQ0B,QAAQC,aAAa,mBAAK,QAAO;MACxE;IACF,CAAA;EACF;EAEAE,QAAQ;AACN,SAAK7B,OAAO6B,MAAK;EACnB;EAEAC,UAAU;AACR,SAAK9B,OAAO8B,QAAO;EACrB;EAEA,IAAI7B,OAAO;AACT,WAAO,mBAAK,SAAQA,KAAK8B,KAAK,mBAAK,QAAO;EAC5C;EAEA,IAAI5B,cAAc;AAChB,WAAO,mBAAK,SAAQA,YAAY4B,KAAK,mBAAK,QAAO;EACnD;AACF;AA/EE;AAGA;AACA;AARWzB;AAAN,IAAMA,sBAAN","sourcesContent":["import type { Socket } from \"socket.io-client\";\nimport { io } from \"socket.io-client\";\nimport { ZodError, z } from \"zod\";\nimport { EventEmitterLike, ZodMessageValueSchema } from \"./zodMessageHandler\";\nimport { LogLevel, SimpleStructuredLogger, StructuredLogger } from \"./utils/structuredLogger\";\nimport { fromZodError } from \"zod-validation-error\";\n\nexport interface ZodSocketMessageCatalogSchema {\n  [key: string]:\n    | {\n        message: ZodMessageValueSchema<any>;\n      }\n    | {\n        message: ZodMessageValueSchema<any>;\n        callback?: ZodMessageValueSchema<any>;\n      };\n}\n\nexport type ZodMessageCatalogToSocketIoEvents<TCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TCatalog]: SocketMessageHasCallback<TCatalog, K> extends true\n    ? (\n        message: z.infer<GetSocketMessageSchema<TCatalog, K>>,\n        callback: (ack: z.infer<GetSocketCallbackSchema<TCatalog, K>>) => void\n      ) => void\n    : (message: z.infer<GetSocketMessageSchema<TCatalog, K>>) => void;\n};\n\nexport type GetSocketMessageSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = TRPCCatalog[TMessageType][\"message\"];\n\nexport type InferSocketMessageSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = z.infer<GetSocketMessageSchema<TRPCCatalog, TMessageType>>;\n\nexport type GetSocketCallbackSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = TRPCCatalog[TMessageType] extends { callback: any }\n  ? TRPCCatalog[TMessageType][\"callback\"]\n  : never;\n\nexport type InferSocketCallbackSchema<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = z.infer<GetSocketCallbackSchema<TRPCCatalog, TMessageType>>;\n\nexport type SocketMessageHasCallback<\n  TRPCCatalog extends ZodSocketMessageCatalogSchema,\n  TMessageType extends keyof TRPCCatalog,\n> = GetSocketCallbackSchema<TRPCCatalog, TMessageType> extends never ? false : true;\n\nexport type ZodSocketMessageHandlers<TCatalogSchema extends ZodSocketMessageCatalogSchema> =\n  Partial<{\n    [K in keyof TCatalogSchema]: (\n      payload: z.infer<GetSocketMessageSchema<TCatalogSchema, K>>\n    ) => Promise<\n      SocketMessageHasCallback<TCatalogSchema, K> extends true\n        ? z.input<GetSocketCallbackSchema<TCatalogSchema, K>>\n        : void\n    >;\n  }>;\n\nexport type ZodSocketMessageHandlerOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> =\n  {\n    schema: TMessageCatalog;\n    handlers?: ZodSocketMessageHandlers<TMessageCatalog>;\n  };\n\ntype MessageFromSocketSchema<\n  K extends keyof TMessageCatalog,\n  TMessageCatalog extends ZodSocketMessageCatalogSchema,\n> = {\n  type: K;\n  payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>;\n};\n\nexport type MessagesFromSocketCatalog<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TMessageCatalog]: MessageFromSocketSchema<K, TMessageCatalog>;\n}[keyof TMessageCatalog];\n\nconst messageSchema = z.object({\n  version: z.string(),\n  type: z.string(),\n  payload: z.unknown(),\n});\n\nexport class ZodSocketMessageHandler<TRPCCatalog extends ZodSocketMessageCatalogSchema> {\n  #schema: TRPCCatalog;\n  #handlers: ZodSocketMessageHandlers<TRPCCatalog> | undefined;\n\n  constructor(options: ZodSocketMessageHandlerOptions<TRPCCatalog>) {\n    this.#schema = options.schema;\n    this.#handlers = options.handlers;\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): MessagesFromSocketCatalog<TRPCCatalog> {\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 messageWithVersion = {\n      version: parsedMessage.data.version,\n      ...(typeof parsedMessage.data.payload === \"object\" ? parsedMessage.data.payload : {}),\n    };\n\n    const parsedPayload = schema.safeParse(messageWithVersion);\n\n    if (!parsedPayload.success) {\n      console.error(\"Failed to parse message payload\", {\n        message,\n        payload: messageWithVersion,\n      });\n\n      throw parsedPayload.error instanceof ZodError\n        ? fromZodError(parsedPayload.error)\n        : 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.#handlers)) {\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        try {\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        } catch (error) {\n          log.error(\"Error while handling message\", {\n            error:\n              error instanceof Error\n                ? {\n                    message: error.message,\n                    stack: error.stack,\n                  }\n                : error,\n          });\n          return;\n        }\n\n        if (callback && typeof callback === \"function\") {\n          callback(ack);\n        }\n      });\n    }\n  }\n}\n\nexport type ZodSocketMessageSenderOptions<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  schema: TMessageCatalog;\n  socket: ZodSocket<any, TMessageCatalog>;\n};\n\nexport type GetSocketMessagesWithCallback<TMessageCatalog extends ZodSocketMessageCatalogSchema> = {\n  [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true\n    ? K\n    : never;\n}[keyof TMessageCatalog];\n\nexport type GetSocketMessagesWithoutCallback<\n  TMessageCatalog extends ZodSocketMessageCatalogSchema,\n> = {\n  [K in keyof TMessageCatalog]: SocketMessageHasCallback<TMessageCatalog, K> extends true\n    ? never\n    : K;\n}[keyof TMessageCatalog];\n\nexport class ZodSocketMessageSender<TMessageCatalog extends ZodSocketMessageCatalogSchema> {\n  #schema: TMessageCatalog;\n  #socket: ZodSocket<any, TMessageCatalog>;\n\n  constructor(options: ZodSocketMessageSenderOptions<TMessageCatalog>) {\n    this.#schema = options.schema;\n    this.#socket = options.socket;\n  }\n\n  public send<K extends GetSocketMessagesWithoutCallback<TMessageCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>\n  ): void {\n    const schema = this.#schema[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 Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    // @ts-expect-error\n    this.#socket.emit(type, { payload, version: \"v1\" });\n\n    return;\n  }\n\n  public async sendWithAck<K extends GetSocketMessagesWithCallback<TMessageCatalog>>(\n    type: K,\n    payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>\n  ): Promise<z.infer<GetSocketCallbackSchema<TMessageCatalog, K>>> {\n    const schema = this.#schema[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 Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);\n    }\n\n    // @ts-expect-error\n    const callbackResult = await this.#socket.emitWithAck(type, { payload, version: \"v1\" });\n\n    return callbackResult;\n  }\n}\n\nexport type ZodSocket<\n  TListenEvents extends ZodSocketMessageCatalogSchema,\n  TEmitEvents extends ZodSocketMessageCatalogSchema,\n> = Socket<\n  ZodMessageCatalogToSocketIoEvents<TListenEvents>,\n  ZodMessageCatalogToSocketIoEvents<TEmitEvents>\n>;\n\ninterface ZodSocketConnectionOptions<\n  TClientMessages extends ZodSocketMessageCatalogSchema,\n  TServerMessages extends ZodSocketMessageCatalogSchema,\n> {\n  host: string;\n  port?: number;\n  secure?: boolean;\n  namespace: string;\n  clientMessages: TClientMessages;\n  serverMessages: TServerMessages;\n  extraHeaders?: {\n    [header: string]: string;\n  };\n  handlers?: ZodSocketMessageHandlers<TServerMessages>;\n  authToken?: string;\n  onConnection?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    handler: ZodSocketMessageHandler<TServerMessages>,\n    sender: ZodSocketMessageSender<TClientMessages>,\n    logger: StructuredLogger\n  ) => Promise<void>;\n  onDisconnect?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    reason: Socket.DisconnectReason,\n    description: any,\n    logger: StructuredLogger\n  ) => Promise<void>;\n  onError?: (\n    socket: ZodSocket<TServerMessages, TClientMessages>,\n    err: Error,\n    logger: StructuredLogger\n  ) => Promise<void>;\n}\n\nexport class ZodSocketConnection<\n  TClientMessages extends ZodSocketMessageCatalogSchema,\n  TServerMessages extends ZodSocketMessageCatalogSchema,\n> {\n  #sender: ZodSocketMessageSender<TClientMessages>;\n  socket: ZodSocket<TServerMessages, TClientMessages>;\n\n  #handler: ZodSocketMessageHandler<TServerMessages>;\n  #logger: StructuredLogger;\n\n  constructor(opts: ZodSocketConnectionOptions<TClientMessages, TServerMessages>) {\n    const uri = `${opts.secure ? \"wss\" : \"ws\"}://${opts.host}:${\n      opts.port ?? (opts.secure ? \"443\" : \"80\")\n    }/${opts.namespace}`;\n\n    const logger = new SimpleStructuredLogger(opts.namespace, LogLevel.info);\n    logger.log(\"new zod socket\", { uri });\n\n    this.socket = io(uri, {\n      transports: [\"websocket\"],\n      auth: {\n        token: opts.authToken,\n      },\n      extraHeaders: opts.extraHeaders,\n      reconnectionDelay: 500,\n      reconnectionDelayMax: 1000,\n    });\n\n    this.#logger = logger.child({\n      socketId: this.socket.id,\n    });\n\n    this.#handler = new ZodSocketMessageHandler({\n      schema: opts.serverMessages,\n      handlers: opts.handlers,\n    });\n    this.#handler.registerHandlers(this.socket, this.#logger);\n\n    this.#sender = new ZodSocketMessageSender({\n      schema: opts.clientMessages,\n      socket: this.socket,\n    });\n\n    this.socket.on(\"connect_error\", async (error) => {\n      this.#logger.error(`connect_error: ${error}`);\n\n      if (opts.onError) {\n        await opts.onError(this.socket, error, this.#logger);\n      }\n    });\n\n    this.socket.on(\"connect\", async () => {\n      this.#logger.info(\"connect\");\n\n      if (opts.onConnection) {\n        await opts.onConnection(this.socket, this.#handler, this.#sender, this.#logger);\n      }\n    });\n\n    this.socket.on(\"disconnect\", async (reason, description) => {\n      this.#logger.info(\"disconnect\", { reason, description });\n\n      if (opts.onDisconnect) {\n        await opts.onDisconnect(this.socket, reason, description, this.#logger);\n      }\n    });\n  }\n\n  close() {\n    this.socket.close();\n  }\n\n  connect() {\n    this.socket.connect();\n  }\n\n  get send() {\n    return this.#sender.send.bind(this.#sender);\n  }\n\n  get sendWithAck() {\n    return this.#sender.sendWithAck.bind(this.#sender);\n  }\n}\n\nfunction createLogger(prefix: string) {\n  return (...args: any[]) => console.log(prefix, ...args);\n}\n","type StructuredArgs = (Record<string, unknown> | undefined)[];\n\nexport interface StructuredLogger {\n  log: (message: string, ...args: StructuredArgs) => any;\n  error: (message: string, ...args: StructuredArgs) => any;\n  warn: (message: string, ...args: StructuredArgs) => any;\n  info: (message: string, ...args: StructuredArgs) => any;\n  debug: (message: string, ...args: StructuredArgs) => any;\n  child: (fields: Record<string, unknown>) => StructuredLogger;\n}\n\nexport enum LogLevel {\n  \"log\",\n  \"error\",\n  \"warn\",\n  \"info\",\n  \"debug\",\n}\n\nexport class SimpleStructuredLogger implements StructuredLogger {\n  constructor(\n    private name: string,\n    private level: LogLevel = [\"1\", \"true\"].includes(process.env.DEBUG ?? \"\")\n      ? LogLevel.debug\n      : LogLevel.info,\n    private fields?: Record<string, unknown>\n  ) {}\n\n  child(fields: Record<string, unknown>, level?: LogLevel) {\n    return new SimpleStructuredLogger(this.name, level, { ...this.fields, ...fields });\n  }\n\n  log(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.log) return;\n\n    this.#structuredLog(console.log, message, \"log\", ...args);\n  }\n\n  error(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.error) return;\n\n    this.#structuredLog(console.error, message, \"error\", ...args);\n  }\n\n  warn(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.warn) return;\n\n    this.#structuredLog(console.warn, message, \"warn\", ...args);\n  }\n\n  info(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.info) return;\n\n    this.#structuredLog(console.info, message, \"info\", ...args);\n  }\n\n  debug(message: string, ...args: StructuredArgs) {\n    if (this.level < LogLevel.debug) return;\n\n    this.#structuredLog(console.debug, message, \"debug\", ...args);\n  }\n\n  #structuredLog(\n    loggerFunction: (message: string, ...args: any[]) => void,\n    message: string,\n    level: string,\n    ...args: StructuredArgs\n  ) {\n    const structuredLog = {\n      ...(args.length === 1 ? args[0] : args),\n      ...this.fields,\n      timestamp: new Date(),\n      name: this.name,\n      message,\n      level,\n    };\n\n    loggerFunction(JSON.stringify(structuredLog));\n  }\n}\n"]}

package/dist/v3/zodfetch.d.mts

@@ -0,0 +1,262 @@
+import { z } from 'zod';
+import { R as RetryOptions } from '../schemas-r4ZP9S-F.mjs';
+import { Readable } from 'node:stream';
+
+interface CursorPageParams {
+    limit?: number;
+    after?: string;
+    before?: string;
+}
+interface OffsetLimitPageParams {
+    limit?: number;
+    page?: number;
+}
+interface PageResponse<Item> {
+    data: Array<Item>;
+}
+interface CursorPageResponse<Item> extends PageResponse<Item> {
+    pagination: {
+        next?: string;
+        previous?: string;
+    };
+}
+interface OffsetLimitPageResponse<Item> extends PageResponse<Item> {
+    pagination: {
+        currentPage: number;
+        totalPages: number;
+        count: number;
+    };
+}
+interface Page<Item> {
+    getPaginatedItems(): Item[];
+    hasNextPage(): boolean;
+    hasPreviousPage(): boolean;
+}
+declare class CursorPage<Item> implements CursorPageResponse<Item>, Page<Item>, AsyncIterable<Item> {
+    private pageFetcher;
+    data: Array<Item>;
+    pagination: {
+        next?: string;
+        previous?: string;
+    };
+    constructor(data: Array<Item>, pagination: {
+        next?: string;
+        previous?: string;
+    }, pageFetcher: (params: Omit<CursorPageParams, "limit">) => Promise<CursorPage<Item>>);
+    getPaginatedItems(): Item[];
+    hasNextPage(): boolean;
+    hasPreviousPage(): boolean;
+    getNextPage(): Promise<CursorPage<Item>>;
+    getPreviousPage(): Promise<CursorPage<Item>>;
+    iterPages(): AsyncGenerator<CursorPage<Item>, void, unknown>;
+    [Symbol.asyncIterator](): AsyncGenerator<Awaited<Item>, void, unknown>;
+}
+declare class OffsetLimitPage<Item> implements OffsetLimitPageResponse<Item>, Page<Item>, AsyncIterable<Item> {
+    private pageFetcher;
+    data: Array<Item>;
+    pagination: {
+        currentPage: number;
+        totalPages: number;
+        count: number;
+    };
+    constructor(data: Array<Item>, pagination: {
+        currentPage: number;
+        totalPages: number;
+        count: number;
+    }, pageFetcher: (params: Omit<OffsetLimitPageParams, "limit">) => Promise<OffsetLimitPage<Item>>);
+    getPaginatedItems(): Item[];
+    hasNextPage(): boolean;
+    hasPreviousPage(): boolean;
+    getNextPage(): Promise<OffsetLimitPage<Item>>;
+    getPreviousPage(): Promise<OffsetLimitPage<Item>>;
+    iterPages(): AsyncGenerator<OffsetLimitPage<Item>, void, unknown>;
+    [Symbol.asyncIterator](): AsyncGenerator<Awaited<Item>, void, unknown>;
+}
+
+declare const defaultRetryOptions: {
+    maxAttempts: number;
+    factor: number;
+    minTimeoutInMs: number;
+    maxTimeoutInMs: number;
+    randomize: false;
+};
+type ZodFetchOptions = {
+    retry?: RetryOptions;
+};
+interface FetchCursorPageParams extends CursorPageParams {
+    query?: URLSearchParams;
+}
+interface FetchOffsetLimitPageParams extends OffsetLimitPageParams {
+    query?: URLSearchParams;
+}
+declare function zodfetch<TResponseBodySchema extends z.ZodTypeAny>(schema: TResponseBodySchema, url: string, requestInit?: RequestInit, options?: ZodFetchOptions): ApiPromise<z.output<TResponseBodySchema>>;
+declare function zodfetchCursorPage<TItemSchema extends z.ZodTypeAny>(schema: TItemSchema, url: string, params: FetchCursorPageParams, requestInit?: RequestInit, options?: ZodFetchOptions): CursorPagePromise<TItemSchema>;
+declare function zodfetchOffsetLimitPage<TItemSchema extends z.ZodTypeAny>(schema: TItemSchema, url: string, params: FetchOffsetLimitPageParams, requestInit?: RequestInit, options?: ZodFetchOptions): OffsetLimitPagePromise<TItemSchema>;
+declare function zodupload<TResponseBodySchema extends z.ZodTypeAny, TBody = Record<string, unknown>>(schema: TResponseBodySchema, url: string, body: TBody, requestInit?: RequestInit, options?: ZodFetchOptions): ApiPromise<z.output<TResponseBodySchema>>;
+type ZodFetchResult<T> = {
+    data: T;
+    response: Response;
+};
+type ToFileInput = Uploadable | Exclude<BlobLikePart, string> | AsyncIterable<BlobLikePart>;
+/**
+ * Helper for creating a {@link File} to pass to an SDK upload method from a variety of different data formats
+ * @param value the raw content of the file.  Can be an {@link Uploadable}, {@link BlobLikePart}, or {@link AsyncIterable} of {@link BlobLikePart}s
+ * @param {string=} name the name of the file. If omitted, toFile will try to determine a file name from bits if possible
+ * @param {Object=} options additional properties
+ * @param {string=} options.type the MIME type of the content
+ * @param {number=} options.lastModified the last modified timestamp
+ * @returns a {@link File} with the given properties
+ */
+declare function toFile(value: ToFileInput | PromiseLike<ToFileInput>, name?: string | null | undefined, options?: FilePropertyBag | undefined): Promise<FileLike>;
+/**
+ * Intended to match web.Blob, node.Blob, node-fetch.Blob, etc.
+ */
+interface BlobLike {
+    /** [MDN Reference](https://developer.mozilla.org/docs/Web/API/Blob/size) */
+    readonly size: number;
+    /** [MDN Reference](https://developer.mozilla.org/docs/Web/API/Blob/type) */
+    readonly type: string;
+    /** [MDN Reference](https://developer.mozilla.org/docs/Web/API/Blob/text) */
+    text(): Promise<string>;
+    /** [MDN Reference](https://developer.mozilla.org/docs/Web/API/Blob/slice) */
+    slice(start?: number, end?: number): BlobLike;
+}
+/**
+ * Intended to match web.File, node.File, node-fetch.File, etc.
+ */
+interface FileLike extends BlobLike {
+    /** [MDN Reference](https://developer.mozilla.org/docs/Web/API/File/lastModified) */
+    readonly lastModified: number;
+    /** [MDN Reference](https://developer.mozilla.org/docs/Web/API/File/name) */
+    readonly name: string;
+}
+/**
+ * Intended to match web.Response, node.Response, node-fetch.Response, etc.
+ */
+interface ResponseLike {
+    url: string;
+    blob(): Promise<BlobLike>;
+}
+type Uploadable = FileLike | ResponseLike | Readable;
+declare const isResponseLike: (value: any) => value is ResponseLike;
+declare const isFileLike: (value: any) => value is FileLike;
+/**
+ * The BlobLike type omits arrayBuffer() because @types/node-fetch@^2.6.4 lacks it; but this check
+ * adds the arrayBuffer() method type because it is available and used at runtime
+ */
+declare const isBlobLike: (value: any) => value is BlobLike & {
+    arrayBuffer(): Promise<ArrayBuffer>;
+};
+declare const isFsReadStream: (value: any) => value is Readable;
+declare const isUploadable: (value: any) => value is Uploadable;
+type BlobLikePart = string | ArrayBuffer | ArrayBufferView | BlobLike | Uint8Array | DataView;
+declare const isRecordLike: (value: any) => value is Record<string, string>;
+/**
+ * A subclass of `Promise` providing additional helper methods
+ * for interacting with the SDK.
+ */
+declare class ApiPromise<T> extends Promise<T> {
+    private responsePromise;
+    constructor(responsePromise: Promise<ZodFetchResult<T>>);
+    /**
+     * Gets the raw `Response` instance instead of parsing the response
+     * data.
+     *
+     * If you want to parse the response body but still get the `Response`
+     * instance, you can use {@link withResponse()}.
+     */
+    asResponse(): Promise<Response>;
+    /**
+     * Gets the parsed response data and the raw `Response` instance.
+     *
+     * If you just want to get the raw `Response` instance without parsing it,
+     * you can use {@link asResponse()}.
+     */
+    withResponse(): Promise<{
+        data: T;
+        response: Response;
+    }>;
+    private parse;
+    then<TResult1 = T, TResult2 = never>(onfulfilled?: ((value: T) => TResult1 | PromiseLike<TResult1>) | undefined | null, onrejected?: ((reason: any) => TResult2 | PromiseLike<TResult2>) | undefined | null): Promise<TResult1 | TResult2>;
+    catch<TResult = never>(onrejected?: ((reason: any) => TResult | PromiseLike<TResult>) | undefined | null): Promise<T | TResult>;
+    finally(onfinally?: (() => void) | undefined | null): Promise<T>;
+}
+declare class CursorPagePromise<TItemSchema extends z.ZodTypeAny> extends ApiPromise<CursorPage<z.output<TItemSchema>>> implements AsyncIterable<z.output<TItemSchema>> {
+    #private;
+    private schema;
+    private url;
+    private params;
+    private requestInit?;
+    private options?;
+    constructor(result: Promise<ZodFetchResult<CursorPageResponse<z.output<TItemSchema>>>>, schema: TItemSchema, url: string, params: FetchCursorPageParams, requestInit?: RequestInit | undefined, options?: ZodFetchOptions | undefined);
+    /**
+     * Allow auto-paginating iteration on an unawaited list call, eg:
+     *
+     *    for await (const item of client.items.list()) {
+     *      console.log(item)
+     *    }
+     */
+    [Symbol.asyncIterator](): AsyncGenerator<Awaited<z.output<TItemSchema>>, void, unknown>;
+}
+declare class OffsetLimitPagePromise<TItemSchema extends z.ZodTypeAny> extends ApiPromise<OffsetLimitPage<z.output<TItemSchema>>> implements AsyncIterable<z.output<TItemSchema>> {
+    #private;
+    private schema;
+    private url;
+    private params;
+    private requestInit?;
+    private options?;
+    constructor(result: Promise<ZodFetchResult<OffsetLimitPageResponse<z.output<TItemSchema>>>>, schema: TItemSchema, url: string, params: FetchOffsetLimitPageParams, requestInit?: RequestInit | undefined, options?: ZodFetchOptions | undefined);
+    /**
+     * Allow auto-paginating iteration on an unawaited list call, eg:
+     *
+     *    for await (const item of client.items.list()) {
+     *      console.log(item)
+     *    }
+     */
+    [Symbol.asyncIterator](): AsyncGenerator<Awaited<z.output<TItemSchema>>, void, unknown>;
+}
+
+type APIHeaders = Record<string, string | null | undefined>;
+declare class ApiError extends Error {
+    readonly status: number | undefined;
+    readonly headers: APIHeaders | undefined;
+    readonly error: Object | undefined;
+    readonly code: string | null | undefined;
+    readonly param: string | null | undefined;
+    readonly type: string | undefined;
+    constructor(status: number | undefined, error: Object | undefined, message: string | undefined, headers: APIHeaders | undefined);
+    private static makeMessage;
+    static generate(status: number | undefined, errorResponse: Object | undefined, message: string | undefined, headers: APIHeaders | undefined): ApiError;
+}
+declare class ApiConnectionError extends ApiError {
+    readonly status: undefined;
+    constructor({ message, cause }: {
+        message?: string;
+        cause?: Error | undefined;
+    });
+}
+declare class BadRequestError extends ApiError {
+    readonly status: 400;
+}
+declare class AuthenticationError extends ApiError {
+    readonly status: 401;
+}
+declare class PermissionDeniedError extends ApiError {
+    readonly status: 403;
+}
+declare class NotFoundError extends ApiError {
+    readonly status: 404;
+}
+declare class ConflictError extends ApiError {
+    readonly status: 409;
+}
+declare class UnprocessableEntityError extends ApiError {
+    readonly status: 422;
+}
+declare class RateLimitError extends ApiError {
+    readonly status: 429;
+}
+declare class InternalServerError extends ApiError {
+}
+
+export { type APIHeaders, ApiConnectionError, ApiError, ApiPromise, AuthenticationError, BadRequestError, type BlobLike, type BlobLikePart, ConflictError, CursorPage, type CursorPageParams, CursorPagePromise, type CursorPageResponse, type FileLike, InternalServerError, NotFoundError, OffsetLimitPage, type OffsetLimitPageParams, OffsetLimitPagePromise, type OffsetLimitPageResponse, type Page, type PageResponse, PermissionDeniedError, RateLimitError, type ResponseLike, type ToFileInput, UnprocessableEntityError, type Uploadable, type ZodFetchOptions, defaultRetryOptions, isBlobLike, isFileLike, isFsReadStream, isRecordLike, isResponseLike, isUploadable, toFile, zodfetch, zodfetchCursorPage, zodfetchOffsetLimitPage, zodupload };