local lsp = vim.lsp
local au = vim.api.nvim_create_autocmd
local ag = vim.api.nvim_create_augroup

local keymap = require('utils.keymap')
local mapopts = keymap.new_opts
local map_prefix = '<leader>;'

local ag_lsp = ag('lsp', { clear = true })

au('LspAttach', {
    group = ag_lsp,
    callback = function(ev)
        local client = lsp.get_client_by_id(ev.data.client_id)
        local bufnr = ev.buf

        ---@diagnostic disable-next-line: need-check-nil
        if client:supports_method('textDocument/documentHighlight', bufnr) then
            local ag_lsp_doc_highlight = ag('lsp_doc_highlight', { clear = true })
            au({ 'CursorHold' }, {
                group = ag_lsp_doc_highlight,
                buffer = bufnr,
                callback = lsp.buf.document_highlight,
            })

            au({ 'CursorMoved', 'InsertEnter' }, {
                group = ag_lsp_doc_highlight,
                buffer = bufnr,
                callback = lsp.buf.clear_references,
            })
        end

        ---@diagnostic disable-next-line: need-check-nil
        if client:supports_method('textDocument/documentColor', bufnr) then
            local document_color = lsp.document_color
            document_color.enable(false, { bufnr = bufnr })
            keymap.nmap({
                map_prefix .. 'c',
                function()
                    document_color.enable(not document_color.is_enabled())
                end,
                mapopts(bufnr, 'Lsp: Toggle document color'),
            })
        end

        vim.diagnostic.config({
            underline = false,
            virtual_lines = false,
            virtual_text = false,
            float = {
                source = true,
            },
        })

        keymap.nmap({
            -- stylua: ignore start
            { 'gd', lsp.buf.definition, mapopts(bufnr, 'Lsp: Goto definition') },
            { ']d', function() vim.diagnostic.jump({ count = 1, float = true }) end, mapopts(bufnr) },
            { '[d', function() vim.diagnostic.jump({ count = -1, float = true }) end, mapopts(bufnr) },
            { map_prefix .. 'a', lsp.buf.code_action, mapopts(bufnr, 'Lsp: select code action at curpos') },
            { map_prefix .. 'd', vim.diagnostic.setloclist, mapopts(bufnr, 'Lsp: Populate location list with diagnotics') },
            { map_prefix .. 'i', function() lsp.inlay_hint.enable(not lsp.inlay_hint.is_enabled(), { bufnr }) end, mapopts(bufnr, 'Lsp: Toggle inlay hints') },
            -- stylua: ignore end
        })
    end,
})

lsp.enable({
    'luals',
    'ts_ls',
    'clangd',
    'cssls',
    'nixd',
    'tailwindcss',
    'jsonnet_ls',
    'astro',
})