nvim: Preserve registers when transposing chars

1 files changed, 16 insertions, 0 deletions

diff --git a/.config/nvim/lua/mango/lib.lua b/.config/nvim/lua/mango/lib.lua
index 4c38986..9182a06 100644
--- a/.config/nvim/lua/mango/lib.lua
+++ b/.config/nvim/lua/mango/lib.lua
@@ -18,4 +18,20 @@ function M.remap(modes, from, to, opts)
 	vim.keymap.set(ct, from, to, opts)
 end
 
+function M.save_regs(regs, callback)
+	local rs = {}
+	for r in regs:gmatch('.') do
+		rs[r] = {
+			s = vim.fn.getreg(r),
+			t = vim.fn.getregtype(r),
+		}
+	end
+
+	callback()
+
+	for k, v in pairs(rs) do
+		vim.fn.setreg(k, v.s, v.t)
+	end
+end
+
 return M