From 4a091bd206ed882251849a1b0c9c4519cc0fdd99 Mon Sep 17 00:00:00 2001
From: mars <mars@tebibyte.media>
Date: Sun, 6 Nov 2022 19:00:56 -0700
Subject: [PATCH] Label text update methods

---
 scripts/sao-ui/src/widgets/button.rs | 6 ++++++
 scripts/sao-ui/src/widgets/text.rs   | 8 ++++++++
 2 files changed, 14 insertions(+)

diff --git a/scripts/sao-ui/src/widgets/button.rs b/scripts/sao-ui/src/widgets/button.rs
index 6ecf242..d2b8859 100644
--- a/scripts/sao-ui/src/widgets/button.rs
+++ b/scripts/sao-ui/src/widgets/button.rs
@@ -48,6 +48,12 @@ impl RoundButton {
             icon,
         }
     }
+
+    pub fn set_text(&mut self, text: &str) {
+        if let Some(icon) = self.icon.as_mut() {
+            icon.set_text(text);
+        }
+    }
 }
 
 impl Button for RoundButton {
diff --git a/scripts/sao-ui/src/widgets/text.rs b/scripts/sao-ui/src/widgets/text.rs
index c0e8e22..6b9cf54 100644
--- a/scripts/sao-ui/src/widgets/text.rs
+++ b/scripts/sao-ui/src/widgets/text.rs
@@ -128,6 +128,14 @@ impl Icon {
             offset: Vec2::ZERO,
         }
     }
+
+    pub fn set_text(&mut self, text: &str) {
+        if self.text.text != text {
+            self.text.text = text.to_string();
+            self.layout = None;
+            self.dirty = true;
+        }
+    }
 }
 
 impl Widget for Icon {