local set = vim.keymap.set
---@param mode string|table
---@param keymap string
local function unbind(mode, keymap)
    set(mode, keymap, function() end)
end

local function toggle_zen()
    vim.cmd('Neotree close')
    vim.cmd('Trouble todo close')
    vim.cmd('UndotreeHide')
end

local function yank_filepath_to_clipboard(absolute)
    return function()
        local filepath = vim.fn.expand('%:p')
        if not absolute then
            local home_dir = vim.fn.expand('~')
            filepath = filepath:gsub(home_dir, '~')
        end
        vim.fn.setreg('+', filepath)
    end
end

local function paste()
    local value = vim.fn.getreg('+')
    if vim.fn.mode() == 'c' then
        -- value = vim.split(value, '\n')[1] -- only use first line (default with normal pasting)
        value = value:gsub('\n', ' ') -- replace newlines with spaces
        vim.api.nvim_feedkeys(value, 'c', false)
    else
        vim.api.nvim_put(vim.split(value, '\n'), 'c', false, true)
    end
end

-- "Greatest Keymap ever" - ThePrimeagen
-- https://www.youtube.com/watch?v=qZO9A5F6BZs&list=PLm323Lc7iSW_wuxqmKx_xxNtJC_hJbQ7R&index=4
set('x', '<leader>p', '"_dP')

set('n', '!', ':!')

-- Move lines around
set('n', '<A-j>', ':m .+1<cr>', { noremap = true })
set('n', '<A-k>', ':m .-2<cr>', { noremap = true })
set('v', '<A-j>', ":m '>+1<cr>gv", { noremap = true })
set('v', '<A-k>', ":m '<-2<cr>gv", { noremap = true })

-- Misc
set({ 'n', 'i' }, '<C-a>', '<Esc>ggVG', { desc = 'Visually Highlight [A]ll' })
set('i', '<C-H>', '<C-w>', { desc = 'Ctrl + Backspace to delete word' })
set('n', '<esc>', '<cmd>nohlsearch<cr>', { silent = true })
set({ 'i', 'c' }, '<C-v>', paste)
set('n', '<leader>z', toggle_zen, { desc = '[Z]en Mode', silent = true })
set('n', '<leader><tab>', '<cmd>e #<cr>', { desc = 'Switch to Other Buffer' })
unbind({ 'i', 'n', 'v' }, '<C-r>')

set('n', 'U', '<cmd>redo<cr>')
set('i', '<C-z>', '<cmd>undo<cr>')
set({ 'i', 'n', 'v' }, '<C-q>', '<cmd>wqa<cr>')
set('n', '<leader>yp', yank_filepath_to_clipboard(true), { desc = '[Y]oink File [P]ath (linux)' })
set('n', '<leader>yP', yank_filepath_to_clipboard(false), { desc = '[Y]oink File [P]ath (windows)' })

-- Terminal
set('t', '<C-t>', '<cmd>close<cr>', { desc = 'Hide Terminal' })

-- Window resizing
local resize = 2
set({ 'n', 't' }, '<C-Left>', string.format('<cmd>vertical resize -%d<CR>', resize), { desc = 'Resize Left' })
set({ 'n', 't' }, '<C-Up>', string.format('<cmd>resize -%d<CR>', resize), { desc = 'Resize Up' })
set({ 'n', 't' }, '<C-Down>', string.format('<cmd>resize +%d<CR>', resize), { desc = 'Resize Down' })
set({ 'n', 't' }, '<C-Right>', string.format('<cmd>vertical resize +%d<CR>', resize), { desc = 'Resize Right' })

-- highlights under cursor
set('n', '<leader>ui', vim.show_pos, { desc = 'Inspect Pos' })
set('n', '<leader>uI', function()
    vim.treesitter.inspect_tree()
    vim.api.nvim_input('I')
end, { desc = 'Inspect Tree' })

-- Keep selection after < and > in visual mode
set('v', '<', '<gv')
set('v', '>', '>gv')
-- Execute lua file or line
set('n', '<leader>x', '<cmd>.lua<CR>', { desc = 'Execute the current line' })
set('n', '<leader><leader>x', '<cmd>source %<CR>', { desc = 'Execute the current file' })

-- Reformats file using lsp
set('n', '=', vim.lsp.buf.format, { desc = 'Format File' })

-- LSP keymaps
local severity = vim.diagnostic.severity
local function next_diagnostic(diagnostic_severity)
    return function()
        require('lspsaga.diagnostic'):goto_next({ severity = diagnostic_severity })
    end
end
local function prev_diagnostic(diagnostic_severity)
    return function()
        require('lspsaga.diagnostic'):goto_prev({ severity = diagnostic_severity })
    end
end
local function cursor_diagnostics()
    vim.diagnostic.open_float({
        scope = 'cursor',
        border = 'single',
    })
end

local keymaps = {
    { 'n', 'gd', '<cmd>Lspsaga goto_definition<cr>', { desc = '[G]oto [D]efinition' } },
    { 'n', 'gu', '<cmd>Lspsaga finder<cr>', { desc = '[G]oto [U]sages' } },
    { { 'n', 'i' }, '<C-p>', vim.lsp.buf.signature_help, { desc = 'Show [P]arameters' } },
    { 'n', 'K', '<cmd>Lspsaga hover_doc<cr>', { desc = 'Documentation' } },
    { 'n', 'R', '<cmd>Lspsaga rename<cr>', { desc = '[R]ename' } },
    { { 'n', 'i' }, '<M-Enter>', '<cmd>Lspsaga code_action<cr>', { desc = 'Code Actions' } },
    { 'n', '<leader>e', next_diagnostic(severity.ERROR), { desc = 'Goto [E]rror' } },
    { 'n', '<leader>E', prev_diagnostic(severity.ERROR), { desc = 'Goto [E]rror (prev)' } },
    { 'n', '<leader>w', next_diagnostic(severity.WARN), { desc = 'Goto [W]arning' } },
    { 'n', '<leader>W', prev_diagnostic(severity.WARN), { desc = 'Goto [W]arning (prev)' } },
    { 'n', '<leader>D', cursor_diagnostics, { desc = '[D]iagnostics under cursor' } },
}

local function set_keymaps(tbl, bufnr)
    for _, keymap in ipairs(tbl) do
        local mode = keymap[1]
        local lhs = keymap[2]
        local rhs = keymap[3]
        local opts = keymap[4] or {}
        opts.buffer = bufnr
        vim.keymap.set(mode, lhs, rhs, opts)
    end
end

vim.api.nvim_create_autocmd('LspAttach', {
    desc = 'Register LSP keymaps',
    group = vim.api.nvim_create_augroup('lsp-keymaps', { clear = true }),
    callback = function(args)
        local bufnr = args.buf

        local filetype = vim.bo[bufnr].filetype
        set_keymaps(keymaps, bufnr)
        for ft, tbl in pairs(keymaps) do
            if type(ft) == 'number' or ft ~= filetype then
                goto continue
            end
            set_keymaps(tbl, bufnr)
            ::continue::
        end
    end,
})