textFileEditorModel.ts 40.6 KB
Newer Older
E
Erich Gamma 已提交
1 2 3 4 5
/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

6
import { join } from 'vs/base/common/path';
7
import * as nls from 'vs/nls';
M
Matt Bierner 已提交
8
import { Event, Emitter } from 'vs/base/common/event';
J
Johannes Rieken 已提交
9 10
import { guessMimeTypes } from 'vs/base/common/mime';
import { toErrorMessage } from 'vs/base/common/errorMessage';
11
import { URI } from 'vs/base/common/uri';
12
import { isUndefinedOrNull, withUndefinedAsNull } from 'vs/base/common/types';
13
import { IWorkspaceContextService } from 'vs/platform/workspace/common/workspace';
14
import { IEnvironmentService } from 'vs/platform/environment/common/environment';
15
import { ITextFileService, IAutoSaveConfiguration, ModelState, ITextFileEditorModel, ISaveOptions, ISaveErrorHandler, ISaveParticipant, StateChange, SaveReason, IRawTextContent, ILoadOptions, LoadReason, IResolvedTextFileEditorModel } from 'vs/workbench/services/textfile/common/textfiles';
B
Benjamin Pasero 已提交
16
import { EncodingMode } from 'vs/workbench/common/editor';
J
Johannes Rieken 已提交
17
import { BaseTextEditorModel } from 'vs/workbench/common/editor/textEditorModel';
18
import { IBackupFileService } from 'vs/workbench/services/backup/common/backup';
19
import { IFileService, FileOperationError, FileOperationResult, CONTENT_CHANGE_EVENT_BUFFER_DELAY, FileChangesEvent, FileChangeType, IFileStatWithMetadata, etag } from 'vs/platform/files/common/files';
J
Johannes Rieken 已提交
20
import { IInstantiationService } from 'vs/platform/instantiation/common/instantiation';
A
Alex Dima 已提交
21
import { IModeService, ILanguageSelection } from 'vs/editor/common/services/modeService';
22
import { IModelService } from 'vs/editor/common/services/modelService';
23
import { ITelemetryService } from 'vs/platform/telemetry/common/telemetry';
24
import { RunOnceScheduler, timeout } from 'vs/base/common/async';
25
import { ITextBufferFactory } from 'vs/editor/common/model';
26
import { hash } from 'vs/base/common/hash';
27
import { createTextBufferFactory } from 'vs/editor/common/model/textModel';
28
import { INotificationService } from 'vs/platform/notification/common/notification';
B
Benjamin Pasero 已提交
29
import { isLinux } from 'vs/base/common/platform';
30
import { IDisposable, toDisposable } from 'vs/base/common/lifecycle';
I
isidor 已提交
31
import { ILogService } from 'vs/platform/log/common/log';
B
Benjamin Pasero 已提交
32
import { isEqual, isEqualOrParent, extname, basename } from 'vs/base/common/resources';
B
Benjamin Pasero 已提交
33
import { onUnexpectedError } from 'vs/base/common/errors';
B
Benjamin Pasero 已提交
34

E
Erich Gamma 已提交
35 36 37
/**
 * The text file editor model listens to changes to its underlying code editor model and saves these changes through the file service back to the disk.
 */
38
export class TextFileEditorModel extends BaseTextEditorModel implements ITextFileEditorModel {
E
Erich Gamma 已提交
39

B
Benjamin Pasero 已提交
40 41
	static DEFAULT_CONTENT_CHANGE_BUFFER_DELAY = CONTENT_CHANGE_EVENT_BUFFER_DELAY;
	static DEFAULT_ORPHANED_CHANGE_BUFFER_DELAY = 100;
K
kieferrm 已提交
42
	static WHITELIST_JSON = ['package.json', 'package-lock.json', 'tsconfig.json', 'jsconfig.json', 'bower.json', '.eslintrc.json', 'tslint.json', 'composer.json'];
43
	static WHITELIST_WORKSPACE_JSON = ['settings.json', 'extensions.json', 'tasks.json', 'launch.json'];
44

E
Erich Gamma 已提交
45
	private static saveErrorHandler: ISaveErrorHandler;
B
Benjamin Pasero 已提交
46 47
	static setSaveErrorHandler(handler: ISaveErrorHandler): void { TextFileEditorModel.saveErrorHandler = handler; }

48 49
	private static saveParticipant: ISaveParticipant | null;
	static setSaveParticipant(handler: ISaveParticipant | null): void { TextFileEditorModel.saveParticipant = handler; }
E
Erich Gamma 已提交
50

B
Benjamin Pasero 已提交
51 52 53 54 55 56
	private readonly _onDidContentChange: Emitter<StateChange> = this._register(new Emitter<StateChange>());
	get onDidContentChange(): Event<StateChange> { return this._onDidContentChange.event; }

	private readonly _onDidStateChange: Emitter<StateChange> = this._register(new Emitter<StateChange>());
	get onDidStateChange(): Event<StateChange> { return this._onDidStateChange.event; }

E
Erich Gamma 已提交
57 58 59 60 61 62
	private resource: URI;
	private contentEncoding: string; 			// encoding as reported from disk
	private preferredEncoding: string;			// encoding as chosen by the user
	private dirty: boolean;
	private versionId: number;
	private bufferSavedVersionId: number;
63
	private lastResolvedDiskStat: IFileStatWithMetadata;
E
Erich Gamma 已提交
64
	private blockModelContentChange: boolean;
65
	private autoSaveAfterMillies?: number;
66
	private autoSaveAfterMilliesEnabled: boolean;
67
	private autoSaveDisposable?: IDisposable;
68
	private contentChangeEventScheduler: RunOnceScheduler;
69
	private orphanedChangeEventScheduler: RunOnceScheduler;
70
	private saveSequentializer: SaveSequentializer;
E
Erich Gamma 已提交
71
	private disposed: boolean;
B
Benjamin Pasero 已提交
72
	private lastSaveAttemptTime: number;
M
Matt Bierner 已提交
73
	private createTextEditorModelPromise: Promise<TextFileEditorModel> | null;
74 75 76 77
	private inConflictMode: boolean;
	private inOrphanMode: boolean;
	private inErrorMode: boolean;

E
Erich Gamma 已提交
78 79 80
	constructor(
		resource: URI,
		preferredEncoding: string,
81
		@INotificationService private readonly notificationService: INotificationService,
E
Erich Gamma 已提交
82 83
		@IModeService modeService: IModeService,
		@IModelService modelService: IModelService,
84 85 86 87 88 89 90 91
		@IFileService private readonly fileService: IFileService,
		@IInstantiationService private readonly instantiationService: IInstantiationService,
		@ITelemetryService private readonly telemetryService: ITelemetryService,
		@ITextFileService private readonly textFileService: ITextFileService,
		@IBackupFileService private readonly backupFileService: IBackupFileService,
		@IEnvironmentService private readonly environmentService: IEnvironmentService,
		@IWorkspaceContextService private readonly contextService: IWorkspaceContextService,
		@ILogService private readonly logService: ILogService
E
Erich Gamma 已提交
92 93
	) {
		super(modelService, modeService);
B
Benjamin Pasero 已提交
94

E
Erich Gamma 已提交
95 96
		this.resource = resource;
		this.preferredEncoding = preferredEncoding;
97
		this.inOrphanMode = false;
E
Erich Gamma 已提交
98 99
		this.dirty = false;
		this.versionId = 0;
B
Benjamin Pasero 已提交
100
		this.lastSaveAttemptTime = 0;
101
		this.saveSequentializer = new SaveSequentializer();
102

B
Benjamin Pasero 已提交
103 104
		this.contentChangeEventScheduler = this._register(new RunOnceScheduler(() => this._onDidContentChange.fire(StateChange.CONTENT_CHANGE), TextFileEditorModel.DEFAULT_CONTENT_CHANGE_BUFFER_DELAY));
		this.orphanedChangeEventScheduler = this._register(new RunOnceScheduler(() => this._onDidStateChange.fire(StateChange.ORPHANED_CHANGE), TextFileEditorModel.DEFAULT_ORPHANED_CHANGE_BUFFER_DELAY));
105

106
		this.updateAutoSaveConfiguration(textFileService.getAutoSaveConfiguration());
B
Benjamin Pasero 已提交
107

108
		this.registerListeners();
E
Erich Gamma 已提交
109 110
	}

111
	private registerListeners(): void {
B
Benjamin Pasero 已提交
112 113 114 115
		this._register(this.fileService.onFileChanges(e => this.onFileChanges(e)));
		this._register(this.textFileService.onAutoSaveConfigurationChange(config => this.updateAutoSaveConfiguration(config)));
		this._register(this.textFileService.onFilesAssociationChange(e => this.onFilesAssociationChange()));
		this._register(this.onDidStateChange(e => this.onStateChange(e)));
B
Benjamin Pasero 已提交
116
	}
117

B
Benjamin Pasero 已提交
118 119
	private onStateChange(e: StateChange): void {
		if (e === StateChange.REVERTED) {
120

B
Benjamin Pasero 已提交
121 122 123 124 125 126
			// Cancel any content change event promises as they are no longer valid.
			this.contentChangeEventScheduler.cancel();

			// Refire state change reverted events as content change events
			this._onDidContentChange.fire(StateChange.REVERTED);
		}
127 128
	}

129
	private onFileChanges(e: FileChangesEvent): void {
130
		let fileEventImpactsModel = false;
131
		let newInOrphanModeGuess: boolean | undefined;
132 133 134 135 136 137 138 139 140

		// If we are currently orphaned, we check if the model file was added back
		if (this.inOrphanMode) {
			const modelFileAdded = e.contains(this.resource, FileChangeType.ADDED);
			if (modelFileAdded) {
				newInOrphanModeGuess = false;
				fileEventImpactsModel = true;
			}
		}
141

142 143 144 145 146 147 148 149
		// Otherwise we check if the model file was deleted
		else {
			const modelFileDeleted = e.contains(this.resource, FileChangeType.DELETED);
			if (modelFileDeleted) {
				newInOrphanModeGuess = true;
				fileEventImpactsModel = true;
			}
		}
150

151
		if (fileEventImpactsModel && this.inOrphanMode !== newInOrphanModeGuess) {
J
Johannes Rieken 已提交
152
			let checkOrphanedPromise: Promise<boolean>;
153 154 155 156 157
			if (newInOrphanModeGuess) {
				// We have received reports of users seeing delete events even though the file still
				// exists (network shares issue: https://github.com/Microsoft/vscode/issues/13665).
				// Since we do not want to mark the model as orphaned, we have to check if the
				// file is really gone and not just a faulty file event.
158
				checkOrphanedPromise = timeout(100).then(() => {
159 160
					if (this.disposed) {
						return true;
161
					}
162

B
Benjamin Pasero 已提交
163
					return this.fileService.exists(this.resource).then(exists => !exists);
164
				});
165
			} else {
166
				checkOrphanedPromise = Promise.resolve(false);
167
			}
168

169
			checkOrphanedPromise.then(newInOrphanModeValidated => {
170 171 172 173
				if (this.inOrphanMode !== newInOrphanModeValidated && !this.disposed) {
					this.setOrphaned(newInOrphanModeValidated);
				}
			});
174 175 176 177 178 179 180 181 182 183
		}
	}

	private setOrphaned(orphaned: boolean): void {
		if (this.inOrphanMode !== orphaned) {
			this.inOrphanMode = orphaned;
			this.orphanedChangeEventScheduler.schedule();
		}
	}

184
	private updateAutoSaveConfiguration(config: IAutoSaveConfiguration): void {
B
Benjamin Pasero 已提交
185
		const autoSaveAfterMilliesEnabled = (typeof config.autoSaveDelay === 'number') && config.autoSaveDelay > 0;
E
Erich Gamma 已提交
186

B
Benjamin Pasero 已提交
187
		this.autoSaveAfterMilliesEnabled = autoSaveAfterMilliesEnabled;
R
Rob Lourens 已提交
188
		this.autoSaveAfterMillies = autoSaveAfterMilliesEnabled ? config.autoSaveDelay : undefined;
189 190
	}

B
Benjamin Pasero 已提交
191
	private onFilesAssociationChange(): void {
B
Benjamin Pasero 已提交
192 193 194 195
		if (!this.textEditorModel) {
			return;
		}

A
Alex Dima 已提交
196
		const firstLineText = this.getFirstLineText(this.textEditorModel);
R
Rob Lourens 已提交
197
		const languageSelection = this.getOrCreateMode(this.modeService, undefined, firstLineText);
B
Benjamin Pasero 已提交
198

A
Alex Dima 已提交
199
		this.modelService.setMode(this.textEditorModel, languageSelection);
B
Benjamin Pasero 已提交
200 201
	}

B
Benjamin Pasero 已提交
202
	getVersionId(): number {
203 204 205
		return this.versionId;
	}

206
	async revert(soft?: boolean): Promise<void> {
E
Erich Gamma 已提交
207
		if (!this.isResolved()) {
R
Rob Lourens 已提交
208
			return Promise.resolve(undefined);
E
Erich Gamma 已提交
209 210
		}

211
		// Cancel any running auto-save
212
		this.cancelPendingAutoSave();
E
Erich Gamma 已提交
213 214

		// Unset flags
B
Benjamin Pasero 已提交
215
		const undo = this.setDirty(false);
E
Erich Gamma 已提交
216

217
		let loadPromise: Promise<unknown>;
218
		if (soft) {
B
Benjamin Pasero 已提交
219
			loadPromise = Promise.resolve();
220
		} else {
221
			loadPromise = this.load({ forceReadFromDisk: true });
222 223
		}

224 225
		try {
			await loadPromise;
E
Erich Gamma 已提交
226 227

			// Emit file change event
228
			this._onDidStateChange.fire(StateChange.REVERTED);
229
		} catch (error) {
E
Erich Gamma 已提交
230

231
			// Set flags back to previous values, we are still dirty if revert failed
232
			undo();
E
Erich Gamma 已提交
233

B
Benjamin Pasero 已提交
234
			return Promise.reject(error);
235
		}
E
Erich Gamma 已提交
236 237
	}

238
	load(options?: ILoadOptions): Promise<ITextFileEditorModel> {
I
isidor 已提交
239
		this.logService.trace('load() - enter', this.resource);
E
Erich Gamma 已提交
240

B
Benjamin Pasero 已提交
241 242 243
		// It is very important to not reload the model when the model is dirty.
		// We also only want to reload the model from the disk if no save is pending
		// to avoid data loss.
244
		if (this.dirty || this.saveSequentializer.hasPendingSave()) {
I
isidor 已提交
245
			this.logService.trace('load() - exit - without loading because model is dirty or being saved', this.resource);
E
Erich Gamma 已提交
246

B
Benjamin Pasero 已提交
247
			return Promise.resolve(this);
E
Erich Gamma 已提交
248 249
		}

250 251
		// Only for new models we support to load from backup
		if (!this.textEditorModel && !this.createTextEditorModelPromise) {
B
Benjamin Pasero 已提交
252
			return this.loadFromBackup(options);
253 254 255
		}

		// Otherwise load from file resource
256
		return this.loadFromFile(options);
257 258
	}

259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278
	private async loadFromBackup(options?: ILoadOptions): Promise<TextFileEditorModel> {
		const backup = await this.backupFileService.loadBackupResource(this.resource);

		// Make sure meanwhile someone else did not suceed or start loading
		if (this.createTextEditorModelPromise || this.textEditorModel) {
			return this.createTextEditorModelPromise || this;
		}

		// If we have a backup, continue loading with it
		if (!!backup) {
			const content: IRawTextContent = {
				resource: this.resource,
				name: basename(this.resource),
				mtime: Date.now(),
				size: 0,
				etag: etag(Date.now(), 0),
				value: createTextBufferFactory(''), /* will be filled later from backup */
				encoding: this.fileService.encoding.getWriteEncoding(this.resource, this.preferredEncoding).encoding,
				isReadonly: false
			};
279

280 281
			return this.loadWithContent(content, options, backup);
		}
282

283 284
		// Otherwise load from file
		return this.loadFromFile(options);
285 286
	}

287
	private async loadFromFile(options?: ILoadOptions): Promise<TextFileEditorModel> {
288 289
		const forceReadFromDisk = options && options.forceReadFromDisk;
		const allowBinary = this.isResolved() /* always allow if we resolved previously */ || (options && options.allowBinary);
290

E
Erich Gamma 已提交
291
		// Decide on etag
292
		let etag: string | undefined;
293
		if (forceReadFromDisk) {
R
Rob Lourens 已提交
294
			etag = undefined; // reset ETag if we enforce to read from disk
295 296
		} else if (this.lastResolvedDiskStat) {
			etag = this.lastResolvedDiskStat.etag; // otherwise respect etag to support caching
E
Erich Gamma 已提交
297 298
		}

B
Benjamin Pasero 已提交
299 300 301 302 303 304 305 306
		// Ensure to track the versionId before doing a long running operation
		// to make sure the model was not changed in the meantime which would
		// indicate that the user or program has made edits. If we would ignore
		// this, we could potentially loose the changes that were made because
		// after resolving the content we update the model and reset the dirty
		// flag.
		const currentVersionId = this.versionId;

T
t-amqi 已提交
307
		// Resolve Content
308 309
		try {
			const content = await this.textFileService.resolve(this.resource, { acceptTextOnly: !allowBinary, etag, encoding: this.preferredEncoding });
310

311 312
			// Clear orphaned state when loading was successful
			this.setOrphaned(false);
313

314 315 316 317
			// Guard against the model having changed in the meantime
			if (currentVersionId === this.versionId) {
				return this.loadWithContent(content, options);
			}
E
Erich Gamma 已提交
318

319 320 321
			return this;
		} catch (error) {
			const result = error.fileOperationResult;
E
Erich Gamma 已提交
322

323 324
			// Apply orphaned state based on error code
			this.setOrphaned(result === FileOperationResult.FILE_NOT_FOUND);
325

326 327
			// NotModified status is expected and can be handled gracefully
			if (result === FileOperationResult.FILE_NOT_MODIFIED_SINCE) {
E
Erich Gamma 已提交
328

329 330 331
				// Guard against the model having changed in the meantime
				if (currentVersionId === this.versionId) {
					this.setDirty(false); // Ensure we are not tracking a stale state
B
Benjamin Pasero 已提交
332
				}
333

334 335
				return this;
			}
336

337 338 339 340 341 342 343 344 345 346
			// Ignore when a model has been resolved once and the file was deleted meanwhile. Since
			// we already have the model loaded, we can return to this state and update the orphaned
			// flag to indicate that this model has no version on disk anymore.
			if (this.isResolved() && result === FileOperationResult.FILE_NOT_FOUND) {
				return this;
			}

			// Otherwise bubble up the error
			throw error;
		}
347
	}
E
Erich Gamma 已提交
348

J
Johannes Rieken 已提交
349
	private loadWithContent(content: IRawTextContent, options?: ILoadOptions, backup?: URI): Promise<TextFileEditorModel> {
350
		return this.doLoadWithContent(content, backup).then(model => {
351

352
			// Telemetry: We log the fileGet telemetry event after the model has been loaded to ensure a good mimetype
353 354
			const settingsType = this.getTypeIfSettings();
			if (settingsType) {
355
				/* __GDPR__
356
					"settingsRead" : {
357
						"settingsType": { "classification": "SystemMetaData", "purpose": "FeatureInsight" }
358
					}
359
				*/
360
				this.telemetryService.publicLog('settingsRead', { settingsType }); // Do not log read to user settings.json and .vscode folder as a fileGet event as it ruins our JSON usage data
361 362 363
			} else {
				/* __GDPR__
					"fileGet" : {
364 365 366
						"${include}": [
							"${FileTelemetryData}"
						]
367 368
					}
				*/
369
				this.telemetryService.publicLog('fileGet', this.getTelemetryData(options && options.reason ? options.reason : LoadReason.OTHER));
370
			}
371 372 373 374 375

			return model;
		});
	}

J
Johannes Rieken 已提交
376
	private doLoadWithContent(content: IRawTextContent, backup?: URI): Promise<TextFileEditorModel> {
I
isidor 已提交
377
		this.logService.trace('load() - resolved content', this.resource);
378 379

		// Update our resolved disk stat model
B
Benjamin Pasero 已提交
380
		this.updateLastResolvedDiskStat({
381 382 383
			resource: this.resource,
			name: content.name,
			mtime: content.mtime,
384
			size: content.size,
385 386
			etag: content.etag,
			isDirectory: false,
387
			isSymbolicLink: false,
I
isidor 已提交
388
			isReadonly: content.isReadonly
389
		});
390 391 392 393 394 395 396 397 398 399 400

		// Keep the original encoding to not loose it when saving
		const oldEncoding = this.contentEncoding;
		this.contentEncoding = content.encoding;

		// Handle events if encoding changed
		if (this.preferredEncoding) {
			this.updatePreferredEncoding(this.contentEncoding); // make sure to reflect the real encoding of the file (never out of sync)
		} else if (oldEncoding !== this.contentEncoding) {
			this._onDidStateChange.fire(StateChange.ENCODING);
		}
E
Erich Gamma 已提交
401

402 403
		// Update Existing Model
		if (this.textEditorModel) {
B
Benjamin Pasero 已提交
404 405 406
			this.doUpdateTextModel(content.value);

			return Promise.resolve(this);
407
		}
E
Erich Gamma 已提交
408

409 410
		// Join an existing request to create the editor model to avoid race conditions
		else if (this.createTextEditorModelPromise) {
I
isidor 已提交
411
			this.logService.trace('load() - join existing text editor model promise', this.resource);
E
Erich Gamma 已提交
412

413 414
			return this.createTextEditorModelPromise;
		}
E
Erich Gamma 已提交
415

416
		// Create New Model
417
		return this.doCreateTextModel(content.resource, content.value, backup);
418
	}
E
Erich Gamma 已提交
419

B
Benjamin Pasero 已提交
420
	private doUpdateTextModel(value: ITextBufferFactory): void {
I
isidor 已提交
421
		this.logService.trace('load() - updated text editor model', this.resource);
422

423 424
		// Ensure we are not tracking a stale state
		this.setDirty(false);
425

426
		// Update model value in a block that ignores model content change events
427 428 429 430 431 432 433
		this.blockModelContentChange = true;
		try {
			this.updateTextEditorModel(value);
		} finally {
			this.blockModelContentChange = false;
		}

434 435
		// Ensure we track the latest saved version ID given that the contents changed
		this.updateSavedVersionId();
436 437
	}

M
Matt Bierner 已提交
438
	private doCreateTextModel(resource: URI, value: ITextBufferFactory, backup: URI | undefined): Promise<TextFileEditorModel> {
I
isidor 已提交
439
		this.logService.trace('load() - created text editor model', this.resource);
440

441
		this.createTextEditorModelPromise = this.doLoadBackup(backup).then(backupContent => {
B
Benjamin Pasero 已提交
442
			this.createTextEditorModelPromise = null;
443

B
Benjamin Pasero 已提交
444 445
			// Create model
			const hasBackupContent = !!backupContent;
M
Matt Bierner 已提交
446
			this.createTextEditorModel(backupContent ? backupContent : value, resource);
B
Benjamin Pasero 已提交
447 448 449 450 451 452 453 454

			// We restored a backup so we have to set the model as being dirty
			// We also want to trigger auto save if it is enabled to simulate the exact same behaviour
			// you would get if manually making the model dirty (fixes https://github.com/Microsoft/vscode/issues/16977)
			if (hasBackupContent) {
				this.makeDirty();
				if (this.autoSaveAfterMilliesEnabled) {
					this.doAutoSave(this.versionId);
455
				}
B
Benjamin Pasero 已提交
456
			}
457

B
Benjamin Pasero 已提交
458 459 460 461
			// Ensure we are not tracking a stale state
			else {
				this.setDirty(false);
			}
462

B
Benjamin Pasero 已提交
463 464
			// Model Listeners
			this.installModelListeners();
465

B
Benjamin Pasero 已提交
466 467 468
			return this;
		}, error => {
			this.createTextEditorModelPromise = null;
469

B
Benjamin Pasero 已提交
470
			return Promise.reject<TextFileEditorModel>(error);
471
		});
472

473 474 475
		return this.createTextEditorModelPromise;
	}

B
Benjamin Pasero 已提交
476 477
	private installModelListeners(): void {

478 479 480
		// See https://github.com/Microsoft/vscode/issues/30189
		// This code has been extracted to a different method because it caused a memory leak
		// where `value` was captured in the content change listener closure scope.
B
Benjamin Pasero 已提交
481 482

		// Content Change
M
Matt Bierner 已提交
483 484 485
		if (this.textEditorModel) {
			this._register(this.textEditorModel.onDidChangeContent(() => this.onModelContentChanged()));
		}
486 487
	}

488
	private async doLoadBackup(backup: URI | undefined): Promise<ITextBufferFactory | null> {
489
		if (!backup) {
490
			return null;
491
		}
492

493 494 495 496 497
		try {
			return withUndefinedAsNull(await this.backupFileService.resolveBackupContent(backup));
		} catch (error) {
			return null; // ignore errors
		}
E
Erich Gamma 已提交
498 499
	}

500
	protected getOrCreateMode(modeService: IModeService, preferredModeIds: string | undefined, firstLineText?: string): ILanguageSelection {
A
Alex Dima 已提交
501
		return modeService.createByFilepathOrFirstLine(this.resource.fsPath, firstLineText);
E
Erich Gamma 已提交
502 503
	}

504
	private onModelContentChanged(): void {
I
isidor 已提交
505
		this.logService.trace(`onModelContentChanged() - enter`, this.resource);
E
Erich Gamma 已提交
506 507 508

		// In any case increment the version id because it tracks the textual content state of the model at all times
		this.versionId++;
I
isidor 已提交
509
		this.logService.trace(`onModelContentChanged() - new versionId ${this.versionId}`, this.resource);
E
Erich Gamma 已提交
510 511 512 513 514 515 516 517 518 519

		// Ignore if blocking model changes
		if (this.blockModelContentChange) {
			return;
		}

		// The contents changed as a matter of Undo and the version reached matches the saved one
		// In this case we clear the dirty flag and emit a SAVED event to indicate this state.
		// Note: we currently only do this check when auto-save is turned off because there you see
		// a dirty indicator that you want to get rid of when undoing to the saved version.
M
Matt Bierner 已提交
520
		if (!this.autoSaveAfterMilliesEnabled && this.textEditorModel && this.textEditorModel.getAlternativeVersionId() === this.bufferSavedVersionId) {
I
isidor 已提交
521
			this.logService.trace('onModelContentChanged() - model content changed back to last saved version', this.resource);
E
Erich Gamma 已提交
522 523

			// Clear flags
524
			const wasDirty = this.dirty;
E
Erich Gamma 已提交
525 526 527
			this.setDirty(false);

			// Emit event
528
			if (wasDirty) {
529
				this._onDidStateChange.fire(StateChange.REVERTED);
530
			}
E
Erich Gamma 已提交
531 532 533 534

			return;
		}

I
isidor 已提交
535
		this.logService.trace('onModelContentChanged() - model content changed and marked as dirty', this.resource);
E
Erich Gamma 已提交
536 537

		// Mark as dirty
B
Benjamin Pasero 已提交
538
		this.makeDirty();
E
Erich Gamma 已提交
539 540

		// Start auto save process unless we are in conflict resolution mode and unless it is disabled
541
		if (this.autoSaveAfterMilliesEnabled) {
542
			if (!this.inConflictMode) {
E
Erich Gamma 已提交
543 544
				this.doAutoSave(this.versionId);
			} else {
I
isidor 已提交
545
				this.logService.trace('makeDirty() - prevented save because we are in conflict resolution mode', this.resource);
E
Erich Gamma 已提交
546 547
			}
		}
548

549 550
		// Handle content change events
		this.contentChangeEventScheduler.schedule();
E
Erich Gamma 已提交
551 552
	}

B
Benjamin Pasero 已提交
553
	private makeDirty(): void {
E
Erich Gamma 已提交
554 555

		// Track dirty state and version id
B
Benjamin Pasero 已提交
556
		const wasDirty = this.dirty;
E
Erich Gamma 已提交
557 558 559 560
		this.setDirty(true);

		// Emit as Event if we turned dirty
		if (!wasDirty) {
561
			this._onDidStateChange.fire(StateChange.DIRTY);
E
Erich Gamma 已提交
562 563 564
		}
	}

565
	private doAutoSave(versionId: number): void {
I
isidor 已提交
566
		this.logService.trace(`doAutoSave() - enter for versionId ${versionId}`, this.resource);
E
Erich Gamma 已提交
567 568

		// Cancel any currently running auto saves to make this the one that succeeds
569
		this.cancelPendingAutoSave();
E
Erich Gamma 已提交
570

571 572
		// Create new save timer and store it for disposal as needed
		const handle = setTimeout(() => {
E
Erich Gamma 已提交
573 574 575

			// Only trigger save if the version id has not changed meanwhile
			if (versionId === this.versionId) {
576
				this.doSave(versionId, { reason: SaveReason.AUTO }); // Very important here to not return the promise because if the timeout promise is canceled it will bubble up the error otherwise - do not change
E
Erich Gamma 已提交
577
			}
578
		}, this.autoSaveAfterMillies);
E
Erich Gamma 已提交
579

580
		this.autoSaveDisposable = toDisposable(() => clearTimeout(handle));
E
Erich Gamma 已提交
581 582
	}

583 584 585
	private cancelPendingAutoSave(): void {
		if (this.autoSaveDisposable) {
			this.autoSaveDisposable.dispose();
R
Rob Lourens 已提交
586
			this.autoSaveDisposable = undefined;
E
Erich Gamma 已提交
587 588 589
		}
	}

J
Johannes Rieken 已提交
590
	save(options: ISaveOptions = Object.create(null)): Promise<void> {
E
Erich Gamma 已提交
591
		if (!this.isResolved()) {
R
Rob Lourens 已提交
592
			return Promise.resolve(undefined);
E
Erich Gamma 已提交
593 594
		}

I
isidor 已提交
595
		this.logService.trace('save() - enter', this.resource);
E
Erich Gamma 已提交
596 597

		// Cancel any currently running auto saves to make this the one that succeeds
598
		this.cancelPendingAutoSave();
E
Erich Gamma 已提交
599

600
		return this.doSave(this.versionId, options);
E
Erich Gamma 已提交
601 602
	}

J
Johannes Rieken 已提交
603
	private doSave(versionId: number, options: ISaveOptions): Promise<void> {
604
		if (isUndefinedOrNull(options.reason)) {
605 606 607
			options.reason = SaveReason.EXPLICIT;
		}

I
isidor 已提交
608
		this.logService.trace(`doSave(${versionId}) - enter with versionId ' + versionId`, this.resource);
E
Erich Gamma 已提交
609 610

		// Lookup any running pending save for this versionId and return it if found
B
Benjamin Pasero 已提交
611 612 613 614
		//
		// Scenario: user invoked the save action multiple times quickly for the same contents
		//           while the save was not yet finished to disk
		//
615
		if (this.saveSequentializer.hasPendingSave(versionId)) {
I
isidor 已提交
616
			this.logService.trace(`doSave(${versionId}) - exit - found a pending save for versionId ${versionId}`, this.resource);
E
Erich Gamma 已提交
617

M
Matt Bierner 已提交
618
			return this.saveSequentializer.pendingSave || Promise.resolve(undefined);
E
Erich Gamma 已提交
619 620
		}

621
		// Return early if not dirty (unless forced) or version changed meanwhile
B
Benjamin Pasero 已提交
622 623 624 625 626 627
		//
		// Scenario A: user invoked save action even though the model is not dirty
		// Scenario B: auto save was triggered for a certain change by the user but meanwhile the user changed
		//             the contents and the version for which auto save was started is no longer the latest.
		//             Thus we avoid spawning multiple auto saves and only take the latest.
		//
628
		if ((!options.force && !this.dirty) || versionId !== this.versionId) {
I
isidor 已提交
629
			this.logService.trace(`doSave(${versionId}) - exit - because not dirty and/or versionId is different (this.isDirty: ${this.dirty}, this.versionId: ${this.versionId})`, this.resource);
E
Erich Gamma 已提交
630

R
Rob Lourens 已提交
631
			return Promise.resolve(undefined);
E
Erich Gamma 已提交
632 633
		}

634
		// Return if currently saving by storing this save request as the next save that should happen.
635
		// Never ever must 2 saves execute at the same time because this can lead to dirty writes and race conditions.
B
Benjamin Pasero 已提交
636
		//
637
		// Scenario A: auto save was triggered and is currently busy saving to disk. this takes long enough that another auto save
638
		//             kicks in.
639 640
		// Scenario B: save is very slow (e.g. network share) and the user manages to change the buffer and trigger another save
		//             while the first save has not returned yet.
B
Benjamin Pasero 已提交
641
		//
642
		if (this.saveSequentializer.hasPendingSave()) {
I
isidor 已提交
643
			this.logService.trace(`doSave(${versionId}) - exit - because busy saving`, this.resource);
E
Erich Gamma 已提交
644

645
			// Register this as the next upcoming save and return
646
			return this.saveSequentializer.setNext(() => this.doSave(this.versionId /* make sure to use latest version id here */, options));
E
Erich Gamma 已提交
647 648 649 650
		}

		// Push all edit operations to the undo stack so that the user has a chance to
		// Ctrl+Z back to the saved version. We only do this when auto-save is turned off
M
Matt Bierner 已提交
651 652
		if (!this.autoSaveAfterMilliesEnabled && this.textEditorModel) {
			this.textEditorModel.pushStackElement();
E
Erich Gamma 已提交
653 654
		}

B
Benjamin Pasero 已提交
655
		// A save participant can still change the model now and since we are so close to saving
E
Erich Gamma 已提交
656 657
		// we do not want to trigger another auto save or similar, so we block this
		// In addition we update our version right after in case it changed because of a model change
658
		// Save participants can also be skipped through API.
J
Johannes Rieken 已提交
659
		let saveParticipantPromise: Promise<number> = Promise.resolve(versionId);
660
		if (TextFileEditorModel.saveParticipant && !options.skipSaveParticipants) {
B
💄  
Benjamin Pasero 已提交
661
			const onCompleteOrError = () => {
J
Johannes Rieken 已提交
662
				this.blockModelContentChange = false;
B
💄  
Benjamin Pasero 已提交
663

664
				return this.versionId;
B
💄  
Benjamin Pasero 已提交
665 666
			};

B
Benjamin Pasero 已提交
667
			this.blockModelContentChange = true;
668
			saveParticipantPromise = TextFileEditorModel.saveParticipant.participate(this as IResolvedTextFileEditorModel, { reason: options.reason }).then(onCompleteOrError, onCompleteOrError);
E
Erich Gamma 已提交
669 670
		}

671
		// mark the save participant as current pending save operation
672
		return this.saveSequentializer.setPending(versionId, saveParticipantPromise.then(newVersionId => {
E
Erich Gamma 已提交
673

674 675 676 677 678
			// We have to protect against being disposed at this point. It could be that the save() operation
			// was triggerd followed by a dispose() operation right after without waiting. Typically we cannot
			// be disposed if we are dirty, but if we are not dirty, save() and dispose() can still be triggered
			// one after the other without waiting for the save() to complete. If we are disposed(), we risk
			// saving contents to disk that are stale (see https://github.com/Microsoft/vscode/issues/50942).
679
			// To fix this issue, we will not store the contents to disk when we got disposed.
680
			if (this.disposed) {
R
Rob Lourens 已提交
681
				return undefined;
682 683
			}

B
Benjamin Pasero 已提交
684 685
			// Under certain conditions we do a short-cut of flushing contents to disk when we can assume that
			// the file has not changed and as such was not dirty before.
686 687 688 689 690 691
			// The conditions are all of:
			// - a forced, explicit save (Ctrl+S)
			// - the model is not dirty (otherwise we know there are changed which needs to go to the file)
			// - the model is not in orphan mode (because in that case we know the file does not exist on disk)
			// - the model version did not change due to save participants running
			if (options.force && !this.dirty && !this.inOrphanMode && options.reason === SaveReason.EXPLICIT && versionId === newVersionId) {
B
Benjamin Pasero 已提交
692
				return this.doTouch(newVersionId);
693 694
			}

695
			// update versionId with its new value (if pre-save changes happened)
J
Johannes Rieken 已提交
696 697 698 699 700 701 702 703 704
			versionId = newVersionId;

			// Clear error flag since we are trying to save again
			this.inErrorMode = false;

			// Remember when this model was saved last
			this.lastSaveAttemptTime = Date.now();

			// Save to Disk
705
			// mark the save operation as currently pending with the versionId (it might have changed from a save participant triggering)
I
isidor 已提交
706
			this.logService.trace(`doSave(${versionId}) - before updateContent()`, this.resource);
M
Matt Bierner 已提交
707 708 709 710
			const snapshot = this.createSnapshot();
			if (!snapshot) {
				throw new Error('Invalid snapshot');
			}
711
			return this.saveSequentializer.setPending(newVersionId, this.textFileService.update(this.lastResolvedDiskStat.resource, snapshot, {
712 713
				overwriteReadonly: options.overwriteReadonly,
				overwriteEncoding: options.overwriteEncoding,
714
				mtime: this.lastResolvedDiskStat.mtime,
J
Johannes Rieken 已提交
715
				encoding: this.getEncoding(),
716 717
				etag: this.lastResolvedDiskStat.etag,
				writeElevated: options.writeElevated
718
			}).then(stat => {
I
isidor 已提交
719
				this.logService.trace(`doSave(${versionId}) - after updateContent()`, this.resource);
J
Johannes Rieken 已提交
720 721 722

				// Update dirty state unless model has changed meanwhile
				if (versionId === this.versionId) {
I
isidor 已提交
723
					this.logService.trace(`doSave(${versionId}) - setting dirty to false because versionId did not change`, this.resource);
J
Johannes Rieken 已提交
724 725
					this.setDirty(false);
				} else {
I
isidor 已提交
726
					this.logService.trace(`doSave(${versionId}) - not setting dirty to false because versionId did change meanwhile`, this.resource);
J
Johannes Rieken 已提交
727
				}
E
Erich Gamma 已提交
728

729
				// Updated resolved stat with updated stat
730
				this.updateLastResolvedDiskStat(stat);
E
Erich Gamma 已提交
731

732 733 734
				// Cancel any content change event promises as they are no longer valid
				this.contentChangeEventScheduler.cancel();

J
Johannes Rieken 已提交
735 736
				// Emit File Saved Event
				this._onDidStateChange.fire(StateChange.SAVED);
737 738 739 740 741 742 743 744 745 746 747

				// Telemetry
				const settingsType = this.getTypeIfSettings();
				if (settingsType) {
					/* __GDPR__
						"settingsWritten" : {
							"settingsType": { "classification": "SystemMetaData", "purpose": "FeatureInsight" }
						}
					*/
					this.telemetryService.publicLog('settingsWritten', { settingsType }); // Do not log write to user settings.json and .vscode folder as a filePUT event as it ruins our JSON usage data
				} else {
748
					/* __GDPR__
749 750 751 752 753 754
							"filePUT" : {
								"${include}": [
									"${FileTelemetryData}"
								]
							}
						*/
755
					this.telemetryService.publicLog('filePUT', this.getTelemetryData(options.reason));
756
				}
757
			}, error => {
758 759
				if (!error) {
					error = new Error('Unknown Save Error'); // TODO@remote we should never get null as error (https://github.com/Microsoft/vscode/issues/55051)
M
Martin Aeschlimann 已提交
760 761
				}

I
isidor 已提交
762
				this.logService.error(`doSave(${versionId}) - exit - resulted in a save error: ${error.toString()}`, this.resource);
E
Erich Gamma 已提交
763

764
				// Flag as error state in the model
J
Johannes Rieken 已提交
765
				this.inErrorMode = true;
E
Erich Gamma 已提交
766

767
				// Look out for a save conflict
768
				if ((<FileOperationError>error).fileOperationResult === FileOperationResult.FILE_MODIFIED_SINCE) {
769 770 771
					this.inConflictMode = true;
				}

J
Johannes Rieken 已提交
772 773
				// Show to user
				this.onSaveError(error);
E
Erich Gamma 已提交
774

J
Johannes Rieken 已提交
775 776
				// Emit as event
				this._onDidStateChange.fire(StateChange.SAVE_ERROR);
777 778
			}));
		}));
E
Erich Gamma 已提交
779 780
	}

781
	private getTypeIfSettings(): string {
B
Benjamin Pasero 已提交
782
		if (extname(this.resource) !== '.json') {
783
			return '';
784
		}
785 786

		// Check for global settings file
S
Sandeep Somavarapu 已提交
787
		if (isEqual(this.resource, URI.file(this.environmentService.appSettingsPath), !isLinux)) {
788 789 790 791 792 793 794 795 796
			return 'global-settings';
		}

		// Check for keybindings file
		if (isEqual(this.resource, URI.file(this.environmentService.appKeybindingsPath), !isLinux)) {
			return 'keybindings';
		}

		// Check for locale file
797
		if (isEqual(this.resource, URI.file(join(this.environmentService.appSettingsHome, 'locale.json')), !isLinux)) {
798 799 800 801
			return 'locale';
		}

		// Check for snippets
802
		if (isEqualOrParent(this.resource, URI.file(join(this.environmentService.appSettingsHome, 'snippets')))) {
803
			return 'snippets';
804 805 806
		}

		// Check for workspace settings file
807
		const folders = this.contextService.getWorkspace().folders;
808 809
		for (const folder of folders) {
			if (isEqualOrParent(this.resource, folder.toResource('.vscode'))) {
B
Benjamin Pasero 已提交
810
				const filename = basename(this.resource);
811 812 813 814 815 816 817
				if (TextFileEditorModel.WHITELIST_WORKSPACE_JSON.indexOf(filename) > -1) {
					return `.vscode/${filename}`;
				}
			}
		}

		return '';
818 819
	}

820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840
	private getTelemetryData(reason: number | undefined): object {
		const ext = extname(this.resource);
		const fileName = basename(this.resource);
		const telemetryData = {
			mimeType: guessMimeTypes(this.resource.fsPath).join(', '),
			ext,
			path: hash(this.resource.fsPath),
			reason
		};

		if (ext === '.json' && TextFileEditorModel.WHITELIST_JSON.indexOf(fileName) > -1) {
			telemetryData['whitelistedjson'] = fileName;
		}

		/* __GDPR__FRAGMENT__
			"FileTelemetryData" : {
				"mimeType" : { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
				"ext": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
				"path": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
				"reason": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "isMeasurement": true },
				"whitelistedjson": { "classification": "SystemMetaData", "purpose": "FeatureInsight" }
841
			}
842 843
		*/
		return telemetryData;
844 845
	}

J
Johannes Rieken 已提交
846
	private doTouch(versionId: number): Promise<void> {
M
Matt Bierner 已提交
847 848 849 850
		const snapshot = this.createSnapshot();
		if (!snapshot) {
			throw new Error('invalid snapshot');
		}
851 852

		return this.saveSequentializer.setPending(versionId, this.textFileService.update(this.lastResolvedDiskStat.resource, snapshot, {
B
Benjamin Pasero 已提交
853 854 855 856
			mtime: this.lastResolvedDiskStat.mtime,
			encoding: this.getEncoding(),
			etag: this.lastResolvedDiskStat.etag
		}).then(stat => {
857 858 859

			// Updated resolved stat with updated stat since touching it might have changed mtime
			this.updateLastResolvedDiskStat(stat);
B
Benjamin Pasero 已提交
860
		}, error => onUnexpectedError(error) /* just log any error but do not notify the user since the file was not dirty */));
861 862
	}

E
Erich Gamma 已提交
863
	private setDirty(dirty: boolean): () => void {
B
Benjamin Pasero 已提交
864
		const wasDirty = this.dirty;
865
		const wasInConflictMode = this.inConflictMode;
B
Benjamin Pasero 已提交
866 867
		const wasInErrorMode = this.inErrorMode;
		const oldBufferSavedVersionId = this.bufferSavedVersionId;
E
Erich Gamma 已提交
868 869 870

		if (!dirty) {
			this.dirty = false;
871
			this.inConflictMode = false;
E
Erich Gamma 已提交
872
			this.inErrorMode = false;
873
			this.updateSavedVersionId();
E
Erich Gamma 已提交
874 875 876 877 878 879 880
		} else {
			this.dirty = true;
		}

		// Return function to revert this call
		return () => {
			this.dirty = wasDirty;
881
			this.inConflictMode = wasInConflictMode;
E
Erich Gamma 已提交
882 883 884 885 886
			this.inErrorMode = wasInErrorMode;
			this.bufferSavedVersionId = oldBufferSavedVersionId;
		};
	}

887 888 889 890 891 892 893 894 895 896 897
	private updateSavedVersionId(): void {
		// we remember the models alternate version id to remember when the version
		// of the model matches with the saved version on disk. we need to keep this
		// in order to find out if the model changed back to a saved version (e.g.
		// when undoing long enough to reach to a version that is saved and then to
		// clear the dirty flag)
		if (this.textEditorModel) {
			this.bufferSavedVersionId = this.textEditorModel.getAlternativeVersionId();
		}
	}

898
	private updateLastResolvedDiskStat(newVersionOnDiskStat: IFileStatWithMetadata): void {
E
Erich Gamma 已提交
899 900

		// First resolve - just take
901 902
		if (!this.lastResolvedDiskStat) {
			this.lastResolvedDiskStat = newVersionOnDiskStat;
E
Erich Gamma 已提交
903 904 905
		}

		// Subsequent resolve - make sure that we only assign it if the mtime is equal or has advanced.
B
Benjamin Pasero 已提交
906 907
		// This prevents race conditions from loading and saving. If a save comes in late after a revert
		// was called, the mtime could be out of sync.
908 909
		else if (this.lastResolvedDiskStat.mtime <= newVersionOnDiskStat.mtime) {
			this.lastResolvedDiskStat = newVersionOnDiskStat;
E
Erich Gamma 已提交
910 911 912
		}
	}

913
	private onSaveError(error: Error): void {
E
Erich Gamma 已提交
914 915 916 917 918 919 920 921 922 923

		// Prepare handler
		if (!TextFileEditorModel.saveErrorHandler) {
			TextFileEditorModel.setSaveErrorHandler(this.instantiationService.createInstance(DefaultSaveErrorHandler));
		}

		// Handle
		TextFileEditorModel.saveErrorHandler.onSaveError(error, this);
	}

B
Benjamin Pasero 已提交
924
	isDirty(): boolean {
E
Erich Gamma 已提交
925 926 927
		return this.dirty;
	}

B
Benjamin Pasero 已提交
928
	getLastSaveAttemptTime(): number {
B
Benjamin Pasero 已提交
929
		return this.lastSaveAttemptTime;
E
Erich Gamma 已提交
930 931
	}

932
	getETag(): string | null {
M
Matt Bierner 已提交
933
		return this.lastResolvedDiskStat ? this.lastResolvedDiskStat.etag || null : null;
E
Erich Gamma 已提交
934 935
	}

B
Benjamin Pasero 已提交
936
	hasState(state: ModelState): boolean {
937 938 939 940 941 942 943 944 945 946 947 948 949
		switch (state) {
			case ModelState.CONFLICT:
				return this.inConflictMode;
			case ModelState.DIRTY:
				return this.dirty;
			case ModelState.ERROR:
				return this.inErrorMode;
			case ModelState.ORPHAN:
				return this.inOrphanMode;
			case ModelState.PENDING_SAVE:
				return this.saveSequentializer.hasPendingSave();
			case ModelState.SAVED:
				return !this.dirty;
E
Erich Gamma 已提交
950 951 952
		}
	}

B
Benjamin Pasero 已提交
953
	getEncoding(): string {
E
Erich Gamma 已提交
954 955 956
		return this.preferredEncoding || this.contentEncoding;
	}

B
Benjamin Pasero 已提交
957
	setEncoding(encoding: string, mode: EncodingMode): void {
E
Erich Gamma 已提交
958 959 960 961 962 963 964 965 966 967 968 969 970 971
		if (!this.isNewEncoding(encoding)) {
			return; // return early if the encoding is already the same
		}

		// Encode: Save with encoding
		if (mode === EncodingMode.Encode) {
			this.updatePreferredEncoding(encoding);

			// Save
			if (!this.isDirty()) {
				this.versionId++; // needs to increment because we change the model potentially
				this.makeDirty();
			}

972
			if (!this.inConflictMode) {
973
				this.save({ overwriteEncoding: true });
E
Erich Gamma 已提交
974 975 976 977 978 979
			}
		}

		// Decode: Load with encoding
		else {
			if (this.isDirty()) {
980
				this.notificationService.info(nls.localize('saveFileFirst', "The file is dirty. Please save it first before reopening it with another encoding."));
E
Erich Gamma 已提交
981 982 983 984 985 986 987

				return;
			}

			this.updatePreferredEncoding(encoding);

			// Load
988 989
			this.load({
				forceReadFromDisk: true	// because encoding has changed
990
			});
E
Erich Gamma 已提交
991 992 993
		}
	}

B
Benjamin Pasero 已提交
994
	updatePreferredEncoding(encoding: string): void {
E
Erich Gamma 已提交
995 996 997 998 999 1000 1001
		if (!this.isNewEncoding(encoding)) {
			return;
		}

		this.preferredEncoding = encoding;

		// Emit
1002
		this._onDidStateChange.fire(StateChange.ENCODING);
E
Erich Gamma 已提交
1003 1004 1005 1006 1007 1008 1009 1010
	}

	private isNewEncoding(encoding: string): boolean {
		if (this.preferredEncoding === encoding) {
			return false; // return early if the encoding is already the same
		}

		if (!this.preferredEncoding && this.contentEncoding === encoding) {
1011
			return false; // also return if we don't have a preferred encoding but the content encoding is already the same
E
Erich Gamma 已提交
1012 1013 1014 1015 1016
		}

		return true;
	}

B
Benjamin Pasero 已提交
1017
	isResolved(): boolean {
1018
		return !isUndefinedOrNull(this.lastResolvedDiskStat);
E
Erich Gamma 已提交
1019 1020
	}

B
Benjamin Pasero 已提交
1021
	isReadonly(): boolean {
1022
		return !!(this.lastResolvedDiskStat && this.lastResolvedDiskStat.isReadonly);
1023 1024
	}

B
Benjamin Pasero 已提交
1025
	isDisposed(): boolean {
E
Erich Gamma 已提交
1026 1027 1028
		return this.disposed;
	}

B
Benjamin Pasero 已提交
1029
	getResource(): URI {
E
Erich Gamma 已提交
1030 1031 1032
		return this.resource;
	}

1033
	getStat(): IFileStatWithMetadata {
B
Benjamin Pasero 已提交
1034 1035 1036
		return this.lastResolvedDiskStat;
	}

B
Benjamin Pasero 已提交
1037
	dispose(): void {
E
Erich Gamma 已提交
1038
		this.disposed = true;
1039 1040
		this.inConflictMode = false;
		this.inOrphanMode = false;
E
Erich Gamma 已提交
1041 1042 1043 1044
		this.inErrorMode = false;

		this.createTextEditorModelPromise = null;

1045
		this.cancelPendingAutoSave();
D
Daniel Imms 已提交
1046

E
Erich Gamma 已提交
1047 1048
		super.dispose();
	}
1049 1050
}

1051 1052
interface IPendingSave {
	versionId: number;
J
Johannes Rieken 已提交
1053
	promise: Promise<void>;
1054 1055
}

1056
interface ISaveOperation {
J
Johannes Rieken 已提交
1057
	promise: Promise<void>;
B
Benjamin Pasero 已提交
1058 1059
	promiseResolve: () => void;
	promiseReject: (error: Error) => void;
J
Johannes Rieken 已提交
1060
	run: () => Promise<void>;
1061 1062 1063
}

export class SaveSequentializer {
1064 1065
	private _pendingSave?: IPendingSave;
	private _nextSave?: ISaveOperation;
1066

B
Benjamin Pasero 已提交
1067
	hasPendingSave(versionId?: number): boolean {
1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078
		if (!this._pendingSave) {
			return false;
		}

		if (typeof versionId === 'number') {
			return this._pendingSave.versionId === versionId;
		}

		return !!this._pendingSave;
	}

1079
	get pendingSave(): Promise<void> | undefined {
R
Rob Lourens 已提交
1080
		return this._pendingSave ? this._pendingSave.promise : undefined;
1081 1082
	}

J
Johannes Rieken 已提交
1083
	setPending(versionId: number, promise: Promise<void>): Promise<void> {
1084 1085
		this._pendingSave = { versionId, promise };

1086
		promise.then(() => this.donePending(versionId), () => this.donePending(versionId));
1087 1088 1089 1090 1091 1092

		return promise;
	}

	private donePending(versionId: number): void {
		if (this._pendingSave && versionId === this._pendingSave.versionId) {
1093 1094

			// only set pending to done if the promise finished that is associated with that versionId
R
Rob Lourens 已提交
1095
			this._pendingSave = undefined;
1096 1097 1098

			// schedule the next save now that we are free if we have any
			this.triggerNextSave();
1099 1100 1101
		}
	}

1102 1103 1104
	private triggerNextSave(): void {
		if (this._nextSave) {
			const saveOperation = this._nextSave;
R
Rob Lourens 已提交
1105
			this._nextSave = undefined;
1106 1107

			// Run next save and complete on the associated promise
B
Benjamin Pasero 已提交
1108
			saveOperation.run().then(saveOperation.promiseResolve, saveOperation.promiseReject);
1109 1110 1111
		}
	}

J
Johannes Rieken 已提交
1112
	setNext(run: () => Promise<void>): Promise<void> {
1113 1114 1115 1116 1117

		// this is our first next save, so we create associated promise with it
		// so that we can return a promise that completes when the save operation
		// has completed.
		if (!this._nextSave) {
M
Matt Bierner 已提交
1118 1119
			let promiseResolve: () => void;
			let promiseReject: (error: Error) => void;
B
Benjamin Pasero 已提交
1120 1121 1122
			const promise = new Promise<void>((resolve, reject) => {
				promiseResolve = resolve;
				promiseReject = reject;
1123 1124 1125 1126 1127
			});

			this._nextSave = {
				run,
				promise,
M
Matt Bierner 已提交
1128 1129
				promiseResolve: promiseResolve!,
				promiseReject: promiseReject!
1130 1131 1132 1133 1134 1135 1136 1137
			};
		}

		// we have a previous next save, just overwrite it
		else {
			this._nextSave.run = run;
		}

M
Matt Bierner 已提交
1138
		return this._nextSave.promise;
1139 1140 1141
	}
}

1142 1143
class DefaultSaveErrorHandler implements ISaveErrorHandler {

1144
	constructor(@INotificationService private readonly notificationService: INotificationService) { }
1145

1146
	onSaveError(error: Error, model: TextFileEditorModel): void {
B
Benjamin Pasero 已提交
1147
		this.notificationService.error(nls.localize('genericSaveError', "Failed to save '{0}': {1}", basename(model.getResource()), toErrorMessage(error, false)));
1148 1149
	}
}