refocus when window closed

src/events/events.cpp

@@ -79,12 +79,18 @@ void Events::eventDestroy(xcb_generic_event_t* event) {
     const auto E = reinterpret_cast<xcb_destroy_notify_event_t*>(event);
 
     g_pWindowManager->closeWindowAllChecks(E->window);
+
+    // refocus on new window
+    g_pWindowManager->refocusWindowOnClosed();
 }
 
 void Events::eventUnmapWindow(xcb_generic_event_t* event) {
     const auto E = reinterpret_cast<xcb_unmap_notify_event_t*>(event);
 
     g_pWindowManager->closeWindowAllChecks(E->window);
+
+    // refocus on new window
+    g_pWindowManager->refocusWindowOnClosed();
 }
 
 CWindow* Events::remapFloatingWindow(int windowID, int forcemonitor) {

src/windowManager.cpp

@@ -1379,3 +1379,14 @@ void CWindowManager::getICCCMWMProtocols(CWindow* pWindow) {
     
     xcb_icccm_get_wm_protocols_reply_wipe(&WMProtocolsReply);
 }
+
+void CWindowManager::refocusWindowOnClosed() {
+    const auto PWINDOW = findWindowAtCursor();
+
+    if (!PWINDOW)
+        return;
+
+    LastWindow = PWINDOW->getDrawable();
+
+    setFocusedWindow(PWINDOW->getDrawable());
+}

src/windowManager.hpp

@@ -67,6 +67,7 @@ public:
     void                        refreshDirtyWindows();
 
     void                        setFocusedWindow(xcb_drawable_t);
+    void                        refocusWindowOnClosed();
 
     void                        calculateNewWindowParams(CWindow*);
     void                        fixWindowOnClose(CWindow*);