@replit/river 0.10.0 → 0.10.1

package/dist/chunk-AJQU4AZG.js

@@ -0,0 +1,284 @@
+import {
+  OpaqueTransportMessageSchema,
+  TransportAckSchema,
+  isAck,
+  reply
+} from "./chunk-ZE4MX7DF.js";
+import {
+  log
+} from "./chunk-SLUSVGQH.js";
+
+// transport/transport.ts
+import { Value } from "@sinclair/typebox/value";
+
+// transport/events.ts
+var EventDispatcher = class {
+  eventListeners = {};
+  numberOfListeners(eventType) {
+    return this.eventListeners[eventType]?.size ?? 0;
+  }
+  addEventListener(eventType, handler) {
+    if (!this.eventListeners[eventType]) {
+      this.eventListeners[eventType] = /* @__PURE__ */ new Set();
+    }
+    this.eventListeners[eventType]?.add(handler);
+  }
+  removeEventListener(eventType, handler) {
+    const handlers = this.eventListeners[eventType];
+    if (handlers) {
+      this.eventListeners[eventType]?.delete(handler);
+    }
+  }
+  dispatchEvent(eventType, event) {
+    const handlers = this.eventListeners[eventType];
+    if (handlers) {
+      for (const handler of handlers) {
+        handler(event);
+      }
+    }
+  }
+};
+
+// transport/transport.ts
+var Connection = class {
+  connectedTo;
+  transport;
+  constructor(transport, connectedTo) {
+    this.connectedTo = connectedTo;
+    this.transport = transport;
+  }
+};
+var Transport = class {
+  /**
+   * A flag indicating whether the transport has been destroyed.
+   * A destroyed transport will not attempt to reconnect and cannot be used again.
+   */
+  state;
+  /**
+   * The {@link Codec} used to encode and decode messages.
+   */
+  codec;
+  /**
+   * The client ID of this transport.
+   */
+  clientId;
+  /**
+   * An array of message IDs that are waiting to be sent over the WebSocket connection.
+   * This builds up if the WebSocket is down for a period of time.
+   */
+  sendQueue;
+  /**
+   * The buffer of messages that have been sent but not yet acknowledged.
+   */
+  sendBuffer;
+  /**
+   * The map of {@link Connection}s managed by this transport.
+   */
+  connections;
+  /**
+   * The event dispatcher for handling events of type EventTypes.
+   */
+  eventDispatcher;
+  /**
+   * Creates a new Transport instance.
+   * @param codec The codec used to encode and decode messages.
+   * @param clientId The client ID of this transport.
+   */
+  constructor(codec, clientId) {
+    this.eventDispatcher = new EventDispatcher();
+    this.sendBuffer = /* @__PURE__ */ new Map();
+    this.sendQueue = /* @__PURE__ */ new Map();
+    this.connections = /* @__PURE__ */ new Map();
+    this.codec = codec;
+    this.clientId = clientId;
+    this.state = "open";
+  }
+  /**
+   * The downstream implementation needs to call this when a new connection is established.
+   * @param conn The connection object.
+   */
+  onConnect(conn) {
+    log?.info(`${this.clientId} -- new connection to ${conn.connectedTo}`);
+    this.connections.set(conn.connectedTo, conn);
+    this.eventDispatcher.dispatchEvent("connectionStatus", {
+      status: "connect",
+      conn
+    });
+    const outstanding = this.sendQueue.get(conn.connectedTo);
+    if (!outstanding) {
+      return;
+    }
+    for (const id of outstanding) {
+      const msg = this.sendBuffer.get(id);
+      if (!msg) {
+        log?.warn(
+          `${this.clientId} -- tried to resend a message we received an ack for`
+        );
+        continue;
+      }
+      this.send(msg);
+    }
+    this.sendQueue.delete(conn.connectedTo);
+  }
+  /**
+   * The downstream implementation needs to call this when a connection is closed.
+   * @param conn The connection object.
+   */
+  onDisconnect(conn) {
+    log?.info(`${this.clientId} -- disconnect from ${conn.connectedTo}`);
+    conn.close();
+    this.connections.delete(conn.connectedTo);
+    this.eventDispatcher.dispatchEvent("connectionStatus", {
+      status: "disconnect",
+      conn
+    });
+  }
+  /**
+   * Handles a message received by this transport. Thin wrapper around {@link handleMsg} and {@link parseMsg}.
+   * @param msg The message to handle.
+   */
+  onMessage(msg) {
+    return this.handleMsg(this.parseMsg(msg));
+  }
+  /**
+   * Parses a message from a Uint8Array into a {@link OpaqueTransportMessage}.
+   * @param msg The message to parse.
+   * @returns The parsed message, or null if the message is malformed or invalid.
+   */
+  parseMsg(msg) {
+    const parsedMsg = this.codec.fromBuffer(msg);
+    if (parsedMsg === null) {
+      const decodedBuffer = new TextDecoder().decode(msg);
+      log?.warn(`${this.clientId} -- received malformed msg: ${decodedBuffer}`);
+      return null;
+    }
+    if (Value.Check(OpaqueTransportMessageSchema, parsedMsg)) {
+      return {
+        ...parsedMsg,
+        serviceName: parsedMsg.serviceName === null ? void 0 : parsedMsg.serviceName,
+        procedureName: parsedMsg.procedureName === null ? void 0 : parsedMsg.procedureName
+      };
+    } else {
+      log?.warn(
+        `${this.clientId} -- received invalid msg: ${JSON.stringify(
+          parsedMsg
+        )}`
+      );
+      return null;
+    }
+  }
+  /**
+   * Called when a message is received by this transport.
+   * You generally shouldn't need to override this in downstream transport implementations.
+   * @param msg The received message.
+   */
+  handleMsg(msg) {
+    if (!msg) {
+      return;
+    }
+    if (isAck(msg.controlFlags) && Value.Check(TransportAckSchema, msg)) {
+      log?.info(`${this.clientId} -- received ack: ${JSON.stringify(msg)}`);
+      if (this.sendBuffer.has(msg.payload.ack)) {
+        this.sendBuffer.delete(msg.payload.ack);
+      }
+    } else {
+      log?.info(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
+      if (msg.to !== this.clientId) {
+        return;
+      }
+      this.eventDispatcher.dispatchEvent("message", msg);
+      if (!isAck(msg.controlFlags)) {
+        const ackMsg = reply(msg, { ack: msg.id });
+        ackMsg.controlFlags = 1 /* AckBit */;
+        ackMsg.from = this.clientId;
+        this.send(ackMsg);
+      }
+    }
+  }
+  /**
+   * Adds a listener to this transport.
+   * @param the type of event to listen for
+   * @param handler The message handler to add.
+   */
+  addEventListener(type, handler) {
+    this.eventDispatcher.addEventListener(type, handler);
+  }
+  /**
+   * Removes a listener from this transport.
+   * @param the type of event to unlisten on
+   * @param handler The message handler to remove.
+   */
+  removeEventListener(type, handler) {
+    this.eventDispatcher.removeEventListener(type, handler);
+  }
+  /**
+   * Sends a message over this transport, delegating to the appropriate connection to actually
+   * send the message.
+   * @param msg The message to send.
+   * @returns The ID of the sent message.
+   */
+  send(msg) {
+    if (this.state === "destroyed") {
+      const err = "transport is destroyed, cant send";
+      log?.error(`${this.clientId} -- ` + err + `: ${JSON.stringify(msg)}`);
+      throw new Error(err);
+    } else if (this.state === "closed") {
+      log?.info(
+        `${this.clientId} -- transport closed when sending, discarding : ${JSON.stringify(
+          msg
+        )}`
+      );
+      return msg.id;
+    }
+    let conn = this.connections.get(msg.to);
+    if (!isAck(msg.controlFlags)) {
+      this.sendBuffer.set(msg.id, msg);
+    }
+    if (conn) {
+      log?.info(`${this.clientId} -- sending ${JSON.stringify(msg)}`);
+      const ok = conn.send(this.codec.toBuffer(msg));
+      if (ok) {
+        return msg.id;
+      }
+    }
+    log?.info(
+      `${this.clientId} -- connection to ${msg.to} not ready, attempting reconnect and queuing ${JSON.stringify(msg)}`
+    );
+    const outstanding = this.sendQueue.get(msg.to) || [];
+    outstanding.push(msg.id);
+    this.sendQueue.set(msg.to, outstanding);
+    this.createNewConnection(msg.to);
+    return msg.id;
+  }
+  /**
+   * Default close implementation for transports. You should override this in the downstream
+   * implementation if you need to do any additional cleanup and call super.close() at the end.
+   * Closes the transport. Any messages sent while the transport is closed will be silently discarded.
+   */
+  async close() {
+    for (const conn of this.connections.values()) {
+      conn.close();
+    }
+    this.connections.clear();
+    this.state = "closed";
+    log?.info(`${this.clientId} -- closed transport`);
+  }
+  /**
+   * Default destroy implementation for transports. You should override this in the downstream
+   * implementation if you need to do any additional cleanup and call super.destroy() at the end.
+   * Destroys the transport. Any messages sent while the transport is destroyed will throw an error.
+   */
+  async destroy() {
+    for (const conn of this.connections.values()) {
+      conn.close();
+    }
+    this.connections.clear();
+    this.state = "destroyed";
+    log?.info(`${this.clientId} -- destroyed transport`);
+  }
+};
+
+export {
+  Connection,
+  Transport
+};

package/dist/chunk-PC65ZFWJ.js

@@ -0,0 +1,29 @@
+import {
+  Connection
+} from "./chunk-AJQU4AZG.js";
+
+// transport/impls/ws/connection.ts
+var WebSocketConnection = class extends Connection {
+  ws;
+  constructor(transport, connectedTo, ws) {
+    super(transport, connectedTo);
+    this.ws = ws;
+    ws.binaryType = "arraybuffer";
+    this.ws.onmessage = (msg) => transport.onMessage(msg.data);
+  }
+  send(payload) {
+    if (this.ws.readyState === this.ws.OPEN) {
+      this.ws.send(payload);
+      return true;
+    } else {
+      return false;
+    }
+  }
+  async close() {
+    this.ws.close();
+  }
+};
+
+export {
+  WebSocketConnection
+};

package/dist/chunk-R6H2BIMC.js

@@ -0,0 +1,49 @@
+// codec/json.ts
+var encoder = new TextEncoder();
+var decoder = new TextDecoder();
+function uint8ArrayToBase64(uint8Array) {
+  let binary = "";
+  uint8Array.forEach((byte) => {
+    binary += String.fromCharCode(byte);
+  });
+  return btoa(binary);
+}
+function base64ToUint8Array(base64) {
+  const binaryString = atob(base64);
+  const uint8Array = new Uint8Array(binaryString.length);
+  for (let i = 0; i < binaryString.length; i++) {
+    uint8Array[i] = binaryString.charCodeAt(i);
+  }
+  return uint8Array;
+}
+var NaiveJsonCodec = {
+  toBuffer: (obj) => {
+    return encoder.encode(
+      JSON.stringify(obj, function replacer(key) {
+        let val = this[key];
+        if (val instanceof Uint8Array) {
+          return { $t: uint8ArrayToBase64(val) };
+        } else {
+          return val;
+        }
+      })
+    );
+  },
+  fromBuffer: (buff) => {
+    try {
+      return JSON.parse(decoder.decode(buff), function reviver(_key, val) {
+        if (val?.$t) {
+          return base64ToUint8Array(val.$t);
+        } else {
+          return val;
+        }
+      });
+    } catch {
+      return null;
+    }
+  }
+};
+
+export {
+  NaiveJsonCodec
+};

package/dist/chunk-RGMHF6PF.js

@@ -0,0 +1,65 @@
+import {
+  WebSocketConnection
+} from "./chunk-PC65ZFWJ.js";
+import {
+  NaiveJsonCodec
+} from "./chunk-R6H2BIMC.js";
+import {
+  Transport
+} from "./chunk-AJQU4AZG.js";
+import {
+  log
+} from "./chunk-SLUSVGQH.js";
+
+// transport/impls/ws/server.ts
+var defaultOptions = {
+  codec: NaiveJsonCodec
+};
+var WebSocketServerTransport = class extends Transport {
+  wss;
+  clientId;
+  constructor(wss, clientId, providedOptions) {
+    const options = { ...defaultOptions, ...providedOptions };
+    super(options.codec, clientId);
+    this.wss = wss;
+    this.clientId = clientId;
+    this.setupConnectionStatusListeners();
+  }
+  connectionHandler = (ws) => {
+    let conn = void 0;
+    ws.onmessage = (msg) => {
+      const parsedMsg = this.parseMsg(msg.data);
+      if (parsedMsg && !conn) {
+        conn = new WebSocketConnection(this, parsedMsg.from, ws);
+        this.onConnect(conn);
+        this.handleMsg(parsedMsg);
+      }
+    };
+    ws.onclose = () => {
+      if (conn) {
+        this.onDisconnect(conn);
+      }
+    };
+    ws.onerror = (msg) => {
+      log?.warn(
+        `${this.clientId} -- ws error from client ${conn?.connectedTo ?? "unknown"}: ${msg}`
+      );
+    };
+  };
+  setupConnectionStatusListeners() {
+    this.wss.on("connection", this.connectionHandler);
+  }
+  async createNewConnection(to) {
+    const err = `${this.clientId} -- failed to send msg to ${to}, client probably dropped`;
+    log?.warn(err);
+    return;
+  }
+  async close() {
+    super.close();
+    this.wss.off("connection", this.connectionHandler);
+  }
+};
+
+export {
+  WebSocketServerTransport
+};

package/dist/chunk-SLUSVGQH.js

@@ -0,0 +1,30 @@
+// logging/index.ts
+var LoggingLevels = {
+  info: 0,
+  warn: 1,
+  error: 2
+};
+var log;
+var defaultLoggingLevel = "warn";
+function bindLogger(write, color) {
+  const info = color ? "\x1B[37minfo\x1B[0m" : "info";
+  const warn = color ? "\x1B[33mwarn\x1B[0m" : "warn";
+  const error = color ? "\x1B[31merr\x1B[0m" : "err";
+  log = {
+    info: (msg) => log && LoggingLevels[log.minLevel] <= 0 && write(`[river:${info}] ${msg}`),
+    warn: (msg) => log && LoggingLevels[log.minLevel] <= 1 && write(`[river:${warn}] ${msg}`),
+    error: (msg) => log && LoggingLevels[log.minLevel] <= 2 && write(`[river:${error}] ${msg}`),
+    minLevel: log?.minLevel ?? defaultLoggingLevel
+  };
+}
+function setLevel(level) {
+  if (log) {
+    log.minLevel = level;
+  }
+}
+
+export {
+  log,
+  bindLogger,
+  setLevel
+};

package/dist/chunk-UU2Z7LDR.js

@@ -0,0 +1,113 @@
+import {
+  WebSocketConnection
+} from "./chunk-PC65ZFWJ.js";
+import {
+  NaiveJsonCodec
+} from "./chunk-R6H2BIMC.js";
+import {
+  Transport
+} from "./chunk-AJQU4AZG.js";
+import {
+  log
+} from "./chunk-SLUSVGQH.js";
+
+// transport/impls/ws/client.ts
+var defaultOptions = {
+  retryIntervalMs: 250,
+  retryAttemptsMax: 5,
+  codec: NaiveJsonCodec
+};
+var WebSocketClientTransport = class extends Transport {
+  /**
+   * A function that returns a Promise that resolves to a WebSocket instance.
+   */
+  wsGetter;
+  options;
+  serverId;
+  reconnectPromises;
+  tryReconnecting = true;
+  /**
+   * Creates a new WebSocketTransport instance.
+   * @param wsGetter A function that returns a Promise that resolves to a WebSocket instance.
+   * @param clientId The ID of the client using the transport.
+   * @param providedOptions An optional object containing configuration options for the transport.
+   */
+  constructor(wsGetter, clientId, serverId, providedOptions) {
+    const options = { ...defaultOptions, ...providedOptions };
+    super(options.codec, clientId);
+    this.wsGetter = wsGetter;
+    this.serverId = serverId;
+    this.options = options;
+    this.reconnectPromises = /* @__PURE__ */ new Map();
+    this.setupConnectionStatusListeners();
+  }
+  setupConnectionStatusListeners() {
+    this.createNewConnection(this.serverId);
+  }
+  async createNewConnection(to, attempt = 0) {
+    if (this.state === "destroyed") {
+      throw new Error("cant reopen a destroyed connection");
+    }
+    let reconnectPromise = this.reconnectPromises.get(to);
+    if (!reconnectPromise) {
+      if (!this.tryReconnecting) {
+        log?.info(
+          `${this.clientId} -- tryReconnecting is false, not attempting reconnect`
+        );
+        return;
+      }
+      reconnectPromise = new Promise(async (resolve) => {
+        log?.info(`${this.clientId} -- establishing a new websocket to ${to}`);
+        const ws = await this.wsGetter(to);
+        if (ws.readyState === ws.OPEN) {
+          return resolve({ ws });
+        }
+        if (ws.readyState === ws.CLOSING || ws.readyState === ws.CLOSED) {
+          return resolve({ err: "ws is closing or closed" });
+        }
+        ws.addEventListener("open", function onOpen() {
+          ws.removeEventListener("open", onOpen);
+          resolve({ ws });
+        });
+        ws.addEventListener("close", function onClose(evt) {
+          ws.removeEventListener("close", onClose);
+          resolve({ err: evt.reason });
+        });
+      });
+      this.reconnectPromises.set(to, reconnectPromise);
+    }
+    const res = await reconnectPromise;
+    if ("ws" in res && res.ws.readyState === res.ws.OPEN) {
+      if (res.ws === this.connections.get(to)?.ws) {
+        return;
+      }
+      log?.info(`${this.clientId} -- websocket ok`);
+      const conn = new WebSocketConnection(this, to, res.ws);
+      this.onConnect(conn);
+      res.ws.onclose = () => {
+        this.reconnectPromises.delete(to);
+        this.onDisconnect(conn);
+      };
+      this.state = "open";
+      return;
+    }
+    this.reconnectPromises.delete(to);
+    if (attempt >= this.options.retryAttemptsMax) {
+      throw new Error(
+        `${this.clientId} -- websocket to ${to} failed after ${attempt} attempts, giving up`
+      );
+    } else {
+      log?.warn(
+        `${this.clientId} -- websocket to ${to} failed, trying again in ${this.options.retryIntervalMs * attempt}ms`
+      );
+      setTimeout(
+        () => this.createNewConnection(to, attempt + 1),
+        this.options.retryIntervalMs * attempt
+      );
+    }
+  }
+};
+
+export {
+  WebSocketClientTransport
+};

package/dist/chunk-WVT5QXMZ.js

@@ -0,0 +1,20 @@
+// codec/binary.ts
+import { decode, encode } from "@msgpack/msgpack";
+var BinaryCodec = {
+  toBuffer: encode,
+  fromBuffer: (buff) => {
+    try {
+      const res = decode(buff);
+      if (typeof res !== "object") {
+        return null;
+      }
+      return res;
+    } catch {
+      return null;
+    }
+  }
+};
+
+export {
+  BinaryCodec
+};

package/dist/chunk-ZE4MX7DF.js

@@ -0,0 +1,75 @@
+// transport/message.ts
+import { Type } from "@sinclair/typebox";
+import { nanoid } from "nanoid";
+var TransportMessageSchema = (t) => Type.Object({
+  id: Type.String(),
+  from: Type.String(),
+  to: Type.String(),
+  serviceName: Type.Optional(Type.Union([Type.String(), Type.Null()])),
+  procedureName: Type.Optional(Type.Union([Type.String(), Type.Null()])),
+  streamId: Type.String(),
+  controlFlags: Type.Integer(),
+  payload: t
+});
+var TransportAckSchema = TransportMessageSchema(
+  Type.Object({
+    ack: Type.String()
+  })
+);
+var ControlMessagePayloadSchema = Type.Object({
+  type: Type.Literal("CLOSE")
+});
+var OpaqueTransportMessageSchema = TransportMessageSchema(
+  Type.Unknown()
+);
+function msg(from, to, streamId, payload, serviceName, procedureName) {
+  return {
+    id: nanoid(),
+    to,
+    from,
+    serviceName,
+    procedureName,
+    streamId,
+    controlFlags: 0,
+    payload
+  };
+}
+function reply(msg2, response) {
+  return {
+    id: nanoid(),
+    streamId: msg2.streamId,
+    controlFlags: 0,
+    to: msg2.from,
+    from: msg2.to,
+    payload: response
+  };
+}
+function closeStream(from, to, stream) {
+  const closeMessage = msg(from, to, stream, {
+    type: "CLOSE"
+  });
+  closeMessage.controlFlags |= 4 /* StreamClosedBit */;
+  return closeMessage;
+}
+function isAck(controlFlag) {
+  return (controlFlag & 1 /* AckBit */) === 1 /* AckBit */;
+}
+function isStreamOpen(controlFlag) {
+  return (controlFlag & 2 /* StreamOpenBit */) === 2 /* StreamOpenBit */;
+}
+function isStreamClose(controlFlag) {
+  return (controlFlag & 4 /* StreamClosedBit */) === 4 /* StreamClosedBit */;
+}
+
+export {
+  TransportMessageSchema,
+  TransportAckSchema,
+  ControlMessagePayloadSchema,
+  OpaqueTransportMessageSchema,
+  msg,
+  reply,
+  closeStream,
+  isAck,
+  isStreamOpen,
+  isStreamClose
+};