add BulkEditPreviewHandler

jrieken · jrieken · commit 94a469319585 · 2019-12-18T11:25:45.000+01:00
diff --git a/src/vs/editor/browser/services/bulkEditService.ts b/src/vs/editor/browser/services/bulkEditService.ts
@@ -7,10 +7,10 @@ import { ICodeEditor } from 'vs/editor/browser/editorBrowser';
 import { WorkspaceEdit } from 'vs/editor/common/modes';
 import { createDecorator } from 'vs/platform/instantiation/common/instantiation';
 import { IProgress, IProgressStep } from 'vs/platform/progress/common/progress';
+import { IDisposable } from 'vs/base/common/lifecycle';
 
 export const IBulkEditService = createDecorator<IBulkEditService>('IWorkspaceEditService');
 
-
 export interface IBulkEditOptions {
 	editor?: ICodeEditor;
 	progress?: IProgress<IProgressStep>;
@@ -20,9 +20,13 @@ export interface IBulkEditResult {
 	ariaSummary: string;
 }
 
+export type IBulkEditPreviewHandler = (edit: WorkspaceEdit, options?: IBulkEditOptions) => Promise<WorkspaceEdit>;
+
 export interface IBulkEditService {
 	_serviceBrand: undefined;
 
+	setPreviewHandler(handler: IBulkEditPreviewHandler): IDisposable;
+
 	apply(edit: WorkspaceEdit, options?: IBulkEditOptions): Promise<IBulkEditResult>;
 }
 
diff --git a/src/vs/editor/standalone/browser/simpleServices.ts b/src/vs/editor/standalone/browser/simpleServices.ts
@@ -631,6 +631,10 @@ export class SimpleBulkEditService implements IBulkEditService {
 		//
 	}
 
+	setPreviewHandler(): IDisposable {
+		return Disposable.None;
+	}
+
 	apply(workspaceEdit: WorkspaceEdit, options?: IBulkEditOptions): Promise<IBulkEditResult> {
 
 		let edits = new Map<ITextModel, TextEdit[]>();
diff --git a/src/vs/workbench/services/bulkEdit/browser/bulkEditService.ts b/src/vs/workbench/services/bulkEdit/browser/bulkEditService.ts
@@ -4,10 +4,10 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { mergeSort } from 'vs/base/common/arrays';
-import { dispose, IDisposable, IReference } from 'vs/base/common/lifecycle';
+import { dispose, IDisposable, IReference, toDisposable } from 'vs/base/common/lifecycle';
 import { URI } from 'vs/base/common/uri';
 import { ICodeEditor, isCodeEditor } from 'vs/editor/browser/editorBrowser';
-import { IBulkEditOptions, IBulkEditResult, IBulkEditService } from 'vs/editor/browser/services/bulkEditService';
+import { IBulkEditOptions, IBulkEditResult, IBulkEditService, IBulkEditPreviewHandler } from 'vs/editor/browser/services/bulkEditService';
 import { EditOperation } from 'vs/editor/common/core/editOperation';
 import { Range } from 'vs/editor/common/core/range';
 import { EndOfLineSequence, IIdentifiedSingleEditOperation, ITextModel } from 'vs/editor/common/model';
@@ -381,6 +381,8 @@ export class BulkEditService implements IBulkEditService {
 
 	_serviceBrand: undefined;
 
+	private _previewHandler?: IBulkEditPreviewHandler;
+
 	constructor(
 		@ILogService private readonly _logService: ILogService,
 		@IModelService private readonly _modelService: IModelService,
@@ -393,7 +395,20 @@ export class BulkEditService implements IBulkEditService {
 		@IConfigurationService private readonly _configurationService: IConfigurationService
 	) { }
 
-	apply(edit: WorkspaceEdit, options: IBulkEditOptions = {}): Promise<IBulkEditResult> {
+	setPreviewHandler(handler: IBulkEditPreviewHandler): IDisposable {
+		this._previewHandler = handler;
+		return toDisposable(() => {
+			if (this._previewHandler === handler) {
+				this._previewHandler = undefined;
+			}
+		});
+	}
+
+	async apply(edit: WorkspaceEdit, options: IBulkEditOptions = {}): Promise<IBulkEditResult> {
+
+		if (this._previewHandler) {
+			edit = await this._previewHandler(edit, options);
+		}
 
 		let { edits } = edit;
 		let codeEditor = options.editor;

Original file line number	Diff line number	Diff line change
`@@ -631,6 +631,10 @@ export class SimpleBulkEditService implements IBulkEditService {`
`631`	`631`	`//`
`632`	`632`	`}`
`633`	`633`
	`634`	`+ setPreviewHandler(): IDisposable {`
	`635`	`+ return Disposable.None;`
	`636`	`+ }`
	`637`	`+`
`634`	`638`	`apply(workspaceEdit: WorkspaceEdit, options?: IBulkEditOptions): Promise<IBulkEditResult> {`
`635`	`639`
`636`	`640`	`let edits = new Map<ITextModel, TextEdit[]>();`