-- attach using autocommand and setup keybindings
vim.api.nvim_create_autocmd('LspAttach', {
  group = vim.api.nvim_create_augroup('sl.lsp', {}),
  callback = function(args)
    -- See `:help vim.lsp.*` for documentation on any of the below functions
    local bufopts = { noremap = true, silent = true, buffer = args.buf }
    vim.keymap.set('n', 'K', vim.lsp.buf.hover, { desc = "LSP show signature", unpack(bufopts) })
    vim.keymap.set('n', '<space>wa', vim.lsp.buf.add_workspace_folder,
      { desc = "Workspace add folder", unpack(bufopts) })
    vim.keymap.set('n', '<space>wr', vim.lsp.buf.remove_workspace_folder,
      { desc = "Workspace remove folder", unpack(bufopts) })
    vim.keymap.set('n', '<space>wl', function()
      print(vim.inspect(vim.lsp.buf.list_workspace_folders()))
    end, { desc = "Workspace list folders", unpack(bufopts) })
    vim.keymap.set('n', '<space>rn', vim.lsp.buf.rename, { desc = "LSP Rename", unpack(bufopts) })
    vim.keymap.set("n", "<space>f", function()
      vim.lsp.buf.format({ async = true })
    end, { desc = "LSP Format buffer", unpack(bufopts) })
    vim.keymap.set("n", '<space>F', ":silent !dx fmt --file %<cr>", { desc = "DX format rsx! regions", unpack(bufopts) })
    vim.keymap.set("v", "<space>f", LSPRangeFormatFunction, { desc = "LSP Format region" })

    -- 2024-09-09 - some ccflow commands for diagnostics, symbols and code actions
    local fzf = require('fzf-lua')
    vim.keymap.set('n', '<space>d', fzf.diagnostics_document, { desc = "Document diagnostics" })
    vim.keymap.set('n', '<space>D', fzf.diagnostics_workspace, { desc = "Workspace diagnostics" })
    vim.keymap.set('n', '<space>e', function()
      local new_config = not vim.diagnostic.config().virtual_lines
      vim.diagnostic.config({ virtual_lines = new_config })
    end, { desc = "Virtual line diagnostics" })
    vim.keymap.set('n', '<space>E', vim.diagnostic.open_float, { desc = "Popup diagnostics" })
    vim.keymap.set('n', '<space>s', fzf.lsp_document_symbols, { desc = "Doc symbols" })
    vim.keymap.set('n', '<space>c', fzf.lsp_code_actions, { desc = "Code Actions" })
    vim.keymap.set('n', '<space>[', vim.diagnostic.goto_prev, { desc = "Previous diagnostics" })
    vim.keymap.set('n', '<space>]', vim.diagnostic.goto_next, { desc = "Previous diagnostics" })

    local gitsigns = require('gitsigns')
    vim.keymap.set('n', '<space>==', gitsigns.preview_hunk_inline, { desc = "Git hunk preview" })
    vim.keymap.set('n', '<space>=B', gitsigns.blame, { desc = "Git blame file" })
    vim.keymap.set('n', '<space>=D', function() gitsigns.diffthis('~') end, { desc = "Git diff this (~)" })
    vim.keymap.set('n', '<space>=R', gitsigns.reset_buffer, { desc = "Git reset file" })
    vim.keymap.set('n', '<space>=S', gitsigns.stage_buffer, { desc = "Git stage file" })
    vim.keymap.set('n', '<space>=[', gitsigns.prev_hunk, { desc = "Git previous hunk" })
    vim.keymap.set('n', '<space>=]', gitsigns.next_hunk, { desc = "Git next hunk" })
    vim.keymap.set('n', '<space>=b', gitsigns.blame_line, { desc = "Git blame line" })
    vim.keymap.set('n', '<space>=d', gitsigns.diffthis, { desc = "Git diff this" })
    vim.keymap.set('n', '<space>=r', gitsigns.reset_hunk, { desc = "Git reset hunk" })
    vim.keymap.set('n', '<space>=s', gitsigns.stage_hunk, { desc = "Git stage hunk" })

    -- vim.keymap.set('n', 'gr', vim.lsp.buf.references, bufopts)
    vim.keymap.set('n', '<C-,>', fzf.lsp_references, { desc = "Find References" })
    vim.keymap.set('n', '<C-.>', fzf.lsp_definitions, { desc = "Find References" })
  end
})

-- lsp format selected range
function LSPRangeFormatFunction()
  vim.lsp.buf.format({
    async = true,
    range = {
      ["start"] = vim.api.nvim_buf_get_mark(0, "<"),
      ["end"] = vim.api.nvim_buf_get_mark(0, ">"),
    }
  })
end

-- Set different settings for different languages' LSP

-- use globally installed vue-language-server (instead of mason provided)
vim.lsp.config('vue_ls', {
  -- add filetypes for typescript, javascript and vue
  -- filetypes = { 'typescript', 'javascript', 'javascriptreact', 'typescriptreact', 'vue' },
  init_options = {
    vue = {
      -- disable hybrid mode, will thus use ts_ls internally
      hybridMode = false,
    },
  },
})

local enable_this_lsp = {
  'vue_ls',
  'gopls',
  'bashls',
  'html',
  'lua_ls',
}

for l in pairs(enable_this_lsp) do 
  vim.lsp.enable(enable_this_lsp[l])
end


-- add autoformat to Dioxus projects
vim.api.nvim_create_autocmd("BufWritePost", {
  pattern = "*.rs",
  callback = function()
    local cwd = vim.lsp.buf.list_workspace_folders()
    if not (cwd == nil) then
      if vim.fn.filereadable(cwd[1] .. "/Dioxus.toml") == 1 then
        local command = "dx fmt --file %"
        vim.cmd("silent ! " .. command)
        -- vim.notify(command)
      end
    end
  end,
})