diff --git a/src/helpers/WLClasses.hpp b/src/helpers/WLClasses.hpp index 4d47ea25..5d93b91d 100644 --- a/src/helpers/WLClasses.hpp +++ b/src/helpers/WLClasses.hpp @@ -270,8 +270,9 @@ struct STabletPad { }; struct SIdleInhibitor { - wlr_idle_inhibitor_v1* pWlrInhibitor = nullptr; - CWindow* pWindow = nullptr; + wlr_idle_inhibitor_v1* pWlrInhibitor = nullptr; + CWindow* pWindow = nullptr; + HOOK_CALLBACK_FN* onWindowDestroy = nullptr; DYNLISTENER(Destroy); diff --git a/src/managers/input/IdleInhibitor.cpp b/src/managers/input/IdleInhibitor.cpp index 7e7ee3c9..72f60702 100644 --- a/src/managers/input/IdleInhibitor.cpp +++ b/src/managers/input/IdleInhibitor.cpp @@ -10,6 +10,16 @@ void Events::listener_newIdleInhibitor(wl_listener* listener, void* data) { g_pInputManager->newIdleInhibitor(WLRIDLEINHIBITOR); } +static void destroyInhibitor(SIdleInhibitor* inhibitor) { + g_pHookSystem->unhook(inhibitor->onWindowDestroy); + + g_pInputManager->m_lIdleInhibitors.remove(*inhibitor); + + Debug::log(LOG, "Destroyed an idleinhibitor"); + + g_pInputManager->recheckIdleInhibitorStatus(); +} + void CInputManager::newIdleInhibitor(wlr_idle_inhibitor_v1* pInhibitor) { const auto PINHIBIT = &m_lIdleInhibitors.emplace_back(); @@ -17,16 +27,17 @@ void CInputManager::newIdleInhibitor(wlr_idle_inhibitor_v1* pInhibitor) { PINHIBIT->pWlrInhibitor = pInhibitor; + PINHIBIT->onWindowDestroy = g_pHookSystem->hookDynamic("closeWindow", [PINHIBIT](void* self, SCallbackInfo& info, std::any data) { + if (PINHIBIT->pWindow == std::any_cast(data)) + destroyInhibitor(PINHIBIT); + }); + PINHIBIT->hyprListener_Destroy.initCallback( &pInhibitor->events.destroy, [](void* owner, void* data) { const auto PINH = (SIdleInhibitor*)owner; - g_pInputManager->m_lIdleInhibitors.remove(*PINH); - - Debug::log(LOG, "Destroyed an idleinhibitor"); - - g_pInputManager->recheckIdleInhibitorStatus(); + destroyInhibitor(PINH); }, PINHIBIT, "IdleInhibitor");