bitzdots

1#!/bin/bash
2# Rofi wallpaper selector with automatic theme generation via matugen
3 
4CONFIG_DIR="${XDG_CONFIG_HOME:-$HOME/.config}"
5WALL_DIR="${WALLPAPER_DIR:-$HOME/Pictures/Wallpapers}"
6CACHE_DIR="${XDG_CACHE_HOME:-$HOME/.cache}"
7THEME_CACHE="$CACHE_DIR/matugen/themes"
8ROFI_THEME="$CONFIG_DIR/rofi/themes/wallpaper-grid.rasi"
9ROFI_POWER="$CONFIG_DIR/rofi/themes/power.rasi"
10LIVE_DIR="$WALL_DIR/live"
11THUMB_DIR="$CACHE_DIR/matugen-thumbs"
12LOG_DIR="$CACHE_DIR/matugen"
13LOG_FILE="$LOG_DIR/wallpaper-select.log"
14 
15mkdir -p "$CACHE_DIR" "$THUMB_DIR" "$THEME_CACHE" "$LOG_DIR"
16 
17alert() {
18    local urgency="${2:-normal}"
19    local title="$1"
20    local msg="$3"
21    echo "[$(date '+%Y-%m-%d %H:%M:%S')] [$urgency] $title${msg:+ - $msg}" >> "$LOG_FILE"
22    echo "[wallpaper-select] $title${msg:+ - $msg}" >&2
23    notify-send -u "$urgency" "$title" "$msg" 2>/dev/null || true
24}
25 
26if [ ! -d "$WALL_DIR" ]; then
27    alert "Wallpapers not found" critical "Directory $WALL_DIR does not exist"
28    exit 1
29fi
30 
31MATUGEN_OUTPUTS=(
32    "waybar/style.css"
33    "waybar/config.jsonc"
34    "swaync/style.css"
35    "wlogout/style.css"
36    "hypr/colors.lua"
37    "kitty/colors.conf"
38    "cava/themes/generated"
39    "rofi/theme-generated.rasi"
40    "matugen/env"
41    "matugen/browser-colors.css"
42    "matugen/chromium-theme.json"
43    "rofi/themes/wallpaper-grid.rasi"
44    "rofi/themes/power.rasi"
45    "rofi/themes/launcher.rasi"
46    "rofi/themes/clipboard.rasi"
47    "rofi/icons/lock.svg"
48    "rofi/icons/logout.svg"
49    "rofi/icons/sleep.svg"
50    "rofi/icons/reboot.svg"
51    "rofi/icons/shutdown.svg"
52    "rofi/icons/cancel.svg"
53    "rofi/icons/static.svg"
54    "rofi/icons/live.svg"
55    "kdeglobals"
56    "bitzdots.colors"
57    "qt6ct/qt6ct.conf"
58)
59 
60backup_theme() {
61    local backup_dir="$CACHE_DIR/matugen-backup"
62    rm -rf "$backup_dir"
63    for output in "${MATUGEN_OUTPUTS[@]}"; do
64        local src="$CONFIG_DIR/$output"
65        if [ -f "$src" ]; then
66            local dst="$backup_dir/$output"
67            mkdir -p "$(dirname "$dst")"
68            cp "$src" "$dst"
69        fi
70    done
71}
72 
73restore_theme() {
74    local backup_dir="$CACHE_DIR/matugen-backup"
75    if [ ! -d "$backup_dir" ]; then
76        return
77    fi
78    for output in "${MATUGEN_OUTPUTS[@]}"; do
79        local src="$backup_dir/$output"
80        local dst="$CONFIG_DIR/$output"
81        if [ -f "$src" ]; then
82            mkdir -p "$(dirname "$dst")"
83            cp "$src" "$dst"
84        fi
85    done
86    rm -rf "$backup_dir"
87}
88 
89clear_backup() {
90    rm -rf "$CACHE_DIR/matugen-backup"
91}
92 
93apply_cached_theme() {
94    local name="$1"
95    local cache_dir="$THEME_CACHE/$name"
96    local marker="$cache_dir/.done"
97 
98    if [ ! -f "$marker" ]; then
99        return 1
100    fi
101 
102    # Invalidate cache if templates changed since it was made
103    local template_dir="$CONFIG_DIR/matugen/templates"
104    local latest=0
105    for f in "$template_dir"/*; do
106        [ -f "$f" ] || continue
107        local t
108        t=$(stat -c '%Y' "$f")
109        [ "$t" -gt "$latest" ] && latest=$t
110    done
111    local cache_time
112    cache_time=$(stat -c '%Y' "$marker")
113    if [ "$latest" -gt "$cache_time" ]; then
114        rm -rf "$cache_dir"
115        return 1
116    fi
117 
118    for output in "${MATUGEN_OUTPUTS[@]}"; do
119        local src="$cache_dir/$output"
120        local dst="$CONFIG_DIR/$output"
121        if [ -f "$src" ]; then
122            mkdir -p "$(dirname "$dst")"
123            cp "$src" "$dst"
124        fi
125    done
126 
127    return 0
128}
129 
130cache_current_theme() {
131    local name="$1"
132    local cache_dir="$THEME_CACHE/$name"
133 
134    mkdir -p "$cache_dir"
135    for output in "${MATUGEN_OUTPUTS[@]}"; do
136        local src="$CONFIG_DIR/$output"
137        if [ -f "$src" ]; then
138            mkdir -p "$(dirname "$cache_dir/$output")"
139            cp "$src" "$cache_dir/$output"
140        fi
141    done
142    touch "$cache_dir/.done"
143}
144 
145# ── Helper: set static wallpaper ─────────────────────────────────
146set_static() {
147    local img="$1"
148    local name
149    name="$(basename "$img" | sed 's/\.[^.]*$//')"
150 
151    if ! command -v matugen &>/dev/null; then
152        alert "matugen not installed" critical
153        exit 1
154    fi
155 
156    echo "$img" > "$CACHE_DIR/current_wallpaper.txt"
157    ln -sf "$img" "$CACHE_DIR/current_wallpaper.png" 2>/dev/null || true
158 
159    # Stop any running live wallpaper first (awww can't override mpvpaper)
160    pkill mpvpaper 2>/dev/null || true
161 
162    if command -v awww &>/dev/null; then
163        if ! pgrep -x awww-daemon > /dev/null; then
164            awww-daemon &
165            sleep 0.5
166        fi
167        awww img "$img" \
168            --transition-type grow \
169            --transition-pos center \
170            --transition-duration 2 \
171            --transition-fps 60 || true
172    fi
173 
174    if ! apply_cached_theme "$name"; then
175        alert "Generating theme..." normal "Calculating color palette..."
176        backup_theme
177        if timeout 30 matugen image "$img" --config "$CONFIG_DIR/matugen/config.toml"; then
178            cache_current_theme "$name" || true
179            clear_backup
180        else
181            restore_theme
182            alert "Theme generation failed" critical "Could not generate colors for this wallpaper"
183            return 1
184        fi
185    fi
186 
187    "$CONFIG_DIR/matugen/reload-theme.sh" || true
188 
189    alert "Theme updated" normal "Wallpaper and colors applied"
190 
191    # Restart waybar to pick up new config.jsonc and CSS
192    if pgrep -x waybar > /dev/null; then
193        pkill -x waybar 2>/dev/null || true
194        sleep 0.3
195        waybar &>/dev/null &
196    fi
197}
198 
199# ── Helper: set live wallpaper ───────────────────────────────────
200set_live() {
201    local video="$1"
202    local name
203    name="$(basename "$video" | sed 's/\.[^.]*$//')"
204 
205    if ! command -v mpvpaper &>/dev/null; then
206        alert "mpvpaper not installed" critical "Install mpvpaper for live wallpapers"
207        exit 1
208    fi
209 
210    echo "$video" > "$CACHE_DIR/current_wallpaper.txt"
211 
212    pkill mpvpaper 2>/dev/null || true
213    mpvpaper -o "loop no-audio" '*' "$video"
214 
215    if command -v ffmpeg &>/dev/null && command -v matugen &>/dev/null; then
216        local frame="$CACHE_DIR/live-frame.jpg"
217        ffmpeg -i "$video" -vframes 1 "$frame" -y 2>/dev/null || true
218        if [ -f "$frame" ]; then
219            ln -sf "$frame" "$CACHE_DIR/current_wallpaper.png" 2>/dev/null || true
220            if ! apply_cached_theme "$name"; then
221                alert "Generating theme..." normal "Calculating color palette..."
222                backup_theme
223                if timeout 30 matugen image "$frame" --config "$CONFIG_DIR/matugen/config.toml"; then
224                    cache_current_theme "$name" || true
225                    clear_backup
226                else
227                    restore_theme
228                    alert "Theme generation failed" critical "Could not generate colors for this wallpaper"
229                    return 1
230                fi
231            fi
232            "$CONFIG_DIR/matugen/reload-theme.sh" || true
233 
234            alert "Live wallpaper" normal "$(basename "$video")"
235 
236            # Restart waybar to pick up new config.jsonc and CSS
237            if pgrep -x waybar > /dev/null; then
238                pkill -x waybar 2>/dev/null || true
239                sleep 0.3
240                waybar &>/dev/null &
241            fi
242        fi
243    fi
244}
245 
246# ── Generate thumbnail for video ─────────────────────────────────
247generate_thumb() {
248    local video="$1"
249    local thumb_name
250    thumb_name="$(basename "$video" | sed 's/\.[^.]*$//').jpg"
251    local thumb_path="$THUMB_DIR/$thumb_name"
252 
253    if [ ! -f "$thumb_path" ] && command -v ffmpeg &>/dev/null; then
254        mkdir -p "$(dirname "$thumb_path")"
255        ffmpeg -i "$video" -vframes 1 -vf "scale=180:-1" "$thumb_path" -y 2>/dev/null || true
256    fi
257 
258    if [ -f "$thumb_path" ]; then
259        echo "$thumb_path"
260    else
261        echo "$video"
262    fi
263}
264 
265# ── Pick static wallpaper ────────────────────────────────────────
266pick_static() {
267    local images
268    mapfile -t images < <(find "$WALL_DIR" -maxdepth 1 -type f \( -iname "*.jpg" -o -iname "*.jpeg" -o -iname "*.png" -o -iname "*.webp" \) | sort)
269 
270    if [ ${#images[@]} -eq 0 ]; then
271        alert "No wallpapers found" normal "Add images to $WALL_DIR/"
272        return 1
273    fi
274 
275    local selected
276    selected=$(
277        for f in "${images[@]}"; do
278            local thumb="$THUMB_DIR/$(basename "$f" | sed 's/\.[^.]*$//').jpg"
279            local display_name
280            display_name="$(basename "$f")"
281            if [ -f "$thumb" ]; then
282                echo -ne "$display_name\0icon\x1f$thumb\n"
283            else
284                echo -ne "$display_name\0icon\x1f$f\n"
285            fi
286        done \
287        | rofi -dmenu -i -p "Wallpapers" -theme "$ROFI_THEME" -show-icons
288    )
289 
290    if [ -z "$selected" ]; then
291        return 1
292    fi
293 
294    local target_file=""
295    for f in "${images[@]}"; do
296        if [ "$(basename "$f")" = "$selected" ] || [ "$f" = "$selected" ]; then
297            target_file="$f"
298            break
299        fi
300    done
301 
302    if [ -z "$target_file" ] && [ -f "$WALL_DIR/$selected" ]; then
303        target_file="$WALL_DIR/$selected"
304    fi
305 
306    if [ -z "$target_file" ] || [ ! -f "$target_file" ]; then
307        alert "Wallpaper not found" critical "$selected"
308        return 1
309    fi
310 
311    set_static "$target_file"
312}
313 
314# ── Pick live wallpaper ──────────────────────────────────────────
315pick_live() {
316    if [ ! -d "$LIVE_DIR" ]; then
317        alert "No live wallpapers" critical "Directory $LIVE_DIR does not exist"
318        return 1
319    fi
320 
321    local videos
322    mapfile -t videos < <(find "$LIVE_DIR" -maxdepth 1 -type f \( -iname "*.mp4" -o -iname "*.webm" -o -iname "*.mkv" -o -iname "*.gif" \) | sort)
323 
324    if [ ${#videos[@]} -eq 0 ]; then
325        alert "No live wallpapers" normal "Add videos to $LIVE_DIR/"
326        return 1
327    fi
328 
329    local selected
330    selected=$(
331        for f in "${videos[@]}"; do
332            thumb=$(generate_thumb "$f")
333            local display_name
334            display_name="$(basename "$f")"
335            echo -ne "$display_name\0icon\x1f$thumb\n"
336        done \
337        | rofi -dmenu -i -p "Live Wallpapers" -theme "$ROFI_THEME" -show-icons
338    )
339 
340    if [ -z "$selected" ]; then
341        return 1
342    fi
343 
344    local target_file=""
345    for f in "${videos[@]}"; do
346        if [ "$(basename "$f")" = "$selected" ] || [ "$f" = "$selected" ]; then
347            target_file="$f"
348            break
349        fi
350    done
351 
352    if [ -z "$target_file" ] && [ -f "$LIVE_DIR/$selected" ]; then
353        target_file="$LIVE_DIR/$selected"
354    fi
355 
356    if [ -z "$target_file" ] || [ ! -f "$target_file" ]; then
357        alert "Video not found" critical "$selected"
358        return 1
359    fi
360 
361    set_live "$target_file"
362}
363 
364# ── Main menu ────────────────────────────────────────────────────
365main_menu() {
366    local has_live=false
367    [ -d "$LIVE_DIR" ] && [ "$(find "$LIVE_DIR" -maxdepth 1 -type f 2>/dev/null | wc -l)" -gt 0 ] && has_live=true
368 
369    local icons_dir="$CONFIG_DIR/rofi/icons"
370    local choice
371    choice=$(
372        {
373            printf "Static wallpaper\0icon\x1f${icons_dir}/static.svg\n"
374            $has_live && printf "Live wallpaper\0icon\x1f${icons_dir}/live.svg\n"
375        } | rofi -dmenu -i -p "Wallpaper" -theme "$ROFI_POWER" -show-icons \
376            -theme-str 'listview { lines: 2; }'
377    )
378 
379    case "$choice" in
380        "Static wallpaper") pick_static ;;
381        "Live wallpaper")   pick_live ;;
382        *) exit 0 ;;
383    esac
384}
385 
386# ── Entry point ──────────────────────────────────────────────────
387if [ $# -eq 1 ] && [ -f "$1" ]; then
388    set_static "$1"
389elif [ $# -eq 1 ] && [ "$1" = "--live" ]; then
390    pick_live
391else
392    main_menu
393fi
394