Add root tasks getter

Author: schaable

v-next/core/src/internal/tasks/task-manager.ts

@@ -48,6 +48,10 @@ export class TaskManagerImplementation implements TaskManager {
     }
   }
 
+  public get rootTasks(): Map<string, Task> {
+    return this.#rootTasks;
+  }
+
   public getTask(taskId: string | string[]): Task {
     taskId = Array.isArray(taskId) ? taskId : [taskId];
     if (taskId.length === 0) {

v-next/core/src/types/tasks.ts

@@ -379,6 +379,10 @@ export interface Task {
  * can be used to retrieve tasks and run them.
  */
 export interface TaskManager {
+  /**
+   * Returns the root tasks of the task manager.
+   */
+  get rootTasks(): Map<string, Task>;
   /**
    * Returns a task by its id, throwing if it doesn't exist.
    */

v-next/core/test/internal/tasks/task-manager.ts

@@ -25,7 +25,9 @@ import { TaskDefinitionType } from "../../../src/types/tasks.js";
  */
 describe("TaskManagerImplementation", () => {
   it("should initialize the task manager with an empty set of tasks if no plugins or tasks are provided", async () => {
-    await assert.doesNotReject(createHardhatRuntimeEnvironment({}));
+    const hre = await createHardhatRuntimeEnvironment({});
+
+    assert.equal(hre.tasks.rootTasks.size, 0);
   });
 
   it("should initialize the task manager with the tasks from the plugins", async () => {
@@ -79,6 +81,12 @@ describe("TaskManagerImplementation", () => {
     const task3 = hre.tasks.getTask("task3");
     assert.deepEqual(task3.id, ["task3"]);
     assert.equal(task3.pluginId, "plugin2");
+
+    // task1, task2 and task3 should be root tasks
+    assert.equal(hre.tasks.rootTasks.size, 3);
+    assert.deepEqual(hre.tasks.rootTasks.get("task1")?.id, ["task1"]);
+    assert.deepEqual(hre.tasks.rootTasks.get("task2")?.id, ["task2"]);
+    assert.deepEqual(hre.tasks.rootTasks.get("task3")?.id, ["task3"]);
   });
 
   it("should initialize the task manager with the tasks from the config", async () => {
@@ -114,6 +122,12 @@ describe("TaskManagerImplementation", () => {
     const task3 = hre.tasks.getTask("task3");
     assert.deepEqual(task3.id, ["task3"]);
     assert.equal(task3.pluginId, undefined);
+
+    // task1, task2 and task3 should be root tasks
+    assert.equal(hre.tasks.rootTasks.size, 3);
+    assert.deepEqual(hre.tasks.rootTasks.get("task1")?.id, ["task1"]);
+    assert.deepEqual(hre.tasks.rootTasks.get("task2")?.id, ["task2"]);
+    assert.deepEqual(hre.tasks.rootTasks.get("task3")?.id, ["task3"]);
   });
 
   it("should override a task within the same plugin", async () => {
@@ -350,6 +364,12 @@ describe("TaskManagerImplementation", () => {
     const subsubtask1 = hre.tasks.getTask(["task1", "subtask1", "subsubtask1"]);
     assert.deepEqual(subsubtask1.id, ["task1", "subtask1", "subsubtask1"]);
     assert.equal(subsubtask1.pluginId, "plugin2");
+
+    // task1 should be a root task, but subtask1 and subsubtask1 should not
+    assert.equal(hre.tasks.rootTasks.size, 1);
+    assert.deepEqual(hre.tasks.rootTasks.get("task1")?.id, ["task1"]);
+    assert.equal(hre.tasks.rootTasks.get("subtask1"), undefined);
+    assert.equal(hre.tasks.rootTasks.get("subsubtask1"), undefined);
   });
 
   /**