add method to add & remove tracked origins

yjs · Mar 25, 2022 · d3e1177 · d3e1177
1 parent ff5067e
commit d3e1177
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/utils/UndoManager.js b/src/utils/UndoManager.js
@@ -247,6 +247,20 @@ export class UndoManager extends Observable {
     })
   }
 
+  /**
+   * @param {any} origin
+   */
+  addTrackedOrigin (origin) {
+    this.trackedOrigins.add(origin)
+  }
+
+  /**
+   * @param {any} origin
+   */
+  removeTrackedOrigin (origin) {
+    this.trackedOrigins.delete(origin)
+  }
+
   clear (clearUndoStack = true, clearRedoStack = true) {
     if ((clearUndoStack && this.canUndo()) || (clearRedoStack && this.canRedo())) {
       this.doc.transact(tr => {