1 files changed, 161 insertions, 0 deletions
diff --git a/.local/bin/waldl b/.local/bin/waldl
new file mode 100755
index 0000000..d90dec2
--- /dev/null
+++ b/.local/bin/waldl
@@ -0,0 +1,161 @@
+#!/bin/sh
+
+# https://git.surgot.tech/waldl
+# script to find and download wallpapers from wallhaven
+version="0.0.1"
+
+# Usage:
+# 	waldl <query>
+# if query left empty then sh_menu will be used (dmenu by default)
+# after the thumbnails are cached, sxiv would open up with the thumbnails
+# Select the wallpapers using `m` on the image. ( marking the image in sxiv )
+# press `q` to quit sxiv, the marked images would start downloading
+
+####################
+## User variables ##
+####################
+
+[ -z $VIEWER ] && VIEWER=sxiv
+# the dir where wallpapers are stored
+walldir="$HOME/pix/wall"
+# the dir used to cache thumbnails
+cachedir="$HOME/.cache/wall"
+# sxiv options
+sxiv_otps=" -tfpo -z 200" # o is needed for selection
+# number of pages to show in search results
+# each page contains 24 results
+max_pages=4
+# sorting : date_added, relevance, random, views, favorites, toplist
+sorting=relevance
+# quality : large original small
+quality=large
+# atleast : least res
+atleast=1920x1080
+
+# allow the user to customize the defaults
+[ -e "$HOME/.config/waldlrc" ] && . "$HOME/.config/waldlrc"
+
+# the menu command used when no query is provided
+sh_menu () {
+	: | dmenu -p "search wallhaven:"
+	# ROFI: comment the previous line and uncomment the next line for rofi
+	# rofi -dmenu -l 0 -p "search wallpapers"
+}
+
+##########################
+## getting search query ##
+##########################
+
+[ -n "$*" ] && query="$*" || query=$( sh_menu )
+[ -z "$query" ] && exit 1
+query=$(printf '%s' "$query" | tr ' ' '+' )
+
+######################
+## start up commands #
+######################
+
+rm -rf "$cachedir"
+mkdir -p "$walldir" "$cachedir"
+
+# progress display command
+sh_info () {
+	printf "%s\n" "$1" >&2
+	notify-send "wallhaven" "$1"
+	[ -n "$2" ] && exit "$2"
+}
+
+# dependency checking
+dep_ck () {
+	for pr; do
+		command -v $pr >/dev/null 2>&1 || sh_info "command $pr not found, install: $pr" 1
+	done
+}
+dep_ck "$VIEWER" "curl" "jq"
+
+
+# clean up command that would be called when the program exits
+clean_up () {
+	printf "%s\n" "cleaning up..." >&2
+	rm -rf "$datafile" "$cachedir"
+}
+
+# data file to store the api information
+datafile="/tmp/wald.$$"
+
+# clean up if killed
+trap "exit" INT TERM
+trap "clean_up" EXIT
+
+##################
+## getting data ##
+##################
+
+# request the search results for each page
+get_results () {
+	for page_no in $(seq $max_pages)
+	do
+		{
+			json=$(curl -s -G "https://wallhaven.cc/api/v1/search" \
+					-d "q=$1" \
+					-d "page=$page_no" \
+					-d "atleast=$atleast" \
+					-d "sorting=$sorting"
+				)
+			printf "%s\n" "$json" >> "$datafile"
+		} &
+		sleep 0.001
+	done
+	wait
+}
+
+# search wallpapers
+sh_info "getting data..."
+get_results "$query"
+
+# check if data file is empty, if so then exit
+[ -s "$datafile" ] || sh_info "no images found" 1
+
+############################
+## downloading thumbnails ##
+############################
+
+# get a list of thumnails from the data
+thumbnails=$( jq -r '.data[]?|.thumbs.'"$quality" < "$datafile")
+
+[ -z "$thumbnails" ] && sh_info "no-results found" 1
+
+# download the thumbnails
+sh_info "caching thumbnails..."
+for url in $thumbnails
+do
+		printf "url = %s\n" "$url"
+		printf "output = %s\n" "$cachedir/${url##*/}"
+done | curl -Z -K -
+#sh_info "downloaded thumbnails..."
+
+###########################
+## user selection (sxiv) ##
+###########################
+
+# extract the id's out of the thumbnail name
+image_ids="$($VIEWER $sxiv_otps "$cachedir")"
+[ -z "$image_ids" ] && exit
+
+#########################
+## download wallpapers ##
+#########################
+
+# download the selected wall papers
+cd "$walldir"
+sh_info "downloading wallpapers..."
+for ids in $image_ids
+do
+	ids="${ids##*/}"
+	ids="${ids%.*}"
+	url=$( jq -r '.data[]?|select( .id == "'$ids'" )|.path' < "$datafile" )
+	printf "url = %s\n" "$url"
+	printf -- "-O\n"
+done | curl -K -
+
+sh_info "wallpapers downloaded in:- '$walldir'"
+$VIEWER $(ls -c)