#!/usr/bin/env bash

if [ -z "$1" ]; then
    image=$(zenity --file-selection --title="Open File" --file-filter="*.jpg *.jpeg *.png *.webp *.bmp *.jfif *.tiff *.avif *.heic *.heif")
else
    image="$1"
fi

[ -z "$image" ] && exit 1
[ ! -f "$image" ] && exit 1

ext=${image##*.}
current_dir="$HOME/.config/wallpaper-chooser/current"
image_copied="$current_dir/wallpaper.$ext"

mkdir -p "$current_dir" || (
    notify-send "Error" "Could not create directory $current_dir"
    exit 1
)

temp_img=$(mktemp --suffix=."$ext") || exit 1
cp "$image" "$temp_img" || exit 1
rm -f "$current_dir"/wallpaper.*
cp -f "$temp_img" "$image_copied" || (
    notify-send "Error" "Could not copy image to $current_dir"
    rm -f "$temp_img"
    exit 1
)
rm -f "$temp_img"

if [ "$XDG_CURRENT_DESKTOP" = "Hyprland" ]; then
    hyprctl hyprpaper reload ,"$image_copied" || exit 1
    echo "preload = $image_copied" >"$HOME/.config/hypr/hyprpaper.conf"
    echo "wallpaper = , $image_copied" >>"$HOME/.config/hypr/hyprpaper.conf"

    notify-send "Wallpaper Changed" "$image"
elif [ "$XDG_CURRENT_DESKTOP" = "niri" ]; then
    killall swaybg
    killall swww

    blur_dir="$HOME/.local/share/swaybg"
    mkdir -p "$blur_dir" || (
        notify-send "Error" "Could not create cache directory"
        exit 1
    )
    rm -rf "${blur_dir:?}"/*
    blurred_image="$blur_dir/blurred.$ext"

    # blur
    magick "$image_copied" -blur 0x16 "$blurred_image" || (
        notify-send "Error" "Could not create blurred image"
        exit 1
    )

    (setsid swaybg -i "$blurred_image" -m fill)&
    disown

    (setsid swww img "$image_copied" --transition-type fade --transition-duration 2)&
    disown

    notify-send "Wallpaper Changed" "$image"
fi

notify-send "Extracting colors from wallpaper" "This may take a few seconds..."
change-colortheme -i "$image_copied" || exit 1