X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/394edc388755eff01cc2ea155918bc4379ce933d..12826f3c1e2ad75051d52cd2a81718a0349cc3db:/plugin/black.vim

diff --git a/plugin/black.vim b/plugin/black.vim
index 8e05c2a..385d0e1 100644
--- a/plugin/black.vim
+++ b/plugin/black.vim
@@ -122,12 +122,20 @@ def Black():
   except Exception as exc:
     print(exc)
   else:
-    cursor = vim.current.window.cursor
+    current_buffer = vim.current.window.buffer
+    cursors = []
+    for i, tabpage in enumerate(vim.tabpages):
+      if tabpage.valid:
+        for j, window in enumerate(tabpage.windows):
+          if window.valid and window.buffer == current_buffer:
+            cursors.append((i, j, window.cursor))
     vim.current.buffer[:] = new_buffer_str.split('\n')[:-1]
-    try:
-      vim.current.window.cursor = cursor
-    except vim.error:
-      vim.current.window.cursor = (len(vim.current.buffer), 0)
+    for i, j, cursor in cursors:
+      window = vim.tabpages[i].windows[j]
+      try:
+        window.cursor = cursor
+      except vim.error:
+        window.cursor = (len(window.buffer), 0)
     print(f'Reformatted in {time.time() - start:.4f}s.')
 
 def BlackUpgrade():