azuredatastudio/extensions/data-workspace/src/common/workspaceTreeDataProvider.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the Source EULA. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import * as vscode from 'vscode';
import { IWorkspaceService } from './interfaces';
import { UnknownProjectsErrorMessage } from './constants';
import { WorkspaceTreeItem } from 'dataworkspace';

/**
 * Tree data provider for the workspace main view
 */
export class WorkspaceTreeDataProvider implements vscode.TreeDataProvider<WorkspaceTreeItem>{
	constructor(private _workspaceService: IWorkspaceService) {
		this._workspaceService.onDidWorkspaceProjectsChange(() => {
			this.refresh();
		});
	}

	private _onDidChangeTreeData: vscode.EventEmitter<void | WorkspaceTreeItem | null | undefined> | undefined = new vscode.EventEmitter<WorkspaceTreeItem | undefined | void>();
	readonly onDidChangeTreeData?: vscode.Event<void | WorkspaceTreeItem | null | undefined> | undefined = this._onDidChangeTreeData?.event;

	refresh(): void {
		this._onDidChangeTreeData?.fire();
	}

	getTreeItem(element: WorkspaceTreeItem): vscode.TreeItem | Thenable<vscode.TreeItem> {
		return element.treeDataProvider.getTreeItem(element.element);
	}

	async getChildren(element?: WorkspaceTreeItem | undefined): Promise<WorkspaceTreeItem[]> {
		if (element) {
			const items = await element.treeDataProvider.getChildren(element.element);
			return items ? items.map(item => <WorkspaceTreeItem>{ treeDataProvider: element.treeDataProvider, element: item }) : [];
		}
		else {
			// if the element is undefined return the project tree items
			const projects = await this._workspaceService.getProjectsInWorkspace();
			const unknownProjects: string[] = [];
			const treeItems: WorkspaceTreeItem[] = [];
			for (const project of projects) {
				const projectProvider = await this._workspaceService.getProjectProvider(project);
				if (projectProvider === undefined) {
					unknownProjects.push(project.path);
					continue;
				}
				const treeDataProvider = await projectProvider.getProjectTreeDataProvider(project);
				if (treeDataProvider.onDidChangeTreeData) {
					treeDataProvider.onDidChangeTreeData((e: any) => {
						this._onDidChangeTreeData?.fire(e);
					});
				}
				const children = await treeDataProvider.getChildren(element);
				children?.forEach(child => {
					treeItems.push({
						treeDataProvider: treeDataProvider,
						element: child
					});
				});
			}
			if (unknownProjects.length > 0) {
				vscode.window.showErrorMessage(UnknownProjectsErrorMessage(unknownProjects));
			}
			return treeItems;
		}
	}
}