Refactor drop shadow handling for window border style

Flow.Launcher.Core/Resource/Theme.cs

@@ -477,76 +477,113 @@
 
         public void AddDropShadowEffectToCurrentTheme()
         {
-            var dict = GetCurrentResourceDictionary();
+            // Get current theme's WindowBorderStyle
+            var theme = _settings.Theme;
+            var dict = GetThemeResourceDictionary(theme);
+            var windowBorderStyle = dict.Contains("WindowBorderStyle") ? dict["WindowBorderStyle"] as Style : null;
+            if (windowBorderStyle == null) return;
 
-            var windowBorderStyle = dict["WindowBorderStyle"] as Style;
+            // Get a new unsealed style based on the old one, and copy Resources and Triggers
+            var newWindowBorderStyle = new Style(typeof(Border));
+            CopyStyle(windowBorderStyle, newWindowBorderStyle);
 
-            var effectSetter = new Setter
+            // Identify existing Margin to calculate new Margin, and copy other setters
+            Setter existingMarginSetter = null;
+            foreach (var setterBase in windowBorderStyle.Setters)
             {
-                Property = UIElement.EffectProperty,
-                Value = new DropShadowEffect
+                if (setterBase is Setter setter)
                 {
-                    Opacity = 0.3,
-                    ShadowDepth = 12,
-                    Direction = 270,
-                    BlurRadius = 30
+                    // Skip existing Margin (we'll replace it)
+                    if (setter.Property == FrameworkElement.MarginProperty)
+                    {
+                        existingMarginSetter = setter;
+                        continue;
+                    }
+
+                    // Skip existing Effect (we'll ensure strictly one is added)
+                    if (setter.Property == UIElement.EffectProperty) continue;
                 }
-            };
 
-            if (windowBorderStyle.Setters.FirstOrDefault(setterBase => setterBase is Setter setter && setter.Property == FrameworkElement.MarginProperty) is not Setter marginSetter)
-            {
-                var margin = new Thickness(ShadowExtraMargin, 12, ShadowExtraMargin, ShadowExtraMargin);
-                marginSetter = new Setter()
-                {
-                    Property = FrameworkElement.MarginProperty,
-                    Value = margin,
-                };
-                windowBorderStyle.Setters.Add(marginSetter);
+                // Add other setters (e.g. Background, BorderThickness)
+                newWindowBorderStyle.Setters.Add(setterBase);
+            }
 
-                SetResizeBoarderThickness(margin);
+            // Calculate new Margin
+            Thickness newMargin;
+            if (existingMarginSetter == null)
+            {
+                newMargin = new Thickness(ShadowExtraMargin, 12, ShadowExtraMargin, ShadowExtraMargin);
             }
             else
             {
-                var baseMargin = (Thickness)marginSetter.Value;
-                var newMargin = new Thickness(
+                var baseMargin = (Thickness)existingMarginSetter.Value;
+                newMargin = new Thickness(
                     baseMargin.Left + ShadowExtraMargin,
                     baseMargin.Top + ShadowExtraMargin,
                     baseMargin.Right + ShadowExtraMargin,
                     baseMargin.Bottom + ShadowExtraMargin);
-                marginSetter.Value = newMargin;
-
-                SetResizeBoarderThickness(newMargin);
             }
 
-            windowBorderStyle.Setters.Add(effectSetter);
+            // Add new Margin Setter
+            newWindowBorderStyle.Setters.Add(new Setter(FrameworkElement.MarginProperty, newMargin));
 
-            UpdateResourceDictionary(dict);
+            // Add Drop Shadow Effect Setter
+            newWindowBorderStyle.Setters.Add(new Setter
+            {
+                Property = UIElement.EffectProperty,
+                Value = new DropShadowEffect
+                {
+                    Opacity = 0.3,
+                    ShadowDepth = 12,
+                    Direction = 270,
+                    BlurRadius = 30
+                }
+            });
+
+            SetResizeBoarderThickness(newMargin);
+
+            Application.Current.Resources["WindowBorderStyle"] = newWindowBorderStyle;
         }
 
         public void RemoveDropShadowEffectFromCurrentTheme()
         {
-            var dict = GetCurrentResourceDictionary();
-            var windowBorderStyle = dict["WindowBorderStyle"] as Style;
+            // Get current theme's WindowBorderStyle
+            var theme = _settings.Theme;
+            var dict = GetThemeResourceDictionary(theme);
+            var windowBorderStyle = dict.Contains("WindowBorderStyle") ? dict["WindowBorderStyle"] as Style : null;
+            if (windowBorderStyle == null) return;
 
-            if (windowBorderStyle.Setters.FirstOrDefault(setterBase => setterBase is Setter setter && setter.Property == UIElement.EffectProperty) is Setter effectSetter)
-            {
-                windowBorderStyle.Setters.Remove(effectSetter);
-            }
+            // Get a new unsealed style based on the old one, and copy Resources and Triggers
+            var newWindowBorderStyle = new Style(typeof(Border));
+            CopyStyle(windowBorderStyle, newWindowBorderStyle);
 
-            if (windowBorderStyle.Setters.FirstOrDefault(setterBase => setterBase is Setter setter && setter.Property == FrameworkElement.MarginProperty) is Setter marginSetter)
+            // Copy Setters, excluding the Effect setter and updating the Margin setter
+            foreach (var setterBase in windowBorderStyle.Setters)
             {
-                var currentMargin = (Thickness)marginSetter.Value;
-                var newMargin = new Thickness(
-                    currentMargin.Left - ShadowExtraMargin,
-                    currentMargin.Top - ShadowExtraMargin,
-                    currentMargin.Right - ShadowExtraMargin,
-                    currentMargin.Bottom - ShadowExtraMargin);
-                marginSetter.Value = newMargin;
+                if (setterBase is Setter setter)
+                {
+                    // Skip existing Effect (We'll remove it)
+                    if (setter.Property == UIElement.EffectProperty) continue;
+
+                    // Update Margin by subtracting the extra margin we added for the shadow
+                    if (setter.Property == FrameworkElement.MarginProperty)
+                    {
+                        var currentMargin = (Thickness)setter.Value;
+                        var newMargin = new Thickness(
+                            currentMargin.Left - ShadowExtraMargin,
+                            currentMargin.Top - ShadowExtraMargin,
+                            currentMargin.Right - ShadowExtraMargin,
+                            currentMargin.Bottom - ShadowExtraMargin);
+                        newWindowBorderStyle.Setters.Add(new Setter(FrameworkElement.MarginProperty, newMargin));
+                        continue;
+                    }
+                }
+                newWindowBorderStyle.Setters.Add(setterBase);
             }
 
             SetResizeBoarderThickness(null);
 
-            UpdateResourceDictionary(dict);
+            Application.Current.Resources["WindowBorderStyle"] = newWindowBorderStyle;
         }
 
         public void SetResizeBorderThickness(WindowChrome windowChrome, bool fixedWindowSize)
@@ -602,7 +639,7 @@
                 var (backdropType, useDropShadowEffect) = GetActualValue();
 
                 // Remove OS minimizing/maximizing animation
                 // Methods.SetWindowAttribute(new WindowInteropHelper(mainWindow).Handle, DWMWINDOWATTRIBUTE.DWMWA_TRANSITIONS_FORCEDISABLED, 3);
 
                 // The timing of adding the shadow effect should vary depending on whether the theme is transparent.
                 if (BlurEnabled)
@@ -646,7 +683,7 @@
                 backdropType = BackdropTypes.None;
             }
 
             // Dropshadow on and control disabled.(user can't change dropshadow with blur theme)
             if (BlurEnabled)
             {
                 useDropShadowEffect = true;
@@ -670,7 +707,7 @@
             bool hasBlur = dict.Contains("ThemeBlurEnabled") && dict["ThemeBlurEnabled"] is bool b && b;
             if (BlurEnabled && hasBlur && Win32Helper.IsBackdropSupported())
             {
                 // If the BackdropType is Mica or MicaAlt, set the windowborderstyle's background to transparent
                 if (backdropType == BackdropTypes.Mica || backdropType == BackdropTypes.MicaAlt)
                 {
                     windowBorderStyle.Setters.Remove(windowBorderStyle.Setters.OfType<Setter>().FirstOrDefault(x => x.Property.Name == "Background"));