Skip to content

Commit

Permalink
notify clients of TaskDefinitionRegistry on change
Browse files Browse the repository at this point in the history 
- In current theia, task definitions from plugins are registered after the initialization of the task extension, causing task-loading, which counts on the existence of task definitions, not to function correctly. With changes in this pull request, events are emitted so that the clients of TaskDefinitionRegistry are notified and therefore have the flexibility to re-organize the loaded tasks.

Signed-off-by: Liang Huang <liang.huang@ericsson.com>
  • Loading branch information
Liang Huang committed Aug 18, 2019
1 parent 0eebcf3 commit a520f99
Show file tree
Hide file tree
Showing 2 changed files with 85 additions and 19 deletions.
97 changes: 78 additions & 19 deletions packages/task/src/browser/task-configurations.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
* SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
********************************************************************************/

import { inject, injectable } from 'inversify';
import { inject, injectable, postConstruct } from 'inversify';
import { ContributedTaskConfiguration, TaskConfiguration, TaskCustomization, TaskDefinition } from '../common';
import { TaskDefinitionRegistry } from './task-definition-registry';
import { ProvidedTaskConfigurations } from './provided-task-configurations';
Expand Down Expand Up @@ -84,30 +84,41 @@ export class TaskConfigurations implements Disposable {
@inject(FileSystem) protected readonly fileSystem: FileSystem
) {
this.toDispose.push(watcherServer);
this.watcherServer.onFilesChanged(async changes => {
try {
const watchedConfigFileChanges = changes.filter(change =>
this.watchedConfigFileUris.some(fileUri => FileChangeEvent.isAffected([change], new URI(fileUri)))
).map(relatedChange => (
{ uri: relatedChange.uri.toString(), type: relatedChange.type }
));
if (watchedConfigFileChanges.length >= 0) {
await this.onDidTaskFileChange(watchedConfigFileChanges);
if (this.client) {
this.client.taskConfigurationChanged(this.getTaskLabels());
this.toDispose.push(
this.watcherServer.onFilesChanged(async changes => {
try {
const watchedConfigFileChanges = changes.filter(change =>
this.watchedConfigFileUris.some(fileUri => FileChangeEvent.isAffected([change], new URI(fileUri)))
).map(relatedChange => (
{ uri: relatedChange.uri.toString(), type: relatedChange.type }
));
if (watchedConfigFileChanges.length >= 0) {
await this.onDidTaskFileChange(watchedConfigFileChanges);
if (this.client) {
this.client.taskConfigurationChanged(this.getTaskLabels());
}
}
} catch (err) {
console.error(err);
}
} catch (err) {
console.error(err);
}
});

})
);
this.toDispose.push(Disposable.create(() => {
this.tasksMap.clear();
this.taskCustomizationMap.clear();
this.watchersMap.clear();
this.client = undefined;
}));
}

@postConstruct()
protected init(): void {
this.reorgnizeTasks();
this.toDispose.push(
this.taskDefinitionRegistry.onDidRegisterTaskDefinition(() => this.reorgnizeTasks())
);
}

setClient(client: TaskConfigurationClient): void {
this.client = client;
}
Expand Down Expand Up @@ -198,6 +209,7 @@ export class TaskConfigurations implements Disposable {
removeTasks(configFileUri: string): void {
const source = this.getSourceFolderFromConfigUri(configFileUri);
this.tasksMap.delete(source);
this.taskCustomizationMap.delete(source);
}

/**
Expand Down Expand Up @@ -424,18 +436,65 @@ export class TaskConfigurations implements Disposable {
return filteredTasks;
}

/**
* This function is called after a change in TaskDefinitionRegistry happens.
* It checks all tasks that have been loaded, and re-organized them in `tasksMap` and `taskCustomizationMap`.
*/
protected reorgnizeTasks(): void {
const newTaskMap = new Map<string, Map<string, TaskConfiguration>>();
const newTaskCustomizationMap = new Map<string, TaskCustomization[]>();
const addCustomization = (rootFolder: string, customization: TaskCustomization) => {
if (newTaskCustomizationMap.has(rootFolder)) {
newTaskCustomizationMap.get(rootFolder)!.push(customization);
} else {
newTaskCustomizationMap.set(rootFolder, [customization]);
}
};
const addConfiguredTask = (rootFolder: string, label: string, configuredTask: TaskCustomization) => {
if (newTaskMap.has(rootFolder)) {
newTaskMap.get(rootFolder)!.set(label, configuredTask as TaskConfiguration);
} else {
const newConfigMap = new Map();
newConfigMap.set(label, configuredTask);
newTaskMap.set(rootFolder, newConfigMap);
}
};
for (const [rootFolder, configuredTaskMap] of this.tasksMap.entries()) {
for (const [label, configuredTask] of configuredTaskMap.entries()) {
if (this.isDetectedTask(configuredTask)) {
addCustomization(rootFolder, configuredTask);
} else {
addConfiguredTask(rootFolder, label, configuredTask);
}
}
}

for (const [rootFolder, taskCustomizations] of this.taskCustomizationMap.entries()) {
for (const customization of taskCustomizations) {
if (this.isDetectedTask(customization)) {
addCustomization(rootFolder, customization);
} else {
addConfiguredTask(rootFolder, customization['label'] as string, customization);
}
}
}

this.taskCustomizationMap = newTaskCustomizationMap;
this.tasksMap = newTaskMap;
}

private getSourceFolderFromConfigUri(configFileUri: string): string {
return new URI(configFileUri).parent.parent.path.toString();
}

/** checks if the config is a detected / contributed task */
private isDetectedTask(task: TaskConfiguration): task is ContributedTaskConfiguration {
private isDetectedTask(task: TaskConfiguration | TaskCustomization): task is ContributedTaskConfiguration {
const taskDefinition = this.getTaskDefinition(task);
// it is considered as a customization if the task definition registry finds a def for the task configuration
return !!taskDefinition;
}

private getTaskDefinition(task: TaskConfiguration): TaskDefinition | undefined {
private getTaskDefinition(task: TaskConfiguration | TaskCustomization): TaskDefinition | undefined {
return this.taskDefinitionRegistry.getDefinition({
...task,
type: task.taskType || task.type
Expand Down
7 changes: 7 additions & 0 deletions packages/task/src/browser/task-definition-registry.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
********************************************************************************/

import { injectable } from 'inversify';
import { Event, Emitter } from '@theia/core/lib/common';
import { TaskConfiguration, TaskCustomization, TaskDefinition } from '../common';

@injectable()
Expand All @@ -23,6 +24,11 @@ export class TaskDefinitionRegistry {
// task type - array of task definitions
private definitions: Map<string, TaskDefinition[]> = new Map();

protected readonly onDidRegisterTaskDefinitionEmitter = new Emitter<void>();
get onDidRegisterTaskDefinition(): Event<void> {
return this.onDidRegisterTaskDefinitionEmitter.event;
}

/**
* Finds the task definition(s) from the registry with the given `taskType`.
*
Expand Down Expand Up @@ -70,5 +76,6 @@ export class TaskDefinitionRegistry {
register(definition: TaskDefinition): void {
const taskType = definition.taskType;
this.definitions.set(taskType, [...this.getDefinitions(taskType), definition]);
this.onDidRegisterTaskDefinitionEmitter.fire(undefined);
}
}

0 comments on commit a520f99

Please sign in to comment.