Listen for Notebook change events

Author: alanhamlett

src/wakatime.ts

@@ -417,6 +417,9 @@ export class WakaTime {
     vscode.window.onDidChangeActiveTextEditor(this.onChangeTab, this, subscriptions);
     vscode.workspace.onDidSaveTextDocument(this.onSave, this, subscriptions);
 
+    vscode.workspace.onDidChangeNotebookDocument(this.onChangeNotebook, this, subscriptions);
+    vscode.workspace.onDidSaveNotebookDocument(this.onSaveNotebook, this, subscriptions);
+
     vscode.tasks.onDidStartTask(this.onDidStartTask, this, subscriptions);
     vscode.tasks.onDidEndTask(this.onDidEndTask, this, subscriptions);
 
@@ -468,6 +471,14 @@ export class WakaTime {
     this.onEvent(true);
   }
 
+  private onChangeNotebook(_e: vscode.NotebookDocumentChangeEvent): void {
+    this.onEvent(false);
+  }
+
+  private onSaveNotebook(_e: vscode.NotebookDocument | undefined): void {
+    this.onEvent(true);
+  }
+
   private onEvent(isWrite: boolean): void {
     clearTimeout(this.debounceTimeoutId);
     this.debounceTimeoutId = setTimeout(() => {