Fix: Link directives and contracts

1 files changed, 46 insertions, 5 deletions

diff --git a/makima/src/orchestration/engine.rs b/makima/src/orchestration/engine.rs
index 470db40..9f7c3b1 100644
--- a/makima/src/orchestration/engine.rs
+++ b/makima/src/orchestration/engine.rs
@@ -99,6 +99,20 @@ pub enum EngineEvent {
     },
 }
 
+/// Result from starting a directive, containing info needed for auto-start.
+pub struct PlanningStartResult {
+    /// The planning task ID that needs to be started on a daemon
+    pub task_id: Uuid,
+    /// The owner ID for finding available daemons
+    pub owner_id: Uuid,
+    /// The planning task details needed for the SpawnTask command
+    pub task_name: String,
+    pub plan: String,
+    pub contract_id: Uuid,
+    pub repository_url: Option<String>,
+    pub base_branch: Option<String>,
+}
+
 /// Main orchestration engine for directives.
 pub struct DirectiveEngine {
     pool: PgPool,
@@ -134,7 +148,8 @@ impl DirectiveEngine {
     // ========================================================================
 
     /// Start a directive: spawn a planning contract+task to generate the chain.
-    pub async fn start_directive(&self, directive_id: Uuid) -> Result<(), EngineError> {
+    /// Returns a `PlanningStartResult` so the caller can auto-start the task on a daemon.
+    pub async fn start_directive(&self, directive_id: Uuid) -> Result<PlanningStartResult, EngineError> {
         let directive = repository::get_directive(&self.pool, directive_id)
             .await?
             .ok_or(EngineError::DirectiveNotFound(directive_id))?;
@@ -201,17 +216,18 @@ impl DirectiveEngine {
         let plan = self.build_planning_task_instructions(&directive);
 
         // Create the planning task
-        let _task = repository::create_task_for_owner(
+        let task_name = format!("{} - Planning", directive.title);
+        let task = repository::create_task_for_owner(
             &self.pool,
             directive.owner_id,
             CreateTaskRequest {
                 contract_id: Some(contract.id),
-                name: format!("{} - Planning", directive.title),
+                name: task_name.clone(),
                 description: Some(format!(
                     "Plan the execution chain for directive: {}",
                     directive.goal
                 )),
-                plan,
+                plan: plan.clone(),
                 parent_task_id: None,
                 is_supervisor: true,
                 priority: 5,
@@ -234,6 +250,22 @@ impl DirectiveEngine {
             EngineError::ContractCreation(format!("Failed to create planning task: {}", e))
         })?;
 
+        // Link the supervisor task to the contract
+        if let Err(e) = repository::update_contract_supervisor(
+            &self.pool,
+            contract.id,
+            task.id,
+        )
+        .await
+        {
+            tracing::warn!(
+                contract_id = %contract.id,
+                task_id = %task.id,
+                error = %e,
+                "Failed to link supervisor task to planning contract"
+            );
+        }
+
         // Link the planning contract to the directive
         repository::set_directive_orchestrator_contract(
             &self.pool,
@@ -248,13 +280,22 @@ impl DirectiveEngine {
             "info",
             serde_json::json!({
                 "contract_id": contract.id,
+                "task_id": task.id,
                 "message": "Planning task spawned, waiting for chain generation",
             }),
             "system",
         )
         .await?;
 
-        Ok(())
+        Ok(PlanningStartResult {
+            task_id: task.id,
+            owner_id: directive.owner_id,
+            task_name,
+            plan,
+            contract_id: contract.id,
+            repository_url: directive.repository_url.clone(),
+            base_branch: directive.base_branch.clone(),
+        })
     }
 
     /// Pause a directive.