(let [mod {}
      cmd vim.cmd
      api vim.api
      keymap (require :nifoc.keymap)]
  (fn maybe-refresh-codelens [client]
    (when client.server_capabilities.codeLensProvider
      (vim.lsp.codelens.refresh)))

  (fn mod.setup []
    (vim.diagnostic.config {:underline true
                            :virtual_text false
                            :signs true
                            :float {:border :rounded :source true}
                            :update_in_insert false
                            :severity_sort true})
    (cmd "sign define DiagnosticSignError text= texthl=DiagnosticSignError linehl= numhl=DiagnosticSignError")
    (cmd "sign define DiagnosticSignWarn text= texthl=DiagnosticSignWarn linehl= numhl=DiagnosticSignWarn")
    (cmd "sign define DiagnosticSignInfo text= texthl=DiagnosticSignInfo linehl= numhl=DiagnosticSignInfo")
    (cmd "sign define DiagnosticSignHint text= texthl=DiagnosticSignHint linehl= numhl=DiagnosticSignHint"))

  (fn mod.maybe-enable-lsp [client bufnr]
    (when (= vim.b.nifoc_lsp_enabled nil)
      (api.nvim_buf_set_var bufnr :nifoc_lsp_enabled 1)
      (keymap.lsp-attach client bufnr)
      (let [augroup (vim.api.nvim_create_augroup :NifocDiagnostic {:clear true})
            aucmd vim.api.nvim_create_autocmd]
        (aucmd [:CursorHold :CursorHoldI]
               {:callback (fn []
                            (vim.diagnostic.open_float nil {:focus false})
                            (maybe-refresh-codelens client))
                :buffer bufnr
                :group augroup})
        (aucmd :InsertLeave {:callback #(maybe-refresh-codelens client)
                             :buffer bufnr
                             :group augroup}))))

  mod)