From 677e7a737897e8c2a60564085821545810369e02 Mon Sep 17 00:00:00 2001
From: Sine Striker <trueful@163.com>
Date: 周一, 11 12月 2023 22:43:14 +0800
Subject: [PATCH] Widgets: Add window state handler

---
 src/widgets/widgetwindowagent.cpp |  124 +++++++++++++++++++++++++++++++----------
 1 files changed, 94 insertions(+), 30 deletions(-)

diff --git a/src/widgets/widgetwindowagent.cpp b/src/widgets/widgetwindowagent.cpp
index 252213e..578ab83 100644
--- a/src/widgets/widgetwindowagent.cpp
+++ b/src/widgets/widgetwindowagent.cpp
@@ -1,9 +1,62 @@
 #include "widgetwindowagent.h"
 #include "widgetwindowagent_p.h"
 
+#include <QtGui/QtEvents>
+#include <QtGui/QPainter>
+#include <QtCore/QDebug>
+
 #include "widgetitemdelegate_p.h"
 
 namespace QWK {
+
+    class WidgetBorderHandler : public QObject {
+    public:
+        WidgetBorderHandler(QWidget *widget, AbstractWindowContext *ctx)
+            : widget(widget), ctx(ctx) {
+            widget->installEventFilter(this);
+        }
+
+        void updateMargins() {
+            if (widget->windowState() & (Qt::WindowMaximized | Qt::WindowFullScreen)) {
+                widget->setContentsMargins({});
+            } else {
+                widget->setContentsMargins({0, 1, 0, 0});
+            }
+        }
+
+    protected:
+        bool eventFilter(QObject *obj, QEvent *event) override {
+            switch (event->type()) {
+                case QEvent::Paint: {
+                    if (widget->windowState() & (Qt::WindowMaximized | Qt::WindowFullScreen))
+                        break;
+
+                    auto paintEvent = static_cast<QPaintEvent *>(event);
+                    QPainter painter(widget);
+                    QRect rect = paintEvent->rect();
+                    QRegion region = paintEvent->region();
+                    void *args[] = {
+                        &painter,
+                        &rect,
+                        &region,
+                    };
+                    ctx->virtual_hook(AbstractWindowContext::DrawBordersHook, args);
+                    return true;
+                }
+
+                case QEvent::WindowStateChange: {
+                    updateMargins();
+                    break;
+                }
+                default:
+                    break;
+            }
+            return false;
+        }
+
+        QWidget *widget;
+        AbstractWindowContext *ctx;
+    };
 
     WidgetWindowAgentPrivate::WidgetWindowAgentPrivate() {
     }
@@ -32,40 +85,23 @@
             return false;
         }
 
-        std::ignore = w->winId(); // Make sure the window handle is created
-        if (!d->setup(w->windowHandle(), new WidgetItemDelegate())) {
+        w->setAttribute(Qt::WA_DontCreateNativeAncestors);
+        w->setAttribute(Qt::WA_NativeWindow);
+
+        if (!d->setup(w, new WidgetItemDelegate())) {
             return false;
         }
         d->hostWidget = w;
-        return true;
-    }
 
-    bool WidgetWindowAgent::isHitTestVisible(QWidget *w) const {
-        Q_D(const WidgetWindowAgent);
-        return d->context->isHitTestVisible(w);
-    }
-
-    void WidgetWindowAgent::setHitTestVisible(QWidget *w, bool visible) {
-        Q_D(WidgetWindowAgent);
-        d->context->setHitTestVisible(w, visible);
-    }
-
-    void WidgetWindowAgent::setHitTestVisible(const QRect &rect, bool visible) {
-        Q_D(WidgetWindowAgent);
-        d->context->setHitTestVisible(rect, visible);
-    }
-
-    QWidget *WidgetWindowAgent::systemButton(CoreWindowAgent::SystemButton button) const {
-        Q_D(const WidgetWindowAgent);
-        return static_cast<QWidget *>(d->context->systemButton(button));
-    }
-
-    void WidgetWindowAgent::setSystemButton(CoreWindowAgent::SystemButton button, QWidget *w) {
-        Q_D(WidgetWindowAgent);
-        if (!d->context->setSystemButton(button, w)) {
-            return;
+        // Install painting hook
+        if (bool needPaintBorder = false;
+            d->context->virtual_hook(AbstractWindowContext::NeedsDrawBordersHook, &needPaintBorder),
+            needPaintBorder) {
+            auto borderHandler = std::make_unique<WidgetBorderHandler>(w, d->context.get());
+            borderHandler->updateMargins();
+            d->borderHandler = std::move(borderHandler);
         }
-        Q_EMIT systemButtonChanged(button, w);
+        return true;
     }
 
     QWidget *WidgetWindowAgent::titleBar() const {
@@ -81,8 +117,36 @@
         Q_EMIT titleBarWidgetChanged(w);
     }
 
+    QWidget *WidgetWindowAgent::systemButton(SystemButton button) const {
+        Q_D(const WidgetWindowAgent);
+        return static_cast<QWidget *>(d->context->systemButton(button));
+    }
+
+    void WidgetWindowAgent::setSystemButton(SystemButton button, QWidget *w) {
+        Q_D(WidgetWindowAgent);
+        if (!d->context->setSystemButton(button, w)) {
+            return;
+        }
+        Q_EMIT systemButtonChanged(button, w);
+    }
+
+    bool WidgetWindowAgent::isHitTestVisible(const QWidget *w) const {
+        Q_D(const WidgetWindowAgent);
+        return d->context->isHitTestVisible(w);
+    }
+
+    void WidgetWindowAgent::setHitTestVisible(const QWidget *w, bool visible) {
+        Q_D(WidgetWindowAgent);
+        d->context->setHitTestVisible(w, visible);
+    }
+
+    void WidgetWindowAgent::setHitTestVisible(const QRect &rect, bool visible) {
+        Q_D(WidgetWindowAgent);
+        d->context->setHitTestVisible(rect, visible);
+    }
+
     WidgetWindowAgent::WidgetWindowAgent(WidgetWindowAgentPrivate &d, QObject *parent)
-        : CoreWindowAgent(d, parent) {
+        : WindowAgentBase(d, parent) {
         d.init();
     }
 }

--
Gitblit v1.9.1