fix(zen): properly handle close

lua/snacks/zen.lua

@@ -181,10 +181,7 @@ function M.zen(opts)
   })
 
   -- restore toggle states when window is closed
-  vim.api.nvim_create_autocmd("WinClosed", {
+  win:on("WinClosed", function()
-    group = win.augroup,
-    pattern = tostring(win.win),
-    callback = vim.schedule_wrap(function()
     if zoom_indicator then
       zoom_indicator:close()
     end
@@ -192,32 +189,28 @@ function M.zen(opts)
       state.toggle:set(state.state)
     end
     opts.on_close(win)
-    end),
+  end, { win = true })
-  })
 
   -- update the buffer of the parent window
   -- when the zen buffer changes
-  vim.api.nvim_create_autocmd("BufWinEnter", {
+  win:on("BufWinEnter", function()
-    group = win.augroup,
-    callback = function()
     vim.api.nvim_win_set_buf(parent_win, win.buf)
-    end,
+  end, { buf = true })
-  })
 
   -- close when entering another window
-  vim.api.nvim_create_autocmd("WinEnter", {
+  win:on("WinEnter", function()
-    group = win.augroup,
-    callback = function()
     local w = vim.api.nvim_get_current_win()
     if w == win.win then
       return
     end
     -- exit if other window is not a floating window
     if vim.api.nvim_win_get_config(w).relative == "" then
+      -- schedule so that WinClosed is properly triggered
+      vim.schedule(function()
         win:close()
+      end)
     end
-    end,
+  end)
-  })
   return win
 end