if not vim.lsp.enable then
  return
end

local lsp = vim.lsp

lsp.enable({
  'bashls',
  'ccls',
  'clangd',
  'gopls',
  'harper_ls',
  'pylsp',
  'rust_analyzer',
  'superhtml',
  'tinymist',
  'zk',
  'zls',
})

local capabilities = {}
if pcall(require, 'blink.cmp') then
  capabilities = require('blink.cmp').get_lsp_capabilities()
end

lsp.config('*', {
  root_markers = { '.git' },
  capabilities = vim.tbl_deep_extend('error', capabilities, {
    textDocument = {
      semanticTokens = { multilineTokenSupport = true },
    },
  }),
})

vim.api.nvim_create_autocmd({ 'LspAttach' }, {
  group = vim.api.nvim_create_augroup('lspattach', {}),
  callback = function(args)
    local client = assert(vim.lsp.get_client_by_id(args.data.client_id))

    -- TODO: what does this do?
    vim.opt_local.omnifunc = 'v:lua.vim.lsp.omnifunc'

    vim.keymap.set('n', 'gd', vim.lsp.buf.definition, { desc = 'goto_definition' })
    vim.keymap.set('n', 'gD', vim.lsp.buf.declaration, { desc = 'goto declaration' })
    vim.keymap.set('n', 'gT', vim.lsp.buf.type_definition, { desc = 'goto type definition' })
    vim.keymap.set('n', 'gR', vim.lsp.buf.references, { desc = 'show references' })
    vim.keymap.set({ 'n', 'v' }, 'ga', vim.lsp.buf.code_action, { desc = 'show code actions' })
    vim.keymap.set('n', 'grr', vim.lsp.buf.rename, { desc = 'rename' })

    if client:supports_method('textDocument/implementation') then
      vim.keymap.set('n', 'gi', vim.lsp.buf.implementation, { desc = 'goto implementation' })
    end

    if client:supports_method('textDocument/completion') then
      vim.lsp.completion.enable(true, client.id, args.buf, { autotrigger = false })
    end

    if client:supports_method('textDocument/formatting') then
      vim.keymap.set('n', '<leader>f', vim.lsp.buf.format, { desc = 'format' })
    end

    if
      not client:supports_method('textDocument/willSaveWaitUntil')
      and client:supports_method('textDocument/formatting')
    then
      vim.api.nvim_create_autocmd({ 'BufWritePre' }, {
        group = vim.api.nvim_create_augroup('lspattach', {}),
        buffer = args.buf,
        callback = function()
          if vim.b.nofmt then
            return
          end
          vim.lsp.buf.format({ bufnr = args.buf, id = client.id, timeout_ms = 1000 })
        end,
      })
    end
  end,
})