;;; lsp.fnl - lsp configurations

(import-macros {: bind!} :macros)

(fn on-attach [client buf]
  ((. (. (require :cmp) :setup) :buffer) {
    :sources [{ :name "nvim_lsp" :priority 1 :group 1 }
              { :name "snippy" :priority 1 :group 2 }
              { :name "buffer" :priority 0 :group 3 }
              { :name "path" :priority 0 :group 4 }]})
  (bind! :n :gD vim.lsp.buf.declaration buf)
  ; (bind! :n :gd vim.lsp.buf.definition buf)
  (bind! :n :K vim.lsp.buf.hover buf)
  ; (bind! :n :gI vim.lsp.buf.implementation buf)
  (bind! :n :<C-k> vim.lsp.buf.signature_help buf)
  (bind! :n :<leader>wa vim.lsp.buf.add_workspace_folder buf)
  (bind! :n :<leader>wr vim.lsp.buf.remove_workspace_folder buf)
  (bind! :n :<leader>wl #(print (vim.inspect
                                (vim.lsp.buf.list_workspace_folders))) buf)
  ; (bind! :n :<leader>D vim.lsp.buf.type_definition buf)
  (bind! :n :<leader>rn vim.lsp.buf.rename buf)
  (bind! :n :<leader>cn vim.lsp.buf.code_action buf)
  ; (bind! :n :gr vim.lsp.buf.references buf)
  (bind! [ :n :v ] :<leader>o
         #(vim.lsp.buf.format { :filter
                                ;; use null-ls for formatting
                                (fn [client] (or
                                               (= client.name :null-ls)
                                               (= client.name :rust_analyzer)))
                                :async true }) buf)
  (vim.api.nvim_buf_set_option buf :formatexpr "v:lua.vim.lsp.formatexpr()")

  ;; Some telescope commands
  (bind! :n :<leader>s "<cmd>Telescope lsp_dynamic_workspace_symbols<cr>" buf)
  (bind! :n :<leader>fs "<cmd>Telescope lsp_dynamic_workspace_symbols<cr>" buf)
  (bind! :n :<leader>fS "<cmd>Telescope lsp_workspace_symbols<cr>" buf)
  ;(print (vim.inspect client.server_capabilities))
  (when client.server_capabilities.documentSymbolProvider
    (bind! :n :<leader>d "<cmd>Telescope lsp_document_symbols<cr>" buf)
    (bind! :n :<leader>fd "<cmd>Telescope lsp_document_symbols<cr>" buf))
  (bind! :n :gr "<cmd>Telescope lsp_references<cr>" buf)
  (bind! :n :gd "<cmd>Telescope lsp_definitions<cr>" buf)
  (bind! :n :gI "<cmd>Telescope lsp_implementations<cr>" buf)
  (bind! :n :D "<cmd>Telescope lsp_type_definitions<cr>" buf)

  ;; Some trouble commands
  (bind! :n :<leader>q "<cmd>TroubleToggle document_diagnostics<cr>")
  (bind! :n :<leader>Q "<cmd>TroubleToggle workspace_diagnostics<cr>"))

(fn get-data-dir [server root]
  (let [resolved_path (vim.fn.resolve root)
        joined_path (vim.fn.substitute resolved_path "\\/" "@" :g)]
    (.. (vim.fn.fnamemodify (.. "~/.local/nvim/lsp-cache/"
                                server
                                "/")
                            ":p") joined_path)))

(fn configure []
    (let [lsp (require :lspconfig)
          ;configs (require :lspconfig.configs)
          lsp-cap ((. (require :cmp_nvim_lsp) :default_capabilities))
          lsp-utils (require :plugin.lsp)]
      (macro setup-server! [name ...]
            (let [opts { :on_attach `lsp-utils.on-attach
                         :capabilities `lsp-cap }]
              (var last-key nil)
              (each [_ val (ipairs [...])]
                (if last-key
                    (do (tset opts last-key val)
                        (set last-key nil))
                    (set last-key val)))
              `((. (. lsp ,name) :setup) ,opts)))
      (setup-server! :clangd
                     :on_attach (fn [client buf]
                                  (lsp-utils.on-attach client buf)
                                  (bind! :n :go :<cmd>ClangdSwitchSourceHeader<cr> buf))
                     :cmd [ "clangd"
                            "--header-insertion-decorators=0"
                            "--background-index"
                            "--clang-tidy"
                            "--completion-style=bundled"
                            "--function-arg-placeholders"
                            "--header-insertion=never"
                            "--malloc-trim"
                            "--pch-storage=memory"
                            "--offset-encoding=utf-16" ])
      (setup-server! :cmake)
      (setup-server! :gopls)
      (setup-server! :rust_analyzer)
      (setup-server! :texlab)
      (setup-server! :pylsp)
      (setup-server! :lua_ls
                     :settings {
                       :Lua {
                         :runtime {
                           :version "LuaJIT" }
                         :diagnostics {
                           :globals [ "vim" ] }
                         :workspace {
                           :checkThirdParty false
                           :library (vim.api.nvim_get_runtime_file "" true) }
                         :telemetry {
                           :enable false }}})
      (setup-server! :fennel_language_server
                     :settings {
                       :fennel {
                         :workspace {
                           :library (vim.api.nvim_list_runtime_paths) }
                         :diagnostics {
                           :globals [ "vim" ] }}})))

{ : configure : on-attach : get-data-dir }