@hocuspocus/provider 1.1.1 → 2.0.0-alpha.1

package/src/HocuspocusProviderWebsocket.ts

@@ -0,0 +1,475 @@
+import * as time from 'lib0/time'
+import * as mutex from 'lib0/mutex'
+import * as url from 'lib0/url'
+import type { MessageEvent } from 'ws'
+import { retry } from '@lifeomic/attempt'
+import {
+  Forbidden, Unauthorized, WsReadyStates,
+} from '@hocuspocus/common'
+import { Event } from 'ws'
+import EventEmitter from './EventEmitter'
+import {
+  onCloseParameters, onDisconnectParameters, onMessageParameters, onOpenParameters, onOutgoingMessageParameters, onStatusParameters, WebSocketStatus,
+} from './types'
+import { HocuspocusProvider, onAwarenessChangeParameters, onAwarenessUpdateParameters } from '.'
+
+export type HocuspocusProviderWebsocketConfiguration =
+  Required<Pick<CompleteHocuspocusProviderWebsocketConfiguration, 'url'>>
+  & Partial<CompleteHocuspocusProviderWebsocketConfiguration>
+
+export interface CompleteHocuspocusProviderWebsocketConfiguration {
+  /**
+   * URL of your @hocuspocus/server instance
+   */
+   url: string,
+
+  /**
+   * Pass `false` to start the connection manually.
+   */
+  connect: boolean,
+
+  /**
+   * URL parameters that should be added.
+   */
+  parameters: { [key: string]: any },
+  /**
+   * An optional WebSocket polyfill, for example for Node.js
+   */
+  WebSocketPolyfill: any,
+
+  /**
+   * Disconnect when no message is received for the defined amount of milliseconds.
+   */
+  messageReconnectTimeout: number,
+  /**
+   * The delay between each attempt in milliseconds. You can provide a factor to have the delay grow exponentially.
+   */
+  delay: number,
+  /**
+   * The intialDelay is the amount of time to wait before making the first attempt. This option should typically be 0 since you typically want the first attempt to happen immediately.
+   */
+  initialDelay: number,
+  /**
+   * The factor option is used to grow the delay exponentially.
+   */
+  factor: number,
+  /**
+   * The maximum number of attempts or 0 if there is no limit on number of attempts.
+   */
+  maxAttempts: number,
+  /**
+   * minDelay is used to set a lower bound of delay when jitter is enabled. This property has no effect if jitter is disabled.
+   */
+  minDelay: number,
+  /**
+   * The maxDelay option is used to set an upper bound for the delay when factor is enabled. A value of 0 can be provided if there should be no upper bound when calculating delay.
+   */
+  maxDelay: number,
+  /**
+   * If jitter is true then the calculated delay will be a random integer value between minDelay and the calculated delay for the current iteration.
+   */
+  jitter: boolean,
+  /**
+   * A timeout in milliseconds. If timeout is non-zero then a timer is set using setTimeout. If the timeout is triggered then future attempts will be aborted.
+   */
+  timeout: number,
+  onOpen: (data: onOpenParameters) => void,
+  onConnect: () => void,
+  onMessage: (data: onMessageParameters) => void,
+  onOutgoingMessage: (data: onOutgoingMessageParameters) => void,
+  onStatus: (data: onStatusParameters) => void,
+  onDisconnect: (data: onDisconnectParameters) => void,
+  onClose: (data: onCloseParameters) => void,
+  onDestroy: () => void,
+  onAwarenessUpdate: (data: onAwarenessUpdateParameters) => void,
+  onAwarenessChange: (data: onAwarenessChangeParameters) => void,
+  /**
+   * Don’t output any warnings.
+   */
+  quiet: boolean,
+}
+
+export class HocuspocusProviderWebsocket extends EventEmitter {
+  public configuration: CompleteHocuspocusProviderWebsocketConfiguration = {
+    url: '',
+    // @ts-ignore
+    document: undefined,
+    // @ts-ignore
+    awareness: undefined,
+    WebSocketPolyfill: undefined,
+    parameters: {},
+    connect: true,
+    broadcast: true,
+    forceSyncInterval: false,
+    // TODO: this should depend on awareness.outdatedTime
+    messageReconnectTimeout: 30000,
+    // 1 second
+    delay: 1000,
+    // instant
+    initialDelay: 0,
+    // double the delay each time
+    factor: 2,
+    // unlimited retries
+    maxAttempts: 0,
+    // wait at least 1 second
+    minDelay: 1000,
+    // at least every 30 seconds
+    maxDelay: 30000,
+    // randomize
+    jitter: true,
+    // retry forever
+    timeout: 0,
+    onOpen: () => null,
+    onConnect: () => null,
+    onMessage: () => null,
+    onOutgoingMessage: () => null,
+    onStatus: () => null,
+    onDisconnect: () => null,
+    onClose: () => null,
+    onDestroy: () => null,
+    onAwarenessUpdate: () => null,
+    onAwarenessChange: () => null,
+    quiet: false,
+  }
+
+  subscribedToBroadcastChannel = false
+
+  webSocket: WebSocket | null = null
+
+  shouldConnect = true
+
+  status = WebSocketStatus.Disconnected
+
+  lastMessageReceived = 0
+
+  mux = mutex.createMutex()
+
+  intervals: any = {
+    forceSync: null,
+    connectionChecker: null,
+  }
+
+  connectionAttempt: {
+    resolve: (value?: any) => void
+    reject: (reason?: any) => void
+  } | null = null
+
+  constructor(configuration: HocuspocusProviderWebsocketConfiguration) {
+    super()
+    this.setConfiguration(configuration)
+
+    this.configuration.WebSocketPolyfill = configuration.WebSocketPolyfill ? configuration.WebSocketPolyfill : WebSocket
+
+    this.on('open', this.configuration.onOpen)
+    this.on('open', this.onOpen.bind(this))
+    this.on('connect', this.configuration.onConnect)
+    this.on('message', this.configuration.onMessage)
+    this.on('outgoingMessage', this.configuration.onOutgoingMessage)
+    this.on('status', this.configuration.onStatus)
+    this.on('status', this.onStatus.bind(this))
+    this.on('disconnect', this.configuration.onDisconnect)
+    this.on('close', this.configuration.onClose)
+    this.on('destroy', this.configuration.onDestroy)
+    this.on('awarenessUpdate', this.configuration.onAwarenessUpdate)
+    this.on('awarenessChange', this.configuration.onAwarenessChange)
+
+    this.on('close', this.onClose.bind(this))
+    this.on('message', this.onMessage.bind(this))
+
+    this.registerEventListeners()
+
+    this.intervals.connectionChecker = setInterval(
+      this.checkConnection.bind(this),
+      this.configuration.messageReconnectTimeout / 10,
+    )
+
+    if (typeof configuration.connect !== 'undefined') {
+      this.shouldConnect = configuration.connect
+    }
+
+    if (!this.shouldConnect) {
+      return
+    }
+
+    this.connect()
+  }
+
+  receivedOnOpenPayload?: Event | undefined = undefined
+
+  receivedOnStatusPayload?: onStatusParameters | undefined = undefined
+
+  async onOpen(event: Event) {
+    this.receivedOnOpenPayload = event
+  }
+
+  async onStatus(data: onStatusParameters) {
+    this.receivedOnStatusPayload = data
+  }
+
+  attach(provider: HocuspocusProvider) {
+    if (this.receivedOnOpenPayload) {
+      provider.onOpen(this.receivedOnOpenPayload)
+    }
+
+    if (this.receivedOnStatusPayload) {
+      provider.onStatus(this.receivedOnStatusPayload)
+    }
+  }
+
+  detach(provider: HocuspocusProvider) {
+    // tell the server to remove the listener
+
+  }
+
+  public setConfiguration(configuration: Partial<HocuspocusProviderWebsocketConfiguration> = {}): void {
+    this.configuration = { ...this.configuration, ...configuration }
+  }
+
+  boundConnect = this.connect.bind(this)
+
+  cancelWebsocketRetry?: () => void
+
+  async connect() {
+    if (this.status === WebSocketStatus.Connected) {
+      return
+    }
+
+    // Always cancel any previously initiated connection retryer instances
+    if (this.cancelWebsocketRetry) {
+      this.cancelWebsocketRetry()
+      this.cancelWebsocketRetry = undefined
+    }
+
+    this.shouldConnect = true
+
+    const abortableRetry = () => {
+      let cancelAttempt = false
+
+      const retryPromise = retry(this.createWebSocketConnection.bind(this), {
+        delay: this.configuration.delay,
+        initialDelay: this.configuration.initialDelay,
+        factor: this.configuration.factor,
+        maxAttempts: this.configuration.maxAttempts,
+        minDelay: this.configuration.minDelay,
+        maxDelay: this.configuration.maxDelay,
+        jitter: this.configuration.jitter,
+        timeout: this.configuration.timeout,
+        beforeAttempt: context => {
+          if (!this.shouldConnect || cancelAttempt) {
+            context.abort()
+          }
+        },
+      }).catch((error: any) => {
+        // If we aborted the connection attempt then don’t throw an error
+        // ref: https://github.com/lifeomic/attempt/blob/master/src/index.ts#L136
+        if (error && error.code !== 'ATTEMPT_ABORTED') {
+          throw error
+        }
+      })
+
+      return {
+        retryPromise,
+        cancelFunc: () => {
+          cancelAttempt = true
+        },
+      }
+    }
+
+    const { retryPromise, cancelFunc } = abortableRetry()
+    this.cancelWebsocketRetry = cancelFunc
+
+    return retryPromise
+  }
+
+  createWebSocketConnection() {
+    return new Promise((resolve, reject) => {
+      if (this.webSocket) {
+        this.webSocket.close()
+        this.webSocket = null
+      }
+
+      // Init the WebSocket connection
+      const ws = new this.configuration.WebSocketPolyfill(this.url)
+      ws.binaryType = 'arraybuffer'
+      ws.onmessage = (payload: any) => this.emit('message', payload)
+      ws.onclose = (payload: any) => this.emit('close', { event: payload })
+      ws.onopen = (payload: any) => this.emit('open', payload)
+      ws.onerror = (err: any) => {
+        reject(err)
+      }
+      this.webSocket = ws
+
+      // Reset the status
+      this.status = WebSocketStatus.Connecting
+      this.emit('status', { status: WebSocketStatus.Connecting })
+
+      // Store resolve/reject for later use
+      this.connectionAttempt = {
+        resolve,
+        reject,
+      }
+    })
+  }
+
+  onMessage(event: MessageEvent) {
+    this.resolveConnectionAttempt()
+  }
+
+  resolveConnectionAttempt() {
+    if (this.connectionAttempt) {
+      this.connectionAttempt.resolve()
+      this.connectionAttempt = null
+
+      this.status = WebSocketStatus.Connected
+      this.emit('status', { status: WebSocketStatus.Connected })
+      this.emit('connect')
+    }
+  }
+
+  stopConnectionAttempt() {
+    this.connectionAttempt = null
+  }
+
+  rejectConnectionAttempt() {
+    this.connectionAttempt?.reject()
+    this.connectionAttempt = null
+  }
+
+  checkConnection() {
+    // Don’t check the connection when it’s not even established
+    if (this.status !== WebSocketStatus.Connected) {
+      return
+    }
+
+    // Don’t close then connection while waiting for the first message
+    if (!this.lastMessageReceived) {
+      return
+    }
+
+    // Don’t close the connection when a message was received recently
+    if (this.configuration.messageReconnectTimeout >= time.getUnixTime() - this.lastMessageReceived) {
+      return
+    }
+
+    // No message received in a long time, not even your own
+    // Awareness updates, which are updated every 15 seconds.
+    this.webSocket?.close()
+  }
+
+  registerEventListeners() {
+    if (typeof window === 'undefined') {
+      return
+    }
+
+    window.addEventListener('online', this.boundConnect)
+  }
+
+  // Ensure that the URL always ends with /
+  get serverUrl() {
+    while (this.configuration.url[this.configuration.url.length - 1] === '/') {
+      return this.configuration.url.slice(0, this.configuration.url.length - 1)
+    }
+
+    return this.configuration.url
+  }
+
+  get url() {
+    const encodedParams = url.encodeQueryParams(this.configuration.parameters)
+
+    return `${this.serverUrl}${encodedParams.length === 0 ? '' : `?${encodedParams}`}`
+  }
+
+  disconnect() {
+    this.shouldConnect = false
+
+    if (this.webSocket === null) {
+      return
+    }
+
+    try {
+      this.webSocket.close()
+    } catch {
+      //
+    }
+  }
+
+  send(message: any) {
+    if (this.webSocket?.readyState === WsReadyStates.Open) {
+      this.webSocket.send(message)
+    }
+  }
+
+  onClose({ event }: onCloseParameters) {
+    this.webSocket = null
+
+    if (this.status === WebSocketStatus.Connected) {
+      this.status = WebSocketStatus.Disconnected
+      this.emit('status', { status: WebSocketStatus.Disconnected })
+      this.emit('disconnect', { event })
+    }
+
+    if (event.code === Unauthorized.code) {
+      if (!this.configuration.quiet) {
+        console.warn('[HocuspocusProvider] An authentication token is required, but you didn’t send one. Try adding a `token` to your HocuspocusProvider configuration. Won’t try again.')
+      }
+
+      this.shouldConnect = false
+    }
+
+    if (event.code === Forbidden.code) {
+      if (!this.configuration.quiet) {
+        console.warn('[HocuspocusProvider] The provided authentication token isn’t allowed to connect to this server. Will try again.')
+        return // TODO REMOVE ME
+      }
+    }
+
+    if (this.connectionAttempt) {
+      // That connection attempt failed.
+      this.rejectConnectionAttempt()
+    } else if (this.shouldConnect) {
+      // The connection was closed by the server. Let’s just try again.
+      this.connect()
+    }
+
+    // If we’ll reconnect, we’re done for now.
+    if (this.shouldConnect) {
+      return
+    }
+
+    // The status is set correctly already.
+    if (this.status === WebSocketStatus.Disconnected) {
+      return
+    }
+
+    // Let’s update the connection status.
+    this.status = WebSocketStatus.Disconnected
+    this.emit('status', { status: WebSocketStatus.Disconnected })
+    this.emit('disconnect', { event })
+  }
+
+  destroy() {
+    this.emit('destroy')
+
+    if (this.intervals.forceSync) {
+      clearInterval(this.intervals.forceSync)
+    }
+
+    clearInterval(this.intervals.connectionChecker)
+
+    // If there is still a connection attempt outstanding then we should stop
+    // it before calling disconnect, otherwise it will be rejected in the onClose
+    // handler and trigger a retry
+    this.stopConnectionAttempt()
+
+    this.disconnect()
+
+    this.removeAllListeners()
+
+    if (typeof window === 'undefined') {
+      return
+    }
+
+    window.removeEventListener('online', this.boundConnect)
+  }
+
+}

package/src/IncomingMessage.ts

@@ -2,6 +2,7 @@ import {
   createDecoder,
   readVarUint,
   readVarUint8Array,
+  readVarString,
   Decoder,
 } from 'lib0/decoding'
 import {
@@ -9,6 +10,7 @@ import {
   createEncoder,
   writeVarUint,
   writeVarUint8Array,
+  writeVarString,
   length,
 } from 'lib0/encoding'
 import { MessageType } from './types'
@@ -31,6 +33,10 @@ export class IncomingMessage {
     return readVarUint(this.decoder)
   }
 
+  readVarString(): string {
+    return readVarString(this.decoder)
+  }
+
   readVarUint8Array() {
     return readVarUint8Array(this.decoder)
   }
@@ -39,6 +45,10 @@ export class IncomingMessage {
     return writeVarUint(this.encoder, type)
   }
 
+  writeVarString(string: string) {
+    return writeVarString(this.encoder, string)
+  }
+
   writeVarUint8Array(data: Uint8Array) {
     return writeVarUint8Array(this.encoder, data)
   }

package/src/MessageReceiver.ts

@@ -27,6 +27,8 @@ export class MessageReceiver {
     const { message } = this
     const type = message.readVarUint()
 
+    const emptyMessageLength = message.length()
+
     switch (type) {
       case MessageType.Sync:
         this.applySyncMessage(provider, emitSynced)
@@ -53,7 +55,7 @@ export class MessageReceiver {
     }
 
     // Reply
-    if (message.length() > 1) {
+    if (message.length() > emptyMessageLength + 1) { // length of documentName (considered in emptyMessageLength plus length of yjs sync type, set in applySyncMessage)
       if (this.broadcasted) {
         // TODO: Some weird TypeScript error
         // @ts-ignore
@@ -80,7 +82,7 @@ export class MessageReceiver {
     )
 
     // Synced once we receive Step2
-    if (emitSynced && (syncMessageType === messageYjsSyncStep2)) {
+    if (emitSynced && syncMessageType === messageYjsSyncStep2) {
       provider.synced = true
     }
 

package/src/OutgoingMessages/AuthenticationMessage.ts

@@ -1,4 +1,4 @@
-import { writeVarUint } from 'lib0/encoding'
+import { writeVarString, writeVarUint } from 'lib0/encoding'
 import { writeAuthentication } from '@hocuspocus/common'
 import { MessageType, OutgoingMessageArguments } from '../types'
 import { OutgoingMessage } from '../OutgoingMessage'
@@ -13,6 +13,7 @@ export class AuthenticationMessage extends OutgoingMessage {
       throw new Error('The authentication message requires `token` as an argument.')
     }
 
+    writeVarString(this.encoder, args.documentName!)
     writeVarUint(this.encoder, this.type)
     writeAuthentication(this.encoder, args.token)
 

package/src/OutgoingMessages/AwarenessMessage.ts

@@ -17,6 +17,7 @@ export class AwarenessMessage extends OutgoingMessage {
       throw new Error('The awareness message requires clients as an argument')
     }
 
+    encoding.writeVarString(this.encoder, args.documentName!)
     encoding.writeVarUint(this.encoder, this.type)
 
     let awarenessUpdate

package/src/OutgoingMessages/CloseMessage.ts

@@ -0,0 +1,16 @@
+import * as encoding from 'lib0/encoding'
+import { MessageType, OutgoingMessageArguments } from '../types'
+import { OutgoingMessage } from '../OutgoingMessage'
+
+export class CloseMessage extends OutgoingMessage {
+  type = MessageType.CLOSE
+
+  description = 'Ask the server to close the connection'
+
+  get(args: Partial<OutgoingMessageArguments>) {
+    encoding.writeVarString(this.encoder, args.documentName!)
+    encoding.writeVarUint(this.encoder, this.type)
+
+    return this.encoder
+  }
+}

package/src/OutgoingMessages/QueryAwarenessMessage.ts

@@ -8,6 +8,11 @@ export class QueryAwarenessMessage extends OutgoingMessage {
   description = 'Queries awareness states'
 
   get(args: Partial<OutgoingMessageArguments>) {
+
+    console.log('queryAwareness: writing string docName', args.documentName)
+    console.log(this.encoder.cpos)
+
+    encoding.writeVarString(this.encoder, args.documentName!)
     encoding.writeVarUint(this.encoder, this.type)
 
     return this.encoder

package/src/OutgoingMessages/StatelessMessage.ts

@@ -8,6 +8,7 @@ export class StatelessMessage extends OutgoingMessage {
   description = 'A stateless message'
 
   get(args: Partial<OutgoingMessageArguments>) {
+    writeVarString(this.encoder, args.documentName!)
     writeVarUint(this.encoder, this.type)
     writeVarString(this.encoder, args.payload ?? '')
 

package/src/OutgoingMessages/SyncStepOneMessage.ts

@@ -13,6 +13,7 @@ export class SyncStepOneMessage extends OutgoingMessage {
       throw new Error('The sync step one message requires document as an argument')
     }
 
+    encoding.writeVarString(this.encoder, args.documentName!)
     encoding.writeVarUint(this.encoder, this.type)
     syncProtocol.writeSyncStep1(this.encoder, args.document)
 

package/src/OutgoingMessages/SyncStepTwoMessage.ts

@@ -1,4 +1,3 @@
-import * as Y from 'yjs'
 import * as encoding from 'lib0/encoding'
 import * as syncProtocol from 'y-protocols/sync'
 import { MessageType, OutgoingMessageArguments } from '../types'
@@ -14,6 +13,7 @@ export class SyncStepTwoMessage extends OutgoingMessage {
       throw new Error('The sync step two message requires document as an argument')
     }
 
+    encoding.writeVarString(this.encoder, args.documentName!)
     encoding.writeVarUint(this.encoder, this.type)
     syncProtocol.writeSyncStep2(this.encoder, args.document)
 

package/src/OutgoingMessages/UpdateMessage.ts

@@ -1,4 +1,4 @@
-import { writeVarUint } from 'lib0/encoding'
+import { writeVarString, writeVarUint } from 'lib0/encoding'
 import { writeUpdate } from 'y-protocols/sync'
 import { MessageType, OutgoingMessageArguments } from '../types'
 import { OutgoingMessage } from '../OutgoingMessage'
@@ -9,7 +9,9 @@ export class UpdateMessage extends OutgoingMessage {
   description = 'A document update'
 
   get(args: Partial<OutgoingMessageArguments>) {
+    writeVarString(this.encoder, args.documentName!)
     writeVarUint(this.encoder, this.type)
+
     writeUpdate(this.encoder, args.update)
 
     return this.encoder

package/src/index.ts

@@ -1,3 +1,4 @@
 export * from './HocuspocusProvider'
 export * from './HocuspocusCloudProvider'
+export * from './HocuspocusProviderWebsocket'
 export * from './types'

package/src/types.ts

@@ -17,6 +17,7 @@ export enum MessageType {
   Auth = 2,
   QueryAwareness = 3,
   Stateless = 5,
+  CLOSE = 7,
 }
 
 export enum WebSocketStatus {
@@ -31,6 +32,7 @@ export interface OutgoingMessageInterface {
 }
 
 export interface OutgoingMessageArguments {
+  documentName: string,
   token: string,
   document: Y.Doc,
   awareness: Awareness,

package/dist/tests/server/getDocumentName.d.ts

@@ -1 +0,0 @@
-export {};