local lsp        = require('lspconfig')
local nnoremap   = require('astronauta.keymap').nnoremap
local spinner_lsp= require('spinner.lsp')

require('lspconfig/configs').zk = {
  default_config = {
    cmd = {'zk', 'lsp'},
    filetypes = {'markdown'},
    root_dir = function()
      return vim.loop.cwd()
    end,
    settings = {},
  }
}

local my_attach = function(client, bufnr)
  spinner_lsp.on_attach(client, bufnr)

  if client.resolved_capabilities.goto_definition then
    nnoremap { '<C-]>'     , vim.lsp.buf.definition }
    nnoremap { 'gd'        , vim.lsp.buf.definition }
    nnoremap { 'gR'        , vim.lsp.buf.references }
  end
  if client.resolved_capabilities.hover then
    nnoremap { 'K'         , vim.lsp.buf.hover }
  end
  if client.resolved_capabilities.rename then
    nnoremap { 'grr'       , vim.lsp.buf.rename }
  end
  if client.resolved_capabilities.signature_help then
    nnoremap { '<c-h>'     , vim.lsp.buf.signature_help }
    -- nnoremap <silent> ;s <cmd> lua vim.lsp.buf.signature_help()<CR>
  end
  if client.resolved_capabilities.document_formatting then
    nnoremap { '<leader>f' , vim.lsp.buf.formatting_sync }
  end
  if client.resolved_capabilities.type_definition then
    nnoremap { 'gT'        , vim.lsp.buf.type_definition }
  end
  if client.resolved_capabilities.declaration then
    nnoremap { 'gD'        , vim.lsp.buf.declaration }
  end
  if client.resolved_capabilities.implementation then
    nnoremap { 'gI'        , vim.lsp.buf.implementation }
  end
  if client.resolved_capabilities.workspace_symbol then
    nnoremap { 'gW'        , vim.lsp.buf.workspace_symbol }
  end
  -- if client.resolved_capabilities.document_symbol then
  --   nnoremap { 'g0', vim.lsp.buf.document_symbol }
  -- end

  nnoremap { ']d'        , vim.lsp.diagnostic.goto_next }
  nnoremap { '[d'        , vim.lsp.diagnostic.goto_prev }
  nnoremap { '<leader>d' , vim.lsp.diagnostic.show_line_diagnostics }

  augroup {
    lsp_user = {
      {'BufWritePre' , '*'    ,  'lua require("lsp").format(vim.fn.expand("<afile>:p"))'},
      {'BufWritePre' , '*.go' ,  'lua vim.lsp.buf.code_action({ only = {"source.organizeImports"} })'},
      -- {'CursorHold'  , '*'    ,  'lua vim.lsp.diagnostic.show_line_diagnostics()'},
    }
  }
end

local my_exit = function(code, signal, client_id)
  spinner_lsp.on_exit(code, signal, client_id)
end

local setup = function()
  local caps = vim.lsp.protocol.make_client_capabilities()
  -- enable lsp-based snippets
  caps.textDocument.completion.completionItem.snippetSupport = true
  caps.textDocument.completion.completionItem.resolveSupport = {
    properties = {
      'documentation',
      'detail',
      'additionalTextEdits',
    }
  }

  spinner_lsp.setup()
  spinner_lsp.init_capabilities(caps)

  -- handler overwrites
  -- configure builtin diagnositics
  vim.lsp.handlers['textDocument/publishDiagnostics'] = vim.lsp.with( -- {{{
    vim.lsp.diagnostic.on_publish_diagnostics, {
      virtual_text     = false,
      signs            = true,
      underline        = true,
      update_in_insert = false,
    }
  )
  -- }}}

  -- overwrite codeAction handler to not ask for confirmation if only a single
  -- action is given
  vim.lsp.handlers['textDocument/codeAction'] = function(_, _, actions) -- {{{
    if actions == nil or #actions == 0 then
      -- skip printing 'no code actions available'
      return
    end

    local action_chosen
    if #actions == 1 then
      action_chosen = actions[1]
    else
      local option_strings = {'Code Actions:'}
      for i, action in ipairs(actions) do
        local title = action.title:gsub('\r\n', '\\r\\n')
        title = title:gsub('\n', '\\n')
        table.insert(option_strings, string.format('%d. %s', i, title))
      end

      local choice = vim.fn.inputlist(option_strings)
      if choice < 1 or choice > #actions then
        return
      end
      action_chosen = actions[choice]
    end

    if action_chosen.edit or type(action_chosen.command) == 'table' then
      if action_chosen.edit then
        vim.lsp.util.apply_workspace_edit(action_chosen.edit)
      end
      if type(action_chosen.command) == 'table' then
        vim.lsp.buf.execute_command(action_chosen.command)
      end
    else
      vim.lsp.buf.execute_command(action_chosen)
    end
  end
  -- }}}

  vim.fn.sign_define({
    {name='LspDiagnosticsSignError', text='⚡', numhl='Error'},
    {name='LspDiagnosticsSignWarning', text='⯈', numhl='Warning'},
    {name='LspDiagnosticsSignInformation', text='ℹ'},
    {name='LspDiagnosticsSignHint', text='🞶'}
  })

  lsp.ccls.setup {
    on_attach = my_attach,
    on_exit = my_exit,
    capabilities = caps,
    root_dir = lsp.util.root_pattern('.ccls','meson.build', 'Makefile','.git'),
    init_options = {
      compilationDatabaseDirectory = "build",
      completion = {
        filterAndSort = false,
      }
    }
  }

  lsp.gopls.setup {
    on_attach = my_attach,
    on_exit = my_exit,
    capabilities = caps,
    settings = {
      gopls = {
        -- analyses = {
        --  composites = false
        -- },
        codelenses = {
          test = true,
          tidy = true,
        },
        gofumpt = true,
      }
    }
  }

  lsp.sumneko_lua.setup {
    on_attach = my_attach,
    on_exit = my_exit,
    capabilities = caps,
    cmd = { 'lua-language-server' };
    settings = {
      Lua = {
        runtime = {
          version = 'LuaJIT',
        },
        diagnostics = {
          globals = {'vim'},
        },
        workspace = {
          library = {
            [vim.fn.expand('$VIMRUNTIME/lua')] = true,
            [vim.fn.expand('$VIMRUNTIME/lua/vim/lsp')] = true,
          }
        }
      }
    }
  }

  lsp.rust_analyzer.setup {
    on_attach = my_attach,
    on_exit = my_exit,
    capabilities = caps,
    cmd = { 'env', 'CARGO_TARGET_DIR=/tmp', 'RUST_SRC_PATH=/usr/src/rustlib/library', 'rust-analyzer' },
  }

  lsp.zls.setup {
    on_attach = my_attach,
    on_exit = my_exit,
    capabilities = caps,
  }

  lsp.zk.setup {
    on_attach = my_attach,
    on_exit = my_exit,
    capabilities = caps,
    root_dir = lsp.util.root_pattern('.zk'),
  }

end

return {
  my_attach = my_attach,
  setup = setup,
  format = function(afile)
    local doit = true
    doit = doit and not string.match(afile, "^/home/robert/devel/upstream")
    if doit then
      print('Formatting using LSP:', afile)
      vim.lsp.buf.formatting_seq_sync(nil, 1000)
    end
  end,
}