fix: use vim.ui.input in CodyTask

Closes #219
sourcegraph · May 25, 2024 · 8bdd4d1 · 8bdd4d1
1 parent 92a9f5f
commit 8bdd4d1
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/lua/sg/cody/plugin/commands.lua b/lua/sg/cody/plugin/commands.lua
@@ -129,9 +129,20 @@ commands.CodyToggle = {
 commands.CodyTask = {
   function(command)
     local bufnr = vim.api.nvim_get_current_buf()
-    cody_commands.do_task(bufnr, command.line1 - 1, command.line2, command.args)
+
+    if command.args == "" then
+      vim.ui.input({ prompt = "Cody Task: " }, function(input)
+        if not input or input == "" then
+          return
+        end
+
+        cody_commands.do_task(bufnr, command.line1 - 1, command.line2, input)
+      end)
+    else
+      cody_commands.do_task(bufnr, command.line1 - 1, command.line2, command.args)
+    end
   end,
-  { range = 2, nargs = 1 },
+  { range = 2, nargs = "*" },
 }
 
 ---@command :CodyRestart [[

diff --git a/lua/sg/cody/tasks.lua b/lua/sg/cody/tasks.lua
@@ -41,7 +41,7 @@ CodyTask.init = function(opts)
 
       local bufnr = task.windows.history_bufnr
       vim.keymap.set("n", "<CR>", function()
-        local text = task.transcript:last_message():text()
+        local text = task.transcript:last_message():text() or ""
         local lines = vim.split(vim.trim(text), "\n")
         local to_insert = {}