-- Custom keymaps

require("utils.functions")
require("custom.uni")

local season = "S2"

local links = require("utils.linker")
local user = vim.fn.system('whoami'):gsub('\n', '')
local api = require("nvim-tree.api")
local builtin = require('telescope.builtin')

local function open_cal()
	local current_date = os.date("%Y-%m-%d")
	local week_number = os.date("%V")
	local day_of_week = os.date("%a")
	local path = "~/synced/brainstore/calendar/calendar_" .. os.date("%Y") .. ".txt"
	local keys = ":e " ..
		path .. "<CR>/" .. current_date .. " w" .. tonumber(week_number) .. " " .. day_of_week .. "<CR>$"
	vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes(keys, true, false, true), 'n', true)
end

-------------------------------------------------------
--------------------- KEYMAPS -------------------------
-------------------------------------------------------

-- Fast quitter
vim.keymap.set('n', '<leader>q', function()
	pcall(function()
		vim.cmd('wa')
	end)
	vim.cmd('qa!')
end)

-- Fast window switch
vim.keymap.set("n", "<leader>w", "<C-w>w")

--------------------- NORMAL -------------------------

vim.keymap.set("n", "L", ":BufferNext<CR>", { silent = true })
vim.keymap.set("n", "n", "nzz", { silent = true })
vim.keymap.set("n", "N", "Nzz", { silent = true })
vim.keymap.set("n", "H", ":BufferPrevious<CR>", { silent = true })
vim.keymap.set("n", "<C-o>", "<C-o>zz", { silent = true })
vim.keymap.set("n", "<C-i>", "<C-i>zz", { silent = true })
vim.keymap.set("n", "<C-u>", "<C-u>zz", { silent = true })
vim.keymap.set("n", "<C-d>", "<C-d>zz", { silent = true })

vim.keymap.set('n', '<leader>a', 'm9ggVG"+y`9')
vim.keymap.set('n', '<leader>va', 'ggVG')

-- Launch panel if nothing is typed after <leader>z
vim.keymap.set("n", "<leader>z", "<cmd>Telekasten panel<CR>")

-- Most used functions
vim.keymap.set("n", "<leader>zf", "<cmd>Telekasten find_notes<CR>")
vim.keymap.set("n", "<leader>zg", "<cmd>Telekasten search_notes<CR>")
vim.keymap.set('n', '<leader>zq', ':e ~/synced/brainstore/zettelkasten/input.txt<CR>`.zz')
vim.keymap.set("n", "<leader>zd", "<cmd>Telekasten goto_today<CR>")
vim.keymap.set("n", "<leader>zr", "<cmd>Telekasten rename_note<CR>")
vim.keymap.set("n", "<leader>zz", "<cmd>Telekasten follow_link<CR>")
vim.keymap.set("n", "<leader>zn", "<cmd>Telekasten new_note<CR>")
vim.keymap.set("n", "<leader>zb", "<cmd>Telekasten show_backlinks<CR>")
vim.keymap.set("n", "<leader>zw", "<cmd>Telekasten find_weekly_notes<CR>")
vim.keymap.set("n", "<leader>zI", "<cmd>Telekasten insert_img_link<CR>")

vim.keymap.set("n", "<leader>me", ":mes<CR>")
vim.keymap.set("n", "<C-/>", ":ToggleTerm<CR>")
vim.keymap.set("t", "<C-/>", "<C-\\><C-n>:ToggleTerm<CR>")

vim.keymap.set("n", "<leader>snt", "<cmd>set nu<CR>")
vim.keymap.set("n", "<leader>snf", "<cmd>set nonu<CR>")

-- Call insert link automatically when we start typing a link
vim.keymap.set("n", "<leader>il", "<cmd>Telekasten insert_link<CR>")

vim.keymap.set("n", "<leader>nv", function()
	select_course_directory()
end, { desc = "Open UniCourse menu" })


vim.keymap.set('n', '<leader>ca', 'ggVGd')
vim.keymap.set("n", "<leader>bd", ":BufferDelete<CR>", { silent = true })

-- Typstar stuff
vim.keymap.set("n", "<leader>ti", ":TypstarInsertRnote<CR>", { silent = true })
vim.keymap.set("n", "<leader>to", ":TypstarOpenDrawing<CR>", { silent = true })

-- Get a ready to use terminal
vim.keymap.set('n', '<leader>tr', ':tabnew<CR>:term<CR>i')
vim.keymap.set("n", "<leader>tt", ":Telescope<CR>", { desc = "Follow Link" })

-- This needs to be refined for quick access to a new file or a recently edited one
vim.keymap.set('n', '<leader>ov', open_vorlesung)
-- new quick note file
-- TODO: make this smarter
vim.keymap.set("n", "<leader>nn", ":e ~/synced/brainstore/zettelkasten/quick<CR>", { silent = true })

vim.keymap.set("n", "<leader>r", set_root)

-- Quickly open some buffers
vim.keymap.set('n', '<leader>occ', ':e ~/.config/nvim/init.lua<CR>`.zz')
vim.keymap.set('n', '<leader>oct', ':e ~/synced/vault/contacts/contacts.txt<CR>`.zz')
vim.keymap.set('n', '<leader>ock', ':e ~/.config/nvim/lua/config/keymaps.lua<CR>`.zz')
vim.keymap.set('n', '<leader>ocd', ':e ~/.config/nvim/lua/config/autocmds.lua<CR>`.zz')
vim.keymap.set('n', '<leader>oco', ':e ~/.config/nvim/lua/config/options.lua<CR>`.zz')
vim.keymap.set('n', '<leader>ocl', ':e ~/.config/nvim/lua/config/lazy.lua<CR>`.zz')
vim.keymap.set('n', '<leader>oczl', ':e ~/.config/nvim/lua/config/lsp.lua<CR>`.zz')
vim.keymap.set('n', '<leader>ocp', ':e ~/.config/nvim/lua/plugins/main.lua<CR>`.zz')
vim.keymap.set('n', '<leader>ocf', ':e ~/.config/nvim/lua/utils/functions.lua<CR>`.zz')
vim.keymap.set('n', '<leader>oca', ':e ~/.config/nvim/lua/utils/after.lua<CR>`.zz')
vim.keymap.set('n', '<leader>oq', ':e ~/synced/brainstore/input.txt<CR>`.zz')
vim.keymap.set('n', '<leader>ohh', ':e ~/configuration/nixos/users/' .. user .. '/home.nix<CR>`.zz')
vim.keymap.set('n', '<leader>op', ':e ~/configuration/nixos/users/' .. user .. '/packages.nix<CR>`.zz')
vim.keymap.set('n', '<leader>on', ':e ~/configuration/nixos/configuration.nix<CR>`.zz')
vim.keymap.set('n', '<leader>om', ':e ~/configuration/nixos/modules<CR>')
vim.keymap.set('n', '<leader>ow', ':e ~/synced/brainstore/waste.txt<CR>')
vim.keymap.set('n', '<leader>oho', ':e ~/configuration/nixos/hosts<CR>')
vim.keymap.set('n', '<leader>os', ':e ~/configuration/nixos/modules/server<CR>')
vim.keymap.set('n', '<leader>ot', ':e ~/synced/brainstore/todos/todo.txt<CR>`.zz')
vim.keymap.set('n', '<leader>od', ':e ~/synced/brainstore/todos/done.txt<CR>`.zz')
vim.keymap.set('n', '<leader>ou', ':e ~/projects/university/' .. season .. '/input.txt<CR>`.zz')
vim.keymap.set('n', '<leader>oz', ':e ~/.zshrc<CR>`.zz')
vim.keymap.set('n', '<leader>oaa', ':e ~/.common_shell<CR>`.zz')
vim.keymap.set("n", "<leader>or", "<cmd>lua open_last_file()<CR>", { noremap = true, silent = true })


vim.keymap.set('n', '<leader>ok', open_cal)
-------------------------------------------------------------------------------------

vim.keymap.set("n", "<leader>lf", links.insert_brainstore_link, { desc = "Link Brainstore file" })
vim.keymap.set("n", "<leader>lm", links.insert_mail_link, { desc = "Link Mail" })
vim.keymap.set('n', '<leader>ll', ':Lazy<CR>')
vim.keymap.set("n", "<leader>lp", links.insert_project_link, { desc = "Link Project" })
vim.keymap.set("n", "<leader>lc", links.insert_contact_link, { desc = "Link Contact" })
vim.keymap.set("n", "<leader>ld", links.insert_date_link, { desc = "Link Contact" })

-- Nvim tree
vim.keymap.set("n", "<leader>e", function()
	api.tree.toggle({ find_file = true, update_root = true, focus = true, })
end, { silent = true }) -- also update the root with the bang

vim.keymap.set('n', '<leader>ia', 'gg=G<C-o>zz')
vim.keymap.set('n', '<leader>ya', 'ggVG"+y<C-o>')

vim.keymap.set('n', '<leader>ss', ':wa<CR>')
vim.keymap.set('n', '<leader>sw', function()
	local word = vim.fn.expand("<cword>")
	local replacement = vim.fn.input("Replace '" .. word .. "' with: ")
	if replacement ~= "" then
		vim.cmd(string.format("%%s/\\<%s\\>/%s/gI", vim.fn.escape(word, '\\/'), vim.fn.escape(replacement, '\\/')))
	end
end, { desc = "Substitute word under cursor (prompt)" })
vim.keymap.set('v', '<leader>sv', function()
	-- Save the current selection
	local save_reg = vim.fn.getreg('"')
	local save_regtype = vim.fn.getregtype('"')

	-- Yank the visual selection into the " register
	vim.cmd('normal! ""y')

	local selection = vim.fn.getreg('"')
	-- Escape magic characters for the search
	selection = vim.fn.escape(selection, '\\/.*$^~[]')

	-- Prompt for the replacement text
	local replacement = vim.fn.input("Replace '" .. selection .. "' with: ")
	if replacement ~= "" then
		vim.cmd(string.format("%%s/%s/%s/gI", selection, replacement))
	end

	-- Restore previous register
	vim.fn.setreg('"', save_reg, save_regtype)
end, { desc = "Substitute selection in file" })

vim.keymap.set('n', '<leader>pp', function()
	vim.api.nvim_command('normal! "+p')
end, { desc = 'Paste from system clipboard' })

vim.keymap.set('v', '<leader>p', function()
	vim.cmd('normal! "+p')
end, { desc = 'Yank to clipboard and keep the selection' })

vim.keymap.set('n', '<leader>fr', function()
	require('telescope.builtin').oldfiles({
		disable_devicons = true,
	})
end, { noremap = true, silent = true })
vim.keymap.set('n', '<leader>fb', builtin.buffers, { desc = 'Telescope buffers' })
vim.keymap.set('n', '<leader>fh', builtin.help_tags, { desc = 'Telescope help tags' })
vim.keymap.set("n", "<leader>fl", links.follow_link, { desc = "Follow Link" })


vim.keymap.set('n', '<leader>g', function()
	require('telescope.builtin').live_grep({
		disable_devicons = true,
		cwd = vim.fn.getcwd(),                -- set the starting directory
		additional_args = function()
			return { '--hidden', '--glob', '!.git/*' } -- include hidden files but exclude .git
		end,
	})
end, { noremap = true, silent = true })

vim.keymap.set('n', '<leader><leader>', find_eff,
	{ desc = 'Telescope find files (with dotfiles and folders but excluding .git, .cache, .local, and large files)' })

------------------------ VISUAL ------------------

vim.keymap.set('v', 'p', function()
	local unnamed_content = vim.fn.getreg('""')
	vim.api.nvim_command('normal! p')
	vim.fn.setreg('""', unnamed_content)
end, { desc = 'Paste from unnamed register (don\'t overwrite it) in visual mode' })
vim.keymap.set('v', '<leader>y', function()
	vim.cmd('normal! "+y')
	vim.cmd('normal! gv')
end, { desc = 'Yank to clipboard and keep the selection' })


---------------------------- INSERT ---------------------------

vim.keymap.set('i', '<C-k>', function()
	local col = vim.fn.col('.')
	local line = vim.fn.line('.')
	local line_len = vim.fn.col('$') - 1
	if col <= line_len then
		vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<Right>', true, false, true), 'n', true)
	else
		if line < vim.fn.line('$') then
			vim.cmd('normal! j^')
		end
	end
end)
-- Move left with wrapping
vim.keymap.set('i', '<C-j>', function()
	local col = vim.fn.col('.')
	local line = vim.fn.line('.')
	if col > vim.fn.indent(line) + 1 then
		-- not at very beginning (after indent), move left
		vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<Left>', true, false, true), 'n', true)
	else
		if line > 1 then
			vim.cmd('normal! k$')
			vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes('<Right>', true, false, true), 'n', true)
		end
	end
end, { noremap = true, silent = true })