add error handling for styles

ChrisTitusTech · Aug 20, 2024 · 1c2b301 · 1c2b301
1 parent cbb5691
commit 1c2b301
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/functions/public/Invoke-WPFUIElements.ps1 b/functions/public/Invoke-WPFUIElements.ps1
@@ -31,6 +31,10 @@ function Invoke-WPFUIElements {
     $borderstyle = $window.FindResource("BorderStyle")
     $HoverTextBlockStyle = $window.FindResource("HoverTextBlockStyle")
     $ColorfulToggleSwitchStyle = $window.FindResource("ColorfulToggleSwitchStyle")
+
+    if (!$borderstyle -or !$HoverTextBlockStyle -or !$ColorfulToggleSwitchStyle) {
+        throw "Failed to retrieve Styles using 'FindResource' from main window element."
+    }
 
     $targetGrid = $window.FindName($targetGridName)