Add PR button to directives

3 files changed, 84 insertions, 0 deletions

diff --git a/makima/src/server/handlers/directives.rs b/makima/src/server/handlers/directives.rs
index f03dccf..960da94 100644
--- a/makima/src/server/handlers/directives.rs
+++ b/makima/src/server/handlers/directives.rs
@@ -930,6 +930,88 @@ pub async fn cleanup_tasks(
 }
 
 // =============================================================================
+// PR Creation
+// =============================================================================
+
+/// Trigger PR creation or update for a directive.
+#[utoipa::path(
+    post,
+    path = "/api/v1/directives/{id}/create-pr",
+    params(("id" = Uuid, Path, description = "Directive ID")),
+    responses(
+        (status = 200, description = "PR task spawned", body = DirectiveWithSteps),
+        (status = 404, description = "Not found", body = ApiError),
+        (status = 409, description = "Completion task already running", body = ApiError),
+        (status = 503, description = "Database not configured", body = ApiError),
+    ),
+    security(("bearer_auth" = []), ("api_key" = [])),
+    tag = "Directives"
+)]
+pub async fn create_pr(
+    State(state): State<SharedState>,
+    Authenticated(auth): Authenticated,
+    Path(id): Path<Uuid>,
+) -> impl IntoResponse {
+    let Some(ref pool) = state.db_pool else {
+        return (
+            StatusCode::SERVICE_UNAVAILABLE,
+            Json(ApiError::new("DB_UNAVAILABLE", "Database not configured")),
+        )
+            .into_response();
+    };
+
+    match crate::orchestration::directive::trigger_completion_task(
+        pool,
+        &state,
+        id,
+        auth.owner_id,
+    )
+    .await
+    {
+        Ok(_task_id) => {
+            // Return the updated directive with steps
+            match repository::get_directive_with_steps_for_owner(pool, auth.owner_id, id).await {
+                Ok(Some((directive, steps))) => {
+                    Json(DirectiveWithSteps { directive, steps }).into_response()
+                }
+                Ok(None) => (
+                    StatusCode::NOT_FOUND,
+                    Json(ApiError::new("NOT_FOUND", "Directive not found")),
+                )
+                    .into_response(),
+                Err(e) => (
+                    StatusCode::INTERNAL_SERVER_ERROR,
+                    Json(ApiError::new("GET_FAILED", &e.to_string())),
+                )
+                    .into_response(),
+            }
+        }
+        Err(e) => {
+            let msg = e.to_string();
+            if msg.contains("not found") {
+                (
+                    StatusCode::NOT_FOUND,
+                    Json(ApiError::new("NOT_FOUND", &msg)),
+                )
+                    .into_response()
+            } else if msg.contains("already running") || msg.contains("already claimed") {
+                (
+                    StatusCode::CONFLICT,
+                    Json(ApiError::new("COMPLETION_IN_PROGRESS", &msg)),
+                )
+                    .into_response()
+            } else {
+                (
+                    StatusCode::INTERNAL_SERVER_ERROR,
+                    Json(ApiError::new("CREATE_PR_FAILED", &msg)),
+                )
+                    .into_response()
+            }
+        }
+    }
+}
+
+// =============================================================================
 // Order Pickup
 // =============================================================================
 
diff --git a/makima/src/server/mod.rs b/makima/src/server/mod.rs
index ce18620..c963618 100644
--- a/makima/src/server/mod.rs
+++ b/makima/src/server/mod.rs
@@ -238,6 +238,7 @@ pub fn make_router(state: SharedState) -> Router {
         .route("/directives/{id}/steps/{step_id}/skip", post(directives::skip_step))
         .route("/directives/{id}/goal", put(directives::update_goal))
         .route("/directives/{id}/cleanup-tasks", post(directives::cleanup_tasks))
+        .route("/directives/{id}/create-pr", post(directives::create_pr))
         .route("/directives/{id}/pick-up-orders", post(directives::pick_up_orders))
         // Order endpoints
         .route(
diff --git a/makima/src/server/openapi.rs b/makima/src/server/openapi.rs
index b21dab9..9c6463a 100644
--- a/makima/src/server/openapi.rs
+++ b/makima/src/server/openapi.rs
@@ -129,6 +129,7 @@ use crate::server::messages::{ApiError, AudioEncoding, StartMessage, StopMessage
         directives::skip_step,
         directives::update_goal,
         directives::cleanup_tasks,
+        directives::create_pr,
         // Order endpoints
         orders::list_orders,
         orders::create_order,