y-partyserver 2.1.0 → 2.1.2

package/dist/provider/react.cjs

@@ -0,0 +1,34 @@
+const require_chunk = require("../chunk-C0xms8kb.cjs");
+const require_provider_index = require("./index.cjs");
+let react = require("react");
+
+//#region src/provider/react.tsx
+function useYProvider(yProviderOptions) {
+  const { host, room, party, doc, options, prefix } = yProviderOptions;
+  const [provider] = (0, react.useState)(
+    () =>
+      new require_provider_index.default(
+        host ||
+          (typeof window !== "undefined"
+            ? window.location.host
+            : "dummy-domain.com"),
+        room,
+        doc,
+        {
+          connect: false,
+          party,
+          prefix,
+          ...options
+        }
+      )
+  );
+  (0, react.useEffect)(() => {
+    provider.connect();
+    return () => provider.disconnect();
+  }, [provider]);
+  return provider;
+}
+
+//#endregion
+module.exports = useYProvider;
+//# sourceMappingURL=react.cjs.map

package/dist/provider/react.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"react.cjs","names":["YProvider"],"sources":["../../src/provider/react.tsx"],"sourcesContent":["import { useEffect, useState } from \"react\";\n\nimport YProvider from \"./index\";\n\nimport type * as Y from \"yjs\";\n\ntype UseYProviderOptions = {\n  host?: string | undefined;\n  room: string;\n  party?: string;\n  doc?: Y.Doc;\n  prefix?: string;\n  options?: ConstructorParameters<typeof YProvider>[3];\n};\n\nexport default function useYProvider(yProviderOptions: UseYProviderOptions) {\n  const { host, room, party, doc, options, prefix } = yProviderOptions;\n  const [provider] = useState<YProvider>(\n    () =>\n      new YProvider(\n        host ||\n          (typeof window !== \"undefined\"\n            ? window.location.host\n            : \"dummy-domain.com\"),\n        room,\n        doc,\n        {\n          connect: false,\n          party,\n          prefix,\n          ...options\n        }\n      )\n  );\n\n  useEffect(() => {\n    void provider.connect();\n    return () => provider.disconnect();\n  }, [provider]);\n  return provider;\n}\n"],"mappings":";;;;;AAeA,SAAwB,aAAa,kBAAuC;CAC1E,MAAM,EAAE,MAAM,MAAM,OAAO,KAAK,SAAS,WAAW;CACpD,MAAM,CAAC,sCAEH,IAAIA,+BACF,SACG,OAAO,WAAW,cACf,OAAO,SAAS,OAChB,qBACN,MACA,KACA;EACE,SAAS;EACT;EACA;EACA,GAAG;EACJ,CACF,CACJ;AAED,4BAAgB;AACd,EAAK,SAAS,SAAS;AACvB,eAAa,SAAS,YAAY;IACjC,CAAC,SAAS,CAAC;AACd,QAAO"}

package/dist/provider/react.d.cts

@@ -0,0 +1,15 @@
+import YProvider from "./index.cjs";
+import * as Y from "yjs";
+
+//#region src/provider/react.d.ts
+type UseYProviderOptions = {
+  host?: string | undefined;
+  room: string;
+  party?: string;
+  doc?: Y.Doc;
+  prefix?: string;
+  options?: ConstructorParameters<typeof YProvider>[3];
+};
+declare function useYProvider(yProviderOptions: UseYProviderOptions): YProvider;
+export = useYProvider;
+//# sourceMappingURL=react.d.cts.map

package/dist/provider/react.js

@@ -3,20 +3,31 @@ import { useEffect, useState } from "react";
 
 //#region src/provider/react.tsx
 function useYProvider(yProviderOptions) {
-	const { host, room, party, doc, options, prefix } = yProviderOptions;
-	const [provider] = useState(() => new YProvider(host || (typeof window !== "undefined" ? window.location.host : "dummy-domain.com"), room, doc, {
-		connect: false,
-		party,
-		prefix,
-		...options
-	}));
-	useEffect(() => {
-		provider.connect();
-		return () => provider.disconnect();
-	}, [provider]);
-	return provider;
+  const { host, room, party, doc, options, prefix } = yProviderOptions;
+  const [provider] = useState(
+    () =>
+      new YProvider(
+        host ||
+          (typeof window !== "undefined"
+            ? window.location.host
+            : "dummy-domain.com"),
+        room,
+        doc,
+        {
+          connect: false,
+          party,
+          prefix,
+          ...options
+        }
+      )
+  );
+  useEffect(() => {
+    provider.connect();
+    return () => provider.disconnect();
+  }, [provider]);
+  return provider;
 }
 
 //#endregion
 export { useYProvider as default };
-//# sourceMappingURL=react.js.map
+//# sourceMappingURL=react.js.map

package/dist/server/index.cjs

@@ -0,0 +1,366 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const require_chunk = require("../chunk-C0xms8kb.cjs");
+let lib0_decoding = require("lib0/decoding");
+lib0_decoding = require_chunk.__toESM(lib0_decoding);
+let lib0_encoding = require("lib0/encoding");
+lib0_encoding = require_chunk.__toESM(lib0_encoding);
+let lodash_debounce = require("lodash.debounce");
+lodash_debounce = require_chunk.__toESM(lodash_debounce);
+let partyserver = require("partyserver");
+let y_protocols_awareness = require("y-protocols/awareness");
+y_protocols_awareness = require_chunk.__toESM(y_protocols_awareness);
+let y_protocols_sync = require("y-protocols/sync");
+y_protocols_sync = require_chunk.__toESM(y_protocols_sync);
+let yjs = require("yjs");
+
+//#region src/server/index.ts
+const snapshotOrigin = Symbol("snapshot-origin");
+const wsReadyStateConnecting = 0;
+const wsReadyStateOpen = 1;
+const messageSync = 0;
+const messageAwareness = 1;
+/**
+ * Internal key used in connection.setState() to track which awareness
+ * client IDs are controlled by each connection. This survives hibernation
+ * because connection state is persisted to WebSocket attachments.
+ */
+const AWARENESS_IDS_KEY = "__ypsAwarenessIds";
+function getAwarenessIds(conn) {
+  try {
+    return conn.state?.[AWARENESS_IDS_KEY] ?? [];
+  } catch {
+    return [];
+  }
+}
+function setAwarenessIds(conn, ids) {
+  try {
+    conn.setState((prev) => ({
+      ...prev,
+      [AWARENESS_IDS_KEY]: ids
+    }));
+  } catch {}
+}
+var WSSharedDoc = class extends yjs.Doc {
+  awareness;
+  constructor() {
+    super({ gc: true });
+    this.awareness = new y_protocols_awareness.Awareness(this);
+    this.awareness.setLocalState(null);
+    clearInterval(this.awareness._checkInterval);
+  }
+};
+const CALLBACK_DEFAULTS = {
+  debounceWait: 2e3,
+  debounceMaxWait: 1e4,
+  timeout: 5e3
+};
+function readSyncMessage(
+  decoder,
+  encoder,
+  doc,
+  transactionOrigin,
+  readOnly = false
+) {
+  const messageType = lib0_decoding.readVarUint(decoder);
+  switch (messageType) {
+    case y_protocols_sync.messageYjsSyncStep1:
+      y_protocols_sync.readSyncStep1(decoder, encoder, doc);
+      break;
+    case y_protocols_sync.messageYjsSyncStep2:
+      if (!readOnly)
+        y_protocols_sync.readSyncStep2(decoder, doc, transactionOrigin);
+      break;
+    case y_protocols_sync.messageYjsUpdate:
+      if (!readOnly)
+        y_protocols_sync.readUpdate(decoder, doc, transactionOrigin);
+      break;
+    default:
+      throw new Error("Unknown message type");
+  }
+  return messageType;
+}
+function send(conn, m) {
+  if (
+    conn.readyState !== void 0 &&
+    conn.readyState !== wsReadyStateConnecting &&
+    conn.readyState !== wsReadyStateOpen
+  )
+    return;
+  try {
+    conn.send(m);
+  } catch {}
+}
+function withYjs(Base) {
+  class YjsMixin extends Base {
+    static callbackOptions = {};
+    document = new WSSharedDoc();
+    async onLoad() {}
+    async onSave() {}
+    /**
+     * Replaces the document with a different state using Yjs UndoManager key remapping.
+     *
+     * @param snapshotUpdate - The snapshot update to replace the document with.
+     * @param getMetadata (optional) - A function that returns the type of the root for a given key.
+     */
+    unstable_replaceDocument(snapshotUpdate, getMetadata = () => "Map") {
+      try {
+        const doc = this.document;
+        const snapshotDoc = new yjs.Doc();
+        (0, yjs.applyUpdate)(snapshotDoc, snapshotUpdate, snapshotOrigin);
+        const currentStateVector = (0, yjs.encodeStateVector)(doc);
+        const changesSinceSnapshotUpdate = (0, yjs.encodeStateAsUpdate)(
+          doc,
+          (0, yjs.encodeStateVector)(snapshotDoc)
+        );
+        const undoManager = new yjs.UndoManager(
+          [...snapshotDoc.share.keys()].map((key) => {
+            const type = getMetadata(key);
+            if (type === "Text") return snapshotDoc.getText(key);
+            else if (type === "Map") return snapshotDoc.getMap(key);
+            else if (type === "Array") return snapshotDoc.getArray(key);
+            else if (type === "XmlText")
+              return snapshotDoc.get(key, yjs.XmlText);
+            else if (type === "XmlElement")
+              return snapshotDoc.get(key, yjs.XmlElement);
+            else if (type === "XmlFragment")
+              return snapshotDoc.get(key, yjs.XmlFragment);
+            throw new Error(`Unknown root type: ${type} for key: ${key}`);
+          }),
+          { trackedOrigins: new Set([snapshotOrigin]) }
+        );
+        (0, yjs.applyUpdate)(
+          snapshotDoc,
+          changesSinceSnapshotUpdate,
+          snapshotOrigin
+        );
+        undoManager.undo();
+        const documentChangesSinceSnapshotUpdate = (0, yjs.encodeStateAsUpdate)(
+          snapshotDoc,
+          currentStateVector
+        );
+        (0, yjs.applyUpdate)(this.document, documentChangesSinceSnapshotUpdate);
+      } catch (error) {
+        throw new Error(
+          `Failed to replace document: ${error instanceof Error ? error.message : "Unknown error"}`
+        );
+      }
+    }
+    async onStart() {
+      const src = await this.onLoad();
+      if (src != null) {
+        const state = (0, yjs.encodeStateAsUpdate)(src);
+        (0, yjs.applyUpdate)(this.document, state);
+      }
+      this.document.on("update", (update) => {
+        const encoder = lib0_encoding.createEncoder();
+        lib0_encoding.writeVarUint(encoder, messageSync);
+        y_protocols_sync.writeUpdate(encoder, update);
+        const message = lib0_encoding.toUint8Array(encoder);
+        for (const conn of this.getConnections()) send(conn, message);
+      });
+      this.document.awareness.on(
+        "update",
+        ({ added, updated, removed }, conn) => {
+          if (conn !== null)
+            try {
+              const currentIds = new Set(getAwarenessIds(conn));
+              for (const clientID of added) currentIds.add(clientID);
+              for (const clientID of removed) currentIds.delete(clientID);
+              setAwarenessIds(conn, [...currentIds]);
+            } catch (_e) {}
+          else {
+            const changedClients = added.concat(updated, removed);
+            const encoder = lib0_encoding.createEncoder();
+            lib0_encoding.writeVarUint(encoder, messageAwareness);
+            lib0_encoding.writeVarUint8Array(
+              encoder,
+              y_protocols_awareness.encodeAwarenessUpdate(
+                this.document.awareness,
+                changedClients
+              )
+            );
+            const buff = lib0_encoding.toUint8Array(encoder);
+            for (const c of this.getConnections()) send(c, buff);
+          }
+        }
+      );
+      const ctor = this.constructor;
+      this.document.on(
+        "update",
+        (0, lodash_debounce.default)(
+          (_update, _origin, _doc) => {
+            try {
+              this.onSave().catch((err) => {
+                console.error("failed to persist:", err);
+              });
+            } catch (err) {
+              console.error("failed to persist:", err);
+            }
+          },
+          ctor.callbackOptions.debounceWait || CALLBACK_DEFAULTS.debounceWait,
+          {
+            maxWait:
+              ctor.callbackOptions.debounceMaxWait ||
+              CALLBACK_DEFAULTS.debounceMaxWait
+          }
+        )
+      );
+      const syncEncoder = lib0_encoding.createEncoder();
+      lib0_encoding.writeVarUint(syncEncoder, messageSync);
+      y_protocols_sync.writeSyncStep1(syncEncoder, this.document);
+      const syncMessage = lib0_encoding.toUint8Array(syncEncoder);
+      for (const conn of this.getConnections()) send(conn, syncMessage);
+    }
+    isReadOnly(connection) {
+      return false;
+    }
+    /**
+     * Handle custom string messages from the client.
+     * Override this method to implement custom message handling.
+     * @param connection - The connection that sent the message
+     * @param message - The custom message string (without the __YPS: prefix)
+     */
+    onCustomMessage(connection, message) {
+      console.warn(
+        `Received custom message but onCustomMessage is not implemented in ${this.constructor.name}:`,
+        message
+      );
+    }
+    /**
+     * Send a custom string message to a specific connection.
+     * @param connection - The connection to send the message to
+     * @param message - The custom message string to send
+     */
+    sendCustomMessage(connection, message) {
+      if (
+        connection.readyState !== void 0 &&
+        connection.readyState !== wsReadyStateConnecting &&
+        connection.readyState !== wsReadyStateOpen
+      )
+        return;
+      try {
+        connection.send(`__YPS:${message}`);
+      } catch (e) {
+        console.warn("Failed to send custom message", e);
+      }
+    }
+    /**
+     * Broadcast a custom string message to all connected clients.
+     * @param message - The custom message string to broadcast
+     * @param excludeConnection - Optional connection to exclude from the broadcast
+     */
+    broadcastCustomMessage(message, excludeConnection) {
+      const formattedMessage = `__YPS:${message}`;
+      for (const conn of this.getConnections()) {
+        if (excludeConnection && conn === excludeConnection) continue;
+        if (
+          conn.readyState !== void 0 &&
+          conn.readyState !== wsReadyStateConnecting &&
+          conn.readyState !== wsReadyStateOpen
+        )
+          continue;
+        try {
+          conn.send(formattedMessage);
+        } catch (e) {
+          console.warn("Failed to broadcast custom message", e);
+        }
+      }
+    }
+    handleMessage(connection, message) {
+      if (typeof message === "string") {
+        if (message.startsWith("__YPS:")) {
+          const customMessage = message.slice(6);
+          this.onCustomMessage(connection, customMessage);
+          return;
+        }
+        console.warn(
+          `Received non-prefixed string message. Custom messages should be sent using sendMessage() on the provider.`
+        );
+        return;
+      }
+      try {
+        const encoder = lib0_encoding.createEncoder();
+        const uint8Array =
+          message instanceof Uint8Array
+            ? message
+            : message instanceof ArrayBuffer
+              ? new Uint8Array(message)
+              : new Uint8Array(
+                  message.buffer,
+                  message.byteOffset,
+                  message.byteLength
+                );
+        const decoder = lib0_decoding.createDecoder(uint8Array);
+        switch (lib0_decoding.readVarUint(decoder)) {
+          case messageSync:
+            lib0_encoding.writeVarUint(encoder, messageSync);
+            readSyncMessage(
+              decoder,
+              encoder,
+              this.document,
+              connection,
+              this.isReadOnly(connection)
+            );
+            if (lib0_encoding.length(encoder) > 1)
+              send(connection, lib0_encoding.toUint8Array(encoder));
+            break;
+          case messageAwareness: {
+            const awarenessData = lib0_decoding.readVarUint8Array(decoder);
+            y_protocols_awareness.applyAwarenessUpdate(
+              this.document.awareness,
+              awarenessData,
+              connection
+            );
+            const awarenessEncoder = lib0_encoding.createEncoder();
+            lib0_encoding.writeVarUint(awarenessEncoder, messageAwareness);
+            lib0_encoding.writeVarUint8Array(awarenessEncoder, awarenessData);
+            const awarenessBuff = lib0_encoding.toUint8Array(awarenessEncoder);
+            for (const c of this.getConnections()) send(c, awarenessBuff);
+            break;
+          }
+        }
+      } catch (err) {
+        console.error(err);
+        this.document.emit("error", [err]);
+      }
+    }
+    onMessage(conn, message) {
+      this.handleMessage(conn, message);
+    }
+    onClose(connection, _code, _reason, _wasClean) {
+      const controlledIds = getAwarenessIds(connection);
+      if (controlledIds.length > 0)
+        y_protocols_awareness.removeAwarenessStates(
+          this.document.awareness,
+          controlledIds,
+          null
+        );
+    }
+    onConnect(conn, _ctx) {
+      const encoder = lib0_encoding.createEncoder();
+      lib0_encoding.writeVarUint(encoder, messageSync);
+      y_protocols_sync.writeSyncStep1(encoder, this.document);
+      send(conn, lib0_encoding.toUint8Array(encoder));
+      const awarenessStates = this.document.awareness.getStates();
+      if (awarenessStates.size > 0) {
+        const encoder = lib0_encoding.createEncoder();
+        lib0_encoding.writeVarUint(encoder, messageAwareness);
+        lib0_encoding.writeVarUint8Array(
+          encoder,
+          y_protocols_awareness.encodeAwarenessUpdate(
+            this.document.awareness,
+            Array.from(awarenessStates.keys())
+          )
+        );
+        send(conn, lib0_encoding.toUint8Array(encoder));
+      }
+    }
+  }
+  return YjsMixin;
+}
+const YServer = withYjs(partyserver.Server);
+
+//#endregion
+exports.YServer = YServer;
+exports.withYjs = withYjs;
+//# sourceMappingURL=index.cjs.map

package/dist/server/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","names":["YDoc","awarenessProtocol","decoding","syncProtocol","UndoManager","XmlText","XmlElement","XmlFragment","encoding","Server"],"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 {\n  applyUpdate,\n  Doc as YDoc,\n  encodeStateAsUpdate,\n  encodeStateVector,\n  UndoManager,\n  XmlText,\n  XmlElement,\n  XmlFragment\n} from \"yjs\";\n\nconst snapshotOrigin = Symbol(\"snapshot-origin\");\ntype YjsRootType =\n  | \"Text\"\n  | \"Map\"\n  | \"Array\"\n  | \"XmlText\"\n  | \"XmlElement\"\n  | \"XmlFragment\";\n\nconst wsReadyStateConnecting = 0;\nconst wsReadyStateOpen = 1;\n// oxlint-disable-next-line no-unused-vars\nconst wsReadyStateClosing = 2;\n// oxlint-disable-next-line no-unused-vars\nconst wsReadyStateClosed = 3;\n\nconst messageSync = 0;\nconst messageAwareness = 1;\n// oxlint-disable-next-line no-unused-vars\nconst messageAuth = 2;\n\n/**\n * Internal key used in connection.setState() to track which awareness\n * client IDs are controlled by each connection. This survives hibernation\n * because connection state is persisted to WebSocket attachments.\n */\nconst AWARENESS_IDS_KEY = \"__ypsAwarenessIds\";\n\ntype YServerConnectionState = {\n  [AWARENESS_IDS_KEY]?: number[];\n  [key: string]: unknown;\n};\n\nfunction getAwarenessIds(conn: Connection): number[] {\n  try {\n    const state = conn.state as YServerConnectionState | null;\n    return state?.[AWARENESS_IDS_KEY] ?? [];\n  } catch {\n    return [];\n  }\n}\n\nfunction setAwarenessIds(conn: Connection, ids: number[]): void {\n  try {\n    conn.setState((prev: YServerConnectionState | null) => ({\n      ...prev,\n      [AWARENESS_IDS_KEY]: ids\n    }));\n  } catch {\n    // ignore — may fail if connection is already closed\n  }\n}\n\nclass WSSharedDoc extends YDoc {\n  awareness: awarenessProtocol.Awareness;\n\n  constructor() {\n    super({ gc: true });\n    this.awareness = new awarenessProtocol.Awareness(this);\n    this.awareness.setLocalState(null);\n\n    // Disable the awareness protocol's built-in check interval.\n    // It renews the local clock every 15s and removes peers after 30s,\n    // but we handle peer cleanup via onClose instead. Clearing it here\n    // prevents it from defeating Durable Object hibernation.\n    clearInterval(\n      (\n        this.awareness as unknown as {\n          _checkInterval: ReturnType<typeof setInterval>;\n        }\n      )._checkInterval\n    );\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 send(conn: Connection, m: Uint8Array): void {\n  if (\n    conn.readyState !== undefined &&\n    conn.readyState !== wsReadyStateConnecting &&\n    conn.readyState !== wsReadyStateOpen\n  ) {\n    return;\n  }\n  try {\n    conn.send(m);\n  } catch {\n    // connection is broken, ignore\n  }\n}\n\nexport interface CallbackOptions {\n  debounceWait?: number;\n  debounceMaxWait?: number;\n  timeout?: number;\n}\n\ntype ServerClass = new (...args: any[]) => Server;\n\nexport interface YjsInstance {\n  readonly document: WSSharedDoc;\n  onLoad(): Promise<YDoc | void>;\n  onSave(): Promise<void>;\n  unstable_replaceDocument(\n    snapshotUpdate: Uint8Array,\n    getMetadata?: (key: string) => YjsRootType\n  ): void;\n  isReadOnly(connection: Connection): boolean;\n  onCustomMessage(connection: Connection, message: string): void;\n  sendCustomMessage(connection: Connection, message: string): void;\n  broadcastCustomMessage(message: string, excludeConnection?: Connection): void;\n  handleMessage(connection: Connection, message: WSMessage): void;\n}\n\nexport interface YjsStatic {\n  callbackOptions: CallbackOptions;\n}\n\nexport function withYjs<TBase extends ServerClass>(\n  Base: TBase\n): TBase & YjsStatic & (new (...args: any[]) => YjsInstance) {\n  class YjsMixin extends Base {\n    static callbackOptions: CallbackOptions = {};\n\n    readonly document: WSSharedDoc = new WSSharedDoc();\n\n    async onLoad(): Promise<YDoc | void> {\n      // to be implemented by the user\n      return;\n    }\n\n    async onSave(): Promise<void> {\n      // to be implemented by the user\n    }\n\n    /**\n     * Replaces the document with a different state using Yjs UndoManager key remapping.\n     *\n     * @param snapshotUpdate - The snapshot update to replace the document with.\n     * @param getMetadata (optional) - A function that returns the type of the root for a given key.\n     */\n    unstable_replaceDocument(\n      snapshotUpdate: Uint8Array,\n      getMetadata: (key: string) => YjsRootType = () => \"Map\"\n    ): void {\n      try {\n        const doc = this.document;\n        const snapshotDoc = new YDoc();\n        applyUpdate(snapshotDoc, snapshotUpdate, snapshotOrigin);\n\n        const currentStateVector = encodeStateVector(doc);\n        const snapshotStateVector = encodeStateVector(snapshotDoc);\n\n        const changesSinceSnapshotUpdate = encodeStateAsUpdate(\n          doc,\n          snapshotStateVector\n        );\n\n        const undoManager = new UndoManager(\n          [...snapshotDoc.share.keys()].map((key) => {\n            const type = getMetadata(key);\n            if (type === \"Text\") {\n              return snapshotDoc.getText(key);\n            } else if (type === \"Map\") {\n              return snapshotDoc.getMap(key);\n            } else if (type === \"Array\") {\n              return snapshotDoc.getArray(key);\n            } else if (type === \"XmlText\") {\n              return snapshotDoc.get(key, XmlText);\n            } else if (type === \"XmlElement\") {\n              return snapshotDoc.get(key, XmlElement);\n            } else if (type === \"XmlFragment\") {\n              return snapshotDoc.get(key, XmlFragment);\n            }\n            throw new Error(`Unknown root type: ${type} for key: ${key}`);\n          }),\n          {\n            trackedOrigins: new Set([snapshotOrigin])\n          }\n        );\n\n        applyUpdate(snapshotDoc, changesSinceSnapshotUpdate, snapshotOrigin);\n        undoManager.undo();\n\n        const documentChangesSinceSnapshotUpdate = encodeStateAsUpdate(\n          snapshotDoc,\n          currentStateVector\n        );\n\n        applyUpdate(this.document, documentChangesSinceSnapshotUpdate);\n      } catch (error) {\n        throw new Error(\n          `Failed to replace document: ${error instanceof Error ? error.message : \"Unknown error\"}`\n        );\n      }\n    }\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      // Broadcast doc updates to all connections.\n      // Uses this.getConnections() which works for both hibernate and non-hibernate\n      // modes and survives DO hibernation (unlike an in-memory Map).\n      this.document.on(\"update\", (update: Uint8Array) => {\n        const encoder = encoding.createEncoder();\n        encoding.writeVarUint(encoder, messageSync);\n        syncProtocol.writeUpdate(encoder, update);\n        const message = encoding.toUint8Array(encoder);\n        for (const conn of this.getConnections()) {\n          send(conn, message);\n        }\n      });\n\n      // Track which awareness clientIDs each connection controls.\n      // Stored in connection.setState() so it survives hibernation.\n      // When conn is null (internal changes like removeAwarenessStates on close),\n      // broadcast the update to remaining connections.\n      // When conn is non-null (client message), handleMessage broadcasts directly.\n      this.document.awareness.on(\n        \"update\",\n        (\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\n        ) => {\n          if (conn !== null) {\n            // Track which clientIDs this connection controls\n            try {\n              const currentIds = new Set(getAwarenessIds(conn));\n              for (const clientID of added) currentIds.add(clientID);\n              for (const clientID of removed) currentIds.delete(clientID);\n              setAwarenessIds(conn, [...currentIds]);\n            } catch (_e) {\n              // ignore — best-effort tracking\n            }\n          } else {\n            // Internal awareness change (e.g. removeAwarenessStates on close)\n            // — broadcast to all remaining connections\n            const changedClients = added.concat(updated, removed);\n            const encoder = encoding.createEncoder();\n            encoding.writeVarUint(encoder, messageAwareness);\n            encoding.writeVarUint8Array(\n              encoder,\n              awarenessProtocol.encodeAwarenessUpdate(\n                this.document.awareness,\n                changedClients\n              )\n            );\n            const buff = encoding.toUint8Array(encoder);\n            for (const c of this.getConnections()) {\n              send(c, buff);\n            }\n          }\n        }\n      );\n\n      // Debounced persistence handler\n      const ctor = this.constructor as typeof YjsMixin;\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          ctor.callbackOptions.debounceWait || CALLBACK_DEFAULTS.debounceWait,\n          {\n            maxWait:\n              ctor.callbackOptions.debounceMaxWait ||\n              CALLBACK_DEFAULTS.debounceMaxWait\n          }\n        )\n      );\n\n      // After hibernation wake-up, the doc is empty but existing connections\n      // survive. Re-sync by sending sync step 1 to all connections — they'll\n      // respond with sync step 2 containing their full state.\n      // On first start there are no connections, so this is a no-op.\n      const syncEncoder = encoding.createEncoder();\n      encoding.writeVarUint(syncEncoder, messageSync);\n      syncProtocol.writeSyncStep1(syncEncoder, this.document);\n      const syncMessage = encoding.toUint8Array(syncEncoder);\n      for (const conn of this.getConnections()) {\n        send(conn, syncMessage);\n      }\n    }\n\n    // oxlint-disable-next-line no-unused-vars\n    isReadOnly(connection: Connection): boolean {\n      // to be implemented by the user\n      return false;\n    }\n\n    /**\n     * Handle custom string messages from the client.\n     * Override this method to implement custom message handling.\n     * @param connection - The connection that sent the message\n     * @param message - The custom message string (without the __YPS: prefix)\n     */\n    // oxlint-disable-next-line no-unused-vars\n    onCustomMessage(connection: Connection, message: string): void {\n      // to be implemented by the user\n      console.warn(\n        `Received custom message but onCustomMessage is not implemented in ${this.constructor.name}:`,\n        message\n      );\n    }\n\n    /**\n     * Send a custom string message to a specific connection.\n     * @param connection - The connection to send the message to\n     * @param message - The custom message string to send\n     */\n    sendCustomMessage(connection: Connection, message: string): void {\n      if (\n        connection.readyState !== undefined &&\n        connection.readyState !== wsReadyStateConnecting &&\n        connection.readyState !== wsReadyStateOpen\n      ) {\n        return;\n      }\n      try {\n        connection.send(`__YPS:${message}`);\n      } catch (e) {\n        console.warn(\"Failed to send custom message\", e);\n      }\n    }\n\n    /**\n     * Broadcast a custom string message to all connected clients.\n     * @param message - The custom message string to broadcast\n     * @param excludeConnection - Optional connection to exclude from the broadcast\n     */\n    broadcastCustomMessage(\n      message: string,\n      excludeConnection?: Connection\n    ): void {\n      const formattedMessage = `__YPS:${message}`;\n      for (const conn of this.getConnections()) {\n        if (excludeConnection && conn === excludeConnection) {\n          continue;\n        }\n        if (\n          conn.readyState !== undefined &&\n          conn.readyState !== wsReadyStateConnecting &&\n          conn.readyState !== wsReadyStateOpen\n        ) {\n          continue;\n        }\n        try {\n          conn.send(formattedMessage);\n        } catch (e) {\n          console.warn(\"Failed to broadcast custom message\", e);\n        }\n      }\n    }\n\n    handleMessage(connection: Connection, message: WSMessage) {\n      if (typeof message === \"string\") {\n        // Handle custom messages with __YPS: prefix\n        if (message.startsWith(\"__YPS:\")) {\n          const customMessage = message.slice(6); // Remove __YPS: prefix\n          this.onCustomMessage(connection, customMessage);\n          return;\n        }\n        console.warn(\n          `Received non-prefixed string message. Custom messages should be sent using sendMessage() on the provider.`\n        );\n        return;\n      }\n      try {\n        const encoder = encoding.createEncoder();\n        // Convert ArrayBuffer to Uint8Array if needed (ArrayBufferView like Uint8Array can be used directly)\n        const uint8Array =\n          message instanceof Uint8Array\n            ? message\n            : message instanceof ArrayBuffer\n              ? new Uint8Array(message)\n              : new Uint8Array(\n                  message.buffer,\n                  message.byteOffset,\n                  message.byteLength\n                );\n        const decoder = decoding.createDecoder(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(connection, encoding.toUint8Array(encoder));\n            }\n            break;\n          case messageAwareness: {\n            const awarenessData = decoding.readVarUint8Array(decoder);\n            awarenessProtocol.applyAwarenessUpdate(\n              this.document.awareness,\n              awarenessData,\n              connection\n            );\n            // Forward raw awareness bytes to all connections\n            const awarenessEncoder = encoding.createEncoder();\n            encoding.writeVarUint(awarenessEncoder, messageAwareness);\n            encoding.writeVarUint8Array(awarenessEncoder, awarenessData);\n            const awarenessBuff = encoding.toUint8Array(awarenessEncoder);\n            for (const c of this.getConnections()) {\n              send(c, awarenessBuff);\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(conn: Connection, message: WSMessage) {\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      // Read controlled awareness clientIDs from connection state\n      // (survives hibernation unlike an in-memory Map)\n      const controlledIds = getAwarenessIds(connection);\n      if (controlledIds.length > 0) {\n        awarenessProtocol.removeAwarenessStates(\n          this.document.awareness,\n          controlledIds,\n          null\n        );\n      }\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      // Note: awareness IDs are lazily initialized when the first awareness\n      // message is received — no need to call setAwarenessIds(conn, []) here\n\n      // send sync step 1\n      const encoder = encoding.createEncoder();\n      encoding.writeVarUint(encoder, messageSync);\n      syncProtocol.writeSyncStep1(encoder, this.document);\n      send(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(conn, encoding.toUint8Array(encoder));\n      }\n    }\n  }\n\n  return YjsMixin as unknown as TBase &\n    YjsStatic &\n    (new (...args: any[]) => YjsInstance);\n}\n\nexport const YServer = withYjs(Server);\n"],"mappings":";;;;;;;;;;;;;;;;AAkBA,MAAM,iBAAiB,OAAO,kBAAkB;AAShD,MAAM,yBAAyB;AAC/B,MAAM,mBAAmB;AAMzB,MAAM,cAAc;AACpB,MAAM,mBAAmB;;;;;;AASzB,MAAM,oBAAoB;AAO1B,SAAS,gBAAgB,MAA4B;AACnD,KAAI;AAEF,SADc,KAAK,QACJ,sBAAsB,EAAE;SACjC;AACN,SAAO,EAAE;;;AAIb,SAAS,gBAAgB,MAAkB,KAAqB;AAC9D,KAAI;AACF,OAAK,UAAU,UAAyC;GACtD,GAAG;IACF,oBAAoB;GACtB,EAAE;SACG;;AAKV,IAAM,cAAN,cAA0BA,QAAK;CAC7B;CAEA,cAAc;AACZ,QAAM,EAAE,IAAI,MAAM,CAAC;AACnB,OAAK,YAAY,IAAIC,sBAAkB,UAAU,KAAK;AACtD,OAAK,UAAU,cAAc,KAAK;AAMlC,gBAEI,KAAK,UAGL,eACH;;;AAIL,MAAM,oBAAoB;CACxB,cAAc;CACd,iBAAiB;CACjB,SAAS;CACV;AAED,SAAS,gBACP,SACA,SACA,KACA,mBACA,WAAW,OACX;CACA,MAAM,cAAcC,cAAS,YAAY,QAAQ;AACjD,SAAQ,aAAR;EACE,KAAKC,iBAAa;AAChB,oBAAa,cAAc,SAAS,SAAS,IAAI;AACjD;EACF,KAAKA,iBAAa;AAChB,OAAI,CAAC,SACH,kBAAa,cAAc,SAAS,KAAK,kBAAkB;AAC7D;EACF,KAAKA,iBAAa;AAChB,OAAI,CAAC,SAAU,kBAAa,WAAW,SAAS,KAAK,kBAAkB;AACvE;EACF,QACE,OAAM,IAAI,MAAM,uBAAuB;;AAE3C,QAAO;;AAGT,SAAS,KAAK,MAAkB,GAAqB;AACnD,KACE,KAAK,eAAe,UACpB,KAAK,eAAe,0BACpB,KAAK,eAAe,iBAEpB;AAEF,KAAI;AACF,OAAK,KAAK,EAAE;SACN;;AAgCV,SAAgB,QACd,MAC2D;CAC3D,MAAM,iBAAiB,KAAK;EAC1B,OAAO,kBAAmC,EAAE;EAE5C,AAAS,WAAwB,IAAI,aAAa;EAElD,MAAM,SAA+B;EAKrC,MAAM,SAAwB;;;;;;;EAU9B,yBACE,gBACA,oBAAkD,OAC5C;AACN,OAAI;IACF,MAAM,MAAM,KAAK;IACjB,MAAM,cAAc,IAAIH,SAAM;AAC9B,yBAAY,aAAa,gBAAgB,eAAe;IAExD,MAAM,gDAAuC,IAAI;IAGjD,MAAM,0DACJ,gCAH4C,YAAY,CAKzD;IAED,MAAM,cAAc,IAAII,gBACtB,CAAC,GAAG,YAAY,MAAM,MAAM,CAAC,CAAC,KAAK,QAAQ;KACzC,MAAM,OAAO,YAAY,IAAI;AAC7B,SAAI,SAAS,OACX,QAAO,YAAY,QAAQ,IAAI;cACtB,SAAS,MAClB,QAAO,YAAY,OAAO,IAAI;cACrB,SAAS,QAClB,QAAO,YAAY,SAAS,IAAI;cACvB,SAAS,UAClB,QAAO,YAAY,IAAI,KAAKC,YAAQ;cAC3B,SAAS,aAClB,QAAO,YAAY,IAAI,KAAKC,eAAW;cAC9B,SAAS,cAClB,QAAO,YAAY,IAAI,KAAKC,gBAAY;AAE1C,WAAM,IAAI,MAAM,sBAAsB,KAAK,YAAY,MAAM;MAC7D,EACF,EACE,gBAAgB,IAAI,IAAI,CAAC,eAAe,CAAC,EAC1C,CACF;AAED,yBAAY,aAAa,4BAA4B,eAAe;AACpE,gBAAY,MAAM;IAElB,MAAM,kEACJ,aACA,mBACD;AAED,yBAAY,KAAK,UAAU,mCAAmC;YACvD,OAAO;AACd,UAAM,IAAI,MACR,+BAA+B,iBAAiB,QAAQ,MAAM,UAAU,kBACzE;;;EAIL,MAAM,UAAyB;GAC7B,MAAM,MAAM,MAAM,KAAK,QAAQ;AAC/B,OAAI,OAAO,MAAM;IACf,MAAM,qCAA4B,IAAI;AACtC,yBAAY,KAAK,UAAU,MAAM;;AAMnC,QAAK,SAAS,GAAG,WAAW,WAAuB;IACjD,MAAM,UAAUC,cAAS,eAAe;AACxC,kBAAS,aAAa,SAAS,YAAY;AAC3C,qBAAa,YAAY,SAAS,OAAO;IACzC,MAAM,UAAUA,cAAS,aAAa,QAAQ;AAC9C,SAAK,MAAM,QAAQ,KAAK,gBAAgB,CACtC,MAAK,MAAM,QAAQ;KAErB;AAOF,QAAK,SAAS,UAAU,GACtB,WAEE,EACE,OACA,SACA,WAMF,SACG;AACH,QAAI,SAAS,KAEX,KAAI;KACF,MAAM,aAAa,IAAI,IAAI,gBAAgB,KAAK,CAAC;AACjD,UAAK,MAAM,YAAY,MAAO,YAAW,IAAI,SAAS;AACtD,UAAK,MAAM,YAAY,QAAS,YAAW,OAAO,SAAS;AAC3D,qBAAgB,MAAM,CAAC,GAAG,WAAW,CAAC;aAC/B,IAAI;SAGR;KAGL,MAAM,iBAAiB,MAAM,OAAO,SAAS,QAAQ;KACrD,MAAM,UAAUA,cAAS,eAAe;AACxC,mBAAS,aAAa,SAAS,iBAAiB;AAChD,mBAAS,mBACP,SACAP,sBAAkB,sBAChB,KAAK,SAAS,WACd,eACD,CACF;KACD,MAAM,OAAOO,cAAS,aAAa,QAAQ;AAC3C,UAAK,MAAM,KAAK,KAAK,gBAAgB,CACnC,MAAK,GAAG,KAAK;;KAIpB;GAGD,MAAM,OAAO,KAAK;AAClB,QAAK,SAAS,GACZ,wCAEG,SAAqB,SAAqB,SAAe;AACxD,QAAI;AACF,UAAK,QAAQ,CAAC,OAAO,QAAQ;AAC3B,cAAQ,MAAM,sBAAsB,IAAI;OACxC;aACK,KAAK;AACZ,aAAQ,MAAM,sBAAsB,IAAI;;MAG5C,KAAK,gBAAgB,gBAAgB,kBAAkB,cACvD,EACE,SACE,KAAK,gBAAgB,mBACrB,kBAAkB,iBACrB,CACF,CACF;GAMD,MAAM,cAAcA,cAAS,eAAe;AAC5C,iBAAS,aAAa,aAAa,YAAY;AAC/C,oBAAa,eAAe,aAAa,KAAK,SAAS;GACvD,MAAM,cAAcA,cAAS,aAAa,YAAY;AACtD,QAAK,MAAM,QAAQ,KAAK,gBAAgB,CACtC,MAAK,MAAM,YAAY;;EAK3B,WAAW,YAAiC;AAE1C,UAAO;;;;;;;;EAUT,gBAAgB,YAAwB,SAAuB;AAE7D,WAAQ,KACN,qEAAqE,KAAK,YAAY,KAAK,IAC3F,QACD;;;;;;;EAQH,kBAAkB,YAAwB,SAAuB;AAC/D,OACE,WAAW,eAAe,UAC1B,WAAW,eAAe,0BAC1B,WAAW,eAAe,iBAE1B;AAEF,OAAI;AACF,eAAW,KAAK,SAAS,UAAU;YAC5B,GAAG;AACV,YAAQ,KAAK,iCAAiC,EAAE;;;;;;;;EASpD,uBACE,SACA,mBACM;GACN,MAAM,mBAAmB,SAAS;AAClC,QAAK,MAAM,QAAQ,KAAK,gBAAgB,EAAE;AACxC,QAAI,qBAAqB,SAAS,kBAChC;AAEF,QACE,KAAK,eAAe,UACpB,KAAK,eAAe,0BACpB,KAAK,eAAe,iBAEpB;AAEF,QAAI;AACF,UAAK,KAAK,iBAAiB;aACpB,GAAG;AACV,aAAQ,KAAK,sCAAsC,EAAE;;;;EAK3D,cAAc,YAAwB,SAAoB;AACxD,OAAI,OAAO,YAAY,UAAU;AAE/B,QAAI,QAAQ,WAAW,SAAS,EAAE;KAChC,MAAM,gBAAgB,QAAQ,MAAM,EAAE;AACtC,UAAK,gBAAgB,YAAY,cAAc;AAC/C;;AAEF,YAAQ,KACN,4GACD;AACD;;AAEF,OAAI;IACF,MAAM,UAAUA,cAAS,eAAe;IAExC,MAAM,aACJ,mBAAmB,aACf,UACA,mBAAmB,cACjB,IAAI,WAAW,QAAQ,GACvB,IAAI,WACF,QAAQ,QACR,QAAQ,YACR,QAAQ,WACT;IACT,MAAM,UAAUN,cAAS,cAAc,WAAW;AAElD,YADoBA,cAAS,YAAY,QAAQ,EACjD;KACE,KAAK;AACH,oBAAS,aAAa,SAAS,YAAY;AAC3C,sBACE,SACA,SACA,KAAK,UACL,YACA,KAAK,WAAW,WAAW,CAC5B;AAKD,UAAIM,cAAS,OAAO,QAAQ,GAAG,EAC7B,MAAK,YAAYA,cAAS,aAAa,QAAQ,CAAC;AAElD;KACF,KAAK,kBAAkB;MACrB,MAAM,gBAAgBN,cAAS,kBAAkB,QAAQ;AACzD,4BAAkB,qBAChB,KAAK,SAAS,WACd,eACA,WACD;MAED,MAAM,mBAAmBM,cAAS,eAAe;AACjD,oBAAS,aAAa,kBAAkB,iBAAiB;AACzD,oBAAS,mBAAmB,kBAAkB,cAAc;MAC5D,MAAM,gBAAgBA,cAAS,aAAa,iBAAiB;AAC7D,WAAK,MAAM,KAAK,KAAK,gBAAgB,CACnC,MAAK,GAAG,cAAc;AAExB;;;YAGG,KAAK;AACZ,YAAQ,MAAM,IAAI;AAElB,SAAK,SAAS,KAAK,SAAS,CAAC,IAAI,CAAC;;;EAItC,UAAU,MAAkB,SAAoB;AAC9C,QAAK,cAAc,MAAM,QAAQ;;EAGnC,QACE,YACA,OACA,SACA,WACsB;GAGtB,MAAM,gBAAgB,gBAAgB,WAAW;AACjD,OAAI,cAAc,SAAS,EACzB,uBAAkB,sBAChB,KAAK,SAAS,WACd,eACA,KACD;;EAML,UACE,MACA,MACsB;GAKtB,MAAM,UAAUA,cAAS,eAAe;AACxC,iBAAS,aAAa,SAAS,YAAY;AAC3C,oBAAa,eAAe,SAAS,KAAK,SAAS;AACnD,QAAK,MAAMA,cAAS,aAAa,QAAQ,CAAC;GAC1C,MAAM,kBAAkB,KAAK,SAAS,UAAU,WAAW;AAC3D,OAAI,gBAAgB,OAAO,GAAG;IAC5B,MAAM,UAAUA,cAAS,eAAe;AACxC,kBAAS,aAAa,SAAS,iBAAiB;AAChD,kBAAS,mBACP,SACAP,sBAAkB,sBAChB,KAAK,SAAS,WACd,MAAM,KAAK,gBAAgB,MAAM,CAAC,CACnC,CACF;AACD,SAAK,MAAMO,cAAS,aAAa,QAAQ,CAAC;;;;AAKhD,QAAO;;AAKT,MAAa,UAAU,QAAQC,mBAAO"}

package/dist/server/index.d.cts

@@ -0,0 +1,48 @@
+import * as awarenessProtocol from "y-protocols/awareness";
+import { Doc } from "yjs";
+import { Connection, Server, WSMessage } from "partyserver";
+
+//#region src/server/index.d.ts
+type YjsRootType =
+  | "Text"
+  | "Map"
+  | "Array"
+  | "XmlText"
+  | "XmlElement"
+  | "XmlFragment";
+declare class WSSharedDoc extends Doc {
+  awareness: awarenessProtocol.Awareness;
+  constructor();
+}
+interface CallbackOptions {
+  debounceWait?: number;
+  debounceMaxWait?: number;
+  timeout?: number;
+}
+type ServerClass = new (...args: any[]) => Server;
+interface YjsInstance {
+  readonly document: WSSharedDoc;
+  onLoad(): Promise<Doc | void>;
+  onSave(): Promise<void>;
+  unstable_replaceDocument(
+    snapshotUpdate: Uint8Array,
+    getMetadata?: (key: string) => YjsRootType
+  ): void;
+  isReadOnly(connection: Connection): boolean;
+  onCustomMessage(connection: Connection, message: string): void;
+  sendCustomMessage(connection: Connection, message: string): void;
+  broadcastCustomMessage(message: string, excludeConnection?: Connection): void;
+  handleMessage(connection: Connection, message: WSMessage): void;
+}
+interface YjsStatic {
+  callbackOptions: CallbackOptions;
+}
+declare function withYjs<TBase extends ServerClass>(
+  Base: TBase
+): TBase & YjsStatic & (new (...args: any[]) => YjsInstance);
+declare const YServer: typeof Server &
+  YjsStatic &
+  (new (...args: any[]) => YjsInstance);
+//#endregion
+export { CallbackOptions, YServer, YjsInstance, YjsStatic, withYjs };
+//# sourceMappingURL=index.d.cts.map

package/dist/server/index.d.ts

@@ -1,4 +1,4 @@
-import { Connection, ConnectionContext, Server, WSMessage } from "partyserver";
+import { Connection, Server, WSMessage } from "partyserver";
 import * as awarenessProtocol from "y-protocols/awareness";
 import { Doc } from "yjs";
 
@@ -19,58 +19,30 @@ interface CallbackOptions {
   debounceMaxWait?: number;
   timeout?: number;
 }
-declare class YServer<
-  Env extends Cloudflare.Env = Cloudflare.Env
-> extends Server<Env> {
-  #private;
-  static callbackOptions: CallbackOptions;
+type ServerClass = new (...args: any[]) => Server;
+interface YjsInstance {
   readonly document: WSSharedDoc;
   onLoad(): Promise<Doc | void>;
   onSave(): Promise<void>;
-  /**
-   * Replaces the document with a different state using Yjs UndoManager key remapping.
-   *
-   * @param snapshotUpdate - The snapshot update to replace the document with.
-   * @param getMetadata (optional) - A function that returns the type of the root for a given key.
-   */
   unstable_replaceDocument(
     snapshotUpdate: Uint8Array,
     getMetadata?: (key: string) => YjsRootType
   ): void;
-  onStart(): Promise<void>;
   isReadOnly(connection: Connection): boolean;
-  /**
-   * Handle custom string messages from the client.
-   * Override this method to implement custom message handling.
-   * @param connection - The connection that sent the message
-   * @param message - The custom message string (without the __YPS: prefix)
-   */
   onCustomMessage(connection: Connection, message: string): void;
-  /**
-   * Send a custom string message to a specific connection.
-   * @param connection - The connection to send the message to
-   * @param message - The custom message string to send
-   */
   sendCustomMessage(connection: Connection, message: string): void;
-  /**
-   * Broadcast a custom string message to all connected clients.
-   * @param message - The custom message string to broadcast
-   * @param excludeConnection - Optional connection to exclude from the broadcast
-   */
   broadcastCustomMessage(message: string, excludeConnection?: Connection): void;
   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 | Promise<void>;
 }
+interface YjsStatic {
+  callbackOptions: CallbackOptions;
+}
+declare function withYjs<TBase extends ServerClass>(
+  Base: TBase
+): TBase & YjsStatic & (new (...args: any[]) => YjsInstance);
+declare const YServer: typeof Server &
+  YjsStatic &
+  (new (...args: any[]) => YjsInstance);
 //#endregion
-export { CallbackOptions, YServer };
+export { CallbackOptions, YServer, YjsInstance, YjsStatic, withYjs };
 //# sourceMappingURL=index.d.ts.map