#!/bin/bash
# pywal.sh - apply pywal colors
# Usage: ./pywal.sh [wallpaper_path] [light|dark]

WALLPAPER="$1"
LIGHT_MODE="$2"

echo "Applying pywal colors..."

if [ -n "$WALLPAPER" ]; then
    if [ "$LIGHT_MODE" = "-l" ] || [ "$LIGHT_MODE" = "light" ]; then
        wal -i "$WALLPAPER" -l
        DARK_MODE=0
        GTK_THEME="prefer-light"
        echo "✓ Generated light theme from wallpaper"
    else
        wal -i "$WALLPAPER"
        DARK_MODE=1
        GTK_THEME="prefer-dark"
        echo "✓ Generated dark theme from wallpaper"
    fi
elif [ "$LIGHT_MODE" = "light" ] || [ "$LIGHT_MODE" = "-l" ]; then
    wal -l
    DARK_MODE=0
    GTK_THEME="prefer-light"
    echo "✓ Switched to light theme"
elif [ "$LIGHT_MODE" = "dark" ]; then
    wal
    DARK_MODE=1
    GTK_THEME="prefer-dark"
    echo "✓ Switched to dark theme"
else
    if [ -f ~/.cache/wal/wal ]; then
        wal -R
        BG_COLOR=$(grep -oP 'background.*#\K[0-9A-Fa-f]{6}' ~/.cache/wal/colors.json | head -1)
        if [ -n "$BG_COLOR" ]; then
            R=$((16#${BG_COLOR:0:2}))
            G=$((16#${BG_COLOR:2:2}))
            B=$((16#${BG_COLOR:4:2}))
            BRIGHTNESS=$(( (R + G + B) / 3 ))
            if [ $BRIGHTNESS -gt 128 ]; then
                DARK_MODE=0
                GTK_THEME="prefer-light"
            else
                DARK_MODE=1
                GTK_THEME="prefer-dark"
            fi
        fi
        echo "✓ Refreshed existing theme"
    else
        echo "✗ No wallpaper specified and no existing theme found"
        exit 1
    fi
fi

# Caelestia wallpaper
if [ -f ~/.cache/wal/wal ]; then
    WALLPAPER_PATH=$(cat ~/.cache/wal/wal)
    mkdir -p ~/.local/state/caelestia/wallpaper
    ln -sf "$WALLPAPER_PATH" ~/.local/state/caelestia/wallpaper/current
    echo "$WALLPAPER_PATH" > ~/.local/state/caelestia/wallpaper/path.txt
    echo "✓ Caelestia wallpaper reference updated"
fi

# Caelestia colors
if [ -f ~/.cache/wal/caelestia-scheme.json ]; then
    mkdir -p ~/.local/state/caelestia
    cp ~/.cache/wal/caelestia-scheme.json ~/.local/state/caelestia/scheme.json
    echo "✓ Caelestia colors updated"
fi

# GTK themes
if command -v wal-gtk &> /dev/null; then
    wal-gtk
    echo "✓ GTK themes generated"
else
    echo "ℹ wal-gtk not found - install 'wal-gtk-theme-git' for GTK theme support"
fi

# GTK dark/light mode
if command -v gsettings &> /dev/null; then
    gsettings set org.gnome.desktop.interface color-scheme "$GTK_THEME" 2>/dev/null && \
        echo "✓ System dark mode preference set to: $GTK_THEME" || \
        echo "ℹ Could not set GTK color scheme preference"
fi

# Firefox
if command -v pywalfox &> /dev/null; then
    pywalfox update &>/dev/null &
    echo "✓ Firefox theme update triggered"
else
    echo "ℹ pywalfox not found - install 'python-pywalfox' for Firefox support"
fi

# Hyprland
if command -v hyprctl &> /dev/null; then
    hyprctl reload
    echo "✓ Hyprland configuration reloaded"
fi

# Caelestia
if pgrep -x "caelestia" > /dev/null; then
    pkill caelestia
    sleep 0.5
    caelestia shell -d &
    echo "✓ Caelestia shell restarted"
fi

# Kitty
if command -v kitty &> /dev/null; then
    if pgrep -x "kitty" > /dev/null; then
        killall -SIGUSR1 kitty 2>/dev/null
        echo "✓ Kitty terminal colors reloaded"
    fi
fi

echo ""
echo "Done!"
echo "Current mode: $([ "$DARK_MODE" = "1" ] && echo "Dark" || echo "Light")"