Class WSSharedDoc

packages/yjs-websocket/src/server/utils.ts:68–163

}

class WSSharedDoc extends Y.Doc implements WSSharedDocInterface {
  name: string
  conns: Map<WebSocket.WebSocket, Set<number>>
  awareness: awarenessProtocol.Awareness
  subDocs: Map<string, Y.Doc> = new Map()
  /**
   * @param {string} name
   */
  constructor(name: string, callback?: UpdateCallback) {
    super({ gc: gcEnabled })
    this.name = name

    this.conns = new Map()

    this.awareness = new awarenessProtocol.Awareness(this)
    this.awareness.setLocalState(null)

    const awarenessChangeHandler = (
      { added, updated, removed }: AwarenessChangeHandlerOptions,
      conn: WebSocket.WebSocket | null,
    ) => {
      const changedClients = added.concat(updated, removed)
      if (conn !== null) {
        const connControlledIDs = this.conns.get(conn)
        if (connControlledIDs !== undefined) {
          added.forEach(clientID => {
            connControlledIDs.add(clientID)
          })
          removed.forEach(clientID => {
            connControlledIDs.delete(clientID)
          })
        }
      }
      // broadcast awareness update
      const encoder = encoding.createEncoder()
      encoding.writeVarUint(encoder, messageAwareness)
      encoding.writeVarUint8Array(
        encoder,
        awarenessProtocol.encodeAwarenessUpdate(this.awareness, changedClients),
      )
      const buff = encoding.toUint8Array(encoder)
      this.conns.forEach((_, c) => {
        send(this, c, buff)
      })
    }
    this.awareness.on('update', awarenessChangeHandler)
    this.on('updateV2', updateHandler)

    const handleSubDocUpdate = (update: Uint8Array, origin: string, subDoc: Y.Doc) => {
      updateSubDocHandler(subDoc.guid, update, origin, this)
    }

    const _subDocsHandler = ({
      added,
      removed,
      loaded,
    }: Record<'added' | 'removed' | 'loaded', Y.Doc[]>) => {
      added.forEach(subDoc => {