perf(remote): restore views on TextYankPost

folke · Jun 24, 2023 · d4dadc8 · d4dadc8
1 parent 587a243
commit d4dadc8
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/lua/flash/plugins/remote.lua b/lua/flash/plugins/remote.lua
@@ -37,13 +37,20 @@ function M.restore()
     vim.api.nvim_set_current_win(M.win)
     vim.fn.winrestview(M.view)
   end
+  restore = vim.schedule_wrap(restore)
+
   if M.operator == "c" then
     vim.api.nvim_create_autocmd("InsertLeave", {
       once = true,
       callback = restore,
     })
+  elseif M.operator == "y" then -- need to check for some opt to paste after yank
+    vim.api.nvim_create_autocmd("TextYankPost", {
+      once = true,
+      callback = restore,
+    })
   else
-    vim.schedule(restore)
+    restore()
   end
 end