textFileEditorModel.ts 41.0 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 } 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';
R
Ramya Achutha Rao 已提交
26
import { IHashService } from 'vs/workbench/services/hash/common/hashService';
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 92
		@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,
		@IHashService private readonly hashService: IHashService,
		@ILogService private readonly logService: ILogService
E
Erich Gamma 已提交
93 94
	) {
		super(modelService, modeService);
B
Benjamin Pasero 已提交
95

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

B
Benjamin Pasero 已提交
104 105
		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));
106

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

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

112
	private registerListeners(): void {
B
Benjamin Pasero 已提交
113 114 115 116
		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 已提交
117
	}
118

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

B
Benjamin Pasero 已提交
122 123 124 125 126 127
			// 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);
		}
128 129
	}

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

		// 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;
			}
		}
142

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

152
		if (fileEventImpactsModel && this.inOrphanMode !== newInOrphanModeGuess) {
J
Johannes Rieken 已提交
153
			let checkOrphanedPromise: Promise<boolean>;
154 155 156 157 158
			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.
159
				checkOrphanedPromise = timeout(100).then(() => {
160 161
					if (this.disposed) {
						return true;
162
					}
163 164

					return this.fileService.existsFile(this.resource).then(exists => !exists);
165
				});
166
			} else {
167
				checkOrphanedPromise = Promise.resolve(false);
168
			}
169

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

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

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

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

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

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

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

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

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

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

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

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

		return loadPromise.then(() => {
E
Erich Gamma 已提交
226 227

			// Emit file change event
228
			this._onDidStateChange.fire(StateChange.REVERTED);
229
		}, 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);
E
Erich Gamma 已提交
235 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
	}

J
Johannes Rieken 已提交
259
	private loadFromBackup(options?: ILoadOptions): Promise<TextFileEditorModel> {
260 261 262 263
		return this.backupFileService.loadBackupResource(this.resource).then(backup => {

			// Make sure meanwhile someone else did not suceed or start loading
			if (this.createTextEditorModelPromise || this.textEditorModel) {
B
Benjamin Pasero 已提交
264
				return this.createTextEditorModelPromise || this;
265 266 267 268
			}

			// If we have a backup, continue loading with it
			if (!!backup) {
269
				const content: IRawTextContent = {
270
					resource: this.resource,
B
Benjamin Pasero 已提交
271
					name: basename(this.resource),
272
					mtime: Date.now(),
273
					etag: etag(Date.now(), 0),
274
					value: createTextBufferFactory(''), /* will be filled later from backup */
275
					encoding: this.fileService.encoding.getWriteEncoding(this.resource, this.preferredEncoding).encoding,
I
isidor 已提交
276
					isReadonly: false
277 278
				};

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

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

J
Johannes Rieken 已提交
287
	private 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 308
		// Resolve Content
		return this.textFileService
309
			.resolveTextContent(this.resource, { acceptTextOnly: !allowBinary, etag, encoding: this.preferredEncoding })
B
Benjamin Pasero 已提交
310
			.then(content => {
311

B
Benjamin Pasero 已提交
312 313
				// Clear orphaned state when loading was successful
				this.setOrphaned(false);
314

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

B
Benjamin Pasero 已提交
320 321 322
				return this;
			}, error => {
				const result = error.fileOperationResult;
E
Erich Gamma 已提交
323

B
Benjamin Pasero 已提交
324 325
				// Apply orphaned state based on error code
				this.setOrphaned(result === FileOperationResult.FILE_NOT_FOUND);
E
Erich Gamma 已提交
326

B
Benjamin Pasero 已提交
327 328
				// NotModified status is expected and can be handled gracefully
				if (result === FileOperationResult.FILE_NOT_MODIFIED_SINCE) {
329

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

B
Benjamin Pasero 已提交
335
					return this;
B
Benjamin Pasero 已提交
336
				}
337

B
Benjamin Pasero 已提交
338 339 340 341
				// 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) {
B
Benjamin Pasero 已提交
342
					return this;
B
Benjamin Pasero 已提交
343
				}
344

B
Benjamin Pasero 已提交
345
				// Otherwise bubble up the error
B
Benjamin Pasero 已提交
346
				return Promise.reject<TextFileEditorModel>(error);
B
Benjamin Pasero 已提交
347
			});
348
	}
E
Erich Gamma 已提交
349

J
Johannes Rieken 已提交
350
	private loadWithContent(content: IRawTextContent, options?: ILoadOptions, backup?: URI): Promise<TextFileEditorModel> {
351
		return this.doLoadWithContent(content, backup).then(model => {
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: -1,
385 386
			etag: content.etag,
			isDirectory: false,
387
			isSymbolicLink: false,
R
Rob Lourens 已提交
388
			children: undefined,
I
isidor 已提交
389
			isReadonly: content.isReadonly
390
		});
391 392 393 394 395 396 397 398 399 400 401

		// 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 已提交
402

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

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

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

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

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

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

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

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

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

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

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

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

			// 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);
456
				}
B
Benjamin Pasero 已提交
457
			}
458

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

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

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

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

474 475 476
		return this.createTextEditorModelPromise;
	}

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

479 480 481
		// 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 已提交
482 483

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

M
Matt Bierner 已提交
489
	private doLoadBackup(backup: URI | undefined): Promise<ITextBufferFactory | null> {
490
		if (!backup) {
B
Benjamin Pasero 已提交
491
			return Promise.resolve(null);
492
		}
493

M
Matt Bierner 已提交
494
		return this.backupFileService.resolveBackupContent(backup).then(backupContent => backupContent || null, error => null /* ignore errors */);
E
Erich Gamma 已提交
495 496
	}

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

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

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

		// 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 已提交
517
		if (!this.autoSaveAfterMilliesEnabled && this.textEditorModel && this.textEditorModel.getAlternativeVersionId() === this.bufferSavedVersionId) {
I
isidor 已提交
518
			this.logService.trace('onModelContentChanged() - model content changed back to last saved version', this.resource);
E
Erich Gamma 已提交
519 520

			// Clear flags
521
			const wasDirty = this.dirty;
E
Erich Gamma 已提交
522 523 524
			this.setDirty(false);

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

			return;
		}

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

		// Mark as dirty
B
Benjamin Pasero 已提交
535
		this.makeDirty();
E
Erich Gamma 已提交
536 537

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

546 547
		// Handle content change events
		this.contentChangeEventScheduler.schedule();
E
Erich Gamma 已提交
548 549
	}

B
Benjamin Pasero 已提交
550
	private makeDirty(): void {
E
Erich Gamma 已提交
551 552

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

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

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

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

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

			// Only trigger save if the version id has not changed meanwhile
			if (versionId === this.versionId) {
573
				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 已提交
574
			}
575
		}, this.autoSaveAfterMillies);
E
Erich Gamma 已提交
576

577
		this.autoSaveDisposable = toDisposable(() => clearTimeout(handle));
E
Erich Gamma 已提交
578 579
	}

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

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

I
isidor 已提交
592
		this.logService.trace('save() - enter', this.resource);
E
Erich Gamma 已提交
593 594

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

597
		return this.doSave(this.versionId, options);
E
Erich Gamma 已提交
598 599
	}

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

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

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

M
Matt Bierner 已提交
615
			return this.saveSequentializer.pendingSave || Promise.resolve(undefined);
E
Erich Gamma 已提交
616 617
		}

618
		// Return early if not dirty (unless forced) or version changed meanwhile
B
Benjamin Pasero 已提交
619 620 621 622 623 624
		//
		// 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.
		//
625
		if ((!options.force && !this.dirty) || versionId !== this.versionId) {
I
isidor 已提交
626
			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 已提交
627

R
Rob Lourens 已提交
628
			return Promise.resolve(undefined);
E
Erich Gamma 已提交
629 630
		}

631
		// Return if currently saving by storing this save request as the next save that should happen.
632
		// Never ever must 2 saves execute at the same time because this can lead to dirty writes and race conditions.
B
Benjamin Pasero 已提交
633
		//
634
		// Scenario A: auto save was triggered and is currently busy saving to disk. this takes long enough that another auto save
635
		//             kicks in.
636 637
		// 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 已提交
638
		//
639
		if (this.saveSequentializer.hasPendingSave()) {
I
isidor 已提交
640
			this.logService.trace(`doSave(${versionId}) - exit - because busy saving`, this.resource);
E
Erich Gamma 已提交
641

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

		// 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 已提交
648 649
		if (!this.autoSaveAfterMilliesEnabled && this.textEditorModel) {
			this.textEditorModel.pushStackElement();
E
Erich Gamma 已提交
650 651
		}

B
Benjamin Pasero 已提交
652
		// A save participant can still change the model now and since we are so close to saving
E
Erich Gamma 已提交
653 654
		// 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
655
		// Save participants can also be skipped through API.
J
Johannes Rieken 已提交
656
		let saveParticipantPromise: Promise<number> = Promise.resolve(versionId);
657
		if (TextFileEditorModel.saveParticipant && !options.skipSaveParticipants) {
B
💄  
Benjamin Pasero 已提交
658
			const onCompleteOrError = () => {
J
Johannes Rieken 已提交
659
				this.blockModelContentChange = false;
B
💄  
Benjamin Pasero 已提交
660

661
				return this.versionId;
B
💄  
Benjamin Pasero 已提交
662 663
			};

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

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

671 672 673 674 675
			// 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).
676
			// To fix this issue, we will not store the contents to disk when we got disposed.
677
			if (this.disposed) {
R
Rob Lourens 已提交
678
				return undefined;
679 680
			}

B
Benjamin Pasero 已提交
681 682
			// 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.
683 684 685 686 687 688
			// 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 已提交
689
				return this.doTouch(newVersionId);
690 691
			}

692
			// update versionId with its new value (if pre-save changes happened)
J
Johannes Rieken 已提交
693 694 695 696 697 698 699 700 701
			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
702
			// mark the save operation as currently pending with the versionId (it might have changed from a save participant triggering)
I
isidor 已提交
703
			this.logService.trace(`doSave(${versionId}) - before updateContent()`, this.resource);
M
Matt Bierner 已提交
704 705 706 707 708
			const snapshot = this.createSnapshot();
			if (!snapshot) {
				throw new Error('Invalid snapshot');
			}
			return this.saveSequentializer.setPending(newVersionId, this.fileService.updateContent(this.lastResolvedDiskStat.resource, snapshot, {
709 710
				overwriteReadonly: options.overwriteReadonly,
				overwriteEncoding: options.overwriteEncoding,
711
				mtime: this.lastResolvedDiskStat.mtime,
J
Johannes Rieken 已提交
712
				encoding: this.getEncoding(),
713 714
				etag: this.lastResolvedDiskStat.etag,
				writeElevated: options.writeElevated
715
			}).then(stat => {
I
isidor 已提交
716
				this.logService.trace(`doSave(${versionId}) - after updateContent()`, this.resource);
J
Johannes Rieken 已提交
717 718 719

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

726
				// Updated resolved stat with updated stat
727
				this.updateLastResolvedDiskStat(stat);
E
Erich Gamma 已提交
728

729 730 731
				// Cancel any content change event promises as they are no longer valid
				this.contentChangeEventScheduler.cancel();

J
Johannes Rieken 已提交
732 733
				// Emit File Saved Event
				this._onDidStateChange.fire(StateChange.SAVED);
734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760

				// Telemetry
				let telemetryPromise: Thenable<void>;
				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

					telemetryPromise = Promise.resolve();
				} else {
					telemetryPromise = this.getTelemetryData(options.reason).then(data => {
						/* __GDPR__
							"filePUT" : {
								"${include}": [
									"${FileTelemetryData}"
								]
							}
						*/
						this.telemetryService.publicLog('filePUT', data);
					});
				}

				return telemetryPromise;
761
			}, error => {
762 763
				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 已提交
764 765
				}

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

768
				// Flag as error state in the model
J
Johannes Rieken 已提交
769
				this.inErrorMode = true;
E
Erich Gamma 已提交
770

771
				// Look out for a save conflict
772
				if ((<FileOperationError>error).fileOperationResult === FileOperationResult.FILE_MODIFIED_SINCE) {
773 774 775
					this.inConflictMode = true;
				}

J
Johannes Rieken 已提交
776 777
				// Show to user
				this.onSaveError(error);
E
Erich Gamma 已提交
778

J
Johannes Rieken 已提交
779 780
				// Emit as event
				this._onDidStateChange.fire(StateChange.SAVE_ERROR);
781 782
			}));
		}));
E
Erich Gamma 已提交
783 784
	}

785
	private getTypeIfSettings(): string {
B
Benjamin Pasero 已提交
786
		if (extname(this.resource) !== '.json') {
787
			return '';
788
		}
789 790

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

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

		// Check for locale file
801
		if (isEqual(this.resource, URI.file(join(this.environmentService.appSettingsHome, 'locale.json')), !isLinux)) {
802 803 804 805
			return 'locale';
		}

		// Check for snippets
806
		if (isEqualOrParent(this.resource, URI.file(join(this.environmentService.appSettingsHome, 'snippets')))) {
807
			return 'snippets';
808 809 810
		}

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

		return '';
822 823
	}

824 825 826 827 828 829 830 831 832 833 834 835 836
	private getTelemetryData(reason: number | undefined): Thenable<object> {
		return this.hashService.createSHA1(this.resource.fsPath).then(hashedPath => {
			const ext = extname(this.resource);
			const fileName = basename(this.resource);
			const telemetryData = {
				mimeType: guessMimeTypes(this.resource.fsPath).join(', '),
				ext,
				path: hashedPath,
				reason
			};

			if (ext === '.json' && TextFileEditorModel.WHITELIST_JSON.indexOf(fileName) > -1) {
				telemetryData['whitelistedjson'] = fileName;
837
			}
838 839 840 841 842 843 844 845 846 847 848 849

			/* __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" }
				}
			*/
			return telemetryData;
		});
850 851
	}

J
Johannes Rieken 已提交
852
	private doTouch(versionId: number): Promise<void> {
M
Matt Bierner 已提交
853 854 855 856 857
		const snapshot = this.createSnapshot();
		if (!snapshot) {
			throw new Error('invalid snapshot');
		}
		return this.saveSequentializer.setPending(versionId, this.fileService.updateContent(this.lastResolvedDiskStat.resource, snapshot, {
B
Benjamin Pasero 已提交
858 859 860 861
			mtime: this.lastResolvedDiskStat.mtime,
			encoding: this.getEncoding(),
			etag: this.lastResolvedDiskStat.etag
		}).then(stat => {
862 863 864

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

E
Erich Gamma 已提交
868
	private setDirty(dirty: boolean): () => void {
B
Benjamin Pasero 已提交
869
		const wasDirty = this.dirty;
870
		const wasInConflictMode = this.inConflictMode;
B
Benjamin Pasero 已提交
871 872
		const wasInErrorMode = this.inErrorMode;
		const oldBufferSavedVersionId = this.bufferSavedVersionId;
E
Erich Gamma 已提交
873 874 875

		if (!dirty) {
			this.dirty = false;
876
			this.inConflictMode = false;
E
Erich Gamma 已提交
877
			this.inErrorMode = false;
878
			this.updateSavedVersionId();
E
Erich Gamma 已提交
879 880 881 882 883 884 885
		} else {
			this.dirty = true;
		}

		// Return function to revert this call
		return () => {
			this.dirty = wasDirty;
886
			this.inConflictMode = wasInConflictMode;
E
Erich Gamma 已提交
887 888 889 890 891
			this.inErrorMode = wasInErrorMode;
			this.bufferSavedVersionId = oldBufferSavedVersionId;
		};
	}

892 893 894 895 896 897 898 899 900 901 902
	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();
		}
	}

903
	private updateLastResolvedDiskStat(newVersionOnDiskStat: IFileStatWithMetadata): void {
E
Erich Gamma 已提交
904 905

		// First resolve - just take
906 907
		if (!this.lastResolvedDiskStat) {
			this.lastResolvedDiskStat = newVersionOnDiskStat;
E
Erich Gamma 已提交
908 909 910
		}

		// Subsequent resolve - make sure that we only assign it if the mtime is equal or has advanced.
B
Benjamin Pasero 已提交
911 912
		// 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.
913 914
		else if (this.lastResolvedDiskStat.mtime <= newVersionOnDiskStat.mtime) {
			this.lastResolvedDiskStat = newVersionOnDiskStat;
E
Erich Gamma 已提交
915 916 917 918 919 920 921 922 923 924 925 926 927 928
		}
	}

	private onSaveError(error: any): void {

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

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

B
Benjamin Pasero 已提交
929
	isDirty(): boolean {
E
Erich Gamma 已提交
930 931 932
		return this.dirty;
	}

B
Benjamin Pasero 已提交
933
	getLastSaveAttemptTime(): number {
B
Benjamin Pasero 已提交
934
		return this.lastSaveAttemptTime;
E
Erich Gamma 已提交
935 936
	}

937
	getETag(): string | null {
M
Matt Bierner 已提交
938
		return this.lastResolvedDiskStat ? this.lastResolvedDiskStat.etag || null : null;
E
Erich Gamma 已提交
939 940
	}

B
Benjamin Pasero 已提交
941
	hasState(state: ModelState): boolean {
942 943 944 945 946 947 948 949 950 951 952 953 954
		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 已提交
955 956 957
		}
	}

B
Benjamin Pasero 已提交
958
	getEncoding(): string {
E
Erich Gamma 已提交
959 960 961
		return this.preferredEncoding || this.contentEncoding;
	}

B
Benjamin Pasero 已提交
962
	setEncoding(encoding: string, mode: EncodingMode): void {
E
Erich Gamma 已提交
963 964 965 966 967 968 969 970 971 972 973 974 975 976
		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();
			}

977
			if (!this.inConflictMode) {
978
				this.save({ overwriteEncoding: true });
E
Erich Gamma 已提交
979 980 981 982 983 984
			}
		}

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

				return;
			}

			this.updatePreferredEncoding(encoding);

			// Load
993 994
			this.load({
				forceReadFromDisk: true	// because encoding has changed
995
			});
E
Erich Gamma 已提交
996 997 998
		}
	}

B
Benjamin Pasero 已提交
999
	updatePreferredEncoding(encoding: string): void {
E
Erich Gamma 已提交
1000 1001 1002 1003 1004 1005 1006
		if (!this.isNewEncoding(encoding)) {
			return;
		}

		this.preferredEncoding = encoding;

		// Emit
1007
		this._onDidStateChange.fire(StateChange.ENCODING);
E
Erich Gamma 已提交
1008 1009 1010 1011 1012 1013 1014 1015
	}

	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) {
1016
			return false; // also return if we don't have a preferred encoding but the content encoding is already the same
E
Erich Gamma 已提交
1017 1018 1019 1020 1021
		}

		return true;
	}

B
Benjamin Pasero 已提交
1022
	isResolved(): boolean {
1023
		return !isUndefinedOrNull(this.lastResolvedDiskStat);
E
Erich Gamma 已提交
1024 1025
	}

B
Benjamin Pasero 已提交
1026
	isReadonly(): boolean {
1027
		return !!(this.lastResolvedDiskStat && this.lastResolvedDiskStat.isReadonly);
1028 1029
	}

B
Benjamin Pasero 已提交
1030
	isDisposed(): boolean {
E
Erich Gamma 已提交
1031 1032 1033
		return this.disposed;
	}

B
Benjamin Pasero 已提交
1034
	getResource(): URI {
E
Erich Gamma 已提交
1035 1036 1037
		return this.resource;
	}

1038
	getStat(): IFileStatWithMetadata {
B
Benjamin Pasero 已提交
1039 1040 1041
		return this.lastResolvedDiskStat;
	}

B
Benjamin Pasero 已提交
1042
	dispose(): void {
E
Erich Gamma 已提交
1043
		this.disposed = true;
1044 1045
		this.inConflictMode = false;
		this.inOrphanMode = false;
E
Erich Gamma 已提交
1046 1047 1048 1049
		this.inErrorMode = false;

		this.createTextEditorModelPromise = null;

1050
		this.cancelPendingAutoSave();
D
Daniel Imms 已提交
1051

E
Erich Gamma 已提交
1052 1053
		super.dispose();
	}
1054 1055
}

1056 1057
interface IPendingSave {
	versionId: number;
J
Johannes Rieken 已提交
1058
	promise: Promise<void>;
1059 1060
}

1061
interface ISaveOperation {
J
Johannes Rieken 已提交
1062
	promise: Promise<void>;
B
Benjamin Pasero 已提交
1063 1064
	promiseResolve: () => void;
	promiseReject: (error: Error) => void;
J
Johannes Rieken 已提交
1065
	run: () => Promise<void>;
1066 1067 1068
}

export class SaveSequentializer {
1069 1070
	private _pendingSave?: IPendingSave;
	private _nextSave?: ISaveOperation;
1071

B
Benjamin Pasero 已提交
1072
	hasPendingSave(versionId?: number): boolean {
1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083
		if (!this._pendingSave) {
			return false;
		}

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

		return !!this._pendingSave;
	}

1084
	get pendingSave(): Promise<void> | undefined {
R
Rob Lourens 已提交
1085
		return this._pendingSave ? this._pendingSave.promise : undefined;
1086 1087
	}

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

1091
		promise.then(() => this.donePending(versionId), () => this.donePending(versionId));
1092 1093 1094 1095 1096 1097

		return promise;
	}

	private donePending(versionId: number): void {
		if (this._pendingSave && versionId === this._pendingSave.versionId) {
1098 1099

			// only set pending to done if the promise finished that is associated with that versionId
R
Rob Lourens 已提交
1100
			this._pendingSave = undefined;
1101 1102 1103

			// schedule the next save now that we are free if we have any
			this.triggerNextSave();
1104 1105 1106
		}
	}

1107 1108 1109
	private triggerNextSave(): void {
		if (this._nextSave) {
			const saveOperation = this._nextSave;
R
Rob Lourens 已提交
1110
			this._nextSave = undefined;
1111 1112

			// Run next save and complete on the associated promise
B
Benjamin Pasero 已提交
1113
			saveOperation.run().then(saveOperation.promiseResolve, saveOperation.promiseReject);
1114 1115 1116
		}
	}

J
Johannes Rieken 已提交
1117
	setNext(run: () => Promise<void>): Promise<void> {
1118 1119 1120 1121 1122

		// 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 已提交
1123 1124
			let promiseResolve: () => void;
			let promiseReject: (error: Error) => void;
B
Benjamin Pasero 已提交
1125 1126 1127
			const promise = new Promise<void>((resolve, reject) => {
				promiseResolve = resolve;
				promiseReject = reject;
1128 1129 1130 1131 1132
			});

			this._nextSave = {
				run,
				promise,
M
Matt Bierner 已提交
1133 1134
				promiseResolve: promiseResolve!,
				promiseReject: promiseReject!
1135 1136 1137 1138 1139 1140 1141 1142
			};
		}

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

M
Matt Bierner 已提交
1143
		return this._nextSave.promise;
1144 1145 1146
	}
}

1147 1148
class DefaultSaveErrorHandler implements ISaveErrorHandler {

1149
	constructor(@INotificationService private readonly notificationService: INotificationService) { }
1150

B
Benjamin Pasero 已提交
1151
	onSaveError(error: any, model: TextFileEditorModel): void {
B
Benjamin Pasero 已提交
1152
		this.notificationService.error(nls.localize('genericSaveError', "Failed to save '{0}': {1}", basename(model.getResource()), toErrorMessage(error, false)));
1153 1154
	}
}