local utils = require("utils")
local inlay_hints_handler = require("inlay_hints_handler")
local format_handler = require("format_handler")

local function chain_on_attach(...)
    local funcs = { ... }
    return function(client, bufnr)
        for _, func in ipairs(funcs) do
            func(client, bufnr)
        end
    end
end

local function global_on_attach(client, bufnr)
    inlay_hints_handler.add_buffer(bufnr)

    if client.server_capabilities.documentFormattingProvider then
        vim.api.nvim_buf_create_user_command(bufnr, "Format", vim.lsp.buf.format, { nargs = 0 })
        vim.api.nvim_create_autocmd("BufWritePre", {
            buffer = bufnr,
            callback = function()
                format_handler.format()
            end,
        })
    end

    utils.set_keymap_list({
        { "gd", function() vim.lsp.buf.definition({ reuse_win = true, }) end,  { noremap = true, buffer = bufnr } },
        { "gD", function() vim.lsp.buf.declaration({ reuse_win = true, }) end, { noremap = true, buffer = bufnr } },
    })
end

local global_capabilities = require("blink.cmp").get_lsp_capabilities()
global_capabilities.offsetEncoding = { "utf-16" }

vim.lsp.config("*", {
    capabilities = global_capabilities,
    handlers = {
        ["textDocument/publishDiagnostics"] = vim.lsp.diagnostic.on_publish_diagnostics,
    },
    root_markers = { ".git" },
})

-- Find all files in lua/lsp/servers and require them
-- We use them to ensure that the servers are installed and configured
local errors = {}
local dir_path = "lsp/servers"
utils.foreach(utils.get_file_names_in_dir(dir_path, "*.lua", true), function(server_name)
    local server_path = dir_path .. "/" .. server_name
    local result, conf = utils.xpcallmsg(
        function() return require(server_path) end,
        "Failed to require " .. server_path,
        errors
    )

    if not result or type(conf) ~= "table" or vim.tbl_isempty(conf) or conf.cmd == nil then
        error("Invalid configuration for " .. server_name)
        return
    end

    conf.on_attach = (function()
        if conf.on_attach then
            return chain_on_attach(global_on_attach, conf.on_attach)
        end

        return global_on_attach
    end)()

    -- These still throw errors when wrapped by xpcall.
    -- Wanted it to just handle incorrect input and let the runtime continue
    -- as it would if the require was successful when wrapped. That would be great
    -- for WIP LSP configuration, instead we have the ugly if statements above.
    vim.lsp.config(server_name, conf)
    vim.lsp.enable(server_name)
end)

if #errors > 0 then
    error(table.concat(errors, "\n"))
end