Support document forking

Author: davidbrochart

javascript/src/api.ts

@@ -20,7 +20,7 @@ import type {
   JSONValue,
   PartialJSONValue
 } from '@lumino/coreutils';
-import type { IObservableDisposable } from '@lumino/disposable';
+import type { IDisposable, IObservableDisposable } from '@lumino/disposable';
 import type { ISignal } from '@lumino/signaling';
 import * as Y from 'yjs';
 
@@ -81,6 +81,26 @@ export interface ISharedBase extends IObservableDisposable {
   transact(f: () => void, undoable?: boolean, origin?: any): void;
 }
 
+/**
+ * An interface for a document provider.
+ */
+export interface IDocumentProvider extends IDisposable {
+  /**
+   * Returns a Promise that resolves when the document provider is ready.
+   */
+  readonly ready: Promise<void>;
+
+  /**
+   * Request to fork the room in the backend, and connect the shared document to the forked room.
+   */
+  fork(): Promise<void>;
+
+  /**
+   * Connect a shared document to a forked room with forkId and return a document provider.
+   */
+  connectFork(forkId: string, sharedDocument: ISharedDocument): IDocumentProvider;
+}
+
 /**
  * Implement an API for Context information on the shared information.
  * This is used by, for example, docregistry to share the file-path of the edited content.
@@ -115,6 +135,28 @@ interface ISharedDocumentNoSource extends ISharedBase {
    * The changed signal.
    */
   readonly changed: ISignal<this, DocumentChange>;
+
+  /**
+   * Get a provider with a given ID
+   *
+   * @param providerId The provider ID
+   */
+  getProvider(providerId: string, sharedModel?: ISharedDocument): IDocumentProvider;
+
+  /**
+   * Set a provider for this document
+   *
+   * @param providerId Provider ID (either 'root' or a UUID)
+   * @param provider The document provider
+   */
+  setProvider(providerId: string, provider: IDocumentProvider): void;
+
+  /**
+   * Add a fork ID to the document's ystate, as a new key 'fork_{forkId}'
+   *
+   * @param forkId The fork ID to add to the document
+   */
+  addFork(forkId: string): void;
 }
 
 /**

javascript/src/ydocument.ts

@@ -7,7 +7,7 @@ import { JSONExt, JSONObject, JSONValue } from '@lumino/coreutils';
 import { ISignal, Signal } from '@lumino/signaling';
 import { Awareness } from 'y-protocols/awareness';
 import * as Y from 'yjs';
-import type { DocumentChange, ISharedDocument, StateChange } from './api.js';
+import type { DocumentChange, IDocumentProvider, ISharedDocument, StateChange } from './api';
 
 /**
  * Generic shareable document.
@@ -28,13 +28,37 @@ export abstract class YDocument<T extends DocumentChange>
     this._awareness = new Awareness(this._ydoc);
 
     this._ystate.observe(this.onStateChanged);
+
+    this._providers = {};
   }
 
   /**
    * Document version
    */
   abstract readonly version: string;
 
+  addFork(forkId: string) {
+    this.ystate.set(`fork_${forkId}`, 'new');
+  }
+
+  getProvider(providerId: string, sharedModel?: ISharedDocument): IDocumentProvider {
+    if (!(providerId in this._providers)) {
+      if (providerId === 'root') {
+        throw new Error('Cannot get a new provider for root document');
+      }
+      if (sharedModel === undefined) {
+        throw new Error('New provider needs a shared document');
+      }
+      const root_provider = this._providers['root'];
+      this._providers[providerId] = root_provider.connectFork(providerId, sharedModel!);
+    }
+    return this._providers[providerId];
+  }
+
+  setProvider(providerId: string, provider: IDocumentProvider) {
+    this._providers[providerId] = provider;
+  }
+
   /**
    * YJS document.
    */
@@ -232,6 +256,7 @@ export abstract class YDocument<T extends DocumentChange>
   private _awareness: Awareness;
   private _isDisposed = false;
   private _disposed = new Signal<this, void>(this);
+  private _providers: { [key: string]: IDocumentProvider };
 }
 
 /**