Add completion phases

1 files changed, 55 insertions, 0 deletions

diff --git a/makima/src/db/repository.rs b/makima/src/db/repository.rs
index d3e4c56..b55b05e 100644
--- a/makima/src/db/repository.rs
+++ b/makima/src/db/repository.rs
@@ -3018,6 +3018,61 @@ pub async fn update_contract_supervisor(
     .await
 }
 
+/// Mark a deliverable as complete for a specific phase.
+/// Uses JSONB operations to append the deliverable_id to the phase's array.
+pub async fn mark_deliverable_complete(
+    pool: &PgPool,
+    contract_id: Uuid,
+    phase: &str,
+    deliverable_id: &str,
+) -> Result<Contract, sqlx::Error> {
+    // Use jsonb_set to add the deliverable to the phase's array
+    // If the phase key doesn't exist, create an empty array first
+    // COALESCE handles the case where the phase array doesn't exist yet
+    sqlx::query_as::<_, Contract>(
+        r#"
+        UPDATE contracts
+        SET completed_deliverables = jsonb_set(
+            completed_deliverables,
+            ARRAY[$2::text],
+            COALESCE(completed_deliverables->$2, '[]'::jsonb) || to_jsonb($3::text),
+            true
+        ),
+        updated_at = NOW()
+        WHERE id = $1
+        AND NOT (COALESCE(completed_deliverables->$2, '[]'::jsonb) ? $3)
+        RETURNING *
+        "#,
+    )
+    .bind(contract_id)
+    .bind(phase)
+    .bind(deliverable_id)
+    .fetch_one(pool)
+    .await
+}
+
+/// Clear all completed deliverables for a specific phase.
+/// Used when phase changes or deliverables need to be reset.
+pub async fn clear_phase_deliverables(
+    pool: &PgPool,
+    contract_id: Uuid,
+    phase: &str,
+) -> Result<Contract, sqlx::Error> {
+    sqlx::query_as::<_, Contract>(
+        r#"
+        UPDATE contracts
+        SET completed_deliverables = completed_deliverables - $2,
+            updated_at = NOW()
+        WHERE id = $1
+        RETURNING *
+        "#,
+    )
+    .bind(contract_id)
+    .bind(phase)
+    .fetch_one(pool)
+    .await
+}
+
 /// Get the supervisor task for a contract.
 pub async fn get_contract_supervisor_task(
     pool: &PgPool,