Close dirty editors: Provide fast action to save all or revert all (fixes microsoft#8009)

Benjamin Pasero · Benjamin Pasero · commit 7e6dbe9017e1 · 2016-12-13T15:26:41.000+01:00
diff --git a/src/vs/workbench/browser/parts/editor/editorActions.ts b/src/vs/workbench/browser/parts/editor/editorActions.ts
@@ -9,7 +9,7 @@ import nls = require('vs/nls');
 import { Action } from 'vs/base/common/actions';
 import { mixin } from 'vs/base/common/objects';
 import { getCodeEditor } from 'vs/editor/common/services/codeEditorService';
-import { EditorInput, getUntitledOrFileResource, TextEditorOptions, EditorOptions, IEditorIdentifier, IEditorContext, ActiveEditorMoveArguments, ActiveEditorMovePositioning, EditorCommands } from 'vs/workbench/common/editor';
+import { EditorInput, getUntitledOrFileResource, TextEditorOptions, EditorOptions, IEditorIdentifier, IEditorContext, ActiveEditorMoveArguments, ActiveEditorMovePositioning, EditorCommands, ConfirmResult } from 'vs/workbench/common/editor';
 import { QuickOpenEntryGroup } from 'vs/base/parts/quickopen/browser/quickOpenModel';
 import { EditorQuickOpenEntry, EditorQuickOpenEntryGroup, IEditorQuickOpenEntry, QuickOpenAction } from 'vs/workbench/browser/quickopen';
 import { IWorkbenchEditorService } from 'vs/workbench/services/editor/common/editorService';
@@ -21,6 +21,7 @@ import { IHistoryService } from 'vs/workbench/services/history/common/history';
 import { IKeybindingService } from 'vs/platform/keybinding/common/keybinding';
 import { IEditorGroupService, GroupArrangement } from 'vs/workbench/services/group/common/groupService';
 import { ICommandService } from 'vs/platform/commands/common/commands';
+import { ITextFileService } from 'vs/workbench/services/textfile/common/textfiles';
 
 export class SplitEditorAction extends Action {
 
@@ -565,12 +566,40 @@ export class CloseAllEditorsAction extends Action {
 	public static ID = 'workbench.action.closeAllEditors';
 	public static LABEL = nls.localize('closeAllEditors', "Close All Editors");
 
-	constructor(id: string, label: string, @IWorkbenchEditorService private editorService: IWorkbenchEditorService) {
+	constructor(
+		id: string,
+		label: string,
+		@ITextFileService private textFileService: ITextFileService,
+		@IWorkbenchEditorService private editorService: IWorkbenchEditorService
+	) {
 		super(id, label, 'action-close-all-files');
 	}
 
 	public run(): TPromise<any> {
-		return this.editorService.closeAllEditors();
+
+		// Just close all if there are no or one dirty editor
+		if (this.textFileService.getDirty().length < 2) {
+			return this.editorService.closeAllEditors();
+		}
+
+		// Otherwise ask for combined confirmation
+		const confirm = this.textFileService.confirmSave();
+		if (confirm === ConfirmResult.CANCEL) {
+			return;
+		}
+
+		let saveOrRevertPromise: TPromise<boolean>;
+		if (confirm === ConfirmResult.DONT_SAVE) {
+			saveOrRevertPromise = this.textFileService.revertAll().then(() => true);
+		} else {
+			saveOrRevertPromise = this.textFileService.saveAll(true).then(res => res.results.every(r => r.success));
+		}
+
+		return saveOrRevertPromise.then(success => {
+			if (success) {
+				return this.editorService.closeAllEditors();
+			}
+		});
 	}
 }
 
