blob: 0e2b64d5e768c0ab33bda78a99d241f73c0fec9b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
|
#!/bin/sh
PROG="$(basename "$0")"
## VARIABLES
# copy command and deps variable
deps="pup curl clipp transmission-cli"
MODULES_PATH=$HOME/.local/share/tsh
module='nyaa.sh' # default module
# Files
export tmp="/tmp/$PROG"
files="seeds sizes names html tmp_types"
# Use export so that these variables can be used inside of modules
for file in $files; do
eval "export $file=$tmp/$file"
done
# Files not in $files won't be deleted
export results="$tmp/results"
export links="$tmp/links"
categories="music anime movies shows other software games isos books"
logn() { >&2 printf '%s\n' "$*"; }
die() {
logn "$@"
exit 1
}
confirm() {
>&2 printf '%s' "$1"
head -n 1 | grep -q "[yY]"
}
# Remove temp files and quit
cleanup() {
for file in $files; do
eval "rm -f \$$file"
done
}
list_modules() { find -L "$MODULES_PATH" -type f -printf "%f\n"; }
trap "exit 1" INT
trap "cleanup" EXIT
## OPTIONS
skip=0
while getopts ":hm:rs:c:fd" opt; do
case $opt in
h)
>&2 cat <<-EOF
Usage: $PROG [options] query
Options:
-h Show this help message and exit
-r Skip getting pages and use last results
-s OPTION Sort results based on the specified OPTION
Available options: seeds, size, name
-m MODULE Select a module, if MODULE is 'list',
lists out available modules
-c CATEGORY Select category
-f Do not list files
-d Download without confirming
EOF
exit
;;
c)
[ "$OPTARG" = "list" ] && >&2 printf '%s\n' "$categories" && exit
category="$(printf '%s\n' "$categories" | tr ' ' '\n' | grep -m 1 "^$OPTARG")"
[ -z "$category" ] && die "No valid category for '$OPTARG'"
logn "category: $category"
;;
f) noaskfiles="1" ;;
d) noaskdownload="1" ;;
m)
[ "$OPTARG" = "list" ] && list_modules && exit
module="$(list_modules | grep -m 1 "^$OPTARG")"
[ -z "$module" ] && die "No valid module for '$OPTARG'"
logn "module: $module"
;;
r)
[ ! -r "$results" ] && die "No previous results found."
skip=1
;;
s)
case $OPTARG in
"seeds") sort_results() { sort -k3 -n -r; } ;;
"size") sort_results() { sort -k2 -h -r; } ;;
"name") sort_results() { sort -k4; } ;;
*) die "argument '$OPTARG' not seeds,size,name" ;;
esac
;;
:) die "Option -$OPTARG requires an argument" ;;
?) die "Invalid option: -$OPTARG" ;;
esac
done
shift $((OPTIND - 1))
for p in $deps; do
if ! command -v "$p" >/dev/null; then
logn "E: Program '$p' not found."
error=1
fi
done
[ "${error:-0}" -eq 1 ] && exit 1
# Get the torrents with module
if [ $skip -eq 1 ]; then
module="$(tail -n 1 "$links")"
else
mkdir -p "$tmp"
query="$(
if [ "${query:="$*"}" ]; then
printf "%s" "$query"
else
>&2 printf ' > '
head -n 1
fi | tr ' ' '+'
)"
[ "$query" ] || exit 1
# shellcheck source=/usr/local/lib/$PROG/nyaa.sh disable=SC1091
. "$MODULES_PATH/$module"
rm -f "$results" "$links"
get_torrents
[ -f "$results" ] || die "No results."
# Save which module was used (hack)
printf "%s\n" "$module" >>"$links"
fi
command -v sort_results >/dev/null || sort_results() { sort -k3 -n -r; }
# select result from "$results"
for choice in $(
# Select result(s) from the result file sorted with sort_results
# with fzf
awk '{print NR, $0}' "$results" |
sort_results |
column -t -l 3 |
fzf -m --with-nth 2.. |
awk 'BEGIN{OFS=" "} {print $1}'
); do
printf 'choice: %s\n' "$(sed -n "${choice}p" "$results" | cut -f 3-)"
magnet="$(get_magnet "$choice")"
[ "$magnet" ] || exit 1
if [ -z "$noaskfiles" ] && confirm 'files?'; then
hash="${1##*btih:}"
hash="${hash%%&*}"
# Download the torrent file from a torrent website
curl -s "https://itorrents.org/torrent/${hash}.torrent" >"$tmp"/.torrent
transmission-show "$tmp"/.torrent | sed -n '/^FILES/,$p' | head -n -1 | tail -n +3 >&2
rm -f "$tmp"/.torrent
fi
if [ "$noaskdownload" ] || confirm 'download?'; then
[ "$category" ] || category="$(printf '%s' "$categories" | tr ' ' '\n' | fzf)"
[ "$category" ] || exit 2
transmission-remote debuc.com -a "$magnet" -w "/downloads/$category"
elif confirm "copy?"; then
echo "$magnet" | clipp
fi
done
|
