#!/bin/sh

url_1337x="https://1337x.to"
proxy_1337x="https://1337x.torrentbay.to"
menu="dmenu"

die()
{
	: "${1:?}"

	command -v notify-send > /dev/null &&
		notify-send "  1337x" "$1"

	printf "\033[31;1merr: %b\033[0m\n" "$1"
	exit "${2:-1}"
}

dep_check()
{
	: "${1:?}"

	for dep; do
		command -v "$dep" 1>/dev/null ||
			die "$dep not found, please install it" 127
	done

	unset dep
}

get_1337x()
{
	: "${1:?}"
	dep_check "curl"

	curl --silent -H 'Accept-Encoding: gzip,deflate, br' --compressed "${url_1337x}/$1"
}

search_1337x()
{
	: "${1:?}"

	get_1337x "search/$(echo "$1" | tr ' ' '+')/1/" |
		grep -o "[0-9][0-9][0-9][0-9][0-9][0-9][0-9]/[^/]*"
}

magnet_1337x()
{
	: "${1:?}"

	get_1337x "torrent/$1/" |
		grep -om1 "magnet:?xt=urn:btih:.*announce"
}

main()
{
	fetch=
	select=
	magnet=
	out=
	query=

	[ -z "$DISPLAY" ] &&
		menu="wmenu"
	if [ "$#" -gt 0 ] && [ "$1" = "-o" ]
	then
		out=true
		shift
	fi

	dep_check "$menu"

	curl -Is "$url_1337x" > /dev/null ||
		url_1337x="$proxy_1337x"

	query="${*:-$(printf "" | "$menu" -p "  ")}"
	[ -z "$query" ] &&
		die "please enter a query"

	fetch="$(search_1337x "$query")"
	select="$(echo "$fetch" | sed -e's/^.*\///g' | tr '-' ' ' | "$menu" -l 25 -p "  " | tr ' ' '-')"
	magnet="$(magnet_1337x "$(echo "$fetch" | grep "$select")")"

	[ -z "$magnet" ] &&
		die "empty magnet"

	if [ "$out" = true ]
	then
		echo "$magnet"
	else
		dep_check "pirowatch"
		pirowatch "$magnet"
	fi
}

main "$@"