local api = vim.api

---@class ShortcutsSource : cmp.Source
local ShortcutsSource = {}
ShortcutsSource.__index = ShortcutsSource

function ShortcutsSource:new()
  local instance = setmetatable({}, ShortcutsSource)
  return instance
end

function ShortcutsSource:is_available() return vim.bo.filetype == "AvanteInput" end

function ShortcutsSource.get_position_encoding_kind() return "utf-8" end

function ShortcutsSource:get_trigger_characters() return { "#" } end

function ShortcutsSource:get_keyword_pattern() return [[\%(@\|#\|/\)\k*]] end

---@param params cmp.SourceCompletionApiParams
function ShortcutsSource:complete(params, callback)
  ---@type string?
  local trigger_character
  if params.completion_context.triggerKind == 1 then
    trigger_character = string.match(params.context.cursor_before_line, "%s*(#)%S*$")
  elseif params.completion_context.triggerKind == 2 then
    trigger_character = params.completion_context.triggerCharacter
  end
  if not trigger_character or trigger_character ~= "#" then return callback({ items = {}, isIncomplete = false }) end
  local Utils = require("avante.utils")
  local kind = require("cmp").lsp.CompletionItemKind.Variable
  local shortcuts = Utils.get_shortcuts()

  local items = {}
  for _, shortcut in ipairs(shortcuts) do
    table.insert(items, {
      label = "#" .. shortcut.name,
      kind = kind,
      detail = shortcut.details,
      data = {
        name = shortcut.name,
        prompt = shortcut.prompt,
        details = shortcut.details,
      },
    })
  end

  callback({
    items = items,
    isIncomplete = false,
  })
end

function ShortcutsSource:execute(item, callback)
  -- ShortcutsSource should only provide completion, not perform replacement
  -- The actual shortcut replacement is handled in sidebar.lua handle_submit function
  callback()
end

return ShortcutsSource