-- Leave insert mode with jj
vim.keymap.set("i", "jj", "<esc>")

-- Redo with U
vim.keymap.set("n", "U", "<C-r>")

-- Navigate vim panes better
vim.keymap.set("n", "<c-k>", ":wincmd k<CR>")
vim.keymap.set("n", "<c-j>", ":wincmd j<CR>")
vim.keymap.set("n", "<c-h>", ":wincmd h<CR>")
vim.keymap.set("n", "<c-l>", ":wincmd l<CR>")

-- Which-key Mappings
local wk = require("which-key")
local harpoon = require("harpoon")

wk.add({
  { "<leader>;", "<cmd>Alpha<CR>", desc = "Dashboard", icon = "󰕮" },

  { "<leader>w", "<cmd>w!<CR>", desc = "Save", icon = "" },
  { "<leader>q", "<cmd>confirm q<CR>", desc = "Quit", icon = { icon = "", color = "red" } },

  { "<leader>e", "<cmd>Neotree toggle<CR>", desc = "Toggle Navigation", icon = "" },


  { "<leader>b", group = "Buffers", icon = "" },
  { "<leader>bm", "<cmd>Neotree buffers reveal float<cr>", desc = "Buffer Menu" },
  { "<leader>bp", function() require("cokeline.mappings").pick("focus") end, desc = "Buffer Picker" },

  { "<leader>n", group = "Notifications", icon = "󰆈" },
  { "<leader>nd", "<cmd>NoiceDismiss<CR>", desc = "Dismiss Messages" },

  { "<leader>h", group = "Harpoon", icon = "󰛢" },
  { "<leader>ha", function() harpoon:list():add() end, desc = "Add buffer", icon = "" },
  { "<leader>he", function() harpoon.ui:toggle_quick_menu(harpoon:list()) end, desc = "Open menu" },
  { "<leader>h1", function() harpoon:list():select(1) end, desc = "Buffer 1" },
  { "<leader>h2", function() harpoon:list():select(2) end, desc = "Buffer 2" },
  { "<leader>h3", function() harpoon:list():select(3) end, desc = "Buffer 3" },
  { "<leader>h4", function() harpoon:list():select(4) end, desc = "Buffer 4" },
  { "<leader>hn", function() harpoon:list():next() end, desc = "Next buffer" },
  { "<leader>hp", function() harpoon:list():prev() end, desc = "Prev buffer" },

  { "<leader>c", group = "Code" },
  { "<leader>ca", "<cmd>lua vim.lsp.buf.code_action()<cr>", desc = "Code Action" },
  { "<leader>cd", "<cmd>lua vim.lsp.buf.definition()<cr>", desc = "Code Definition" },
  { "<leader>cf", "<cmd>lua vim.lsp.buf.format()<cr>", desc = "Code Format" },
  { "<leader>ci", "<cmd>Telescope diagnostics<cr>", desc = "Diagnostics" },
  { "<leader>cj", "<cmd>lua vim.diagnostic.jump({count=1, float=true})<cr>", desc = "Next Diagnostics" },
  { "<leader>cr", "<cmd>lua vim.lsp.buf.references()<cr>", desc = "Code References" },

  { "<leader>f", group = "File" },
  { "<leader>ff", "<cmd>Telescope find_files<cr>", desc = "Find File", icon = "" },
  { "<leader>fg", "<cmd>Telescope live_grep<cr>", desc = "Find Word", icon = "󰱼" },

  { "<leader>t", group = "Text", icon = "󰊄" },
  { "<leader>ts", "<cmd>setlocal spell!<cr>", desc = "Toggle Spell Check" },
  { "<leader>tw", "<cmd>setlocal wrap!<cr>", desc = "Toggle Line Wrap" },
  { "<leader>th", "<cmd>nohlsearch<CR>", desc = "Clear Search highlight" },

  { "<leader>d", group = "Debug" },
  { "<leader>dC", "<cmd>lua require'dap'.run_to_cursor()<cr>", desc = "Run To Cursor" },
  { "<leader>dU", "<cmd>lua require'dapui'.toggle({reset = true})<cr>", desc = "Toggle UI" },
  { "<leader>db", "<cmd>lua require'dap'.step_back()<cr>", desc = "Step Back" },
  { "<leader>dc", "<cmd>lua require'dap'.continue()<cr>", desc = "Continue" },
  { "<leader>dd", "<cmd>lua require'dap'.disconnect()<cr>", desc = "Disconnect" },
  { "<leader>dg", "<cmd>lua require'dap'.session()<cr>", desc = "Get Session" },
  { "<leader>di", "<cmd>lua require'dap'.step_into()<cr>", desc = "Step Into" },
  { "<leader>do", "<cmd>lua require'dap'.step_over()<cr>", desc = "Step Over" },
  { "<leader>dp", "<cmd>lua require'dap'.pause()<cr>", desc = "Pause" },
  { "<leader>dq", "<cmd>lua require'dap'.close()<cr>", desc = "Quit" },
  { "<leader>dr", "<cmd>lua require'dap'.repl.toggle()<cr>", desc = "Toggle Repl" },
  { "<leader>ds", "<cmd>lua require'dap'.continue()<cr>", desc = "Start" },
  { "<leader>dt", "<cmd>lua require'dap'.toggle_breakpoint()<cr>", desc = "Toggle Breakpoint" },
  { "<leader>du", "<cmd>lua require'dap'.step_out()<cr>", desc = "Step Out" },

  -- UI shortcuts
  { ",a", "<cmd>Alpha<CR>", desc = "Dashboard", icon = "󰕮" },
  { ",b", "<cmd>Neotree buffers reveal float<cr>", desc = "Buffer Menu", icon = "" },
  { ",e", "<cmd>Neotree toggle<CR>", desc = "Toggle Navigation", icon = "" },
  { ",f", "<cmd>Telescope find_files<cr>", desc = "Find File", icon = "" },
  { ",g", "<cmd>Telescope live_grep<cr>", desc = "Find Word", icon = "󰱼" },
  { ",h", function() harpoon.ui:toggle_quick_menu(harpoon:list()) end, desc = "Harpoon", icon = "󰛢" },
  { ",l", "<cmd>Lazy<cr>", desc = "Lazy", icon = "󰒲" },
  { ",m", "<cmd>Mason<cr>", desc = "Mason", icon = "󱌣" },
  { ",t", "<cmd>TSManager<cr>", desc = "Treesitter Manager", icon = "" },
})