local Base = require("avante.llm_tools.base")

---@class AvanteLLMTool
local M = setmetatable({}, Base)

M.name = "update_todo_status"

M.description = "Update the status of TODO"

---@type AvanteLLMToolParam
M.param = {
  type = "table",
  fields = {
    {
      name = "id",
      description = "The ID of the TODO to update",
      type = "string",
    },
    {
      name = "status",
      description = "The status of the TODO to update",
      type = "string",
      choices = { "todo", "doing", "done", "cancelled" },
    },
  },
}

---@type AvanteLLMToolReturn[]
M.returns = {
  {
    name = "success",
    description = "Whether the TODO was updated successfully",
    type = "boolean",
  },
  {
    name = "error",
    description = "Error message if the TODOs could not be updated",
    type = "string",
    optional = true,
  },
}

M.on_render = function() return {} end

---@type AvanteLLMToolFunc<{ id: string, status: string }>
function M.func(opts, on_log, on_complete, session_ctx)
  local sidebar = require("avante").get()
  if not sidebar then return false, "Avante sidebar not found" end
  local todos = sidebar.chat_history.todos
  if not todos or #todos == 0 then return false, "No todos found" end
  for _, todo in ipairs(todos) do
    if todo.id == opts.id then
      todo.status = opts.status
      break
    end
  end
  sidebar:update_todos(todos)
  if on_complete then
    on_complete(true, nil)
    return nil, nil
  end
  return true, nil
end

return M