pirowatch: rewrite: proper error handling, -s flag to skip index selection

changes
* new -s flag, it can be used speed up the launching when source doesn't
  contain more than one media
* proper exit messages
* follow the xdg base directory specification

2 files changed, 78 insertions, 31 deletions

diff --git a/README.md b/README.md
index b67029a..08904bd 100644
--- a/README.md
+++ b/README.md
@@ -14,7 +14,9 @@ the stream option. mix together varous ambient sounds and stream with damb
 
 ## pirowatch
 pirowatch is a dmenu wrapper for [web torrent](https://webtorrent.io/), provide magnet or torrent files
-as command line argument or input from dmenu and consoom the media using [mpv](https://mpv.io/)
+as command line argument or input from dmenu and consoom the media using [mpv](https://mpv.io/).
+if you pass "-s" as the first argument it'll skip index selection to speed up
+the launching
 
 ## 1337x
 scraper for 1337x.to, has an optional dependency on pirowatch. if you pass "-o"
diff --git a/pirowatch b/pirowatch
index f0fbb88..df4bb3f 100755
--- a/pirowatch
+++ b/pirowatch
@@ -1,42 +1,87 @@
 #!/bin/sh
-# https://sinanmohd.com
 
-####################
-## user variables ##
-####################
-# dir where files are stored
-store_path="${HOME}"/.cache/pirowatch
+# dir where files are cached
+cache_dir="${XDG_CACHE_HOME:-${HOME}/.cache}/pirowatch"
+# dir where torrent files are stored
+torrent_dir="${DOWNLOADS:-$HOME/Downloads}"
 # torrent port
 torrent_port=49110
 # dht port
 dht_port=49130
 
-get_selection() {
-        list=$(webtorrent --mpv -o "$HOME"/.cache/pirowatch --torrent-port "$torrent_port" --dht-port "$dht_port" "$1" --select |
-                grep -i ".mp4\|.mkv\|.webm\|.avi\|.mov\|.flac\|.opus\|.mp3\|.flac\|.wav")
+die()
+{
+	# usage: die "reason" [exit_status]
+	: "${1:?}"
 
-        if [ "$(echo "$list" | wc -l)" = 1 ]; then
-                printf "%s" "$list" | cut -d ' ' -f 1
-        else
-                printf "%s" "$list" | dmenu -p "  " -l 25  | cut -d ' ' -f 1
-        fi
+	notify-send "  pirowatch" "$1"
+	printf "\033[31;1merr: %b\033[0m\n" "$1"
+	exit "${2:-1}"
 }
 
-start_stream() {
-        webtorrent --mpv -o "$store_path" --torrent-port "$torrent_port" --dht-port "$dht_port" "$1" --select "$2"
+dep_check()
+{
+	# usage: dep_check "dep_1" ...
+	: "${1:?}"
+
+	for dep; do
+		command -v "$dep" 1>/dev/null ||
+			die "$dep not found, please install it" 127
+	done
+
+	unset dep
+}
+
+stream() {
+	# usage: stream "torrent"
+	: "${1:?}"
+
+	notify-send " connecting to peers" "wait for few seconds"
+
+	if [ -n "$2" ]
+	then
+		webtorrent --mpv -o "$cache_dir" --torrent-port "$torrent_port" --dht-port "$dht_port" "$1" --select "$2"
+	else
+		webtorrent --mpv -o "$cache_dir" --torrent-port "$torrent_port" --dht-port "$dht_port" "$1"
+	fi
+}
+
+set_index() {
+	# usage: get_index "torrent"
+	: "${1:?}"
+
+        fetch=$(webtorrent --mpv -o "$cache_dir" --torrent-port "$torrent_port" --dht-port "$dht_port" "$1" --select |
+		grep -Ei "\.(mkv|mp4|webm|avi|mov|flv|flac|opus|ogg|mp3|wav) (.*)$")
+        if [ "$(echo "$fetch" | wc -l)" -gt 1 ]
+	then
+		fetch="$(printf "%s" "$fetch" | dmenu -p "  " -l 25)" ||
+			die "empty selection" 66
+	fi
+
+	index="${fetch%%[[:space:]]*}"
+
+	unset fetch
+}
+
+main()
+{
+	index="tobeset"
+	if [ "$1" = "-s" ]
+	then
+		index=
+		shift
+	fi
+	torrent="${1:-$torrent_dir/$(find "$torrent_dir" -type f -name "*.torrent" | sed 's/.*\///g' | dmenu -p "  " -l 25)}"
+
+	dep_check "webtorrent" "mpv" "notify-send"
+	[ -z "${torrent##"${torrent_dir}"/}" ] &&
+		die "magnet or torrent file not entered" 66
+	[ -d "$cache_dir" ] ||
+		mkdir -p "$cache_dir"
+	[ "$index" = "tobeset" ] &&
+		set_index "$torrent"
+
+	stream "$torrent" "$index"
 }
 
-if [ -z "$1" ]; then
-        magnet=$(find "$DOWNLOADS"  -type f -name "*.torrent" | dmenu -p "  " -l 25)
-        [ -z "$magnet" ] && exit
-else
-        magnet="$1"
-fi
-
-selection=$(get_selection "$magnet")
-if [ -z "$selection" ]; then
-        exit
-else
-        notify-send " Connecting to peers" "please wait for few seconds"
-        start_stream "$magnet" "$selection"
-fi
+main "$@"