@trigger.dev/core 3.0.0-beta.4 → 3.0.0-beta.41

package/dist/v3/zodSocket.js

@@ -0,0 +1,319 @@
+'use strict';
+
+var socket_ioClient = require('socket.io-client');
+var zod = require('zod');
+var zodValidationError = require('zod-validation-error');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __accessCheck = (obj, member, msg) => {
+  if (!member.has(obj))
+    throw TypeError("Cannot " + msg);
+};
+var __privateGet = (obj, member, getter) => {
+  __accessCheck(obj, member, "read from private field");
+  return getter ? getter.call(obj) : member.get(obj);
+};
+var __privateAdd = (obj, member, value) => {
+  if (member.has(obj))
+    throw TypeError("Cannot add the same private member more than once");
+  member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
+};
+var __privateSet = (obj, member, value, setter) => {
+  __accessCheck(obj, member, "write to private field");
+  setter ? setter.call(obj, value) : member.set(obj, value);
+  return value;
+};
+var __privateMethod = (obj, member, method) => {
+  __accessCheck(obj, member, "access private method");
+  return method;
+};
+
+// src/v3/utils/structuredLogger.ts
+var LogLevel;
+(function(LogLevel2) {
+  LogLevel2[LogLevel2["log"] = 0] = "log";
+  LogLevel2[LogLevel2["error"] = 1] = "error";
+  LogLevel2[LogLevel2["warn"] = 2] = "warn";
+  LogLevel2[LogLevel2["info"] = 3] = "info";
+  LogLevel2[LogLevel2["debug"] = 4] = "debug";
+})(LogLevel || (LogLevel = {}));
+var _structuredLog, structuredLog_fn;
+var _SimpleStructuredLogger = class _SimpleStructuredLogger {
+  constructor(name, level = [
+    "1",
+    "true"
+  ].includes(process.env.DEBUG ?? "") ? LogLevel.debug : LogLevel.info, fields) {
+    __privateAdd(this, _structuredLog);
+    this.name = name;
+    this.level = level;
+    this.fields = fields;
+  }
+  child(fields, level) {
+    return new _SimpleStructuredLogger(this.name, level, {
+      ...this.fields,
+      ...fields
+    });
+  }
+  log(message, ...args) {
+    if (this.level < LogLevel.log)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.log, message, "log", ...args);
+  }
+  error(message, ...args) {
+    if (this.level < LogLevel.error)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.error, message, "error", ...args);
+  }
+  warn(message, ...args) {
+    if (this.level < LogLevel.warn)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.warn, message, "warn", ...args);
+  }
+  info(message, ...args) {
+    if (this.level < LogLevel.info)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.info, message, "info", ...args);
+  }
+  debug(message, ...args) {
+    if (this.level < LogLevel.debug)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.debug, message, "debug", ...args);
+  }
+};
+_structuredLog = new WeakSet();
+structuredLog_fn = /* @__PURE__ */ __name(function(loggerFunction, message, level, ...args) {
+  const structuredLog = {
+    ...args.length === 1 ? args[0] : args,
+    ...this.fields,
+    timestamp: /* @__PURE__ */ new Date(),
+    name: this.name,
+    message,
+    level
+  };
+  loggerFunction(JSON.stringify(structuredLog));
+}, "#structuredLog");
+__name(_SimpleStructuredLogger, "SimpleStructuredLogger");
+var SimpleStructuredLogger = _SimpleStructuredLogger;
+var messageSchema = zod.z.object({
+  version: zod.z.string(),
+  type: zod.z.string(),
+  payload: zod.z.unknown()
+});
+var _schema, _handlers;
+var _ZodSocketMessageHandler = class _ZodSocketMessageHandler {
+  constructor(options) {
+    __privateAdd(this, _schema, void 0);
+    __privateAdd(this, _handlers, void 0);
+    __privateSet(this, _schema, options.schema);
+    __privateSet(this, _handlers, options.handlers);
+  }
+  async handleMessage(message) {
+    const parsedMessage = this.parseMessage(message);
+    if (!__privateGet(this, _handlers)) {
+      throw new Error("No handlers provided");
+    }
+    const handler = __privateGet(this, _handlers)[parsedMessage.type];
+    if (!handler) {
+      console.error(`No handler for message type: ${String(parsedMessage.type)}`);
+      return;
+    }
+    const ack = await handler(parsedMessage.payload);
+    return ack;
+  }
+  parseMessage(message) {
+    const parsedMessage = messageSchema.safeParse(message);
+    if (!parsedMessage.success) {
+      throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);
+    }
+    const schema = __privateGet(this, _schema)[parsedMessage.data.type]["message"];
+    if (!schema) {
+      throw new Error(`Unknown message type: ${parsedMessage.data.type}`);
+    }
+    const messageWithVersion = {
+      version: parsedMessage.data.version,
+      ...typeof parsedMessage.data.payload === "object" ? parsedMessage.data.payload : {}
+    };
+    const parsedPayload = schema.safeParse(messageWithVersion);
+    if (!parsedPayload.success) {
+      console.error("Failed to parse message payload", {
+        message,
+        payload: messageWithVersion
+      });
+      throw parsedPayload.error instanceof zod.ZodError ? zodValidationError.fromZodError(parsedPayload.error) : parsedPayload.error;
+    }
+    return {
+      type: parsedMessage.data.type,
+      payload: parsedPayload.data
+    };
+  }
+  registerHandlers(emitter, logger) {
+    const log = logger ?? console;
+    if (!__privateGet(this, _handlers)) {
+      log.info("No handlers provided");
+      return;
+    }
+    for (const eventName of Object.keys(__privateGet(this, _handlers))) {
+      emitter.on(eventName, async (message, callback) => {
+        log.info(`handling ${eventName}`, {
+          payload: message,
+          hasCallback: !!callback
+        });
+        let ack;
+        try {
+          if ("payload" in message) {
+            ack = await this.handleMessage({
+              type: eventName,
+              ...message
+            });
+          } else {
+            const { version, ...payload } = message;
+            ack = await this.handleMessage({
+              type: eventName,
+              version,
+              payload
+            });
+          }
+        } catch (error) {
+          log.error("Error while handling message", {
+            error: error instanceof Error ? {
+              message: error.message,
+              stack: error.stack
+            } : error
+          });
+          return;
+        }
+        if (callback && typeof callback === "function") {
+          callback(ack);
+        }
+      });
+    }
+  }
+};
+_schema = new WeakMap();
+_handlers = new WeakMap();
+__name(_ZodSocketMessageHandler, "ZodSocketMessageHandler");
+var ZodSocketMessageHandler = _ZodSocketMessageHandler;
+var _schema2, _socket;
+var _ZodSocketMessageSender = class _ZodSocketMessageSender {
+  constructor(options) {
+    __privateAdd(this, _schema2, void 0);
+    __privateAdd(this, _socket, void 0);
+    __privateSet(this, _schema2, options.schema);
+    __privateSet(this, _socket, options.socket);
+  }
+  send(type, payload) {
+    const schema = __privateGet(this, _schema2)[type]["message"];
+    if (!schema) {
+      throw new Error(`Unknown message type: ${type}`);
+    }
+    const parsedPayload = schema.safeParse(payload);
+    if (!parsedPayload.success) {
+      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
+    }
+    __privateGet(this, _socket).emit(type, {
+      payload,
+      version: "v1"
+    });
+    return;
+  }
+  async sendWithAck(type, payload) {
+    const schema = __privateGet(this, _schema2)[type]["message"];
+    if (!schema) {
+      throw new Error(`Unknown message type: ${type}`);
+    }
+    const parsedPayload = schema.safeParse(payload);
+    if (!parsedPayload.success) {
+      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
+    }
+    const callbackResult = await __privateGet(this, _socket).emitWithAck(type, {
+      payload,
+      version: "v1"
+    });
+    return callbackResult;
+  }
+};
+_schema2 = new WeakMap();
+_socket = new WeakMap();
+__name(_ZodSocketMessageSender, "ZodSocketMessageSender");
+var ZodSocketMessageSender = _ZodSocketMessageSender;
+var _sender, _handler, _logger;
+var _ZodSocketConnection = class _ZodSocketConnection {
+  constructor(opts) {
+    __privateAdd(this, _sender, void 0);
+    __privateAdd(this, _handler, void 0);
+    __privateAdd(this, _logger, void 0);
+    const uri = `${opts.secure ? "wss" : "ws"}://${opts.host}:${opts.port ?? (opts.secure ? "443" : "80")}/${opts.namespace}`;
+    const logger = new SimpleStructuredLogger(opts.namespace, LogLevel.info);
+    logger.log("new zod socket", {
+      uri
+    });
+    this.socket = socket_ioClient.io(uri, {
+      transports: [
+        "websocket"
+      ],
+      auth: {
+        token: opts.authToken
+      },
+      extraHeaders: opts.extraHeaders,
+      reconnectionDelay: 500,
+      reconnectionDelayMax: 1e3
+    });
+    __privateSet(this, _logger, logger.child({
+      socketId: this.socket.id
+    }));
+    __privateSet(this, _handler, new ZodSocketMessageHandler({
+      schema: opts.serverMessages,
+      handlers: opts.handlers
+    }));
+    __privateGet(this, _handler).registerHandlers(this.socket, __privateGet(this, _logger));
+    __privateSet(this, _sender, new ZodSocketMessageSender({
+      schema: opts.clientMessages,
+      socket: this.socket
+    }));
+    this.socket.on("connect_error", async (error) => {
+      __privateGet(this, _logger).error(`connect_error: ${error}`);
+      if (opts.onError) {
+        await opts.onError(this.socket, error, __privateGet(this, _logger));
+      }
+    });
+    this.socket.on("connect", async () => {
+      __privateGet(this, _logger).info("connect");
+      if (opts.onConnection) {
+        await opts.onConnection(this.socket, __privateGet(this, _handler), __privateGet(this, _sender), __privateGet(this, _logger));
+      }
+    });
+    this.socket.on("disconnect", async (reason, description) => {
+      __privateGet(this, _logger).info("disconnect", {
+        reason,
+        description
+      });
+      if (opts.onDisconnect) {
+        await opts.onDisconnect(this.socket, reason, description, __privateGet(this, _logger));
+      }
+    });
+  }
+  close() {
+    this.socket.close();
+  }
+  connect() {
+    this.socket.connect();
+  }
+  get send() {
+    return __privateGet(this, _sender).send.bind(__privateGet(this, _sender));
+  }
+  get sendWithAck() {
+    return __privateGet(this, _sender).sendWithAck.bind(__privateGet(this, _sender));
+  }
+};
+_sender = new WeakMap();
+_handler = new WeakMap();
+_logger = new WeakMap();
+__name(_ZodSocketConnection, "ZodSocketConnection");
+var ZodSocketConnection = _ZodSocketConnection;
+
+exports.ZodSocketConnection = ZodSocketConnection;
+exports.ZodSocketMessageHandler = ZodSocketMessageHandler;
+exports.ZodSocketMessageSender = ZodSocketMessageSender;
+//# sourceMappingURL=out.js.map
+//# sourceMappingURL=zodSocket.js.map

package/dist/v3/zodSocket.js.map

@@ -0,0 +1 @@
+{"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

@@ -0,0 +1,315 @@
+import { io } from 'socket.io-client';
+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 });
+var __accessCheck = (obj, member, msg) => {
+  if (!member.has(obj))
+    throw TypeError("Cannot " + msg);
+};
+var __privateGet = (obj, member, getter) => {
+  __accessCheck(obj, member, "read from private field");
+  return getter ? getter.call(obj) : member.get(obj);
+};
+var __privateAdd = (obj, member, value) => {
+  if (member.has(obj))
+    throw TypeError("Cannot add the same private member more than once");
+  member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
+};
+var __privateSet = (obj, member, value, setter) => {
+  __accessCheck(obj, member, "write to private field");
+  setter ? setter.call(obj, value) : member.set(obj, value);
+  return value;
+};
+var __privateMethod = (obj, member, method) => {
+  __accessCheck(obj, member, "access private method");
+  return method;
+};
+
+// src/v3/utils/structuredLogger.ts
+var LogLevel;
+(function(LogLevel2) {
+  LogLevel2[LogLevel2["log"] = 0] = "log";
+  LogLevel2[LogLevel2["error"] = 1] = "error";
+  LogLevel2[LogLevel2["warn"] = 2] = "warn";
+  LogLevel2[LogLevel2["info"] = 3] = "info";
+  LogLevel2[LogLevel2["debug"] = 4] = "debug";
+})(LogLevel || (LogLevel = {}));
+var _structuredLog, structuredLog_fn;
+var _SimpleStructuredLogger = class _SimpleStructuredLogger {
+  constructor(name, level = [
+    "1",
+    "true"
+  ].includes(process.env.DEBUG ?? "") ? LogLevel.debug : LogLevel.info, fields) {
+    __privateAdd(this, _structuredLog);
+    this.name = name;
+    this.level = level;
+    this.fields = fields;
+  }
+  child(fields, level) {
+    return new _SimpleStructuredLogger(this.name, level, {
+      ...this.fields,
+      ...fields
+    });
+  }
+  log(message, ...args) {
+    if (this.level < LogLevel.log)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.log, message, "log", ...args);
+  }
+  error(message, ...args) {
+    if (this.level < LogLevel.error)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.error, message, "error", ...args);
+  }
+  warn(message, ...args) {
+    if (this.level < LogLevel.warn)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.warn, message, "warn", ...args);
+  }
+  info(message, ...args) {
+    if (this.level < LogLevel.info)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.info, message, "info", ...args);
+  }
+  debug(message, ...args) {
+    if (this.level < LogLevel.debug)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.debug, message, "debug", ...args);
+  }
+};
+_structuredLog = new WeakSet();
+structuredLog_fn = /* @__PURE__ */ __name(function(loggerFunction, message, level, ...args) {
+  const structuredLog = {
+    ...args.length === 1 ? args[0] : args,
+    ...this.fields,
+    timestamp: /* @__PURE__ */ new Date(),
+    name: this.name,
+    message,
+    level
+  };
+  loggerFunction(JSON.stringify(structuredLog));
+}, "#structuredLog");
+__name(_SimpleStructuredLogger, "SimpleStructuredLogger");
+var SimpleStructuredLogger = _SimpleStructuredLogger;
+var messageSchema = z.object({
+  version: z.string(),
+  type: z.string(),
+  payload: z.unknown()
+});
+var _schema, _handlers;
+var _ZodSocketMessageHandler = class _ZodSocketMessageHandler {
+  constructor(options) {
+    __privateAdd(this, _schema, void 0);
+    __privateAdd(this, _handlers, void 0);
+    __privateSet(this, _schema, options.schema);
+    __privateSet(this, _handlers, options.handlers);
+  }
+  async handleMessage(message) {
+    const parsedMessage = this.parseMessage(message);
+    if (!__privateGet(this, _handlers)) {
+      throw new Error("No handlers provided");
+    }
+    const handler = __privateGet(this, _handlers)[parsedMessage.type];
+    if (!handler) {
+      console.error(`No handler for message type: ${String(parsedMessage.type)}`);
+      return;
+    }
+    const ack = await handler(parsedMessage.payload);
+    return ack;
+  }
+  parseMessage(message) {
+    const parsedMessage = messageSchema.safeParse(message);
+    if (!parsedMessage.success) {
+      throw new Error(`Failed to parse message: ${JSON.stringify(parsedMessage.error)}`);
+    }
+    const schema = __privateGet(this, _schema)[parsedMessage.data.type]["message"];
+    if (!schema) {
+      throw new Error(`Unknown message type: ${parsedMessage.data.type}`);
+    }
+    const messageWithVersion = {
+      version: parsedMessage.data.version,
+      ...typeof parsedMessage.data.payload === "object" ? parsedMessage.data.payload : {}
+    };
+    const parsedPayload = schema.safeParse(messageWithVersion);
+    if (!parsedPayload.success) {
+      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,
+      payload: parsedPayload.data
+    };
+  }
+  registerHandlers(emitter, logger) {
+    const log = logger ?? console;
+    if (!__privateGet(this, _handlers)) {
+      log.info("No handlers provided");
+      return;
+    }
+    for (const eventName of Object.keys(__privateGet(this, _handlers))) {
+      emitter.on(eventName, async (message, callback) => {
+        log.info(`handling ${eventName}`, {
+          payload: message,
+          hasCallback: !!callback
+        });
+        let ack;
+        try {
+          if ("payload" in message) {
+            ack = await this.handleMessage({
+              type: eventName,
+              ...message
+            });
+          } else {
+            const { version, ...payload } = message;
+            ack = await this.handleMessage({
+              type: eventName,
+              version,
+              payload
+            });
+          }
+        } catch (error) {
+          log.error("Error while handling message", {
+            error: error instanceof Error ? {
+              message: error.message,
+              stack: error.stack
+            } : error
+          });
+          return;
+        }
+        if (callback && typeof callback === "function") {
+          callback(ack);
+        }
+      });
+    }
+  }
+};
+_schema = new WeakMap();
+_handlers = new WeakMap();
+__name(_ZodSocketMessageHandler, "ZodSocketMessageHandler");
+var ZodSocketMessageHandler = _ZodSocketMessageHandler;
+var _schema2, _socket;
+var _ZodSocketMessageSender = class _ZodSocketMessageSender {
+  constructor(options) {
+    __privateAdd(this, _schema2, void 0);
+    __privateAdd(this, _socket, void 0);
+    __privateSet(this, _schema2, options.schema);
+    __privateSet(this, _socket, options.socket);
+  }
+  send(type, payload) {
+    const schema = __privateGet(this, _schema2)[type]["message"];
+    if (!schema) {
+      throw new Error(`Unknown message type: ${type}`);
+    }
+    const parsedPayload = schema.safeParse(payload);
+    if (!parsedPayload.success) {
+      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
+    }
+    __privateGet(this, _socket).emit(type, {
+      payload,
+      version: "v1"
+    });
+    return;
+  }
+  async sendWithAck(type, payload) {
+    const schema = __privateGet(this, _schema2)[type]["message"];
+    if (!schema) {
+      throw new Error(`Unknown message type: ${type}`);
+    }
+    const parsedPayload = schema.safeParse(payload);
+    if (!parsedPayload.success) {
+      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
+    }
+    const callbackResult = await __privateGet(this, _socket).emitWithAck(type, {
+      payload,
+      version: "v1"
+    });
+    return callbackResult;
+  }
+};
+_schema2 = new WeakMap();
+_socket = new WeakMap();
+__name(_ZodSocketMessageSender, "ZodSocketMessageSender");
+var ZodSocketMessageSender = _ZodSocketMessageSender;
+var _sender, _handler, _logger;
+var _ZodSocketConnection = class _ZodSocketConnection {
+  constructor(opts) {
+    __privateAdd(this, _sender, void 0);
+    __privateAdd(this, _handler, void 0);
+    __privateAdd(this, _logger, void 0);
+    const uri = `${opts.secure ? "wss" : "ws"}://${opts.host}:${opts.port ?? (opts.secure ? "443" : "80")}/${opts.namespace}`;
+    const logger = new SimpleStructuredLogger(opts.namespace, LogLevel.info);
+    logger.log("new zod socket", {
+      uri
+    });
+    this.socket = io(uri, {
+      transports: [
+        "websocket"
+      ],
+      auth: {
+        token: opts.authToken
+      },
+      extraHeaders: opts.extraHeaders,
+      reconnectionDelay: 500,
+      reconnectionDelayMax: 1e3
+    });
+    __privateSet(this, _logger, logger.child({
+      socketId: this.socket.id
+    }));
+    __privateSet(this, _handler, new ZodSocketMessageHandler({
+      schema: opts.serverMessages,
+      handlers: opts.handlers
+    }));
+    __privateGet(this, _handler).registerHandlers(this.socket, __privateGet(this, _logger));
+    __privateSet(this, _sender, new ZodSocketMessageSender({
+      schema: opts.clientMessages,
+      socket: this.socket
+    }));
+    this.socket.on("connect_error", async (error) => {
+      __privateGet(this, _logger).error(`connect_error: ${error}`);
+      if (opts.onError) {
+        await opts.onError(this.socket, error, __privateGet(this, _logger));
+      }
+    });
+    this.socket.on("connect", async () => {
+      __privateGet(this, _logger).info("connect");
+      if (opts.onConnection) {
+        await opts.onConnection(this.socket, __privateGet(this, _handler), __privateGet(this, _sender), __privateGet(this, _logger));
+      }
+    });
+    this.socket.on("disconnect", async (reason, description) => {
+      __privateGet(this, _logger).info("disconnect", {
+        reason,
+        description
+      });
+      if (opts.onDisconnect) {
+        await opts.onDisconnect(this.socket, reason, description, __privateGet(this, _logger));
+      }
+    });
+  }
+  close() {
+    this.socket.close();
+  }
+  connect() {
+    this.socket.connect();
+  }
+  get send() {
+    return __privateGet(this, _sender).send.bind(__privateGet(this, _sender));
+  }
+  get sendWithAck() {
+    return __privateGet(this, _sender).sendWithAck.bind(__privateGet(this, _sender));
+  }
+};
+_sender = new WeakMap();
+_handler = new WeakMap();
+_logger = new WeakMap();
+__name(_ZodSocketConnection, "ZodSocketConnection");
+var ZodSocketConnection = _ZodSocketConnection;
+
+export { ZodSocketConnection, ZodSocketMessageHandler, ZodSocketMessageSender };
+//# sourceMappingURL=out.js.map
+//# sourceMappingURL=zodSocket.mjs.map