handlers.lua

local M = {}
local registry = require('opencode.registry')

local handler_group_modules = {
  'opencode.commands.handlers.window',
  'opencode.commands.handlers.agent',
  'opencode.commands.handlers.workflow',
  'opencode.commands.handlers.session',
  'opencode.commands.handlers.diff',
  'opencode.commands.handlers.permission',
}

---@return OpencodeCommandHandlerMap
local function get_handlers()
  local all_handlers = {}
  local handler_sources = {}

  for _, module_name in ipairs(handler_group_modules) do
    local module_exports = require(module_name)
    local module_handlers = module_exports.handlers or module_exports
    for handler_id, handler in pairs(module_handlers) do
      if handler_sources[handler_id] then
        error(
          string.format(
            "Duplicate handler_id '%s' in modules '%s' and '%s'",
            handler_id,
            handler_sources[handler_id],
            module_name
          )
        )
      end

      handler_sources[handler_id] = module_name
      all_handlers[handler_id] = handler
    end
  end

  for handler_id, handler in pairs(registry.get_handlers()) do
    if handler_sources[handler_id] then
      error(
        string.format(
          "Duplicate handler_id '%s' in modules '%s' and extension registry",
          handler_id,
          handler_sources[handler_id]
        )
      )
    end

    handler_sources[handler_id] = 'extension registry'
    all_handlers[handler_id] = handler
  end

  return all_handlers
end

-- Validate handler graph during module load for fast-fail behavior.
get_handlers()

---@param handler_id string
---@return OpencodeCommandHandler|nil
function M.get(handler_id)
  return get_handlers()[handler_id]
end

---@return string[]
function M.ids()
  local ids = vim.tbl_keys(get_handlers())
  table.sort(ids)
  return ids
end

---@param handler_id string
---@param api OpencodeCommandApi
---@param args string[]
---@param range? OpencodeSelectionRange
---@return boolean, any, OpencodeCommandDispatchError|nil
function M.execute(handler_id, api, args, range)
  local handler = M.get(handler_id)
  if not handler then
    return false, nil, nil
  end

  local ok, result = pcall(handler, api, args or {}, range)
  if ok then
    return true, result, nil
  end

  if type(result) == 'table' and type(result.code) == 'string' and type(result.message) == 'string' then
    if not result.handler_id then
      result.handler_id = handler_id
    end
    return true, nil, result
  end

  return true,
    nil,
    {
      code = 'handler_exception',
      message = 'Command handler failed: ' .. handler_id,
      handler_id = handler_id,
    }
end

return M