From 48c02397a8d715ab7ce80cd05e432745f9f4b358 Mon Sep 17 00:00:00 2001
From: galister <22305755+galister@users.noreply.github.com>
Date: Sun, 21 Dec 2025 18:25:07 +0900
Subject: [PATCH] watch: highlight edit mode button while in edit mode

---
 wlx-overlay-s/src/assets/gui/watch-noset.xml |  2 +-
 wlx-overlay-s/src/assets/gui/watch.xml       |  2 +-
 wlx-overlay-s/src/overlays/watch.rs          | 11 ++++++++++-
 3 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/wlx-overlay-s/src/assets/gui/watch-noset.xml b/wlx-overlay-s/src/assets/gui/watch-noset.xml
index cb72c69..d08f01b 100644
--- a/wlx-overlay-s/src/assets/gui/watch-noset.xml
+++ b/wlx-overlay-s/src/assets/gui/watch-noset.xml
@@ -114,7 +114,7 @@ All overlays are listed on bottom row.
           <Button height="40" macro="button_style" _press="::DashToggle" tooltip="WATCH.DASHBOARD" tooltip_side="top">
             <sprite color="~set_color" width="32" height="32" src="watch/wayvr_dashboard_mono.svg" />
           </Button>
-          <Button height="40" macro="button_style" _press="::EditToggle" tooltip="WATCH.EDIT_MODE" tooltip_side="top">
+          <Button id="btn_edit_mode" height="40" macro="button_style" _press="::EditToggle" tooltip="WATCH.EDIT_MODE" tooltip_side="top">
             <sprite color="~set_color" width="32" height="32" src="watch/edit.svg" />
           </Button>
           <Button height="40" macro="button_style" tooltip="WATCH.TOGGLE_FOR_CURRENT_SET" _press="::OverlayToggle kbd">
diff --git a/wlx-overlay-s/src/assets/gui/watch.xml b/wlx-overlay-s/src/assets/gui/watch.xml
index 960babe..2933a9a 100644
--- a/wlx-overlay-s/src/assets/gui/watch.xml
+++ b/wlx-overlay-s/src/assets/gui/watch.xml
@@ -153,7 +153,7 @@
               <sprite color="~set_color" width="40" height="40" src="edit/add.svg" />
             </Button>
           </div>
-          <Button macro="button_style" _press="::EditToggle" tooltip="WATCH.EDIT_MODE" tooltip_side="top">
+          <Button id="btn_edit_mode" macro="button_style" _press="::EditToggle" tooltip="WATCH.EDIT_MODE" tooltip_side="top">
             <sprite color="~set_color" width="40" height="40" src="watch/edit.svg" />
           </Button>
         </div>
diff --git a/wlx-overlay-s/src/overlays/watch.rs b/wlx-overlay-s/src/overlays/watch.rs
index 631bfff..dff9ab9 100644
--- a/wlx-overlay-s/src/overlays/watch.rs
+++ b/wlx-overlay-s/src/overlays/watch.rs
@@ -306,7 +306,12 @@ pub fn create_watch(app: &mut AppState) -> anyhow::Result<OverlayWindowConfig> {
         },
     )?;
 
-    panel.on_notify = Some(Box::new(|panel, app, event_data| {
+    let btn_edit_mode = panel
+        .parser_state
+        .fetch_component_as::<ComponentButton>("btn_edit_mode")
+        .ok();
+
+    panel.on_notify = Some(Box::new(move |panel, app, event_data| {
         let mut alterables = EventAlterables::default();
         let mut com = CallbackDataCommon {
             alterables: &mut alterables,
@@ -368,6 +373,10 @@ pub fn create_watch(app: &mut AppState) -> anyhow::Result<OverlayWindowConfig> {
                     panel.state.edit_add_widget,
                     StyleSetRequest::Display(display),
                 );
+
+                if let Some(btn) = btn_edit_mode.as_ref() {
+                    btn.set_sticky_state(&mut com, edit_mode);
+                }
             }
             OverlayEventData::OverlaysChanged(metas) => {
                 panel.state.overlay_metas = metas;