fix: return to source window on close (#192)

stevearc · Dec 13, 2022 · a95b63f · a95b63f
1 parent 97279a1
commit a95b63f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lua/aerial/actions.lua b/lua/aerial/actions.lua
@@ -78,7 +78,13 @@ M.next_up = {
 
 M.close = {
   desc = "Close the aerial window",
-  callback = aerial.close,
+  callback = function()
+    local source_win = require("aerial.util").get_winids(0)
+    if source_win then
+      vim.api.nvim_set_current_win(source_win)
+    end
+    aerial.close()
+  end,
 }
 
 M.tree_toggle = {