getPreLaunchTaskChain.ts

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

import * as vscode from 'vscode';
import { getTasks, ITask } from '../vsCodeConfig/tasks';

/**
 * Resolves the full chain of tasks associated with a given `preLaunchTask`.
 * Recursively follows the `dependsOn` references found in the `tasks.json`.
 */
export function getPreLaunchTaskChain(workspace: vscode.WorkspaceFolder, preLaunchTask: string): string[] {
    const allTasks: ITask[] = getTasks(workspace);
    const allTasksMap = new Map<string, ITask>();

    for (const task of allTasks) {
        if (task.label) {
            allTasksMap.set(task.label, task);
        }
    }

    const dependentTasks = new Set<string>();

    function getDependentTasks(name: string): void {
        if (dependentTasks.has(name)) {
            return;
        }
        dependentTasks.add(name);

        const task = allTasksMap.get(name);
        if (!task) {
            return;
        }

        const dependsOn: unknown = task?.dependsOn;
        if (typeof dependsOn === 'string') {
            getDependentTasks(dependsOn);
        } else if (Array.isArray(dependsOn)) {
            for (const dep of dependsOn) {
                if (typeof dep === 'string') {
                    getDependentTasks(dep);
                }
            }
        }
    }

    getDependentTasks(preLaunchTask);
    return Array.from(dependentTasks.values());
}