local config = require('muwiki.config')
local links = require('muwiki.links.detection')
local files = require('muwiki.files')
local handlers = require('muwiki.handlers')
local url_handlers = require('muwiki.links.url_handlers')

local M = {}

local function get_wiki_root_or_notify()
  local wiki_root = config.wiki_root(0)
  if not wiki_root then
    vim.notify('Not in a wiki buffer', vim.log.levels.ERROR)
    return nil
  end
  return wiki_root
end

function M.open_link()
  local link = links.get_link()
  if not link then
    vim.notify('No link found under cursor', vim.log.levels.WARN)
    return
  end

  if link.type == 'web' then
    url_handlers.handle_web_link(link.target)
    return
  end

  local wiki_root = get_wiki_root_or_notify()
  if not wiki_root then
    return
  end

  if link.type == 'file' then
    if vim.startswith(link.target, 'file://') then
      url_handlers.handle_file_url(link.target)
    else
      url_handlers.handle_file_link(link.target, wiki_root)
    end
    return
  end

  url_handlers.handle_wiki_link(link.target, wiki_root)
end

function M.open_link_with()
  local link = links.get_link()
  if not link then
    vim.notify('No link found under cursor', vim.log.levels.WARN)
    return
  end

  if link.type == 'wiki' then
    vim.notify('Menu not available for wiki links', vim.log.levels.WARN)
    return
  end

  local url = link.target
  if link.type == 'file' then
    local wiki_root = get_wiki_root_or_notify()
    if not wiki_root then
      return
    end
    url = files.resolve(url, wiki_root)
  end

  local matching_handlers = handlers.get_matching(url)

  if #matching_handlers == 0 then
    vim.notify('No handlers available for this URL', vim.log.levels.WARN)
    return
  end

  if #matching_handlers == 1 then
    handlers.execute(matching_handlers[1], url)
    return
  end

  local handler_names = {}
  for _, handler in ipairs(matching_handlers) do
    table.insert(handler_names, handler.name)
  end

  vim.ui.select(handler_names, {
    prompt = 'Open with:',
  }, function(choice, idx)
    if choice and idx then
      handlers.execute(matching_handlers[idx], url)
    end
  end)
end

return M