azuredatastudio/src/vs/workbench/services/textfile/browser/browserTextFileService.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the Source EULA. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import { AbstractTextFileService, EncodingOracle } from 'vs/workbench/services/textfile/browser/textFileService';
import { ITextFileService, IResourceEncoding, TextFileEditorModelState } from 'vs/workbench/services/textfile/common/textfiles';
import { registerSingleton } from 'vs/platform/instantiation/common/extensions';
import { ShutdownReason } from 'vs/platform/lifecycle/common/lifecycle';

export class BrowserTextFileService extends AbstractTextFileService {

	private _browserEncoding: EncodingOracle | undefined;

	get encoding(): EncodingOracle {
		if (!this._browserEncoding) {
			this._browserEncoding = this._register(this.instantiationService.createInstance(BrowserEncodingOracle));
		}

		return this._browserEncoding;
	}

	protected registerListeners(): void {
		super.registerListeners();

		// Lifecycle
		this.lifecycleService.onBeforeShutdown(event => event.veto(this.onBeforeShutdown(event.reason)));
	}

	protected onBeforeShutdown(reason: ShutdownReason): boolean {
		if (this.files.models.some(model => model.hasState(TextFileEditorModelState.PENDING_SAVE))) {
			console.warn('Unload prevented: pending file saves');

			return true; // files are pending to be saved: veto
		}

		return false;
	}
}

class BrowserEncodingOracle extends EncodingOracle {
	async getPreferredWriteEncoding(): Promise<IResourceEncoding> {
		return { encoding: 'utf8', hasBOM: false };
	}

	async getWriteEncoding(): Promise<{ encoding: string, addBOM: boolean }> {
		return { encoding: 'utf8', addBOM: false };
	}

	async getReadEncoding(): Promise<string> {
		return 'utf8';
	}
}

registerSingleton(ITextFileService, BrowserTextFileService);