fix task icon in dashboard (#1174)

src/sql/platform/tasks/common/tasks.ts

@@ -2,6 +2,7 @@
  *  Copyright (c) Microsoft Corporation. All rights reserved.
  *  Licensed under the Source EULA. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
+import { IConnectionProfile } from 'sql/parts/connection/common/interfaces';
 
 import * as types from 'vs/base/common/types';
 import { TPromise } from 'vs/base/common/winjs.base';
@@ -14,9 +15,10 @@ import { ILocalizedString, IMenuItem, MenuRegistry, ICommandAction } from 'vs/pl
 import Event, { Emitter } from 'vs/base/common/event';
 import { IDisposable } from 'vs/base/common/lifecycle';
 import { ContextKeyExpr } from 'vs/platform/contextkey/common/contextkey';
-import { IConnectionProfile } from 'sql/parts/connection/common/interfaces';
 import { LinkedList } from 'vs/base/common/linkedList';
-
+import { IdGenerator } from 'vs/base/common/idGenerator';
+import { createCSSRule } from 'vs/base/browser/dom';
+import URI from 'vs/base/common/uri';
 import { CommandsRegistry } from 'vs/platform/commands/common/commands';
 
 export interface ITaskOptions {
@@ -24,6 +26,7 @@ export interface ITaskOptions {
 	title: string;
 	iconPath: { dark: string; light: string; };
 	description?: ITaskHandlerDescription;
+	iconClass?: string;
 }
 
 export abstract class Task {
@@ -31,12 +34,14 @@ export abstract class Task {
 	public readonly title: string;
 	public readonly iconPathDark: string;
 	public readonly iconPath: { dark: string; light: string; };
+	private readonly _iconClass: string;
 	private readonly _description: ITaskHandlerDescription;
 
 	constructor(opts: ITaskOptions) {
 		this.id = opts.id;
 		this.title = opts.title;
 		this.iconPath = opts.iconPath;
+		this._iconClass = opts.iconClass;
 		this._description = opts.description;
 	}
 
@@ -44,7 +49,8 @@ export abstract class Task {
 		return {
 			id: this.id,
 			handler: (accessor, profile, args) => this.runTask(accessor, profile, args),
-			description: this._description
+			description: this._description,
+			iconClass: this._iconClass
 		};
 	}
 
@@ -91,20 +97,25 @@ export interface ITask {
 	handler: ITaskHandler;
 	precondition?: ContextKeyExpr;
 	description?: ITaskHandlerDescription;
+	iconClass?: string;
 }
 
 export interface ITaskRegistry {
 	registerTask(id: string, command: ITaskHandler): IDisposable;
 	registerTask(command: ITask): IDisposable;
 	getTasks(): string[];
+	getOrCreateTaskIconClassName(item: ICommandAction): string;
 	onTaskRegistered: Event<string>;
 }
 
+const ids = new IdGenerator('task-icon-');
+
 export const TaskRegistry: ITaskRegistry = new class implements ITaskRegistry {
 
 	private _tasks = new Array<string>();
 	private _onTaskRegistered = new Emitter<string>();
 	public readonly onTaskRegistered: Event<string> = this._onTaskRegistered.event;
+	private taskIdToIconClassNameMap: Map<string /* task id */, string /* CSS rule */> = new Map<string, string>();
 
 	registerTask(idOrTask: string | ITask, handler?: ITaskHandler): IDisposable {
 		let disposable: IDisposable;
@@ -113,6 +124,9 @@ export const TaskRegistry: ITaskRegistry = new class implements ITaskRegistry {
 			disposable = CommandsRegistry.registerCommand(idOrTask, handler);
 			id = idOrTask;
 		} else {
+			if (idOrTask.iconClass) {
+				this.taskIdToIconClassNameMap.set(idOrTask.id, idOrTask.iconClass);
+			}
 			disposable = CommandsRegistry.registerCommand(idOrTask);
 			id = idOrTask.id;
 		}
@@ -131,6 +145,19 @@ export const TaskRegistry: ITaskRegistry = new class implements ITaskRegistry {
 		};
 	}
 
+	getOrCreateTaskIconClassName(item: ICommandAction): string {
+		let iconClass = null;
+		if (this.taskIdToIconClassNameMap.has(item.id)) {
+			iconClass = this.taskIdToIconClassNameMap.get(item.id);
+		} else if (item.iconPath) {
+			iconClass = ids.nextId();
+			createCSSRule(`.icon.${iconClass}`, `background-image: url("${URI.file(item.iconPath.light || item.iconPath.dark).toString()}")`);
+			createCSSRule(`.vs-dark .icon.${iconClass}, .hc-black .icon.${iconClass}`, `background-image: url("${URI.file(item.iconPath.dark).toString()}")`);
+			this.taskIdToIconClassNameMap.set(item.id, iconClass);
+		}
+		return iconClass;
+	}
+
 	getTasks(): string[] {
 		return this._tasks.slice(0);
 	}