SQL Operations Studio Public Preview 1 (0.23) release source code

src/vs/workbench/parts/backup/common/backupModelTracker.ts

@@ -0,0 +1,99 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+'use strict';
+
+import Uri from 'vs/base/common/uri';
+import errors = require('vs/base/common/errors');
+import { IBackupFileService } from 'vs/workbench/services/backup/common/backup';
+import { IDisposable, dispose } from 'vs/base/common/lifecycle';
+import { ITextFileService, TextFileModelChangeEvent, StateChange } from 'vs/workbench/services/textfile/common/textfiles';
+import { IUntitledEditorService } from 'vs/workbench/services/untitled/common/untitledEditorService';
+import { IWorkbenchContribution } from 'vs/workbench/common/contributions';
+import { IConfigurationService } from 'vs/platform/configuration/common/configuration';
+import { IFilesConfiguration, AutoSaveConfiguration, CONTENT_CHANGE_EVENT_BUFFER_DELAY } from 'vs/platform/files/common/files';
+
+const AUTO_SAVE_AFTER_DELAY_DISABLED_TIME = CONTENT_CHANGE_EVENT_BUFFER_DELAY + 500;
+
+export class BackupModelTracker implements IWorkbenchContribution {
+
+	public _serviceBrand: any;
+
+	private configuredAutoSaveAfterDelay: boolean;
+	private toDispose: IDisposable[];
+
+	constructor(
+		@IBackupFileService private backupFileService: IBackupFileService,
+		@ITextFileService private textFileService: ITextFileService,
+		@IUntitledEditorService private untitledEditorService: IUntitledEditorService,
+		@IConfigurationService private configurationService: IConfigurationService
+	) {
+		this.toDispose = [];
+
+		this.registerListeners();
+	}
+
+	private registerListeners() {
+		if (!this.backupFileService.backupEnabled) {
+			return;
+		}
+
+		// Listen for text file model changes
+		this.toDispose.push(this.textFileService.models.onModelContentChanged((e) => this.onTextFileModelChanged(e)));
+		this.toDispose.push(this.textFileService.models.onModelSaved((e) => this.discardBackup(e.resource)));
+		this.toDispose.push(this.textFileService.models.onModelDisposed((e) => this.discardBackup(e)));
+
+		// Listen for untitled model changes
+		this.toDispose.push(this.untitledEditorService.onDidChangeContent((e) => this.onUntitledModelChanged(e)));
+		this.toDispose.push(this.untitledEditorService.onDidDisposeModel((e) => this.discardBackup(e)));
+
+		// Listen to config changes
+		this.toDispose.push(this.configurationService.onDidUpdateConfiguration(e => this.onConfigurationChange(this.configurationService.getConfiguration<IFilesConfiguration>())));
+	}
+
+	private onConfigurationChange(configuration: IFilesConfiguration): void {
+		if (!configuration || !configuration.files) {
+			this.configuredAutoSaveAfterDelay = false;
+			return;
+		}
+		this.configuredAutoSaveAfterDelay =
+			(configuration.files.autoSave === AutoSaveConfiguration.AFTER_DELAY &&
+				configuration.files.autoSaveDelay <= AUTO_SAVE_AFTER_DELAY_DISABLED_TIME);
+	}
+
+	private onTextFileModelChanged(event: TextFileModelChangeEvent): void {
+		if (event.kind === StateChange.REVERTED) {
+			// This must proceed even if auto save after delay is configured in order to clean up
+			// any backups made before the config change
+			this.discardBackup(event.resource);
+		} else if (event.kind === StateChange.CONTENT_CHANGE) {
+			// Do not backup when auto save after delay is configured
+			if (!this.configuredAutoSaveAfterDelay) {
+				const model = this.textFileService.models.get(event.resource);
+				this.backupFileService.backupResource(model.getResource(), model.getValue(), model.getVersionId()).done(null, errors.onUnexpectedError);
+			}
+		}
+	}
+
+	private onUntitledModelChanged(resource: Uri): void {
+		if (this.untitledEditorService.isDirty(resource)) {
+			this.untitledEditorService.loadOrCreate({ resource }).then(model => this.backupFileService.backupResource(resource, model.getValue(), model.getVersionId())).done(null, errors.onUnexpectedError);
+		} else {
+			this.discardBackup(resource);
+		}
+	}
+
+	private discardBackup(resource: Uri): void {
+		this.backupFileService.discardResourceBackup(resource).done(null, errors.onUnexpectedError);
+	}
+
+	public dispose(): void {
+		this.toDispose = dispose(this.toDispose);
+	}
+
+	public getId(): string {
+		return 'vs.backup.backupModelTracker';
+	}
+}