From 57635a9fbe5e217acd5ad609d7d9b97ce5cef232 Mon Sep 17 00:00:00 2001
From: Omni Worker <bot@omni.agent>
Date: Fri, 21 Nov 2025 22:59:58 -0500
Subject: feat(task): support updating deps in task update

Amp-Thread-ID:
https://ampcode.com/threads/T-79499d9e-f4f4-40de-893c-524c32a45483
Co-authored-by: Amp <amp@ampcode.com>
---
 Omni/Task.hs      | 21 +++++++++++++++++++--
 Omni/Task/Core.hs |  6 +++---
 2 files changed, 22 insertions(+), 5 deletions(-)

(limited to 'Omni')

diff --git a/Omni/Task.hs b/Omni/Task.hs
index 36b318b..f19ab73 100644
--- a/Omni/Task.hs
+++ b/Omni/Task.hs
@@ -44,7 +44,7 @@ Usage:
   task list [options]
   task ready [--json]
   task show <id> [--json]
-  task update <id> <status> [--json]
+  task update <id> <status> [--deps=<ids>] [--dep-type=<type>] [--json]
   task deps <id> [--json]
   task tree [<id>] [--json]
   task progress <id> [--json]
@@ -214,13 +214,30 @@ move args
   | args `Cli.has` Cli.command "update" = do
       tid <- getArgText args "id"
       statusStr <- getArgText args "status"
+      
+      -- Handle update dependencies
+      (deps, depType) <- do
+        -- Parse --deps and --dep-type
+        ids <- case Cli.getArg args (Cli.longOption "deps") of
+          Nothing -> pure []
+          Just depStr -> pure <| T.splitOn "," (T.pack depStr)
+        dtype <- case Cli.getArg args (Cli.longOption "dep-type") of
+          Nothing -> pure Blocks
+          Just "blocks" -> pure Blocks
+          Just "discovered-from" -> pure DiscoveredFrom
+          Just "parent-child" -> pure ParentChild
+          Just "related" -> pure Related
+          Just other -> panic <| "Invalid dependency type: " <> T.pack other <> ". Use: blocks, discovered-from, parent-child, or related"
+        pure (map (\d -> Dependency {depId = d, depType = dtype}) ids, dtype)
+
       let newStatus = case statusStr of
             "open" -> Open
             "in-progress" -> InProgress
             "review" -> Review
             "done" -> Done
             _ -> panic "Invalid status. Use: open, in-progress, review, or done"
-      updateTaskStatus tid newStatus
+            
+      updateTaskStatus tid newStatus deps
       if isJsonMode args
         then outputSuccess <| "Updated task " <> tid
         else do
diff --git a/Omni/Task/Core.hs b/Omni/Task/Core.hs
index bab1912..9e4d2b4 100644
--- a/Omni/Task/Core.hs
+++ b/Omni/Task/Core.hs
@@ -343,15 +343,15 @@ createTask title taskType parent namespace priority deps description =
     pure task
 
 -- Update task status
-updateTaskStatus :: Text -> Status -> IO ()
-updateTaskStatus tid newStatus =
+updateTaskStatus :: Text -> Status -> [Dependency] -> IO ()
+updateTaskStatus tid newStatus newDeps =
   withTaskWriteLock <| do
     tasks <- loadTasksInternal
     now <- getCurrentTime
     let updatedTasks = map updateIfMatch tasks
         updateIfMatch t =
           if matchesId (taskId t) tid
-            then t {taskStatus = newStatus, taskUpdatedAt = now}
+            then t {taskStatus = newStatus, taskUpdatedAt = now, taskDependencies = if null newDeps then taskDependencies t else newDeps}
             else t
     -- Rewrite the entire file (simple approach for MVP)
     tasksFile <- getTasksFilePath
-- 
cgit v1.2.3