From 3e942c3dc8955be577079fbc028ce216e1c594b2 Mon Sep 17 00:00:00 2001
From: SineStriker <55847490+SineStriker@users.noreply.github.com>
Date: 周二, 11 2月 2025 19:07:53 +0800
Subject: [PATCH] Fix numerous bugs (#162)

---
 src/core/contexts/abstractwindowcontext.cpp |  209 +++++++++++++++++++++++++++++-----------------------
 1 files changed, 116 insertions(+), 93 deletions(-)

diff --git a/src/core/contexts/abstractwindowcontext.cpp b/src/core/contexts/abstractwindowcontext.cpp
index 8fa851a..f3a9c7f 100644
--- a/src/core/contexts/abstractwindowcontext.cpp
+++ b/src/core/contexts/abstractwindowcontext.cpp
@@ -7,65 +7,10 @@
 #include <QtGui/QPen>
 #include <QtGui/QPainter>
 #include <QtGui/QScreen>
-#include <memory>
 
 #include "qwkglobal_p.h"
 
 namespace QWK {
-
-    namespace {
-
-        class WinIdChangeEventFilter : public QObject {
-        public:
-            explicit WinIdChangeEventFilter(QObject *widget, AbstractWindowContext *ctx,
-                                            QObject *parent = nullptr)
-                : QObject(parent), ctx(ctx) {
-                widget->installEventFilter(this);
-            }
-
-        protected:
-            bool eventFilter(QObject *obj, QEvent *event) override {
-                Q_UNUSED(obj)
-                if (event->type() == QEvent::WinIdChange) {
-                    ctx->notifyWinIdChange();
-                }
-                return false;
-            }
-
-        protected:
-            AbstractWindowContext *ctx;
-        };
-
-        class WindowEventFilter : public QObject {
-        public:
-            explicit WindowEventFilter(QWindow *window, AbstractWindowContext *ctx,
-                                       QObject *parent = nullptr)
-                : QObject(parent), ctx(ctx), window(window) {
-                if (window)
-                    window->installEventFilter(this);
-            }
-
-            inline void setWindow(QWindow *win) {
-                if (auto oldWin = window.data(); oldWin) {
-                    oldWin->removeEventFilter(this);
-                }
-                window = win;
-                if (win) {
-                    win->installEventFilter(this);
-                }
-            }
-
-        protected:
-            bool eventFilter(QObject *obj, QEvent *event) override {
-                return ctx->sharedDispatch(obj, event);
-            }
-
-        protected:
-            AbstractWindowContext *ctx;
-            QPointer<QWindow> window;
-        };
-
-    }
 
     AbstractWindowContext::AbstractWindowContext() = default;
 
@@ -77,27 +22,31 @@
         }
         m_host = host;
         m_delegate.reset(delegate);
-        m_winIdChangeEventFilter = std::make_unique<WinIdChangeEventFilter>(host, this);
-        m_windowEventFilter = std::make_unique<WindowEventFilter>(m_windowHandle, this);
-
-        m_windowHandle = m_delegate->hostWindow(m_host);
-        if (m_windowHandle) {
-            winIdChanged();
-        }
+        m_winIdChangeEventFilter.reset(delegate->createWinIdEventFilter(host, this));
+        notifyWinIdChange();
     }
 
-
-
-    bool AbstractWindowContext::setHitTestVisible(const QObject *obj, bool visible) {
+    bool AbstractWindowContext::setHitTestVisible(QObject *obj, bool visible) {
         Q_ASSERT(obj);
         if (!obj) {
             return false;
         }
 
+        auto it = m_hitTestVisibleItems.find(obj);
         if (visible) {
+            if (it != m_hitTestVisibleItems.end()) {
+                return true;
+            }
+            connect(obj, &QObject::destroyed, this,
+                    &AbstractWindowContext::_q_hitTestVisibleItemDestroyed);
             m_hitTestVisibleItems.insert(obj);
         } else {
-            m_hitTestVisibleItems.remove(obj);
+            if (it == m_hitTestVisibleItems.end()) {
+                return false;
+            }
+            disconnect(obj, &QObject::destroyed, this,
+                       &AbstractWindowContext::_q_hitTestVisibleItemDestroyed);
+            m_hitTestVisibleItems.erase(it);
         }
         return true;
     }
@@ -109,8 +58,18 @@
             return false;
         }
 
-        if (m_systemButtons[button] == obj) {
-            return false;
+        auto org = m_systemButtons[button];
+        if (org == obj) {
+            return true;
+        }
+
+        if (org) {
+            disconnect(org, &QObject::destroyed, this,
+                       &AbstractWindowContext::_q_systemButtonDestroyed);
+        }
+        if (obj) {
+            connect(obj, &QObject::destroyed, this,
+                    &AbstractWindowContext::_q_systemButtonDestroyed);
         }
         m_systemButtons[button] = obj;
         return true;
@@ -118,18 +77,20 @@
 
     bool AbstractWindowContext::setTitleBar(QObject *item) {
         Q_ASSERT(item);
-        if (m_titleBar == item) {
+        auto org = m_titleBar;
+        if (org == item) {
             return false;
         }
 
-        if (m_titleBar) {
+        if (org) {
             // Since the title bar is changed, all items inside it should be dereferenced right away
-            for (auto &button : m_systemButtons) {
-                button = nullptr;
-            }
-            m_hitTestVisibleItems.clear();
+            removeSystemButtonsAndHitTestItems();
+            disconnect(org, &QObject::destroyed, this,
+                       &AbstractWindowContext::_q_titleBarDistroyed);
         }
-
+        if (item) {
+            connect(item, &QObject::destroyed, this, &AbstractWindowContext::_q_titleBarDistroyed);
+        }
         m_titleBar = item;
         return true;
     }
@@ -213,7 +174,7 @@
     void AbstractWindowContext::virtual_hook(int id, void *data) {
         switch (id) {
             case CentralizeHook: {
-                if (!m_windowHandle)
+                if (!m_windowId)
                     return;
 
                 QRect windowGeometry = m_delegate->getGeometry(m_host);
@@ -227,7 +188,7 @@
             }
 
             case RaiseWindowHook: {
-                if (!m_windowHandle)
+                if (!m_windowId)
                     return;
 
                 m_delegate->setWindowVisible(m_host, true);
@@ -259,33 +220,49 @@
     }
 
     void AbstractWindowContext::notifyWinIdChange() {
-        auto windowEventFilter = static_cast<WindowEventFilter *>(m_windowEventFilter.get());
-        windowEventFilter->setWindow(nullptr);
+        auto oldWinId = m_windowId;
+        m_windowId = m_winIdChangeEventFilter->winId();
 
         // In Qt6, after QWidget::close() is called, the related QWindow's all surfaces and the
         // platform window will be removed, and the WinId will be set to 0. After that, when the
         // QWidget is shown again, the whole things will be recreated again.
         // As a result, we must update our WindowContext each time the WinId changes.
+        if (m_windowHandle) {
+            removeEventFilter(m_windowHandle);
+        }
         m_windowHandle = m_delegate->hostWindow(m_host);
-        winIdChanged();
+
+        if (oldWinId != m_windowId) {
+            winIdChanged(m_windowId, oldWinId);
+        }
 
         if (m_windowHandle) {
-            windowEventFilter->setWindow(m_windowHandle);
+            m_windowHandle->installEventFilter(this);
 
             // Refresh window attributes
-            auto attributes = m_windowAttributes;
-            m_windowAttributes.clear();
-            for (auto it = attributes.begin(); it != attributes.end(); ++it) {
-                if (!windowAttributeChanged(it.key(), it.value(), {})) {
+            for (auto it = m_windowAttributesOrder.begin(); it != m_windowAttributesOrder.end();) {
+                if (!windowAttributeChanged(it->first, it->second, {})) {
+                    m_windowAttributes.remove(it->first);
+                    it = m_windowAttributesOrder.erase(it);
                     continue;
                 }
-                m_windowAttributes.insert(it.key(), it.value());
+                ++it;
             }
+        }
+
+        // Send to shared dispatchers
+        if (oldWinId != m_windowId) {
+            QEvent e(QEvent::WinIdChange);
+            sharedDispatch(m_host, &e);
         }
     }
 
     QVariant AbstractWindowContext::windowAttribute(const QString &key) const {
-        return m_windowAttributes.value(key);
+        auto it = m_windowAttributes.find(key);
+        if (it == m_windowAttributes.end()) {
+            return {};
+        }
+        return it.value()->second;
     }
 
     bool AbstractWindowContext::setWindowAttribute(const QString &key, const QVariant &attribute) {
@@ -294,25 +271,37 @@
             if (!attribute.isValid()) {
                 return true;
             }
-            if (!m_windowHandle || !windowAttributeChanged(key, attribute, {})) {
+            if (m_windowHandle && !windowAttributeChanged(key, attribute, {})) {
                 return false;
             }
-            m_windowAttributes.insert(key, attribute);
+            m_windowAttributes.insert(key,
+                                      m_windowAttributesOrder.insert(m_windowAttributesOrder.end(),
+                                      std::make_pair(key, attribute)));
             return true;
         }
 
-        if (it.value() == attribute)
-            return true;
-        if (!m_windowHandle || !windowAttributeChanged(key, attribute, it.value())) {
+        auto &listIter = it.value();
+        auto &oldAttr = listIter->second;
+        if (m_windowHandle && !windowAttributeChanged(key, attribute, oldAttr)) {
             return false;
         }
 
         if (attribute.isValid()) {
-            it.value() = attribute;
+            oldAttr = attribute;
+            m_windowAttributesOrder.splice(m_windowAttributesOrder.end(), m_windowAttributesOrder,
+                                           listIter);
         } else {
+            m_windowAttributesOrder.erase(listIter);
             m_windowAttributes.erase(it);
         }
         return true;
+    }
+
+    bool AbstractWindowContext::eventFilter(QObject *obj, QEvent *event) {
+        if (obj == m_windowHandle && sharedDispatch(obj, event)) {
+            return true;
+        }
+        return QObject::eventFilter(obj, event);
     }
 
     bool AbstractWindowContext::windowAttributeChanged(const QString &key,
@@ -321,4 +310,38 @@
         return false;
     }
 
+    void AbstractWindowContext::removeSystemButtonsAndHitTestItems() {
+        for (auto &button : m_systemButtons) {
+            if (!button) {
+                continue;
+            }
+            disconnect(button, &QObject::destroyed, this,
+                       &AbstractWindowContext::_q_systemButtonDestroyed);
+            button = nullptr;
+        }
+        for (auto &item : m_hitTestVisibleItems) {
+            disconnect(item, &QObject::destroyed, this,
+                       &AbstractWindowContext::_q_hitTestVisibleItemDestroyed);
+        }
+        m_hitTestVisibleItems.clear();
+    }
+
+    void AbstractWindowContext::_q_titleBarDistroyed(QObject *obj) {
+        Q_UNUSED(obj)
+        removeSystemButtonsAndHitTestItems();
+        m_titleBar = nullptr;
+    }
+
+    void AbstractWindowContext::_q_hitTestVisibleItemDestroyed(QObject *obj) {
+        m_hitTestVisibleItems.remove(obj);
+    }
+
+    void AbstractWindowContext::_q_systemButtonDestroyed(QObject *obj) {
+        for (auto &item : m_systemButtons) {
+            if (item == obj) {
+                item = nullptr;
+            }
+        }
+    }
+
 }

--
Gitblit v1.9.1