From c114fd738b56ab90768fb25feab4108ae7253285 Mon Sep 17 00:00:00 2001
From: glepnir <glephunter@gmail.com>
Date: Sat, 30 Mar 2024 15:34:08 +0800
Subject: [PATCH] better p

---
 lua/keymap/remap.lua | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/lua/keymap/remap.lua b/lua/keymap/remap.lua
index 53ca147c..78ae7c32 100644
--- 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({