local harpoon = require('harpoon')
harpoon:setup()

vim.keymap.set('n', '<M-1>', function() harpoon:list():select(1) end)
vim.keymap.set('n', '<M-2>', function() harpoon:list():select(2) end)
vim.keymap.set('n', '<M-3>', function() harpoon:list():select(3) end)
vim.keymap.set('n', '<M-4>', function() harpoon:list():select(4) end)

local ts = require('telescope.builtin')
require('telescope').load_extension('harpoon')
local wk = require('which-key')
wk.register({
	-- harpoon
	a = {
		name = 'harpoon',
		a = { function() harpoon:list():append() end, 'add file',                        noremap = true, silent = true },
		e = { function() harpoon.ui:toggle_quick_menu(harpoon:list()) end, 'harpoon ui', noremap = true, silent = true },
		f = { ':Telescope harpoon marks<CR>', 'harpoon telescope ui',                    noremap = true, silent = true },
	},
	d = { ':bd<CR>', 'close buffer', noremap = true, silent = true },
	D = { '"-ddk:put = strftime(\' * %a %b %d %T %Y\')<CR>', 'insert date', noremap = true, silent = true },
	h = { ':Startify<CR>', 'startify', noremap = true, silent = true },
	k = { ':w<CR>:bp<CR>:bd #<CR>', 'write and close buffer', noremap = true, silent = false },
	u = { vim.cmd.UndotreeToggle, 'undotree', noremap = true, silent = true },
	w = { ':w<CR>', 'write buffer', noremap = true, silent = false },
	-- trouble
	x = {
		name = 'trouble',
		x = { vim.cmd.TroubleToggle, 'toggle', noremap = true, silent = true },
		w = { ':TroubleToggle workspace_diagnostics<CR>', 'workspace', noremap = true, silent = true },
		d = { ':TroubleToggle document_diagnostics<CR>', 'document', noremap = true, silent = true },
		l = { ':TroubleToggle loclist<CR>', 'local list', noremap = true, silent = true },
		q = { ':TroubleToggle quickfix<CR>', 'quickfix', noremap = true, silent = true },
		r = { ':TroubleToggle lsp_references<CR>', 'lsp', noremap = true, silent = true },
	},
	-- telescope
	f = {
		name = 'telescope',
		b = { ts.buffers, 'buffers', noremap = true, silent = true },
		d = { ts.diagnostics, 'diagnostics', noremap = true, silent = true },
		e = { ':Telescope harpoon marks<CR>', 'harpoon marks', noremap = true, silent = true },
		f = { ts.find_files, 'find files', noremap = true, silent = true },
		g = { ts.git_files, 'git files', noremap = true, silent = true },
		s = { function()
			ts.grep_string({ search = vim.fn.input("Grep > ") })
		end, 'grep string', noremap = true, silent = true },
		t = { ':TodoTelescope keywords=TODO,HACK,WARN,PERF,NOTE,TEST<CR>', 'todo', noremap = true, silent = true }
	},
	-- git
	g = {
		name = 'git',
		g = { vim.cmd.Git, 'git' },
		l = { ':FloatermNew lazygit<CR>', 'lazygit', noremap = true, silent = true },
		-- a = { ':Ag<CR>', 'the_silver_searcher', noremap = true, silent = true },
		-- g = { ':Grep<CR>', 'grep', noremap = true, silent = true },
		-- r = { ':Rg<CR>', 'ripgrep', noremap = true, silent = true },
	},
	-- neovim
	v = {
		name = 'nvim',
		v = { vim.cmd.Ex, 'explorer', noremap = true, silent = true },
		u = { vim.cmd.PackerSync, 'upgrade', noremap = true, silent = true },
	},
}, { prefix = '<leader>', })
wk.setup({
	plugins = {
		presets = {
			operators = false,
			motions = false,
			text_objects = false,
			window = false,
			nav = false,
			z = false,
			g = false
		},
	},
	triggers = { "<leader>" },
	triggers_nowait = { "d" },
})