y-partyserver 0.0.45 → 0.0.47

package/dist/{chunk-TD2NUND2.js → chunk-H4CTOPWX.js}

@@ -1,6 +1,6 @@
 import {
   sendChunked
-} from "./chunk-JFXBVIOX.js";
+} from "./chunk-W2PLHZNF.js";
 
 // src/provider/index.ts
 import * as bc from "lib0/broadcastchannel";
@@ -465,4 +465,4 @@ export {
   WebsocketProvider,
   YProvider
 };
-//# sourceMappingURL=chunk-TD2NUND2.js.map
+//# sourceMappingURL=chunk-H4CTOPWX.js.map

package/dist/{chunk-JFXBVIOX.js → chunk-W2PLHZNF.js}

@@ -123,4 +123,4 @@ export {
   sendChunked,
   handleChunked
 };
-//# sourceMappingURL=chunk-JFXBVIOX.js.map
+//# sourceMappingURL=chunk-W2PLHZNF.js.map

package/dist/chunk-W2PLHZNF.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/shared/chunking.ts"],"sourcesContent":["// This file contains a shared implementation of chunking logic for binary\n// WebSocket messages. Because the Workers platform limits individual\n// WebSocket messages to 1MB, we need to split larger messages into chunks.\n\nimport type { Connection, WSMessage } from \"partyserver\";\n\nexport const CHUNK_MAX_SIZE = 1_000_000;\n\nconst BATCH_SENTINEL = \"y-pk-batch\";\nconst TRACE_ENABLED = false;\nconst trace = (...args: unknown[]) => TRACE_ENABLED && console.log(...args);\n\nlet warnedAboutLargeMessage = false;\n\ntype MessageHandler = (conn: Connection, message: WSMessage) => void;\ntype Batch = {\n  id: string;\n  type: \"start\" | \"end\";\n  size: number;\n  count: number;\n};\n\n/**\n * Takes an ArrayBuffer and sends it in chunks to the provided send function,\n * along with a start and end marker.\n *\n * 1. The sender splits the message into chunks of 1MB or less.\n * 2. The sender sends a chunk start marker\n * 3. The sender sends each chunk as a individual packet\n * 4. The sender sends a chunk end marker\n */\nexport const sendChunked = (data: Uint8Array, ws: WebSocket) => {\n  if (data.byteLength <= CHUNK_MAX_SIZE) {\n    ws.send(data);\n    return;\n  }\n\n  if (!warnedAboutLargeMessage) {\n    console.warn(\n      \"[y-partyserver]\",\n      \"The Y.js update size exceeds 1MB, which is the maximum size for an individual update. The update will be split into chunks. This is an experimental feature.\",\n      `Message size: ${(data.byteLength / 1000 / 1000).toFixed(1)}MB`\n    );\n    warnedAboutLargeMessage = true;\n  }\n\n  const id = (Date.now() + Math.random()).toString(36).substring(10);\n  const chunks = Math.ceil(data.byteLength / CHUNK_MAX_SIZE);\n\n  ws.send(\n    serializeBatchMarker({\n      id,\n      type: \"start\",\n      size: data.byteLength,\n      count: chunks\n    })\n  );\n\n  let sentSize = 0;\n  let sentChunks = 0;\n  for (let i = 0; i < chunks; i++) {\n    const chunk = data.slice(CHUNK_MAX_SIZE * i, CHUNK_MAX_SIZE * (i + 1));\n    ws.send(chunk);\n    sentChunks += 1;\n    sentSize += chunk.byteLength;\n  }\n\n  ws.send(\n    serializeBatchMarker({\n      id,\n      type: \"end\",\n      size: sentSize,\n      count: sentChunks\n    })\n  );\n};\n\n/**\n * Creates a WebSocket message handler that can handle chunked messages.\n *\n * Reassembles the chunks into a single ArrayBuffer and pass it to the\n * provided receive function.\n *\n * 1. The server receives a chunk start marker\n * 2. The server pools each chunk until it receives a chunk end marker\n * 3. The server validates that the received data matches the expected size\n * 4. The server forwards the message to handlers\n */\nexport const handleChunked = (\n  receive: (conn: Connection, data: WSMessage) => void\n): MessageHandler => {\n  let batch: ArrayBuffer[] | undefined;\n  let start: Batch | undefined;\n\n  return (connection, message) => {\n    if (typeof message === \"string\" && !isBatchSentinel(message)) {\n      return;\n    }\n    if (isBatchSentinel(message)) {\n      const marker = parseBatchMarker(message);\n      if (marker.type === \"start\") {\n        batch = [];\n        start = marker;\n      }\n\n      if (marker.type === \"end\") {\n        if (batch) {\n          try {\n            // validate start and end markers match\n            assertEquality(start?.id, marker.id, \"client id\");\n            assertEquality(start?.count, marker.count, \"client counts\");\n            assertEquality(start?.size, marker.size, \"client size\");\n\n            // combine chunks into single buffer\n            const size = batch.reduce(\n              (sum, buffer) => sum + buffer.byteLength,\n              0\n            );\n            const bytes = new Uint8Array(size);\n            let bytesWritten = 0;\n            for (const chunk of batch) {\n              bytes.set(new Uint8Array(chunk), bytesWritten);\n              bytesWritten += chunk.byteLength;\n            }\n\n            // validate data as read matches expected\n            assertEquality(marker.count, batch.length, \"received batch count\");\n            assertEquality(marker.size, bytesWritten, \"client size\");\n            // @ts-ignore typescript hell\n            receive(connection, bytes);\n          } catch (e) {\n            console.error(e);\n            throw e;\n          } finally {\n            batch = undefined;\n            start = undefined;\n          }\n        }\n      }\n    } else if (batch) {\n      // @ts-ignore typescript hell\n      batch.push(message);\n    } else {\n      // @ts-ignore typescript hell\n      receive(connection, new Uint8Array(message));\n    }\n  };\n};\n\nfunction assertEquality(expected: unknown, actual: unknown, label: string) {\n  if (expected !== actual) {\n    throw new Error(\n      `Mismatching ${label}! Expected ${expected as string}, got ${actual as string}`\n    );\n  } else {\n    trace(`Matching ${label}: ${expected as string}`);\n  }\n}\n\n/** Checks whether a message is batch marker */\nfunction isBatchSentinel(msg: WSMessage): msg is string {\n  return typeof msg === \"string\" && msg.startsWith(BATCH_SENTINEL);\n}\n\n/** Encodes a batch marker message so that it can be safely parsed */\nfunction serializeBatchMarker(batch: Batch): string {\n  return `${BATCH_SENTINEL}#${JSON.stringify(batch)}`;\n}\n\n/** Parses a batch marker messages and throws if its invalid */\nexport function parseBatchMarker(msg: string) {\n  const [sentinel, data] = msg.split(\"#\", 2);\n  if (sentinel !== BATCH_SENTINEL) {\n    throw new Error(`Unexpected batch marker: ${msg}`);\n  }\n\n  const batch = JSON.parse(data) as Batch;\n  if (batch.type !== \"start\" && batch.type !== \"end\") {\n    throw new Error(`Unexpected batch data: ${msg}`);\n  }\n\n  return batch;\n}\n"],"mappings":";AAMO,IAAM,iBAAiB;AAE9B,IAAM,iBAAiB;AACvB,IAAM,gBAAgB;AACtB,IAAM,QAAQ,IAAI,SAAoB,iBAAiB,QAAQ,IAAI,GAAG,IAAI;AAE1E,IAAI,0BAA0B;AAmBvB,IAAM,cAAc,CAAC,MAAkB,OAAkB;AAC9D,MAAI,KAAK,cAAc,gBAAgB;AACrC,OAAG,KAAK,IAAI;AACZ;AAAA,EACF;AAEA,MAAI,CAAC,yBAAyB;AAC5B,YAAQ;AAAA,MACN;AAAA,MACA;AAAA,MACA,kBAAkB,KAAK,aAAa,MAAO,KAAM,QAAQ,CAAC,CAAC;AAAA,IAC7D;AACA,8BAA0B;AAAA,EAC5B;AAEA,QAAM,MAAM,KAAK,IAAI,IAAI,KAAK,OAAO,GAAG,SAAS,EAAE,EAAE,UAAU,EAAE;AACjE,QAAM,SAAS,KAAK,KAAK,KAAK,aAAa,cAAc;AAEzD,KAAG;AAAA,IACD,qBAAqB;AAAA,MACnB;AAAA,MACA,MAAM;AAAA,MACN,MAAM,KAAK;AAAA,MACX,OAAO;AAAA,IACT,CAAC;AAAA,EACH;AAEA,MAAI,WAAW;AACf,MAAI,aAAa;AACjB,WAAS,IAAI,GAAG,IAAI,QAAQ,KAAK;AAC/B,UAAM,QAAQ,KAAK,MAAM,iBAAiB,GAAG,kBAAkB,IAAI,EAAE;AACrE,OAAG,KAAK,KAAK;AACb,kBAAc;AACd,gBAAY,MAAM;AAAA,EACpB;AAEA,KAAG;AAAA,IACD,qBAAqB;AAAA,MACnB;AAAA,MACA,MAAM;AAAA,MACN,MAAM;AAAA,MACN,OAAO;AAAA,IACT,CAAC;AAAA,EACH;AACF;AAaO,IAAM,gBAAgB,CAC3B,YACmB;AACnB,MAAI;AACJ,MAAI;AAEJ,SAAO,CAAC,YAAY,YAAY;AAC9B,QAAI,OAAO,YAAY,YAAY,CAAC,gBAAgB,OAAO,GAAG;AAC5D;AAAA,IACF;AACA,QAAI,gBAAgB,OAAO,GAAG;AAC5B,YAAM,SAAS,iBAAiB,OAAO;AACvC,UAAI,OAAO,SAAS,SAAS;AAC3B,gBAAQ,CAAC;AACT,gBAAQ;AAAA,MACV;AAEA,UAAI,OAAO,SAAS,OAAO;AACzB,YAAI,OAAO;AACT,cAAI;AAEF,2BAAe,+BAAO,IAAI,OAAO,IAAI,WAAW;AAChD,2BAAe,+BAAO,OAAO,OAAO,OAAO,eAAe;AAC1D,2BAAe,+BAAO,MAAM,OAAO,MAAM,aAAa;AAGtD,kBAAM,OAAO,MAAM;AAAA,cACjB,CAAC,KAAK,WAAW,MAAM,OAAO;AAAA,cAC9B;AAAA,YACF;AACA,kBAAM,QAAQ,IAAI,WAAW,IAAI;AACjC,gBAAI,eAAe;AACnB,uBAAW,SAAS,OAAO;AACzB,oBAAM,IAAI,IAAI,WAAW,KAAK,GAAG,YAAY;AAC7C,8BAAgB,MAAM;AAAA,YACxB;AAGA,2BAAe,OAAO,OAAO,MAAM,QAAQ,sBAAsB;AACjE,2BAAe,OAAO,MAAM,cAAc,aAAa;AAEvD,oBAAQ,YAAY,KAAK;AAAA,UAC3B,SAAS,GAAG;AACV,oBAAQ,MAAM,CAAC;AACf,kBAAM;AAAA,UACR,UAAE;AACA,oBAAQ;AACR,oBAAQ;AAAA,UACV;AAAA,QACF;AAAA,MACF;AAAA,IACF,WAAW,OAAO;AAEhB,YAAM,KAAK,OAAO;AAAA,IACpB,OAAO;AAEL,cAAQ,YAAY,IAAI,WAAW,OAAO,CAAC;AAAA,IAC7C;AAAA,EACF;AACF;AAEA,SAAS,eAAe,UAAmB,QAAiB,OAAe;AACzE,MAAI,aAAa,QAAQ;AACvB,UAAM,IAAI;AAAA,MACR,eAAe,KAAK,cAAc,QAAkB,SAAS,MAAgB;AAAA,IAC/E;AAAA,EACF,OAAO;AACL,UAAM,YAAY,KAAK,KAAK,QAAkB,EAAE;AAAA,EAClD;AACF;AAGA,SAAS,gBAAgB,KAA+B;AACtD,SAAO,OAAO,QAAQ,YAAY,IAAI,WAAW,cAAc;AACjE;AAGA,SAAS,qBAAqB,OAAsB;AAClD,SAAO,GAAG,cAAc,IAAI,KAAK,UAAU,KAAK,CAAC;AACnD;AAGO,SAAS,iBAAiB,KAAa;AAC5C,QAAM,CAAC,UAAU,IAAI,IAAI,IAAI,MAAM,KAAK,CAAC;AACzC,MAAI,aAAa,gBAAgB;AAC/B,UAAM,IAAI,MAAM,4BAA4B,GAAG,EAAE;AAAA,EACnD;AAEA,QAAM,QAAQ,KAAK,MAAM,IAAI;AAC7B,MAAI,MAAM,SAAS,WAAW,MAAM,SAAS,OAAO;AAClD,UAAM,IAAI,MAAM,0BAA0B,GAAG,EAAE;AAAA,EACjD;AAEA,SAAO;AACT;","names":[]}

package/dist/provider/index.js

@@ -5,8 +5,8 @@ import {
   messageAwareness,
   messageQueryAwareness,
   messageSync
-} from "../chunk-TD2NUND2.js";
-import "../chunk-JFXBVIOX.js";
+} from "../chunk-H4CTOPWX.js";
+import "../chunk-W2PLHZNF.js";
 export {
   WebsocketProvider,
   YProvider as default,

package/dist/provider/react.js

@@ -1,7 +1,7 @@
 import {
   YProvider
-} from "../chunk-TD2NUND2.js";
-import "../chunk-JFXBVIOX.js";
+} from "../chunk-H4CTOPWX.js";
+import "../chunk-W2PLHZNF.js";
 
 // src/provider/react.tsx
 import { useEffect, useState } from "react";

package/dist/server/index.d.ts

@@ -1,4 +1,4 @@
-import { Server, Connection, ConnectionContext } from "partyserver";
+import { Server, Connection, WSMessage, ConnectionContext } from "partyserver";
 import * as awarenessProtocol from "y-protocols/awareness";
 import { Doc } from "yjs";
 
@@ -19,14 +19,19 @@ declare class YServer<Env = unknown> extends Server<Env> {
   onLoad(): Promise<void>;
   onSave(): Promise<void>;
   onStart(): Promise<void>;
-  onMessage: (conn: Connection, message: string | ArrayBufferLike) => void;
+  isReadOnly(connection: Connection): boolean;
+  handleMessage(connection: Connection, message: WSMessage): void;
+  onMessage: (conn: Connection, message: WSMessage) => void;
   onClose(
     connection: Connection<unknown>,
     _code: number,
     _reason: string,
     _wasClean: boolean
   ): void | Promise<void>;
-  onConnect(conn: Connection<unknown>, _ctx: ConnectionContext): void;
+  onConnect(
+    conn: Connection<unknown>,
+    _ctx: ConnectionContext
+  ): void | Promise<void>;
 }
 
 export { type CallbackOptions, YServer };

package/dist/server/index.js

@@ -1,6 +1,6 @@
 import {
   handleChunked
-} from "../chunk-JFXBVIOX.js";
+} from "../chunk-W2PLHZNF.js";
 
 // src/server/index.ts
 import * as decoding from "lib0/decoding";
@@ -9,7 +9,7 @@ import debounce from "lodash.debounce";
 import { Server } from "partyserver";
 import * as awarenessProtocol from "y-protocols/awareness";
 import * as syncProtocol from "y-protocols/sync";
-import { applyUpdate, encodeStateAsUpdate, Doc as YDoc } from "yjs";
+import { applyUpdate, Doc as YDoc, encodeStateAsUpdate } from "yjs";
 var wsReadyStateConnecting = 0;
 var wsReadyStateOpen = 1;
 var messageSync = 0;
@@ -147,8 +147,11 @@ var YServer = class extends Server {
       )
     );
   }
+  isReadOnly(connection) {
+    return false;
+  }
   // @ts-ignore something something typescript
-  onMessage = handleChunked((conn, message) => {
+  handleMessage(connection, message) {
     if (typeof message === "string") {
       console.warn(
         `Received non-binary message. Override onMessage on ${this.#ParentClass.name} to handle string messages if required`
@@ -166,19 +169,18 @@ var YServer = class extends Server {
             decoder,
             encoder,
             this.document,
-            conn,
-            // TODO: readonly conections
-            false
+            connection,
+            this.isReadOnly(connection)
           );
           if (encoding.length(encoder) > 1) {
-            send(this.document, conn, encoding.toUint8Array(encoder));
+            send(this.document, connection, encoding.toUint8Array(encoder));
           }
           break;
         case messageAwareness: {
           awarenessProtocol.applyAwarenessUpdate(
             this.document.awareness,
             decoding.readVarUint8Array(decoder),
-            conn
+            connection
           );
           break;
         }
@@ -187,7 +189,10 @@ var YServer = class extends Server {
       console.error(err);
       this.document.emit("error", [err]);
     }
-  });
+  }
+  onMessage = handleChunked(
+    (conn, message) => this.handleMessage(conn, message)
+  );
   onClose(connection, _code, _reason, _wasClean) {
     closeConn(this.document, connection);
   }

package/dist/server/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/server/index.ts"],"sourcesContent":["import * as decoding from \"lib0/decoding\";\nimport * as encoding from \"lib0/encoding\";\nimport debounce from \"lodash.debounce\";\nimport { Server } from \"partyserver\";\nimport * as awarenessProtocol from \"y-protocols/awareness\";\nimport * as syncProtocol from \"y-protocols/sync\";\nimport { applyUpdate, encodeStateAsUpdate, Doc as YDoc } from \"yjs\";\n\nimport { handleChunked } from \"../shared/chunking\";\n\nimport type { Connection, ConnectionContext } from \"partyserver\";\n\nconst wsReadyStateConnecting = 0;\nconst wsReadyStateOpen = 1;\nconst wsReadyStateClosing = 2;\nconst wsReadyStateClosed = 3;\n\nconst messageSync = 0;\nconst messageAwareness = 1;\nconst messageAuth = 2;\n\nfunction updateHandler(update: Uint8Array, origin: unknown, doc: WSSharedDoc) {\n  const encoder = encoding.createEncoder();\n  encoding.writeVarUint(encoder, messageSync);\n  syncProtocol.writeUpdate(encoder, update);\n  const message = encoding.toUint8Array(encoder);\n  doc.conns.forEach((_, conn) => send(doc, conn, message));\n}\n\nclass WSSharedDoc extends YDoc {\n  conns: Map<Connection, Set<number>>;\n  awareness: awarenessProtocol.Awareness;\n\n  constructor() {\n    super({ gc: true });\n\n    /**\n     * Maps from conn to set of controlled user ids. Delete all user ids from awareness when this conn is closed\n     */\n    this.conns = new Map();\n\n    this.awareness = new awarenessProtocol.Awareness(this);\n    this.awareness.setLocalState(null);\n\n    const awarenessChangeHandler = (\n      {\n        added,\n        updated,\n        removed\n      }: {\n        added: Array<number>;\n        updated: Array<number>;\n        removed: Array<number>;\n      },\n      conn: Connection | null // Origin is the connection that made the change\n    ) => {\n      const changedClients = added.concat(updated, removed);\n      if (conn !== null) {\n        const connControlledIDs =\n          /** @type {Set<number>} */ this.conns.get(conn);\n        if (connControlledIDs !== undefined) {\n          added.forEach((clientID) => {\n            connControlledIDs.add(clientID);\n          });\n          removed.forEach((clientID) => {\n            connControlledIDs.delete(clientID);\n          });\n        }\n      }\n      // broadcast awareness update\n      const encoder = encoding.createEncoder();\n      encoding.writeVarUint(encoder, messageAwareness);\n      encoding.writeVarUint8Array(\n        encoder,\n        awarenessProtocol.encodeAwarenessUpdate(this.awareness, changedClients)\n      );\n      const buff = encoding.toUint8Array(encoder);\n      this.conns.forEach((_, c) => {\n        send(this, c, buff);\n      });\n    };\n    this.awareness.on(\"update\", awarenessChangeHandler);\n    // @ts-expect-error - TODO: fix this\n    this.on(\"update\", updateHandler);\n  }\n}\n\nconst CALLBACK_DEFAULTS = {\n  debounceWait: 2000,\n  debounceMaxWait: 10000,\n  timeout: 5000\n};\n\nfunction readSyncMessage(\n  decoder: decoding.Decoder,\n  encoder: encoding.Encoder,\n  doc: YDoc,\n  transactionOrigin: Connection,\n  readOnly = false\n) {\n  const messageType = decoding.readVarUint(decoder);\n  switch (messageType) {\n    case syncProtocol.messageYjsSyncStep1:\n      syncProtocol.readSyncStep1(decoder, encoder, doc);\n      break;\n    case syncProtocol.messageYjsSyncStep2:\n      if (!readOnly)\n        syncProtocol.readSyncStep2(decoder, doc, transactionOrigin);\n      break;\n    case syncProtocol.messageYjsUpdate:\n      if (!readOnly) syncProtocol.readUpdate(decoder, doc, transactionOrigin);\n      break;\n    default:\n      throw new Error(\"Unknown message type\");\n  }\n  return messageType;\n}\n\nfunction closeConn(doc: WSSharedDoc, conn: Connection): void {\n  if (doc.conns.has(conn)) {\n    const controlledIds: Set<number> = doc.conns.get(conn)!;\n    doc.conns.delete(conn);\n    awarenessProtocol.removeAwarenessStates(\n      doc.awareness,\n      Array.from(controlledIds),\n      null\n    );\n  }\n  try {\n    conn.close();\n  } catch (e) {\n    console.warn(\"failed to close connection\", e);\n  }\n}\n\nfunction send(doc: WSSharedDoc, conn: Connection, m: Uint8Array) {\n  if (\n    conn.readyState !== undefined &&\n    conn.readyState !== wsReadyStateConnecting &&\n    conn.readyState !== wsReadyStateOpen\n  ) {\n    closeConn(doc, conn);\n  }\n  try {\n    conn.send(m);\n  } catch (e) {\n    closeConn(doc, conn);\n  }\n}\n\nexport interface CallbackOptions {\n  debounceWait?: number;\n  debounceMaxWait?: number;\n  timeout?: number;\n}\n\nexport class YServer<Env = unknown> extends Server<Env> {\n  static callbackOptions: CallbackOptions = {};\n\n  #ParentClass: typeof YServer = Object.getPrototypeOf(this).constructor;\n  readonly document = new WSSharedDoc();\n\n  async onLoad(): Promise<void> {\n    // to be implemented by the user\n    return;\n  }\n\n  async onSave(): Promise<void> {}\n\n  async onStart(): Promise<void> {\n    const src = await this.onLoad();\n    if (src != null) {\n      const state = encodeStateAsUpdate(src);\n      applyUpdate(this.document, state);\n    }\n\n    this.document.on(\n      \"update\",\n      debounce(\n        (_update: Uint8Array, _origin: Connection, _doc: YDoc) => {\n          try {\n            this.onSave().catch((err) => {\n              console.error(\"failed to persist:\", err);\n            });\n          } catch (err) {\n            console.error(\"failed to persist:\", err);\n          }\n        },\n        this.#ParentClass.callbackOptions.debounceWait ||\n          CALLBACK_DEFAULTS.debounceWait,\n        {\n          maxWait:\n            this.#ParentClass.callbackOptions.debounceMaxWait ||\n            CALLBACK_DEFAULTS.debounceMaxWait\n        }\n      )\n    );\n  }\n\n  // @ts-ignore something something typescript\n  onMessage = handleChunked((conn, message) => {\n    if (typeof message === \"string\") {\n      console.warn(\n        `Received non-binary message. Override onMessage on ${this.#ParentClass.name} to handle string messages if required`\n      );\n      return;\n    }\n    try {\n      const encoder = encoding.createEncoder();\n      // TODO: this type seems odd\n      const decoder = decoding.createDecoder(message as unknown as Uint8Array);\n      const messageType = decoding.readVarUint(decoder);\n      switch (messageType) {\n        case messageSync:\n          encoding.writeVarUint(encoder, messageSync);\n          readSyncMessage(\n            decoder,\n            encoder,\n            this.document,\n            conn,\n            // TODO: readonly conections\n            false\n          );\n\n          // If the `encoder` only contains the type of reply message and no\n          // message, there is no need to send the message. When `encoder` only\n          // contains the type of reply, its length is 1.\n          if (encoding.length(encoder) > 1) {\n            send(this.document, conn, encoding.toUint8Array(encoder));\n          }\n          break;\n        case messageAwareness: {\n          awarenessProtocol.applyAwarenessUpdate(\n            this.document.awareness,\n            decoding.readVarUint8Array(decoder),\n            conn\n          );\n          break;\n        }\n      }\n    } catch (err) {\n      console.error(err);\n      // @ts-expect-error - TODO: fix this\n      this.document.emit(\"error\", [err]);\n    }\n  });\n\n  onClose(\n    connection: Connection<unknown>,\n    _code: number,\n    _reason: string,\n    _wasClean: boolean\n  ): void | Promise<void> {\n    closeConn(this.document, connection);\n  }\n\n  // TODO: explore why onError gets triggered when a connection closes\n\n  onConnect(conn: Connection<unknown>, _ctx: ConnectionContext) {\n    // conn.binaryType = \"arraybuffer\"; // from y-websocket, breaks in our runtime\n\n    this.document.conns.set(conn, new Set());\n\n    // send sync step 1\n    const encoder = encoding.createEncoder();\n    encoding.writeVarUint(encoder, messageSync);\n    syncProtocol.writeSyncStep1(encoder, this.document);\n    send(this.document, conn, encoding.toUint8Array(encoder));\n    const awarenessStates = this.document.awareness.getStates();\n    if (awarenessStates.size > 0) {\n      const encoder = encoding.createEncoder();\n      encoding.writeVarUint(encoder, messageAwareness);\n      encoding.writeVarUint8Array(\n        encoder,\n        awarenessProtocol.encodeAwarenessUpdate(\n          this.document.awareness,\n          Array.from(awarenessStates.keys())\n        )\n      );\n      send(this.document, conn, encoding.toUint8Array(encoder));\n    }\n  }\n}\n"],"mappings":";;;;;AAAA,YAAY,cAAc;AAC1B,YAAY,cAAc;AAC1B,OAAO,cAAc;AACrB,SAAS,cAAc;AACvB,YAAY,uBAAuB;AACnC,YAAY,kBAAkB;AAC9B,SAAS,aAAa,qBAAqB,OAAO,YAAY;AAM9D,IAAM,yBAAyB;AAC/B,IAAM,mBAAmB;AAIzB,IAAM,cAAc;AACpB,IAAM,mBAAmB;AAGzB,SAAS,cAAc,QAAoB,QAAiB,KAAkB;AAC5E,QAAM,UAAmB,uBAAc;AACvC,EAAS,sBAAa,SAAS,WAAW;AAC1C,EAAa,yBAAY,SAAS,MAAM;AACxC,QAAM,UAAmB,sBAAa,OAAO;AAC7C,MAAI,MAAM,QAAQ,CAAC,GAAG,SAAS,KAAK,KAAK,MAAM,OAAO,CAAC;AACzD;AAEA,IAAM,cAAN,cAA0B,KAAK;AAAA,EAC7B;AAAA,EACA;AAAA,EAEA,cAAc;AACZ,UAAM,EAAE,IAAI,KAAK,CAAC;AAKlB,SAAK,QAAQ,oBAAI,IAAI;AAErB,SAAK,YAAY,IAAsB,4BAAU,IAAI;AACrD,SAAK,UAAU,cAAc,IAAI;AAEjC,UAAM,yBAAyB,CAC7B;AAAA,MACE;AAAA,MACA;AAAA,MACA;AAAA,IACF,GAKA,SACG;AACH,YAAM,iBAAiB,MAAM,OAAO,SAAS,OAAO;AACpD,UAAI,SAAS,MAAM;AACjB,cAAM;AAAA;AAAA,UACuB,KAAK,MAAM,IAAI,IAAI;AAAA;AAChD,YAAI,sBAAsB,QAAW;AACnC,gBAAM,QAAQ,CAAC,aAAa;AAC1B,8BAAkB,IAAI,QAAQ;AAAA,UAChC,CAAC;AACD,kBAAQ,QAAQ,CAAC,aAAa;AAC5B,8BAAkB,OAAO,QAAQ;AAAA,UACnC,CAAC;AAAA,QACH;AAAA,MACF;AAEA,YAAM,UAAmB,uBAAc;AACvC,MAAS,sBAAa,SAAS,gBAAgB;AAC/C,MAAS;AAAA,QACP;AAAA,QACkB,wCAAsB,KAAK,WAAW,cAAc;AAAA,MACxE;AACA,YAAM,OAAgB,sBAAa,OAAO;AAC1C,WAAK,MAAM,QAAQ,CAAC,GAAG,MAAM;AAC3B,aAAK,MAAM,GAAG,IAAI;AAAA,MACpB,CAAC;AAAA,IACH;AACA,SAAK,UAAU,GAAG,UAAU,sBAAsB;AAElD,SAAK,GAAG,UAAU,aAAa;AAAA,EACjC;AACF;AAEA,IAAM,oBAAoB;AAAA,EACxB,cAAc;AAAA,EACd,iBAAiB;AAAA,EACjB,SAAS;AACX;AAEA,SAAS,gBACP,SACA,SACA,KACA,mBACA,WAAW,OACX;AACA,QAAM,cAAuB,qBAAY,OAAO;AAChD,UAAQ,aAAa;AAAA,IACnB,KAAkB;AAChB,MAAa,2BAAc,SAAS,SAAS,GAAG;AAChD;AAAA,IACF,KAAkB;AAChB,UAAI,CAAC;AACH,QAAa,2BAAc,SAAS,KAAK,iBAAiB;AAC5D;AAAA,IACF,KAAkB;AAChB,UAAI,CAAC,SAAU,CAAa,wBAAW,SAAS,KAAK,iBAAiB;AACtE;AAAA,IACF;AACE,YAAM,IAAI,MAAM,sBAAsB;AAAA,EAC1C;AACA,SAAO;AACT;AAEA,SAAS,UAAU,KAAkB,MAAwB;AAC3D,MAAI,IAAI,MAAM,IAAI,IAAI,GAAG;AACvB,UAAM,gBAA6B,IAAI,MAAM,IAAI,IAAI;AACrD,QAAI,MAAM,OAAO,IAAI;AACrB,IAAkB;AAAA,MAChB,IAAI;AAAA,MACJ,MAAM,KAAK,aAAa;AAAA,MACxB;AAAA,IACF;AAAA,EACF;AACA,MAAI;AACF,SAAK,MAAM;AAAA,EACb,SAAS,GAAG;AACV,YAAQ,KAAK,8BAA8B,CAAC;AAAA,EAC9C;AACF;AAEA,SAAS,KAAK,KAAkB,MAAkB,GAAe;AAC/D,MACE,KAAK,eAAe,UACpB,KAAK,eAAe,0BACpB,KAAK,eAAe,kBACpB;AACA,cAAU,KAAK,IAAI;AAAA,EACrB;AACA,MAAI;AACF,SAAK,KAAK,CAAC;AAAA,EACb,SAAS,GAAG;AACV,cAAU,KAAK,IAAI;AAAA,EACrB;AACF;AAQO,IAAM,UAAN,cAAqC,OAAY;AAAA,EACtD,OAAO,kBAAmC,CAAC;AAAA,EAE3C,eAA+B,OAAO,eAAe,IAAI,EAAE;AAAA,EAClD,WAAW,IAAI,YAAY;AAAA,EAEpC,MAAM,SAAwB;AAE5B;AAAA,EACF;AAAA,EAEA,MAAM,SAAwB;AAAA,EAAC;AAAA,EAE/B,MAAM,UAAyB;AAC7B,UAAM,MAAM,MAAM,KAAK,OAAO;AAC9B,QAAI,OAAO,MAAM;AACf,YAAM,QAAQ,oBAAoB,GAAG;AACrC,kBAAY,KAAK,UAAU,KAAK;AAAA,IAClC;AAEA,SAAK,SAAS;AAAA,MACZ;AAAA,MACA;AAAA,QACE,CAAC,SAAqB,SAAqB,SAAe;AACxD,cAAI;AACF,iBAAK,OAAO,EAAE,MAAM,CAAC,QAAQ;AAC3B,sBAAQ,MAAM,sBAAsB,GAAG;AAAA,YACzC,CAAC;AAAA,UACH,SAAS,KAAK;AACZ,oBAAQ,MAAM,sBAAsB,GAAG;AAAA,UACzC;AAAA,QACF;AAAA,QACA,KAAK,aAAa,gBAAgB,gBAChC,kBAAkB;AAAA,QACpB;AAAA,UACE,SACE,KAAK,aAAa,gBAAgB,mBAClC,kBAAkB;AAAA,QACtB;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA,EAGA,YAAY,cAAc,CAAC,MAAM,YAAY;AAC3C,QAAI,OAAO,YAAY,UAAU;AAC/B,cAAQ;AAAA,QACN,sDAAsD,KAAK,aAAa,IAAI;AAAA,MAC9E;AACA;AAAA,IACF;AACA,QAAI;AACF,YAAM,UAAmB,uBAAc;AAEvC,YAAM,UAAmB,uBAAc,OAAgC;AACvE,YAAM,cAAuB,qBAAY,OAAO;AAChD,cAAQ,aAAa;AAAA,QACnB,KAAK;AACH,UAAS,sBAAa,SAAS,WAAW;AAC1C;AAAA,YACE;AAAA,YACA;AAAA,YACA,KAAK;AAAA,YACL;AAAA;AAAA,YAEA;AAAA,UACF;AAKA,cAAa,gBAAO,OAAO,IAAI,GAAG;AAChC,iBAAK,KAAK,UAAU,MAAe,sBAAa,OAAO,CAAC;AAAA,UAC1D;AACA;AAAA,QACF,KAAK,kBAAkB;AACrB,UAAkB;AAAA,YAChB,KAAK,SAAS;AAAA,YACL,2BAAkB,OAAO;AAAA,YAClC;AAAA,UACF;AACA;AAAA,QACF;AAAA,MACF;AAAA,IACF,SAAS,KAAK;AACZ,cAAQ,MAAM,GAAG;AAEjB,WAAK,SAAS,KAAK,SAAS,CAAC,GAAG,CAAC;AAAA,IACnC;AAAA,EACF,CAAC;AAAA,EAED,QACE,YACA,OACA,SACA,WACsB;AACtB,cAAU,KAAK,UAAU,UAAU;AAAA,EACrC;AAAA;AAAA,EAIA,UAAU,MAA2B,MAAyB;AAG5D,SAAK,SAAS,MAAM,IAAI,MAAM,oBAAI,IAAI,CAAC;AAGvC,UAAM,UAAmB,uBAAc;AACvC,IAAS,sBAAa,SAAS,WAAW;AAC1C,IAAa,4BAAe,SAAS,KAAK,QAAQ;AAClD,SAAK,KAAK,UAAU,MAAe,sBAAa,OAAO,CAAC;AACxD,UAAM,kBAAkB,KAAK,SAAS,UAAU,UAAU;AAC1D,QAAI,gBAAgB,OAAO,GAAG;AAC5B,YAAMA,WAAmB,uBAAc;AACvC,MAAS,sBAAaA,UAAS,gBAAgB;AAC/C,MAAS;AAAA,QACPA;AAAA,QACkB;AAAA,UAChB,KAAK,SAAS;AAAA,UACd,MAAM,KAAK,gBAAgB,KAAK,CAAC;AAAA,QACnC;AAAA,MACF;AACA,WAAK,KAAK,UAAU,MAAe,sBAAaA,QAAO,CAAC;AAAA,IAC1D;AAAA,EACF;AACF;","names":["encoder"]}
+{"version":3,"sources":["../../src/server/index.ts"],"sourcesContent":["import * as decoding from \"lib0/decoding\";\nimport * as encoding from \"lib0/encoding\";\nimport debounce from \"lodash.debounce\";\nimport type { Connection, ConnectionContext, WSMessage } from \"partyserver\";\nimport { Server } from \"partyserver\";\nimport * as awarenessProtocol from \"y-protocols/awareness\";\nimport * as syncProtocol from \"y-protocols/sync\";\nimport { applyUpdate, Doc as YDoc, encodeStateAsUpdate } from \"yjs\";\n\nimport { handleChunked } from \"../shared/chunking\";\n\nconst wsReadyStateConnecting = 0;\nconst wsReadyStateOpen = 1;\nconst wsReadyStateClosing = 2;\nconst wsReadyStateClosed = 3;\n\nconst messageSync = 0;\nconst messageAwareness = 1;\nconst messageAuth = 2;\n\nfunction updateHandler(update: Uint8Array, origin: unknown, doc: WSSharedDoc) {\n  const encoder = encoding.createEncoder();\n  encoding.writeVarUint(encoder, messageSync);\n  syncProtocol.writeUpdate(encoder, update);\n  const message = encoding.toUint8Array(encoder);\n  doc.conns.forEach((_, conn) => send(doc, conn, message));\n}\n\nclass WSSharedDoc extends YDoc {\n  conns: Map<Connection, Set<number>>;\n  awareness: awarenessProtocol.Awareness;\n\n  constructor() {\n    super({ gc: true });\n\n    /**\n     * Maps from conn to set of controlled user ids. Delete all user ids from awareness when this conn is closed\n     */\n    this.conns = new Map();\n\n    this.awareness = new awarenessProtocol.Awareness(this);\n    this.awareness.setLocalState(null);\n\n    const awarenessChangeHandler = (\n      {\n        added,\n        updated,\n        removed\n      }: {\n        added: Array<number>;\n        updated: Array<number>;\n        removed: Array<number>;\n      },\n      conn: Connection | null // Origin is the connection that made the change\n    ) => {\n      const changedClients = added.concat(updated, removed);\n      if (conn !== null) {\n        const connControlledIDs =\n          /** @type {Set<number>} */ this.conns.get(conn);\n        if (connControlledIDs !== undefined) {\n          added.forEach((clientID) => {\n            connControlledIDs.add(clientID);\n          });\n          removed.forEach((clientID) => {\n            connControlledIDs.delete(clientID);\n          });\n        }\n      }\n      // broadcast awareness update\n      const encoder = encoding.createEncoder();\n      encoding.writeVarUint(encoder, messageAwareness);\n      encoding.writeVarUint8Array(\n        encoder,\n        awarenessProtocol.encodeAwarenessUpdate(this.awareness, changedClients)\n      );\n      const buff = encoding.toUint8Array(encoder);\n      this.conns.forEach((_, c) => {\n        send(this, c, buff);\n      });\n    };\n    this.awareness.on(\"update\", awarenessChangeHandler);\n    // @ts-expect-error - TODO: fix this\n    this.on(\"update\", updateHandler);\n  }\n}\n\nconst CALLBACK_DEFAULTS = {\n  debounceWait: 2000,\n  debounceMaxWait: 10000,\n  timeout: 5000\n};\n\nfunction readSyncMessage(\n  decoder: decoding.Decoder,\n  encoder: encoding.Encoder,\n  doc: YDoc,\n  transactionOrigin: Connection,\n  readOnly = false\n) {\n  const messageType = decoding.readVarUint(decoder);\n  switch (messageType) {\n    case syncProtocol.messageYjsSyncStep1:\n      syncProtocol.readSyncStep1(decoder, encoder, doc);\n      break;\n    case syncProtocol.messageYjsSyncStep2:\n      if (!readOnly)\n        syncProtocol.readSyncStep2(decoder, doc, transactionOrigin);\n      break;\n    case syncProtocol.messageYjsUpdate:\n      if (!readOnly) syncProtocol.readUpdate(decoder, doc, transactionOrigin);\n      break;\n    default:\n      throw new Error(\"Unknown message type\");\n  }\n  return messageType;\n}\n\nfunction closeConn(doc: WSSharedDoc, conn: Connection): void {\n  if (doc.conns.has(conn)) {\n    const controlledIds: Set<number> = doc.conns.get(conn)!;\n    doc.conns.delete(conn);\n    awarenessProtocol.removeAwarenessStates(\n      doc.awareness,\n      Array.from(controlledIds),\n      null\n    );\n  }\n  try {\n    conn.close();\n  } catch (e) {\n    console.warn(\"failed to close connection\", e);\n  }\n}\n\nfunction send(doc: WSSharedDoc, conn: Connection, m: Uint8Array) {\n  if (\n    conn.readyState !== undefined &&\n    conn.readyState !== wsReadyStateConnecting &&\n    conn.readyState !== wsReadyStateOpen\n  ) {\n    closeConn(doc, conn);\n  }\n  try {\n    conn.send(m);\n  } catch (e) {\n    closeConn(doc, conn);\n  }\n}\n\nexport interface CallbackOptions {\n  debounceWait?: number;\n  debounceMaxWait?: number;\n  timeout?: number;\n}\n\nexport class YServer<Env = unknown> extends Server<Env> {\n  static callbackOptions: CallbackOptions = {};\n\n  #ParentClass: typeof YServer = Object.getPrototypeOf(this).constructor;\n  readonly document = new WSSharedDoc();\n\n  async onLoad(): Promise<void> {\n    // to be implemented by the user\n    return;\n  }\n\n  async onSave(): Promise<void> {}\n\n  async onStart(): Promise<void> {\n    const src = await this.onLoad();\n    if (src != null) {\n      const state = encodeStateAsUpdate(src);\n      applyUpdate(this.document, state);\n    }\n\n    this.document.on(\n      \"update\",\n      debounce(\n        (_update: Uint8Array, _origin: Connection, _doc: YDoc) => {\n          try {\n            this.onSave().catch((err) => {\n              console.error(\"failed to persist:\", err);\n            });\n          } catch (err) {\n            console.error(\"failed to persist:\", err);\n          }\n        },\n        this.#ParentClass.callbackOptions.debounceWait ||\n          CALLBACK_DEFAULTS.debounceWait,\n        {\n          maxWait:\n            this.#ParentClass.callbackOptions.debounceMaxWait ||\n            CALLBACK_DEFAULTS.debounceMaxWait\n        }\n      )\n    );\n  }\n\n  isReadOnly(connection: Connection): boolean {\n    // to be implemented by the user\n    return false;\n  }\n\n  // @ts-ignore something something typescript\n  handleMessage(connection: Connection, message: WSMessage) {\n    if (typeof message === \"string\") {\n      console.warn(\n        `Received non-binary message. Override onMessage on ${this.#ParentClass.name} to handle string messages if required`\n      );\n      return;\n    }\n    try {\n      const encoder = encoding.createEncoder();\n      // TODO: this type seems odd\n      const decoder = decoding.createDecoder(message as unknown as Uint8Array);\n      const messageType = decoding.readVarUint(decoder);\n      switch (messageType) {\n        case messageSync:\n          encoding.writeVarUint(encoder, messageSync);\n          readSyncMessage(\n            decoder,\n            encoder,\n            this.document,\n            connection,\n            this.isReadOnly(connection)\n          );\n\n          // If the `encoder` only contains the type of reply message and no\n          // message, there is no need to send the message. When `encoder` only\n          // contains the type of reply, its length is 1.\n          if (encoding.length(encoder) > 1) {\n            send(this.document, connection, encoding.toUint8Array(encoder));\n          }\n          break;\n        case messageAwareness: {\n          awarenessProtocol.applyAwarenessUpdate(\n            this.document.awareness,\n            decoding.readVarUint8Array(decoder),\n            connection\n          );\n          break;\n        }\n      }\n    } catch (err) {\n      console.error(err);\n      // @ts-expect-error - TODO: fix this\n      this.document.emit(\"error\", [err]);\n    }\n  }\n\n  onMessage = handleChunked((conn, message) =>\n    this.handleMessage(conn, message)\n  );\n\n  onClose(\n    connection: Connection<unknown>,\n    _code: number,\n    _reason: string,\n    _wasClean: boolean\n  ): void | Promise<void> {\n    closeConn(this.document, connection);\n  }\n\n  // TODO: explore why onError gets triggered when a connection closes\n\n  onConnect(\n    conn: Connection<unknown>,\n    _ctx: ConnectionContext\n  ): void | Promise<void> {\n    // conn.binaryType = \"arraybuffer\"; // from y-websocket, breaks in our runtime\n\n    this.document.conns.set(conn, new Set());\n\n    // send sync step 1\n    const encoder = encoding.createEncoder();\n    encoding.writeVarUint(encoder, messageSync);\n    syncProtocol.writeSyncStep1(encoder, this.document);\n    send(this.document, conn, encoding.toUint8Array(encoder));\n    const awarenessStates = this.document.awareness.getStates();\n    if (awarenessStates.size > 0) {\n      const encoder = encoding.createEncoder();\n      encoding.writeVarUint(encoder, messageAwareness);\n      encoding.writeVarUint8Array(\n        encoder,\n        awarenessProtocol.encodeAwarenessUpdate(\n          this.document.awareness,\n          Array.from(awarenessStates.keys())\n        )\n      );\n      send(this.document, conn, encoding.toUint8Array(encoder));\n    }\n  }\n}\n"],"mappings":";;;;;AAAA,YAAY,cAAc;AAC1B,YAAY,cAAc;AAC1B,OAAO,cAAc;AAErB,SAAS,cAAc;AACvB,YAAY,uBAAuB;AACnC,YAAY,kBAAkB;AAC9B,SAAS,aAAa,OAAO,MAAM,2BAA2B;AAI9D,IAAM,yBAAyB;AAC/B,IAAM,mBAAmB;AAIzB,IAAM,cAAc;AACpB,IAAM,mBAAmB;AAGzB,SAAS,cAAc,QAAoB,QAAiB,KAAkB;AAC5E,QAAM,UAAmB,uBAAc;AACvC,EAAS,sBAAa,SAAS,WAAW;AAC1C,EAAa,yBAAY,SAAS,MAAM;AACxC,QAAM,UAAmB,sBAAa,OAAO;AAC7C,MAAI,MAAM,QAAQ,CAAC,GAAG,SAAS,KAAK,KAAK,MAAM,OAAO,CAAC;AACzD;AAEA,IAAM,cAAN,cAA0B,KAAK;AAAA,EAC7B;AAAA,EACA;AAAA,EAEA,cAAc;AACZ,UAAM,EAAE,IAAI,KAAK,CAAC;AAKlB,SAAK,QAAQ,oBAAI,IAAI;AAErB,SAAK,YAAY,IAAsB,4BAAU,IAAI;AACrD,SAAK,UAAU,cAAc,IAAI;AAEjC,UAAM,yBAAyB,CAC7B;AAAA,MACE;AAAA,MACA;AAAA,MACA;AAAA,IACF,GAKA,SACG;AACH,YAAM,iBAAiB,MAAM,OAAO,SAAS,OAAO;AACpD,UAAI,SAAS,MAAM;AACjB,cAAM;AAAA;AAAA,UACuB,KAAK,MAAM,IAAI,IAAI;AAAA;AAChD,YAAI,sBAAsB,QAAW;AACnC,gBAAM,QAAQ,CAAC,aAAa;AAC1B,8BAAkB,IAAI,QAAQ;AAAA,UAChC,CAAC;AACD,kBAAQ,QAAQ,CAAC,aAAa;AAC5B,8BAAkB,OAAO,QAAQ;AAAA,UACnC,CAAC;AAAA,QACH;AAAA,MACF;AAEA,YAAM,UAAmB,uBAAc;AACvC,MAAS,sBAAa,SAAS,gBAAgB;AAC/C,MAAS;AAAA,QACP;AAAA,QACkB,wCAAsB,KAAK,WAAW,cAAc;AAAA,MACxE;AACA,YAAM,OAAgB,sBAAa,OAAO;AAC1C,WAAK,MAAM,QAAQ,CAAC,GAAG,MAAM;AAC3B,aAAK,MAAM,GAAG,IAAI;AAAA,MACpB,CAAC;AAAA,IACH;AACA,SAAK,UAAU,GAAG,UAAU,sBAAsB;AAElD,SAAK,GAAG,UAAU,aAAa;AAAA,EACjC;AACF;AAEA,IAAM,oBAAoB;AAAA,EACxB,cAAc;AAAA,EACd,iBAAiB;AAAA,EACjB,SAAS;AACX;AAEA,SAAS,gBACP,SACA,SACA,KACA,mBACA,WAAW,OACX;AACA,QAAM,cAAuB,qBAAY,OAAO;AAChD,UAAQ,aAAa;AAAA,IACnB,KAAkB;AAChB,MAAa,2BAAc,SAAS,SAAS,GAAG;AAChD;AAAA,IACF,KAAkB;AAChB,UAAI,CAAC;AACH,QAAa,2BAAc,SAAS,KAAK,iBAAiB;AAC5D;AAAA,IACF,KAAkB;AAChB,UAAI,CAAC,SAAU,CAAa,wBAAW,SAAS,KAAK,iBAAiB;AACtE;AAAA,IACF;AACE,YAAM,IAAI,MAAM,sBAAsB;AAAA,EAC1C;AACA,SAAO;AACT;AAEA,SAAS,UAAU,KAAkB,MAAwB;AAC3D,MAAI,IAAI,MAAM,IAAI,IAAI,GAAG;AACvB,UAAM,gBAA6B,IAAI,MAAM,IAAI,IAAI;AACrD,QAAI,MAAM,OAAO,IAAI;AACrB,IAAkB;AAAA,MAChB,IAAI;AAAA,MACJ,MAAM,KAAK,aAAa;AAAA,MACxB;AAAA,IACF;AAAA,EACF;AACA,MAAI;AACF,SAAK,MAAM;AAAA,EACb,SAAS,GAAG;AACV,YAAQ,KAAK,8BAA8B,CAAC;AAAA,EAC9C;AACF;AAEA,SAAS,KAAK,KAAkB,MAAkB,GAAe;AAC/D,MACE,KAAK,eAAe,UACpB,KAAK,eAAe,0BACpB,KAAK,eAAe,kBACpB;AACA,cAAU,KAAK,IAAI;AAAA,EACrB;AACA,MAAI;AACF,SAAK,KAAK,CAAC;AAAA,EACb,SAAS,GAAG;AACV,cAAU,KAAK,IAAI;AAAA,EACrB;AACF;AAQO,IAAM,UAAN,cAAqC,OAAY;AAAA,EACtD,OAAO,kBAAmC,CAAC;AAAA,EAE3C,eAA+B,OAAO,eAAe,IAAI,EAAE;AAAA,EAClD,WAAW,IAAI,YAAY;AAAA,EAEpC,MAAM,SAAwB;AAE5B;AAAA,EACF;AAAA,EAEA,MAAM,SAAwB;AAAA,EAAC;AAAA,EAE/B,MAAM,UAAyB;AAC7B,UAAM,MAAM,MAAM,KAAK,OAAO;AAC9B,QAAI,OAAO,MAAM;AACf,YAAM,QAAQ,oBAAoB,GAAG;AACrC,kBAAY,KAAK,UAAU,KAAK;AAAA,IAClC;AAEA,SAAK,SAAS;AAAA,MACZ;AAAA,MACA;AAAA,QACE,CAAC,SAAqB,SAAqB,SAAe;AACxD,cAAI;AACF,iBAAK,OAAO,EAAE,MAAM,CAAC,QAAQ;AAC3B,sBAAQ,MAAM,sBAAsB,GAAG;AAAA,YACzC,CAAC;AAAA,UACH,SAAS,KAAK;AACZ,oBAAQ,MAAM,sBAAsB,GAAG;AAAA,UACzC;AAAA,QACF;AAAA,QACA,KAAK,aAAa,gBAAgB,gBAChC,kBAAkB;AAAA,QACpB;AAAA,UACE,SACE,KAAK,aAAa,gBAAgB,mBAClC,kBAAkB;AAAA,QACtB;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,WAAW,YAAiC;AAE1C,WAAO;AAAA,EACT;AAAA;AAAA,EAGA,cAAc,YAAwB,SAAoB;AACxD,QAAI,OAAO,YAAY,UAAU;AAC/B,cAAQ;AAAA,QACN,sDAAsD,KAAK,aAAa,IAAI;AAAA,MAC9E;AACA;AAAA,IACF;AACA,QAAI;AACF,YAAM,UAAmB,uBAAc;AAEvC,YAAM,UAAmB,uBAAc,OAAgC;AACvE,YAAM,cAAuB,qBAAY,OAAO;AAChD,cAAQ,aAAa;AAAA,QACnB,KAAK;AACH,UAAS,sBAAa,SAAS,WAAW;AAC1C;AAAA,YACE;AAAA,YACA;AAAA,YACA,KAAK;AAAA,YACL;AAAA,YACA,KAAK,WAAW,UAAU;AAAA,UAC5B;AAKA,cAAa,gBAAO,OAAO,IAAI,GAAG;AAChC,iBAAK,KAAK,UAAU,YAAqB,sBAAa,OAAO,CAAC;AAAA,UAChE;AACA;AAAA,QACF,KAAK,kBAAkB;AACrB,UAAkB;AAAA,YAChB,KAAK,SAAS;AAAA,YACL,2BAAkB,OAAO;AAAA,YAClC;AAAA,UACF;AACA;AAAA,QACF;AAAA,MACF;AAAA,IACF,SAAS,KAAK;AACZ,cAAQ,MAAM,GAAG;AAEjB,WAAK,SAAS,KAAK,SAAS,CAAC,GAAG,CAAC;AAAA,IACnC;AAAA,EACF;AAAA,EAEA,YAAY;AAAA,IAAc,CAAC,MAAM,YAC/B,KAAK,cAAc,MAAM,OAAO;AAAA,EAClC;AAAA,EAEA,QACE,YACA,OACA,SACA,WACsB;AACtB,cAAU,KAAK,UAAU,UAAU;AAAA,EACrC;AAAA;AAAA,EAIA,UACE,MACA,MACsB;AAGtB,SAAK,SAAS,MAAM,IAAI,MAAM,oBAAI,IAAI,CAAC;AAGvC,UAAM,UAAmB,uBAAc;AACvC,IAAS,sBAAa,SAAS,WAAW;AAC1C,IAAa,4BAAe,SAAS,KAAK,QAAQ;AAClD,SAAK,KAAK,UAAU,MAAe,sBAAa,OAAO,CAAC;AACxD,UAAM,kBAAkB,KAAK,SAAS,UAAU,UAAU;AAC1D,QAAI,gBAAgB,OAAO,GAAG;AAC5B,YAAMA,WAAmB,uBAAc;AACvC,MAAS,sBAAaA,UAAS,gBAAgB;AAC/C,MAAS;AAAA,QACPA;AAAA,QACkB;AAAA,UAChB,KAAK,SAAS;AAAA,UACd,MAAM,KAAK,gBAAgB,KAAK,CAAC;AAAA,QACnC;AAAA,MACF;AACA,WAAK,KAAK,UAAU,MAAe,sBAAaA,QAAO,CAAC;AAAA,IAC1D;AAAA,EACF;AACF;","names":["encoder"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "y-partyserver",
-  "version": "0.0.45",
+  "version": "0.0.47",
   "repository": {
     "type": "git",
     "url": "git://github.com/cloudflare/partykit.git"
@@ -47,13 +47,13 @@
   },
   "peerDependencies": {
     "@cloudflare/workers-types": "^4.20240729.0",
-    "partyserver": "^0.0.72",
+    "partyserver": "^0.0.73",
     "yjs": "^13.6.14"
   },
   "devDependencies": {
     "@cloudflare/workers-types": "^4.20250327.0",
     "@types/lodash.debounce": "^4.0.9",
-    "partyserver": "^0.0.72",
+    "partyserver": "^0.0.73",
     "ws": "^8.18.1",
     "yjs": "^13.6.24"
   }

package/dist/chunk-JFXBVIOX.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/shared/chunking.ts"],"sourcesContent":["// This file contains a shared implementation of chunking logic for binary\n// WebSocket messages. Because the Workers platform limits individual\n// WebSocket messages to 1MB, we need to split larger messages into chunks.\n\nimport type { Connection } from \"partyserver\";\n\nexport const CHUNK_MAX_SIZE = 1_000_000;\n\nconst BATCH_SENTINEL = \"y-pk-batch\";\nconst TRACE_ENABLED = false;\nconst trace = (...args: unknown[]) => TRACE_ENABLED && console.log(...args);\n\nlet warnedAboutLargeMessage = false;\n\ntype MessageData = ArrayBufferLike | string;\ntype MessageHandler = (conn: Connection, message: MessageData) => void;\ntype Batch = {\n  id: string;\n  type: \"start\" | \"end\";\n  size: number;\n  count: number;\n};\n\n/**\n * Takes an ArrayBuffer and sends it in chunks to the provided send function,\n * along with a start and end marker.\n *\n * 1. The sender splits the message into chunks of 1MB or less.\n * 2. The sender sends a chunk start marker\n * 3. The sender sends each chunk as a individual packet\n * 4. The sender sends a chunk end marker\n */\nexport const sendChunked = (data: Uint8Array, ws: WebSocket) => {\n  if (data.byteLength <= CHUNK_MAX_SIZE) {\n    ws.send(data);\n    return;\n  }\n\n  if (!warnedAboutLargeMessage) {\n    console.warn(\n      \"[y-partyserver]\",\n      \"The Y.js update size exceeds 1MB, which is the maximum size for an individual update. The update will be split into chunks. This is an experimental feature.\",\n      `Message size: ${(data.byteLength / 1000 / 1000).toFixed(1)}MB`\n    );\n    warnedAboutLargeMessage = true;\n  }\n\n  const id = (Date.now() + Math.random()).toString(36).substring(10);\n  const chunks = Math.ceil(data.byteLength / CHUNK_MAX_SIZE);\n\n  ws.send(\n    serializeBatchMarker({\n      id,\n      type: \"start\",\n      size: data.byteLength,\n      count: chunks\n    })\n  );\n\n  let sentSize = 0;\n  let sentChunks = 0;\n  for (let i = 0; i < chunks; i++) {\n    const chunk = data.slice(CHUNK_MAX_SIZE * i, CHUNK_MAX_SIZE * (i + 1));\n    ws.send(chunk);\n    sentChunks += 1;\n    sentSize += chunk.byteLength;\n  }\n\n  ws.send(\n    serializeBatchMarker({\n      id,\n      type: \"end\",\n      size: sentSize,\n      count: sentChunks\n    })\n  );\n};\n\n/**\n * Creates a WebSocket message handler that can handle chunked messages.\n *\n * Reassembles the chunks into a single ArrayBuffer and pass it to the\n * provided receive function.\n *\n * 1. The server receives a chunk start marker\n * 2. The server pools each chunk until it receives a chunk end marker\n * 3. The server validates that the received data matches the expected size\n * 4. The server forwards the message to handlers\n */\nexport const handleChunked = (\n  receive: (conn: Connection, data: MessageData) => void\n): MessageHandler => {\n  let batch: ArrayBuffer[] | undefined;\n  let start: Batch | undefined;\n\n  return (connection, message) => {\n    if (typeof message === \"string\" && !isBatchSentinel(message)) {\n      return;\n    }\n    if (isBatchSentinel(message)) {\n      const marker = parseBatchMarker(message);\n      if (marker.type === \"start\") {\n        batch = [];\n        start = marker;\n      }\n\n      if (marker.type === \"end\") {\n        if (batch) {\n          try {\n            // validate start and end markers match\n            assertEquality(start?.id, marker.id, \"client id\");\n            assertEquality(start?.count, marker.count, \"client counts\");\n            assertEquality(start?.size, marker.size, \"client size\");\n\n            // combine chunks into single buffer\n            const size = batch.reduce(\n              (sum, buffer) => sum + buffer.byteLength,\n              0\n            );\n            const bytes = new Uint8Array(size);\n            let bytesWritten = 0;\n            for (const chunk of batch) {\n              bytes.set(new Uint8Array(chunk), bytesWritten);\n              bytesWritten += chunk.byteLength;\n            }\n\n            // validate data as read matches expected\n            assertEquality(marker.count, batch.length, \"received batch count\");\n            assertEquality(marker.size, bytesWritten, \"client size\");\n            // @ts-ignore typescript hell\n            receive(connection, bytes);\n          } catch (e) {\n            console.error(e);\n            throw e;\n          } finally {\n            batch = undefined;\n            start = undefined;\n          }\n        }\n      }\n    } else if (batch) {\n      // @ts-ignore typescript hell\n      batch.push(message);\n    } else {\n      // @ts-ignore typescript hell\n      receive(connection, new Uint8Array(message));\n    }\n  };\n};\n\nfunction assertEquality(expected: unknown, actual: unknown, label: string) {\n  if (expected !== actual) {\n    throw new Error(\n      `Mismatching ${label}! Expected ${expected as string}, got ${actual as string}`\n    );\n  } else {\n    trace(`Matching ${label}: ${expected as string}`);\n  }\n}\n\n/** Checks whether a message is batch marker */\nfunction isBatchSentinel(msg: MessageData): msg is string {\n  return typeof msg === \"string\" && msg.startsWith(BATCH_SENTINEL);\n}\n\n/** Encodes a batch marker message so that it can be safely parsed */\nfunction serializeBatchMarker(batch: Batch): string {\n  return `${BATCH_SENTINEL}#${JSON.stringify(batch)}`;\n}\n\n/** Parses a batch marker messages and throws if its invalid */\nexport function parseBatchMarker(msg: string) {\n  const [sentinel, data] = msg.split(\"#\", 2);\n  if (sentinel !== BATCH_SENTINEL) {\n    throw new Error(`Unexpected batch marker: ${msg}`);\n  }\n\n  const batch = JSON.parse(data) as Batch;\n  if (batch.type !== \"start\" && batch.type !== \"end\") {\n    throw new Error(`Unexpected batch data: ${msg}`);\n  }\n\n  return batch;\n}\n"],"mappings":";AAMO,IAAM,iBAAiB;AAE9B,IAAM,iBAAiB;AACvB,IAAM,gBAAgB;AACtB,IAAM,QAAQ,IAAI,SAAoB,iBAAiB,QAAQ,IAAI,GAAG,IAAI;AAE1E,IAAI,0BAA0B;AAoBvB,IAAM,cAAc,CAAC,MAAkB,OAAkB;AAC9D,MAAI,KAAK,cAAc,gBAAgB;AACrC,OAAG,KAAK,IAAI;AACZ;AAAA,EACF;AAEA,MAAI,CAAC,yBAAyB;AAC5B,YAAQ;AAAA,MACN;AAAA,MACA;AAAA,MACA,kBAAkB,KAAK,aAAa,MAAO,KAAM,QAAQ,CAAC,CAAC;AAAA,IAC7D;AACA,8BAA0B;AAAA,EAC5B;AAEA,QAAM,MAAM,KAAK,IAAI,IAAI,KAAK,OAAO,GAAG,SAAS,EAAE,EAAE,UAAU,EAAE;AACjE,QAAM,SAAS,KAAK,KAAK,KAAK,aAAa,cAAc;AAEzD,KAAG;AAAA,IACD,qBAAqB;AAAA,MACnB;AAAA,MACA,MAAM;AAAA,MACN,MAAM,KAAK;AAAA,MACX,OAAO;AAAA,IACT,CAAC;AAAA,EACH;AAEA,MAAI,WAAW;AACf,MAAI,aAAa;AACjB,WAAS,IAAI,GAAG,IAAI,QAAQ,KAAK;AAC/B,UAAM,QAAQ,KAAK,MAAM,iBAAiB,GAAG,kBAAkB,IAAI,EAAE;AACrE,OAAG,KAAK,KAAK;AACb,kBAAc;AACd,gBAAY,MAAM;AAAA,EACpB;AAEA,KAAG;AAAA,IACD,qBAAqB;AAAA,MACnB;AAAA,MACA,MAAM;AAAA,MACN,MAAM;AAAA,MACN,OAAO;AAAA,IACT,CAAC;AAAA,EACH;AACF;AAaO,IAAM,gBAAgB,CAC3B,YACmB;AACnB,MAAI;AACJ,MAAI;AAEJ,SAAO,CAAC,YAAY,YAAY;AAC9B,QAAI,OAAO,YAAY,YAAY,CAAC,gBAAgB,OAAO,GAAG;AAC5D;AAAA,IACF;AACA,QAAI,gBAAgB,OAAO,GAAG;AAC5B,YAAM,SAAS,iBAAiB,OAAO;AACvC,UAAI,OAAO,SAAS,SAAS;AAC3B,gBAAQ,CAAC;AACT,gBAAQ;AAAA,MACV;AAEA,UAAI,OAAO,SAAS,OAAO;AACzB,YAAI,OAAO;AACT,cAAI;AAEF,2BAAe,+BAAO,IAAI,OAAO,IAAI,WAAW;AAChD,2BAAe,+BAAO,OAAO,OAAO,OAAO,eAAe;AAC1D,2BAAe,+BAAO,MAAM,OAAO,MAAM,aAAa;AAGtD,kBAAM,OAAO,MAAM;AAAA,cACjB,CAAC,KAAK,WAAW,MAAM,OAAO;AAAA,cAC9B;AAAA,YACF;AACA,kBAAM,QAAQ,IAAI,WAAW,IAAI;AACjC,gBAAI,eAAe;AACnB,uBAAW,SAAS,OAAO;AACzB,oBAAM,IAAI,IAAI,WAAW,KAAK,GAAG,YAAY;AAC7C,8BAAgB,MAAM;AAAA,YACxB;AAGA,2BAAe,OAAO,OAAO,MAAM,QAAQ,sBAAsB;AACjE,2BAAe,OAAO,MAAM,cAAc,aAAa;AAEvD,oBAAQ,YAAY,KAAK;AAAA,UAC3B,SAAS,GAAG;AACV,oBAAQ,MAAM,CAAC;AACf,kBAAM;AAAA,UACR,UAAE;AACA,oBAAQ;AACR,oBAAQ;AAAA,UACV;AAAA,QACF;AAAA,MACF;AAAA,IACF,WAAW,OAAO;AAEhB,YAAM,KAAK,OAAO;AAAA,IACpB,OAAO;AAEL,cAAQ,YAAY,IAAI,WAAW,OAAO,CAAC;AAAA,IAC7C;AAAA,EACF;AACF;AAEA,SAAS,eAAe,UAAmB,QAAiB,OAAe;AACzE,MAAI,aAAa,QAAQ;AACvB,UAAM,IAAI;AAAA,MACR,eAAe,KAAK,cAAc,QAAkB,SAAS,MAAgB;AAAA,IAC/E;AAAA,EACF,OAAO;AACL,UAAM,YAAY,KAAK,KAAK,QAAkB,EAAE;AAAA,EAClD;AACF;AAGA,SAAS,gBAAgB,KAAiC;AACxD,SAAO,OAAO,QAAQ,YAAY,IAAI,WAAW,cAAc;AACjE;AAGA,SAAS,qBAAqB,OAAsB;AAClD,SAAO,GAAG,cAAc,IAAI,KAAK,UAAU,KAAK,CAAC;AACnD;AAGO,SAAS,iBAAiB,KAAa;AAC5C,QAAM,CAAC,UAAU,IAAI,IAAI,IAAI,MAAM,KAAK,CAAC;AACzC,MAAI,aAAa,gBAAgB;AAC/B,UAAM,IAAI,MAAM,4BAA4B,GAAG,EAAE;AAAA,EACnD;AAEA,QAAM,QAAQ,KAAK,MAAM,IAAI;AAC7B,MAAI,MAAM,SAAS,WAAW,MAAM,SAAS,OAAO;AAClD,UAAM,IAAI,MAAM,0BAA0B,GAAG,EAAE;AAAA,EACjD;AAEA,SAAO;AACT;","names":[]}