From 5c93f6947a04163c6ea8299e8721b923bea2fb9f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stanis=C5=82aw=20Zag=C3=B3rowski?= <hasohasacz@gmail.com>
Date: Wed, 1 Mar 2023 00:11:49 +0100
Subject: Unload plugins on compositor cleanup (#1662)

---
 src/plugins/PluginSystem.cpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src/plugins/PluginSystem.cpp')

diff --git a/src/plugins/PluginSystem.cpp b/src/plugins/PluginSystem.cpp
index 5b15782d..73de9bf5 100644
--- a/src/plugins/PluginSystem.cpp
+++ b/src/plugins/PluginSystem.cpp
@@ -1,6 +1,7 @@
 #include "PluginSystem.hpp"
 
 #include <dlfcn.h>
+#include <ranges>
 #include "../Compositor.hpp"
 
 CPluginSystem::CPluginSystem() {
@@ -113,6 +114,11 @@ void CPluginSystem::unloadPlugin(const CPlugin* plugin, bool eject) {
     std::erase_if(m_vLoadedPlugins, [&](const auto& other) { return other->m_pHandle == plugin->m_pHandle; });
 }
 
+void CPluginSystem::unloadAllPlugins() {
+    for (auto& p : m_vLoadedPlugins | std::views::reverse)
+        unloadPlugin(p.get(), false); // Unload remaining plugins gracefully
+}
+
 CPlugin* CPluginSystem::getPluginByPath(const std::string& path) {
     for (auto& p : m_vLoadedPlugins) {
         if (p->path == path)
@@ -136,4 +142,4 @@ std::vector<CPlugin*> CPluginSystem::getAllPlugins() {
     for (size_t i = 0; i < m_vLoadedPlugins.size(); ++i)
         results[i] = m_vLoadedPlugins[i].get();
     return results;
-}
\ No newline at end of file
+}
-- 
cgit v1.2.3