-- Custom keymaps

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

-- TODO: move this to a config file
local season = "S3"

local links = require("utils.linker")
local user = vim.fn.system('whoami'):gsub('\n', '')

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', '<leader>zq', ':e ~/synced/brainstore/zettelkasten/input.txt<CR>`.zz')

-- Buildin vim

vim.keymap.set("n", "<leader>me", ":mes<CR>")

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

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

vim.keymap.set("n", "n", "nzz", { silent = true })
vim.keymap.set("n", "N", "Nzz", { 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')



-- Get a ready to use terminal
vim.keymap.set('n', '<leader>tr', ':tabnew<CR>:term<CR>i')

-- This needs to be refined for quick access to a new file or a recently edited one
vim.keymap.set('n', '<leader>ov',
	function()
		local uni_dir = vim.fn.expand("~/projects/university/" .. season)

		require('telescope.builtin').find_files {
			prompt_title = "Select Vorlesung in " .. season,
			cwd = uni_dir,
			find_command = {
				"eza", "-1", "-D"
			},
		}
	end
)
-- 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",
	function()
		local current_file = vim.fn.expand('%:p:h') -- get directory of current file
		local cmd = 'git -C ' .. vim.fn.fnameescape(current_file) .. ' status'
		vim.fn.system(cmd)
		if vim.v.shell_error == 0 then
			local git_root = vim.fn.systemlist('git -C ' ..
					vim.fn.fnameescape(current_file) .. ' rev-parse --show-toplevel')
				[1]
			vim.cmd('cd ' .. vim.fn.fnameescape(git_root))
		else
			vim.cmd('cd ' .. vim.fn.fnameescape(current_file))
		end
	end
)

-- 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/misc.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",
	function()
		local last_file_path = vim.fn.stdpath('data') .. "/lastfile.txt"
		local file = io.open(last_file_path, "r")
		if file then
			local last_file = file:read("*line")
			file:close()
			if last_file and vim.fn.filereadable(last_file) == 1 then
				vim.cmd("edit " .. last_file)
				pcall(function()
					vim.cmd('normal! `.') -- Go to the last edit position
					vim.cmd('normal! zz') -- Center the cursor on the screen
				end)
			else
				print("Last file does not exist or is not readable")
			end
		else
			print("No last file found")
		end
	end
	, { 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" })


-- Indent all will be replaced by the formatting of lsp where the lsp is installed
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)" })

-- Substitution in visual mode
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' })

------------------------ 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 })