From 02baef645090f0514fa14b43aecf954fd55257dc Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Wed, 12 May 2021 08:36:43 +0800
Subject: ...

---
 src/ui/host/WindowHost.cpp | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

(limited to 'src/ui/host/WindowHost.cpp')

diff --git a/src/ui/host/WindowHost.cpp b/src/ui/host/WindowHost.cpp
index 5e107733..eac2ef41 100644
--- a/src/ui/host/WindowHost.cpp
+++ b/src/ui/host/WindowHost.cpp
@@ -422,9 +422,13 @@ void WindowHost::DispatchMouseHoverControlChangeEvent(
 
 void WindowHost::UpdateCursor() {
   if (native_window_) {
-    const auto capture = GetMouseCaptureControl();
-    native_window_->SetCursor(
-        (capture ? capture : GetMouseHoverControl())->GetInheritedCursor());
+    if (override_cursor_) {
+      native_window_->SetCursor(override_cursor_);
+    } else {
+      const auto capture = GetMouseCaptureControl();
+      native_window_->SetCursor(
+          (capture ? capture : GetMouseHoverControl())->GetInheritedCursor());
+    }
   }
 }
 
@@ -437,4 +441,15 @@ controls::Control* WindowHost::HitTest(const Point& point) {
   }
   return root_control_;
 }
+
+std::shared_ptr<platform::gui::ICursor> WindowHost::GetOverrideCursor() {
+  return override_cursor_;
+}
+
+void WindowHost::SetOverrideCursor(
+    std::shared_ptr<platform::gui::ICursor> cursor) {
+  if (cursor == override_cursor_) return;
+  override_cursor_ = cursor;
+  UpdateCursor();
+}
 }  // namespace cru::ui::host
-- 
cgit v1.2.3