Add planNext method to PubOrchestrator

SudoPub · Nov 6, 2023 · 62bb031 · 62bb031
1 parent 25d444e
commit 62bb031
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/src/orchestrator/orchestrator.ts b/src/orchestrator/orchestrator.ts
@@ -6,10 +6,12 @@
 
 import { PUB_ACTION_TYPE, PubAction } from "../action/definition/action";
 import { initializeCreateTaskManager } from "../orchestration/initialize/create-tasks";
+import { PUB_PLAN_TYPE, PubPlan } from "../plan/definition/plan";
 import { applyActionOnTaskManager } from "../task/apply/apply";
 import { PubTaskManager } from "../task/task-manager";
 import { PubCachedWorkflowConfiguration } from "../workflow/cache/configuration";
 import { PubWorkflowConfiguration } from "../workflow/definition/configuration";
+import { planForNext } from "./plan/plan-next";
 
 export class PubOrchestrator {
 
@@ -40,4 +42,9 @@ export class PubOrchestrator {
         const applyResult: boolean = applyActionOnTaskManager(action, this._taskManager);
         return applyResult;
     }
+
+    public planNext(): Array<PubPlan<PUB_PLAN_TYPE>> {
+
+        return planForNext(this._configuration, this._taskManager);
+    }
 }
diff --git a/src/orchestrator/plan/plan-next.ts b/src/orchestrator/plan/plan-next.ts
@@ -4,7 +4,10 @@
  * @description Plan Next
  */
 
+import { createPubPlan } from "../../plan/create";
 import { PUB_PLAN_TYPE, PubPlan } from "../../plan/definition/plan";
+import { PUB_PROCEDURE_TYPE, PubProcedureConfiguration } from "../../procedure/definition/configuration";
+import { PUB_TASK_TYPE } from "../../task/definition/task";
 import { PubTaskBase } from "../../task/task-base";
 import { PubTaskManager } from "../../task/task-manager";
 import { PubCachedWorkflowConfiguration } from "../../workflow/cache/configuration";
@@ -20,5 +23,20 @@ export const planForNext = (
         return [];
     }
 
+    const nextPlans: Array<PubPlan<PUB_PLAN_TYPE>> = [];
+    for (const nextExecutableTask of nextExecutableTasks) {
+
+        if (nextExecutableTask.taskType === PUB_TASK_TYPE.DRIVER) {
+
+            nextPlans.push(createPubPlan(
+                PUB_PLAN_TYPE.EXECUTE_DRIVER,
+                {
+                    procedure: nextExecutableTask.procedure as PubProcedureConfiguration<PUB_PROCEDURE_TYPE.DRIVER>,
+                    input: nextExecutableTask.getExecuteInput().getOrThrow(),
+                },
+            ));
+        }
+    }
+
     return [];
 };