Task contribution (#742)

* work in progress * set up necessary code. need to work on getting it working * formatting * work in progress * work in progress * formatting * work in progress * work in progress * work in progress * formatting * needs a lot of work regarding how we do actions vs how extensions do actions * formatting * use connection profile for actions * change action to be
2026-01-14 01:25:37 -05:00 · 2018-02-27 11:40:13 -08:00
parent 5adab4fafb
commit 3432dac261
23 changed files with 518 additions and 323 deletions
--- a/src/sql/workbench/api/node/extHostTasks.ts
+++ b/src/sql/workbench/api/node/extHostTasks.ts
@@ -0,0 +1,91 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+'use strict';
+
+import { validateConstraint } from 'vs/base/common/types';
+import { ILogService } from 'vs/platform/log/common/log';
+import { IMainContext } from 'vs/workbench/api/node/extHost.protocol';
+import * as extHostTypes from 'vs/workbench/api/node/extHostTypes';
+import { TPromise } from 'vs/base/common/winjs.base';
+
+import * as sqlops from 'sqlops';
+
+import { ITaskHandlerDescription } from 'sql/platform/tasks/common/tasks';
+import { SqlMainContext, MainThreadTasksShape, ExtHostTasksShape } from 'sql/workbench/api/node/sqlExtHost.protocol';
+
+interface TaskHandler {
+	callback: Function;
+	thisArg: any;
+	description: ITaskHandlerDescription;
+}
+
+export class ExtHostTasks implements ExtHostTasksShape {
+	private _proxy: MainThreadTasksShape;
+	private _tasks = new Map<string, TaskHandler>();
+
+	constructor(
+		mainContext: IMainContext,
+		private logService: ILogService
+	) {
+		this._proxy = mainContext.get(SqlMainContext.MainThreadTasks);
+	}
+
+	registerTask(id: string, callback: sqlops.tasks.ITaskHandler, thisArg?: any, description?: ITaskHandlerDescription): extHostTypes.Disposable {
+		this.logService.trace('ExtHostTasks#registerTask', id);
+
+		if (!id.trim().length) {
+			throw new Error('invalid id');
+		}
+
+		if (this._tasks.has(id)) {
+			throw new Error(`task '${id}' already exists`);
+		}
+
+		this._tasks.set(id, { callback, thisArg, description });
+		this._proxy.$registerTask(id);
+
+		return new extHostTypes.Disposable(() => {
+			if (this._tasks.delete(id)) {
+				this._proxy.$unregisterTask(id);
+			}
+		});
+	}
+
+	$executeContributedTask<T>(id: string, ...args: any[]): Thenable<T> {
+		let command = this._tasks.get(id);
+		if (!command) {
+			return TPromise.wrapError<T>(new Error(`Contributed task '${id}' does not exist.`));
+		}
+
+		let { callback, thisArg, description } = command;
+
+		if (description) {
+			for (let i = 0; i < description.args.length; i++) {
+				try {
+					validateConstraint(args[i], description.args[i].constraint);
+				} catch (err) {
+					return TPromise.wrapError<T>(new Error(`Running the contributed task:'${id}' failed. Illegal argument '${description.args[i].name}' - ${description.args[i].description}`));
+				}
+			}
+		}
+
+		try {
+			let result = callback.apply(thisArg, args);
+			return TPromise.as(result);
+		} catch (err) {
+			// console.log(err);
+			// try {
+			// 	console.log(toErrorMessage(err));
+			// } catch (err) {
+			// 	//
+			// }
+			return TPromise.wrapError<T>(new Error(`Running the contributed task:'${id}' failed.`));
+		}
+	}
+
+	$getContributedTaskHandlerDescriptions(): TPromise<{ [id: string]: any; }> {
+		throw new Error('Method not implemented.');
+	}
+}