-- Managed by Nix
-- To install everything run `:PackerSync`

-- https://dev.to/voyeg3r/writing-useful-lua-functions-to-my-neovim-14ki
-- function to remove whitespace and preserve spot on save
function _G.preserve(cmd)
    cmd = string.format('keepjumps keeppatterns execute %q', cmd)
    local original_cursor = vim.fn.winsaveview()
    vim.api.nvim_command(cmd)
    vim.fn.winrestview(original_cursor)
end
vim.cmd([[autocmd BufWritePre,FileWritePre,FileAppendPre,FilterWritePre * :lua preserve('%s/\\s\\+$//ge')]])

-- Bootstrap packer if not installed
local fn = vim.fn
local install_path = fn.stdpath('data')..'/site/pack/packer/start/packer.nvim'
if fn.empty(fn.glob(install_path)) > 0 then
    fn.system({'git', 'clone', '--depth', '1', 'https://github.com/wbthomason/packer.nvim', install_path})
    vim.cmd 'packadd packer.nvim'
end

-- Debug function for printing lua tables
function _G.print_table(mytable)
    for k,v in pairs(mytable) do
        if (type(v) == "table") then
            print(k)
            print_table(v)
        else
            print(k, v)
        end
    end
end


-- Set up spacebar as leader
-- https://icyphox.sh/blog/nvim-lua/
vim.api.nvim_set_keymap('n', '<Space>', '', {})
vim.g.mapleader = ' '
-- Set spaces > tabs, 4 as default
vim.o.expandtab = true
vim.o.tabstop = 4
vim.o.shiftwidth = 0
vim.o.relativenumber = true
-- Distable word wrap
vim.o.wrap = false
vim.o.history = 1000
-- Wildmode show list, complete to first result
vim.o.wildignore = "*/app/cache,*/vendor,*/env,*.pyc,*/venv,*/__pycache__,*/venv"
-- Default split directions
vim.o.splitright = true
vim.o.splitbelow = true
-- set spelling language
vim.o.spelllang = "en_ca"

local highlights = {
    "Search ctermfg=166",
    "DiffAdd    cterm=BOLD ctermfg=NONE ctermbg=22",
    "DiffDelete cterm=BOLD ctermfg=NONE ctermbg=52",
    "DiffChange cterm=BOLD ctermfg=NONE ctermbg=23",
    "DiffText   cterm=BOLD ctermfg=NONE ctermbg=23",
    "Normal guibg=NONE ctermbg=NONE",
    "Search ctermbg=None ctermfg=166",
    "PrimaryBlock   ctermfg=06 ctermbg=NONE",
    "SecondaryBlock ctermfg=06 ctermbg=NONE",
    "Blanks   ctermfg=07 ctermbg=NONE",
    "ColorColumn ctermbg=cyan",
    "IndentBlanklineIndent1 ctermbg=234 ctermfg=NONE",
    "IndentBlanklineIndent2 ctermbg=235 ctermfg=NONE",
}
for i, highlight in ipairs(highlights) do
    vim.cmd('au VimEnter * hi ' .. highlight)
end

-- Along with the highlight definition for ColorColumn above, these options
-- will set colored marks at certain line lengths
vim.cmd [[
au BufEnter *.py let w:m1=matchadd('ColorColumn', '\%81v', 100)
au BufEnter *.py let w:m2=matchadd('Error', '\%121v', 100)
au BufLeave *.py call clearmatches()
]]

-- Custom commands
vim.cmd [[
command! MakeTagsPython !ctags --exclude=venv --exclude=.venv --languages=python --python-kinds=-i -R .
]]

-- When using tab, enter will accept the current selection and close the window
-- instead of inserting a newline
vim.cmd [[
inoremap <expr> <CR> pumvisible() ? "\<C-y>" : "\<CR>"
]]

-- Keymaps
--noh gets rid of highlighted search results
vim.api.nvim_set_keymap('n', '<leader><leader>', ':noh<cr>', { noremap = true })
-- `jj` maps to escape
vim.api.nvim_set_keymap('i', 'jj', '<esc>', { noremap = true })
-- Visually select last copied text
vim.api.nvim_set_keymap('n', 'gp', "`[v`]", { noremap = true })
-- Changelist navigation
vim.api.nvim_set_keymap('n', '_', '[c', { noremap = true })
vim.api.nvim_set_keymap('n', '=', ']c', { noremap = true })
vim.api.nvim_set_keymap('n', '<leader>co', '<cmd>copen<cr>', { noremap = true })
vim.api.nvim_set_keymap('n', '<leader>cc', '<cmd>cclose<cr>', { noremap = true })
-- Visually select line without ending
vim.api.nvim_set_keymap('n', '<leader>v', '^v$h', { noremap = true })
-- Ledger Shortcuts
-- Copy the last entry
vim.api.nvim_set_keymap('n', '<leader>ll', 'G{jV}y}p10l', { noremap = true })
-- Copy the current entry to the bottom, copy date from last entry
vim.api.nvim_set_keymap('n', '<leader>lb', '{jV}yGp10l{{jvEy}jvEpl', { noremap = true })
-- Copy the current entry to the next position
vim.api.nvim_set_keymap('n', '<leader>ln', '{jV}y}p10l', { noremap = true })
-- Jump down from line to replace dollar ammount
vim.api.nvim_set_keymap('n', '<leader>ld', 'j^f$lC', { noremap = true })
-- After searching pull entry to current position
vim.api.nvim_set_keymap('n', '<leader>ly', 'vapy<C-o>p{{jvEy}jvEpl', { noremap = true })
-- Shorcut to insert pudb statements for python
vim.api.nvim_set_keymap('n', '<leader>ep', 'ofrom pudb import set_trace; set_trace()<esc>', { noremap = true })
-- Easy new tab creation
vim.api.nvim_set_keymap('n', '<c-w>t', "<cmd>tabnew<cr>", { noremap = true })
vim.api.nvim_set_keymap('n', '<c-w><c-f>', '<c-w>f<c-w>T', { noremap = true })
vim.api.nvim_set_keymap('n', '<c-w><c-]>', '<c-w>v<c-]><c-w>T', { noremap = true })
vim.api.nvim_set_keymap('n', '<c-w><c-t>', '<c-w>v<c-w>T', { noremap = true })
vim.api.nvim_set_keymap('n', '<c-w><c-w>', '<cmd>w<cr>', { noremap = true })
vim.api.nvim_set_keymap('n', '<leader>ee', '<cmd>e!<cr>', { noremap = true })
vim.api.nvim_set_keymap('n', '<expr> k', [[(v:count > 1 ? "m'" . v:count : '') . 'k']], { noremap = true })
vim.api.nvim_set_keymap('n', '<expr> j', [[(v:count > 1 ? "m'" . v:count : '') . 'j']], { noremap = true })


-- Build a custom status line
local status_line = {
    '%#PrimaryBlock#',
    '%#SecondaryBlock#',
    '%#Blanks#',
    '%f',
    '%m',
    '%=',
    '%#SecondaryBlock#',
    '%l,%c ',
    '%#PrimaryBlock#',
    '%{&filetype}',
}
vim.o.statusline = table.concat(status_line)

-- Packer package manage
return require('packer').startup(function()
    use {
        'wbthomason/packer.nvim',
        config = function()
            vim.api.nvim_set_keymap('n', '<leader>ps', '<cmd>source ~/.config/nvim/init.lua<cr><cmd>PackerSync<cr>', { noremap = true })
        end
    }
    use {
        'arcticicestudio/nord-vim',
        config = function()
            vim.cmd [[
            colorscheme nord
            ]]
        end
    }
    use 'p00f/nvim-ts-rainbow'
    use {
        'nvim-treesitter/nvim-treesitter',
        run = ':TSUpdate',
        config = function()
            require('nvim-treesitter.configs').setup {
                ensure_installed = {
                    "python",
                    "lua",
                    "nix",
                    "json",
                    "bash",
                    "html",
                    "toml",
                },
                highlight = {
                    enable = true,
                },
                indent = {
                    enable = true,
                },
                rainbow = {
                    enable = true,
                    -- I use termcolors but this errors if left blank
                    colors = {
                        "#000000",
                        "#000000",
                        "#000000",
                        "#000000",
                    },
                    termcolors = {
                        'darkblue',
                        'magenta',
                        'yellow',
                        'darkcyan',
                    }
                }
            }
        end,
    }
    use {
        "neovim/nvim-lspconfig",
        requires = {
            {'hrsh7th/nvim-cmp'},
            {'hrsh7th/cmp-nvim-lsp'},
        },
        config = function()
            nvim_lsp = require('lspconfig')
            -- Use an on_attach function to only map the following keys
            -- after the language server attaches to the current buffer
            local on_attach = function(client, bufnr)
                local function buf_set_keymap(...) vim.api.nvim_buf_set_keymap(bufnr, ...) end
                local function buf_set_option(...) vim.api.nvim_buf_set_option(bufnr, ...) end
                -- Enable completion triggered by <c-x><c-o>
                buf_set_option('omnifunc', 'v:lua.vim.lsp.omnifunc')
                -- Mappings.
                local opts = { noremap=true, silent=true }
                -- See `:help vim.lsp.*` for documentation on any of the below functions
                buf_set_keymap('n', 'gD', '<cmd>lua vim.lsp.buf.declaration()<CR>', opts)
                buf_set_keymap('n', 'gd', '<cmd>lua vim.lsp.buf.definition()<CR>', opts)
                buf_set_keymap('n', 'K', '<cmd>lua vim.lsp.buf.hover()<CR>', opts)
                buf_set_keymap('n', 'gi', '<cmd>lua vim.lsp.buf.implementation()<CR>', opts)
                buf_set_keymap('n', '<C-k>', '<cmd>lua vim.lsp.buf.signature_help()<CR>', opts)
                buf_set_keymap('n', '<spnce>wa', '<cmd>lua vim.lsp.buf.add_workspace_folder()<CR>', opts)
                buf_set_keymap('n', '<space>wr', '<cmd>lua vim.lsp.buf.remove_workspace_folder()<CR>', opts)
                buf_set_keymap('n', '<space>wl', '<cmd>lua print(vim.inspect(vim.lsp.buf.list_workspace_folders()))<CR>', opts)
                buf_set_keymap('n', '<space>D', '<cmd>lua vim.lsp.buf.type_definition()<CR>', opts)
                buf_set_keymap('n', '<space>rn', '<cmd>lua vim.lsp.buf.rename()<CR>', opts)
                buf_set_keymap('n', '<space>ca', '<cmd>lua vim.lsp.buf.code_action()<CR>', opts)
                buf_set_keymap('n', 'gr', '<cmd>lua vim.lsp.buf.references()<CR>', opts)
                buf_set_keymap('n', '<space>e', '<cmd>lua vim.lsp.diagnostic.show_line_diagnostics()<CR>', opts)
                buf_set_keymap('n', '[d', '<cmd>lua vim.lsp.diagnostic.goto_prev()<CR>', opts)
                buf_set_keymap('n', ']d', '<cmd>lua vim.lsp.diagnostic.goto_next()<CR>', opts)
                buf_set_keymap('n', '<space>q', '<cmd>lua vim.lsp.diagnostic.set_loclist()<CR>', opts)
                buf_set_keymap('n', '<space>f', '<cmd>lua vim.lsp.buf.formatting()<CR>', opts)
            end

            -- Use a loop to conveniently call 'setup' on multiple servers and
            -- map buffer local keybindings when the language server attaches
            -- local servers = { 'pyls', 'rust_analyzer', 'tsserver' }
            local servers = { 'pyright', 'bashls' }
            -- Add additional capabilities supported by nvim-cmp
            local capabilities = vim.lsp.protocol.make_client_capabilities()
            capabilities = require('cmp_nvim_lsp').update_capabilities(capabilities)
            -- Setup language servers
            for _, lsp in ipairs(servers) do
                nvim_lsp[lsp].setup {
                    on_attach = on_attach,
                    flags = {
                        debounce_text_changes = 150,
                    },
                    capabilities = capabilities,
                }
            end
            -- Set completeopt to have a better completion experience
            vim.o.completeopt = 'menuone,noselect'
            -- nvim-cmp setup
            local cmp = require 'cmp'
            cmp.setup {
              mapping = {
                ['<C-p>'] = cmp.mapping.select_prev_item(),
                ['<C-n>'] = cmp.mapping.select_next_item(),
                ['<C-d>'] = cmp.mapping.scroll_docs(-4),
                ['<C-f>'] = cmp.mapping.scroll_docs(4),
                ['<C-Space>'] = cmp.mapping.complete(),
                ['<C-e>'] = cmp.mapping.close(),
                ['<CR>'] = cmp.mapping.confirm {
                  behavior = cmp.ConfirmBehavior.Replace,
                  select = true,
                },
                ['<Tab>'] = function(fallback)
                  if cmp.visible() then
                    cmp.select_next_item()
                  else
                    fallback()
                  end
                end,
                ['<S-Tab>'] = function(fallback)
                  if cmp.visible() then
                    cmp.select_prev_item()
                  else
                    fallback()
                  end
                end,
              },
              sources = {
                { name = 'nvim_lsp' },
              },
            }
        end,
    }
    use 'nvim-treesitter/playground'
    use 'tpope/vim-commentary'
    use 'tpope/vim-surround'
    use 'tpope/vim-repeat'
    use {
        'junegunn/fzf.vim',
        requires = { {'junegunn/fzf'} },
        config = function()
            vim.api.nvim_set_keymap('n', '<leader>ff', "<cmd>Files<cr>", { noremap = true })
            vim.api.nvim_set_keymap('n', '<leader>fb', "<cmd>Buffers<cr>", { noremap = true })
            vim.api.nvim_set_keymap('n', '<leader>ft', "<cmd>Tags<cr>", { noremap = true })
            vim.api.nvim_set_keymap('n', '<leader>fm', "<cmd>Marks<cr>", { noremap = true })
            vim.cmd([[let $FZF_DEFAULT_COMMAND = 'find . -type f -not -path "*/\.git/*" -not -path "*/\.mypy_cache/*" -not -path "*/\.venv/*" ']])
        end,
    }
    use {
        'FooSoft/vim-argwrap',
        config = function()
            vim.api.nvim_set_keymap('n', '<leader>ew', '<cmd>ArgWrap<cr>', { noremap = true })
        end
    }
    use {
        'tpope/vim-fugitive',
        config = function()
            vim.api.nvim_set_keymap('n', '<leader>du', '<cmd>diffupdate<cr>', { noremap = true })
            vim.api.nvim_set_keymap('n', '<leader>dd', '<cmd>diffget<cr>', { noremap = true })
            vim.api.nvim_set_keymap('n', '<leader>df', '<cmd>diffput<cr>', { noremap = true })
            vim.api.nvim_set_keymap('n', '<leader>gs', '<cmd>Git<cr>', { noremap = true })

        end
    }
    use {
        'lukas-reineke/indent-blankline.nvim',
        config = function()
            require("indent_blankline").setup {
                char = " ",
                char_highlight_list = {
                    "IndentBlanklineIndent1",
                    "IndentBlanklineIndent2",
                },
                show_trailing_blankline_indent = false,
            }
        end,
    }
    use {
        'francoiscabrol/ranger.vim',
        -- bclose gets rid of "[Process exited 0]" annoyance
        requires = { {'rbgrouleff/bclose.vim'} },
    }
    use {
        'ptzz/lf.vim',
        -- Adds a floating window
        requires = { {'voldikss/vim-floaterm'} },
        config = function()
            vim.g.floaterm_opener = "edit"
            vim.g.floaterm_width = 0.99
            vim.g.floaterm_height = 0.99
            -- vim.g.lf_replace_netrw = 1;
            vim.api.nvim_set_keymap('n', '<leader>m', '<cmd>Lf<cr>', { noremap = true });
            vim.api.nvim_set_keymap('n', '<leader>n', '<cmd>LfWorkingDirectory<cr>', { noremap = true});
        end
    }
    use 'bkad/CamelCaseMotion'
    use {
        'phaazon/hop.nvim',
        as = 'hop',
        config = function()
            -- you can configure Hop the way you like here; see :h hop-config
            vim.api.nvim_set_keymap('', '<leader>s', '<cmd>HopChar2<cr>', { noremap = true })
            require'hop'.setup { keys = 'etovxqpdygfblzhckisuran' }
        end
    }
    -- The treesitter for nix doesn't do indents
    use 'LnL7/vim-nix'
    use {
        'dense-analysis/ale',
        config = function()
            vim.g.ale_linters = {
                shell = { "shellcheck", },
            }
        end
    }
    use 'junegunn/goyo.vim'
    use 'hashivim/vim-terraform'
end)