fix: closing pum in insert mode

lua/core/mappings.lua

@@ -22,8 +22,8 @@ vim.keymap.set("t", "<Esc>", "<C-\\><C-n>")
 -- Use :diffput/get instead of normal one to allow staging visual selection
 vim.keymap.set({"n", "x"}, "<leader>dp", ":diffput<CR>")
 vim.keymap.set({"n", "x"}, "<leader>do", ":diffget<CR>")
-vim.keymap.set({ "n", "i", }, "<C-e>",
-    function ()
+
+local close_pum = function()
     if vim.fn.pumvisible() ~= 0 then
         return "<cmd>pclose<cr>"
     end
@@ -33,11 +33,15 @@ vim.keymap.set({ "n", "i", }, "<C-e>",
             return "<cmd>fclose<cr>"
         end
     end
+end
 
-        return "<C-e>"
-    end,
-    { expr = true, }
-)
+vim.keymap.set("n", "<C-e>", function()
+    return close_pum() or "<C-e>"
+end, { expr = true })
+
+vim.keymap.set("i", "<C-e>", function()
+    return close_pum() or "<C-o>$"
+end, { expr = true })
 
 vim.keymap.set("n", "<C-w>q", ":bp \\| bd#<CR>")
 
@@ -57,7 +61,7 @@ vim.cmd.aunmenu({ "PopUp.How-to\\ disable\\ mouse", })
 vim.keymap.set('i', '<C-f>', '<Right>')
 vim.keymap.set('i', '<C-b>', '<Left>')
 vim.keymap.set('i', '<C-a>', '<C-o>^')
-vim.keymap.set('i', '<C-e>', '<C-o>$')
+-- vim.keymap.set('i', '<C-e>', '<C-o>$')
 vim.keymap.set('i', '<C-d>', '<C-o>x')
 vim.keymap.set('i', '<M-f>', '<C-o>w')
 vim.keymap.set('i', '<M-b>', '<C-o>b')