Sync dirty property between clients

packages/docregistry/src/context.ts

@@ -676,7 +676,6 @@ export class Context<
         if (this.isDisposed) {
           return;
         }
-        const dirty = false;
         if (contents.format === 'json') {
           model.fromJSON(contents.content);
           if (initializeModel) {
@@ -701,7 +700,7 @@ export class Context<
           }
         }
         this._updateContentsModel(contents);
-        model.dirty = dirty;
+        model.dirty = false;
         if (!this._isPopulated) {
           return this._populate();
         }

packages/docregistry/src/default.ts

@@ -29,6 +29,9 @@ export class DocumentModel
     const filemodel = new models.YFile() as models.ISharedFile;
     this.switchSharedModel(filemodel, true);
     this.value.changed.connect(this.triggerContentChange, this);
+
+    (this.sharedModel as models.YFile).dirty = false;
+    this.sharedModel.changed.connect(this._onStateChanged, this);
   }
 
   /**
@@ -49,15 +52,14 @@ export class DocumentModel
    * The dirty state of the document.
    */
   get dirty(): boolean {
-    return this._dirty;
+    return this.sharedModel.dirty;
   }
   set dirty(newValue: boolean) {
-    if (newValue === this._dirty) {
+    const oldValue = this.sharedModel.dirty;
+    if (newValue === oldValue) {
       return;
     }
-    const oldValue = this._dirty;
-    this._dirty = newValue;
-    this.triggerStateChange({ name: 'dirty', oldValue, newValue });
+    (this.sharedModel as models.YFile).dirty = newValue;
   }
 
   /**
@@ -151,12 +153,22 @@ export class DocumentModel
     this.dirty = true;
   }
 
+  private _onStateChanged(
+    sender: models.ISharedFile,
+    changes: models.NotebookChange
+  ): void {
+    if (changes.stateChange) {
+      changes.stateChange.forEach(value => {
+        this.triggerStateChange(value);
+      });
+    }
+  }
+
   /**
    * The shared notebook model.
    */
   readonly sharedModel: models.ISharedFile;
   private _defaultLang = '';
-  private _dirty = false;
   private _readOnly = false;
   private _contentChanged = new Signal<this, void>(this);
   private _stateChanged = new Signal<this, IChangedArgs<any>>(this);

packages/notebook/src/model.ts

@@ -114,6 +114,7 @@ export class NotebookModel implements INotebookModel {
     metadata.changed.connect(this._onMetadataChanged, this);
     this._deletedCells = [];
 
+    (this.sharedModel as models.YNotebook).dirty = false;
     this.sharedModel.changed.connect(this._onStateChanged, this);
   }
   /**
@@ -134,15 +135,14 @@ export class NotebookModel implements INotebookModel {
    * The dirty state of the document.
    */
   get dirty(): boolean {
-    return this._dirty;
+    return this.sharedModel.dirty;
   }
   set dirty(newValue: boolean) {
-    if (newValue === this._dirty) {
+    const oldValue = this.sharedModel.dirty;
+    if (newValue === oldValue) {
       return;
     }
-    const oldValue = this._dirty;
-    this._dirty = newValue;
-    this.triggerStateChange({ name: 'dirty', oldValue, newValue });
+    (this.sharedModel as models.YNotebook).dirty = newValue;
   }
 
   /**
@@ -428,9 +428,6 @@ close the notebook without saving it.`,
         if (value.name === 'nbformatMinor') {
           this._nbformatMinor = value.newValue;
         }
-        if (value.name === 'dirty') {
-          this._dirty = value.newValue;
-        }
         this.triggerStateChange(value);
       });
     }
@@ -513,7 +510,6 @@ close the notebook without saving it.`,
    */
   readonly modelDB: IModelDB;
 
-  private _dirty = false;
   private _readOnly = false;
   private _contentChanged = new Signal<this, void>(this);
   private _stateChanged = new Signal<this, IChangedArgs<any>>(this);

packages/shared-models/src/api.ts

@@ -58,6 +58,11 @@ export interface ISharedBase extends IDisposable {
  * This is used by, for example, docregistry to share the file-path of the edited content.
  */
 export interface ISharedDocument extends ISharedBase {
+  /**
+   * Whether the document is saved to disk or not.
+   */
+  readonly dirty: boolean;
+
   /**
    * The changed signal.
    */

packages/shared-models/src/ymodels.ts

@@ -26,6 +26,16 @@ export interface IYText extends models.ISharedText {
 export type YCellType = YRawCell | YCodeCell | YMarkdownCell;
 
 export class YDocument<T> implements models.ISharedDocument {
+  get dirty(): boolean {
+    return this.ystate.get('dirty');
+  }
+
+  set dirty(value: boolean) {
+    this.transact(() => {
+      this.ystate.set('dirty', value);
+    }, false);
+  }
+
   /**
    * Perform a transaction. While the function f is called, all changes to the shared
    * document are bundled into a single event.
@@ -128,11 +138,13 @@ export class YFile
 
     event.keysChanged.forEach(key => {
       const change = event.changes.keys.get(key);
-      stateChange.push({
-        name: key,
-        oldValue: change?.oldValue ? change!.oldValue : 0,
-        newValue: this.ystate.get(key)
-      });
+      if (change) {
+        stateChange.push({
+          name: key,
+          oldValue: change.oldValue,
+          newValue: this.ystate.get(key)
+        });
+      }
     });
 
     this._changed.emit({ stateChange });