--
-- Neovim dotfiles
--
--
local execute = vim.api.nvim_command
local fn = vim.fn

-- Bootstrap package manager
local install_path = fn.stdpath('data') .. '/site/pack/packer/opt/packer.nvim'

if fn.empty(fn.glob(install_path)) > 0 then
  execute('!git clone https://github.com/wbthomason/packer.nvim ' .. install_path)
end

-- Packer configuration is compiled and only needs to be loaded on changes
vim.cmd 'packadd packer.nvim'

require('packer').startup(function()
  use { 'wbthomason/packer.nvim', opt = true }

  -- General plugins
  use { 'tpope/vim-sensible' }
  use { 'tpope/vim-repeat' }
  use { 'tpope/vim-rsi' }
  use { 'sgur/vim-editorconfig' }
  use { 'ShikChen/osc52.vim' }
  use { 'einfachtoll/didyoumean' }

  use { 'tpope/vim-obsession' }
  use {
    'tpope/vim-eunuch',
    cmd = {
      'Remove',
      'Unlink',
      'Move',
      'Rename',
      'Mkdir',
      'Chmod',
      'Find',
      'Locate',
      'SudoEdit',
      'SudoWrite',
    },
  }

  use { 'psliwka/vim-smoothie' }

  -- Spelling/autocorrection
  use { 'tpope/vim-abolish' }

  -- Git/VCS
  use { 'vim-scripts/gitignore' }
  use { 'sjl/splice.vim', opt = true, cmd = { 'SpliceInit' } }
  use { 'tpope/vim-git' }

  -- Comments
  use { 'b3nj5m1n/kommentary' }

  -- Undoing
  use { 'sjl/gundo.vim', cmd = { 'GundoToggle' } }

  -- Parentheses etc
  use { 'tpope/vim-surround' }
  use { 'raimondi/delimitMate' }

  -- Moving around within lines
  use { 'wellle/targets.vim', event = 'InsertEnter *' }

  -- Peek at lines
  use { 'nacro90/numb.nvim', config = function() require'numb'.setup() end }

  -- Searching
  use { 'mhinz/vim-grepper', cmd = { 'Grepper' } }

  -- Keymaps TODO: to be removed when #13823 is merged
  use { 'tjdevries/astronauta.nvim', config = function() require 'astronauta.keymap' end }

  -- Indent lines
  use {
    'lukas-reineke/indent-blankline.nvim',
    branch = 'lua',
    config = function()
      vim.g.indent_blankline_buftype_exclude = { 'terminal', 'help', 'nofile' }
      vim.g.indent_blankline_show_first_indent_level = false
    end,
  }

  -- Increment/decrement
  use {
    'zegervdv/nrpattern.nvim',
    requires = 'tpope/vim-repeat',
    config = function()
      local nrpattern = require 'nrpattern'
      local defaults = require 'nrpattern.default'

      defaults[{ 'input', 'output' }] =
          { priority = 12, filetypes = { 'verilog', 'systemverilog' } }
      defaults[{ '\'1', '\'0' }] = { priority = 9, filetypes = { 'verilog', 'systemverilog' } }

      nrpattern.setup(defaults)
    end,
  }

  -- Tmux
  function test_tmux() return os.getenv('TMUX') ~= nil end
  use { 'tmux-plugins/vim-tmux-focus-events', cond = test_tmux }
  use {
    'numtostr/navigator.nvim',
    config = function()
      require('Navigator').setup { auto_save = 'current', disable_on_zoom = true }

      local nnoremap = vim.keymap.nnoremap
      nnoremap { '<c-h>', require'Navigator'.left, silent = true }
      nnoremap { '<c-j>', require'Navigator'.down, silent = true }
      nnoremap { '<c-k>', require'Navigator'.up, silent = true }
      nnoremap { '<c-l>', require'Navigator'.right, silent = true }
    end,
    requires = 'tjdevries/astronauta.nvim',
  }

  -- Completion/snippets/LSP
  use { 'neovim/nvim-lspconfig' }
  use {
    'hrsh7th/nvim-compe',
    config = function()
      require'compe'.setup {
        enabled = true,
        autocomplete = true,
        debug = false,
        min_length = 1,
        preselect = 'enable',
        throttle_time = 80,
        source_timeout = 1000,
        incomplete_delay = 400,
        max_abbr_width = 100,
        max_kind_width = 100,
        max_menu_width = 100,
        documentation = true,

        source = {
          path = true,
          buffer = true,
          nvim_lsp = true,
          nvim_lua = true,
          spell = true,
          ultisnips = true,
          -- TODO add vsnip for LSP snippets
        },
      }

      vim.cmd [[ inoremap <silent><expr> <C-y> compe#complete() ]]
      vim.cmd [[ inoremap <silent><expr> <CR>      compe#confirm({ 'keys': "\<Plug>delimitMateCR", 'mode': '' }) ]]
      vim.cmd [[ inoremap <silent><expr> <C-e>     compe#close('<C-e>') ]]
    end,
  }
  use {
    {
      'nvim-treesitter/nvim-treesitter',
      config = function()
        require 'nvim-treesitter.highlight'

        require'nvim-treesitter.configs'.setup {
          highlight = { enable = false },
          incremental_selection = {
            enable = true,
            keymaps = {
              init_selection = 'gnn',
              node_incremental = 'grn',
              scope_incremental = 'grc',
              node_decremental = 'grm',
            },
          },
          refactor = {
            highlight_definitions = { enable = true },
            smart_rename = { enable = true, keymaps = { smart_rename = 'gsr' } },
            navigation = {
              enable = true,
              keymaps = { goto_definition = 'gnd', list_definitions = 'gnD' },
            },
          },
          textobjects = {
            move = {
              enable = true,
              goto_next_start = { [']]'] = '@block.outer' },
              goto_previous_start = { ['[['] = '@block.outer' },
              goto_next_end = { [']['] = '@block.outer' },
              goto_previous_end = { ['[]'] = '@block.outer' },
            },
          },
          playground = { enable = true, disable = {}, updatetime = 25, persist_queries = false },
        }
      end,
    },
    'nvim-treesitter/nvim-treesitter-refactor',
    'nvim-treesitter/nvim-treesitter-textobjects',
    { 'nvim-treesitter/playground', opt = true },
  }
  use { 'SirVer/ultisnips' }
  use { 'glepnir/lspsaga.nvim', config = function() require'lspsaga'.init_lsp_saga {} end }

  -- Vanity
  use {
    'yamatsum/nvim-web-nonicons',
    requires = 'kyazdani42/nvim-web-devicons',
    config = function() require 'nvim-nonicons' end,
  }

  use {
    'glepnir/galaxyline.nvim',
    branch = 'main',
    -- your statusline
    config = function()
      local gl = require 'galaxyline'
      local colors = require('galaxyline.theme').default
      local condition = require('galaxyline.condition')
      local gls = gl.section

      colors.bg = '#2C323C'

      gls.left[1] = {
        RainbowRed = {
          provider = function() return '▊ ' end,
          highlight = { colors.blue, colors.bg },
        },
      }

      gls.left[2] = {
        FileIcon = {
          provider = 'FileIcon',
          condition = condition.buffer_not_empty,
          highlight = { require('galaxyline.provider_fileinfo').get_file_icon_color, colors.bg },
        },
      }

      gls.left[3] = {
        FileName = {
          provider = 'FileName',
          condition = condition.buffer_not_empty,
          highlight = { colors.magenta, colors.bg, 'bold' },
        },
      }

      gls.left[4] = {
        DiagnosticError = {
          provider = 'DiagnosticError',
          icon = '  ',
          highlight = { colors.red, colors.bg },
        },
      }

      gls.right[1] = {
        ShowLspClient = {
          provider = 'GetLspClient',
          condition = function()
            local tbl = { ['dashboard'] = true, [''] = true }
            if tbl[vim.bo.filetype] then return false end
            return true
          end,
          icon = require'nvim-nonicons'.get('server') .. ' LSP:',
          highlight = { colors.green, colors.bg, 'bold' },
        },
      }

      gls.right[2] = {
        LineInfo = {
          provider = 'LineColumn',
          separator = ' ',
          separator_highlight = { 'NONE', colors.bg },
          highlight = { colors.fg, colors.bg },
        },
      }

      gls.right[3] = {
        PerCent = {
          provider = 'LinePercent',
          separator = ' ',
          separator_highlight = { 'NONE', colors.bg },
          highlight = { colors.fg, colors.bg, 'bold' },
        },
      }
      gls.right[8] = {
        RainbowBlue = {
          provider = function() return ' ▊' end,
          highlight = { colors.blue, colors.bg },
        },
      }

      gls.short_line_left[1] = {
        BufferType = {
          provider = 'FileTypeName',
          separator = ' ',
          separator_highlight = { 'NONE', colors.bg },
          highlight = { colors.blue, colors.bg, 'bold' },
        },
      }

      gls.short_line_left[2] = {
        SFileName = {
          provider = 'SFileName',
          condition = condition.buffer_not_empty,
          highlight = { colors.fg, colors.bg, 'bold' },
        },
      }

      gls.short_line_right[1] = {
        BufferIcon = { provider = 'BufferIcon', highlight = { colors.fg, colors.bg } },
      }
    end,
  }

  -- File navigation
  use { 'justinmk/vim-dirvish' }

  -- Colorscheme
  use {
    'zegervdv/one-lush',
    requires = 'rktjmp/lush.nvim',
    config = function()
      require 'lush_theme.one-lush'
      vim.cmd [[ colorscheme one-lush ]]
    end,
  }

  -- Terminal
  use {
    'akinsho/nvim-toggleterm.lua',
    config = function()
      require'toggleterm'.setup {
        size = 15,
        open_mapping = [[<F12>]],
        shade_filetypes = { 'none' },
        shade_terminals = true,
        persist_size = true,
        direction = 'horizontal',
      }
    end,
  }

  -- Filetypes
  use { 'Glench/Vim-Jinja2-Syntax' }

end)

-- LSP config
local lsp = require 'lspconfig'
local lsputil = require 'lspconfig.util'

local on_attach = function(client)
  local nnoremap = vim.keymap.nnoremap
  local inoremap = vim.keymap.inoremap
  nnoremap { '<CR>', require'lspsaga.diagnostic'.show_line_diagnostics, silent = true }
  nnoremap { 'gd', vim.lsp.buf.declaration, silent = true }
  nnoremap { '<c-]>', vim.lsp.buf.definition, silent = true }
  nnoremap { 'K', require'lspsaga.hover'.render_hover_doc, silent = true }
  nnoremap { 'gD', vim.lsp.buf.implementation, silent = true }
  nnoremap { '1gD', vim.lsp.buf.type_definition, silent = true }
  nnoremap { 'gr', vim.lsp.buf.references, silent = true }
  nnoremap { 'g0', vim.lsp.buf.document_symbol, silent = true }
  nnoremap { '<c-p>', vim.lsp.buf.formatting, silent = true }
  nnoremap { 'gp', require'lspsaga.provider'.preview_definition, silent = true }

  inoremap { '<c-l>', vim.lsp.buf.signature_help, silent = true }

  vim.cmd [[ sign define LspDiagnosticsSignError text=▊ texthl=LspDiagnosticsSignError linehl= numhl= ]]
  vim.cmd [[ sign define LspDiagnosticsSignWarning text=▊ texthl=LspDiagnosticsSignWarning linehl= numhl= ]]
  vim.cmd [[ sign define LspDiagnosticsSignInformation text=▊ texthl=LspDiagnosticsSignInformation linehl= numhl= ]]
  vim.cmd [[ sign define LspDiagnosticsSignHint text=▊ texthl=LspDiagnosticsSignHint linehl= numhl= ]]
end

vim.lsp.handlers['textDocument/formatting'] = function(err, _, result, _, bufnr)
  if err ~= nil or result == nil then return end
  if not vim.api.nvim_buf_get_option(bufnr, 'modified') then
    local view = vim.fn.winsaveview()
    vim.lsp.util.apply_text_edits(result, bufnr)
    -- Fix to reload Treesitter
    -- vim.api.nvim_command("edit")
    vim.fn.winrestview(view)
  end
end

vim.lsp.handlers['textDocument/publishDiagnostics'] =
    function(...)
      vim.lsp.with(vim.lsp.diagnostic.on_publish_diagnostics,
                   { underline = true, update_in_insert = false })(...)
    end

lsp.pyright.setup { on_attach = on_attach }

if (vim.fn.executable('efm-langserver') == 1) then
  require 'efm/python'
  require 'efm/lua'

  -- May not be installed, use pcall to handle errors
  -- pcall(require, 'efm/systemverilog')
  pcall(require, 'efm/flp')

  local language_cfg = require 'efm/languages'

  local filetypes = {}
  for lang, _ in pairs(language_cfg) do table.insert(filetypes, lang) end

  lsp.efm.setup {
    on_attach = on_attach,
    filetypes = filetypes,
    init_options = { documentFormatting = true },
    root_dir = lsputil.root_pattern('.git', '.hg'),
    settings = { rootMarkers = { '.git/', '.hg/' }, languages = language_cfg },
  }
end

-- Try importing local config
local ok, localconfig = pcall(require, 'localconfig')
if ok then localconfig.setup { on_attach = on_attach } end