local lsp_lines = require("lsp_lines")
lsp_lines.setup()
-- Disable virtual_text since it's redundant due to lsp_lines.
vim.diagnostic.config({
  virtual_text = false,
})

vim.o.foldcolumn = "1" -- '0' is not bad
vim.o.foldlevel = 99 -- Using ufo provider need a large value, feel free to decrease the value
vim.o.foldlevelstart = 99
vim.o.foldenable = true
vim.o.fillchars = [[eob: ,fold: ,foldopen:,foldsep: ,foldclose:]]
vim.o.statuscolumn = "%= "
  -- FIXME: figure out how to put on the other side without having to do a lot of shifting
  .. "%s" -- sign column
  .. "%{%" -- evaluate this, and then evaluate what it returns
  .. "&number ?"
  .. "(v:relnum ?"
  -- when showing relative numbers, make sure to pad so things don't shift as you move the cursor
  .. 'printf("%"..len(line("$")).."s", v:relnum)'
  .. ":"
  .. "v:lnum"
  .. ")"
  .. ":"
  .. '""'
  .. " " -- space between lines and fold
  .. "%}"
  .. "%= "
  .. "%#FoldColumn#" -- highlight group for fold
  .. "%{" -- expression for showing fold expand/colapse
  .. "foldlevel(v:lnum) > foldlevel(v:lnum - 1)" -- any folds?
  .. "? (foldclosed(v:lnum) == -1" -- currently open?
  .. '? ""' -- point down
  .. ':  ""' -- point to right
  .. ")"
  .. ': " "' -- blank for no fold, or inside fold
  .. "}"
  .. "%= " -- spacing between end of column and start of text

-- Using ufo provider need remap `zR` and `zM`. If Neovim is 0.6.1, remap yourself
require("which-key").register({
  z = {
    R = { require("ufo").openAllFolds, "Open all folds" },
    M = { require("ufo").closeAllFolds, "Close all folds" },
  },
})
local capabilities = vim.lsp.protocol.make_client_capabilities()
-- Tell the server the capability of foldingRange,
-- Neovim hasn't added foldingRange to default capabilities, users must add it manually
capabilities.textDocument.foldingRange = {
  dynamicRegistration = false,
  lineFoldingOnly = true,
}
require("ufo").setup()

local lspconfig = require("lspconfig")
local on_attach_def = function(client, bufnr)
  require("which-key").register({
    K = { vim.lsp.buf.hover, "Hover" },
    ["<leader>"] = {
      l = {
        d = { vim.diagnostic.open_float, "Open diagnostic window" },
        c = { vim.lsp.buf.code_action, "Code action" },
        r = { vim.lsp.buf.rename, "Rename" },
        f = {
          function()
            vim.lsp.buf.format({ async = true })
          end,
          "Format (lsp)",
          mode = { "n", "v" },
        },
      },
      t = {
        l = { lsp_lines.toggle, "LSP lines" },
        i = {
          function()
            vim.lsp.inlay_hint(bufnr, nil)
          end,
          "LSP inlay hints",
        },
      },
    },
    g = {
      d = {
        function()
          require("telescope.builtin").lsp_definitions({ reuse_win = true })
        end,
        "Goto definition",
      },
      t = {
        function()
          require("telescope.builtin").lsp_type_definitions({ reuse_win = true })
        end,
        "Goto type defininition",
      },
      r = { "<cmd>Telescope lsp_references<cr>", "Goto references" },
      D = { vim.lsp.buf.declaration, "Goto declaration" },
      I = { "<cmd>Telescope lsp_implementations<cr>", "Goto implementation" },
      K = { vim.lsp.buf.signature_help, "Signature help" },
    },
    ["["] = {
      d = { vim.diagnostic.goto_prev, "Previous diagnostic" },
    },
    ["]"] = {
      d = { vim.diagnostic.goto_next, "Next diagnostic" },
    },
  }, { buffer = bufnr, silent = true })

  if client.server_capabilities.inlayHintProvider then
    local slow_lsp_servers = {
      "rust_analyzer",
    }
    local timeout = vim.tbl_contains(slow_lsp_servers, client.name, {}) and 500 or 0
    vim.defer_fn(function()
      vim.lsp.inlay_hint(bufnr, true)
    end, timeout)
  end
end

local lspconfig_default_options = {
  on_attach = on_attach_def,
  capabilities = capabilities,
}

---function to add default options to lspconfig
---@param lsp string
---@param options table
---@return nil
local function lspconfig_setup(lsp, options)
  local coq_options = require("coq").lsp_ensure_capabilities({})
  local merged_options = vim.tbl_deep_extend("force", coq_options, lspconfig_default_options, options)
  lspconfig[lsp].setup(merged_options)
end

local servers = {
  "bashls",
  "nil_ls",
  "pylsp",
  "ruff_lsp",
  "typst_lsp",
  "gopls",
}
for _, lsp in ipairs(servers) do
  lspconfig_setup(lsp, {})
end

lspconfig_setup("rust_analyzer", {
  settings = {
    ["rust-analyzer"] = {
      checkOnSave = {
        command = "clippy",
      },
    },
  },
})

lspconfig_setup("lua_ls", {
  settings = {
    Lua = {
      runtime = {
        -- Tell the language server which version of Lua you're using (most likely LuaJIT in the case of Neovim)
        version = "LuaJIT",
      },
      diagnostics = {
        -- Get the language server to recognize the `vim` global
        globals = { "vim" },
      },
      workspace = {
        -- Make the server aware of Neovim runtime files
        library = vim.api.nvim_get_runtime_file("", true),
        checkThirdParty = false,
      },
      -- Do not send telemetry data containing a randomized but unique identifier
      telemetry = {
        enable = false,
      },
      format = {
        enable = false,
      },
      hint = {
        enable = true,
      },
    },
  },
})