better p

glepnir · Mar 30, 2024 · c114fd7 · c114fd7
1 parent 49f5c16
commit c114fd7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lua/keymap/remap.lua b/lua/keymap/remap.lua
@@ -23,7 +23,10 @@ map.n({
 
 -- paste with correct indent
 map.n('p', function()
-  vim.cmd('normal! p==')
+  local text = vim.fn.getreg('"')
+  -- if have mulitple \n assume before copy from the visual mode
+  local count = #vim.split(text, '\n', { trimempty = true })
+  vim.cmd('normal! ' .. ('p%s'):format((count == 1 and '==' or ('v%sj=='):format(count))))
 end, { remap = true })
 
 map.i({