1 files changed, 18 insertions, 11 deletions
diff --git a/setwll b/setwll
index 1500472..5dae995 100755
--- a/setwll
+++ b/setwll
@@ -65,18 +65,25 @@ fi
 
 
 if [ "$WAYLAND_DISPLAY" ]; then
-    current_wal="$(hyprctl hyprpaper listactive | grep HDMI-A-1 | sed 's/HDMI-A-1 = //')"
-    hyprctl hyprpaper preload $primary
-    hyprctl hyprpaper wallpaper "HDMI-A-1,$primary"
-    if [ $unload == 1 ]; then
-        hyprctl hyprpaper unload "$current_wal"
-    fi
+    if [ "$XDG_CURRENT_DESKTOP" == "sway:wlroots" ]; then
+        PID=$(pidof swaybg)
+        swaybg -o HDMI-A-1 -i $primary -m fill -o DP-1 -i $secondary -m fill & disown
+        sleep 0.4s
+        kill $PID
+    else
+        current_wal="$(hyprctl hyprpaper listactive | grep HDMI-A-1 | sed 's/HDMI-A-1 = //')"
+        hyprctl hyprpaper preload $primary
+        hyprctl hyprpaper wallpaper "HDMI-A-1,$primary"
+        if [ $unload == 1 ]; then
+            hyprctl hyprpaper unload "$current_wal"
+        fi
 
-    current_wal="$(hyprctl hyprpaper listactive | grep DP-1 | sed 's/DP-1 = //')"
-    hyprctl hyprpaper preload $secondary
-    hyprctl hyprpaper wallpaper "DP-1,$secondary"
-    if [ $unload == 1 ]; then
-        hyprctl hyprpaper unload "$current_wal"
+        current_wal="$(hyprctl hyprpaper listactive | grep DP-1 | sed 's/DP-1 = //')"
+        hyprctl hyprpaper preload $secondary
+        hyprctl hyprpaper wallpaper "DP-1,$secondary"
+        if [ $unload == 1 ]; then
+            hyprctl hyprpaper unload "$current_wal"
+        fi
     fi
 else
     if [ -z "$1" ]; then