Corrected typo.
[utils.git] / autodlctl
CommitLineData
aba77738
DC
1#!/bin/bash
2
3paths=(/home/pub/video/anime $HOME/dc/autodl/cur)
4
5function getnext
6{
7 echo "checking $tag"
8 sexpr="`cat "$d/.autodl/sexpr"`"
9 unset badsizes
10 if [ -r "$d/.autodl/badlist" ]; then
11 read curep <"$d/.autodl/badlist"
12 echo "downloading bad files, curep $curep"
13 epfrom=badlist
aba77738
DC
14 elif [ -r "$d/.autodl/curep" ]; then
15 curep="`cat "$d/.autodl/curep"`"
16 echo "downloading series, curep $curep"
17 epfrom=curep
18 else
19 echo "no available episode of $tag" >&2
20 echo "$tag" >>"$HOME/dc/autodl/faulty"
21 return 1
22 fi
6c554f39 23 unset badsizesl
24 if [ -r "$d/.autodl/badsizes" ]; then
8c73da68 25 badsizesl="$(sed -n "s/^0*$curep \([^#]*\)\( *#.*\)\?$/\1/p" "$d/.autodl/badsizes")"
6c554f39 26 fi
45f2d3f3
DC
27 if [ -n "$badsizesl" ]; then
28 read -a badsizes <<<"$badsizesl"
29 unset badsizesl
30 echo "found bad size list: ${badsizes[@]}"
31 fi
aba77738
DC
32 unset args
33 fsexpr="`printf "$sexpr" "$curep"`"
34 if [ "${#badsizes[@]}" -gt 0 ]; then
35 for badsize in "${badsizes[@]}"; do
36 fsexpr="$fsexpr & ! S=$badsize"
37 done
38 fi
34091d15 39 infofile="$d/.autodl/rtinfo"
8c73da68 40 estatfile="$d/.autodl/estat"
32dd9fcf 41 args=(-e "$fsexpr" -t "$tag $curep" -I "$infofile" -E "$estatfile" -x "curep=$curep")
9151870a 42 if [ -e "$d/.autodl/wait" ]; then
43 args=("${args[@]}" -w)
44 fi
aba77738
DC
45 if [ -r "$d/.autodl/uarg" ]; then
46 uarg="`cat "$d/.autodl/uarg"`"
47 elif [ -e "$d/.autodl/autouarg" ]; then
48 uarg="rename:$tag - %02i.avi:move:../autodl/cur/$tag"
49 fi
50 if [ -n "$uarg" ]; then
51 fuarg="`printf "$uarg" "$curep"`"
52 args=("${args[@]}" -a "$fuarg")
53 fi
58a2c684 54 outfile="$d/.autodl/output"
aba77738
DC
55 echo "trying to download -- autodl ${args[@]}"
56 intr=n
57 autodl "${args[@]}" >"$outfile" 2>&1 &
58 pid=$!
59 trap "intr=y; kill -INT $pid" USR1 INT
60 wait $pid
61 stat=$?
8c73da68 62 if [ -r "$estatfile" ]; then
63 estat="`cat "$estatfile"`"
64 rm -f "$estatfile"
65 fi
aba77738
DC
66 if [ "$intr" = y ]; then
67 echo "$tag interrupted"
68 else
69 if [ "$stat" -ne 0 ]; then
58a2c684 70 if [ "$stat" -eq 1 ]; then
71 echo "Failure for $tag" >>"$HOME/dc/autodl/errorlog"
72 tail -n 20 "$outfile" >>"$HOME/dc/autodl/errorlog"
73 elif [ "$stat" -eq 2 ]; then
74 echo "Connection error on $tag"
75 elif [ "$stat" -eq 3 ]; then
76 echo "Configuration error, disabling $tag"
77 cp "$outfile" "$d/.autodl/conferr"
78 touch "$d/.autodl/disabled"
07878795 79 echo "$tag" >>"$HOME/dc/autodl/faulty"
58a2c684 80 fi
aba77738 81 else
8c73da68 82 echo "episode $curep of $tag done (estat: \"$estat\")"
aba77738
DC
83 case "$epfrom" in
84 badlist)
85 echo -en "${tag}\n${curep}\n" >>"$HOME/dc/autodl/baddone"
86 egrep -v "^$curep( |\$)" "$d/.autodl/badlist" >"$d/.autodl/newbadlist"
87 mv -f "$d/.autodl/newbadlist" "$d/.autodl/badlist"
88 if [ `wc -l <"$d/.autodl/badlist"` -eq 0 ]; then
89 rm "$d/.autodl/badlist"
90 if [ -r "$d/.autodl/curep" ]; then
91 if [ -r "$d/.autodl/maxep" ] && [ "`cat "$d/.autodl/curep"`" -gt "`cat "$d/.autodl/maxep"`" ]; then
92 touch "$d/.autodl/disabled"
93 fi
94 else
95 touch "$d/.autodl/disabled"
96 fi
97 echo "$tag has no more bad episodes"
98 echo "$tag" >>"$HOME/dc/autodl/badmaxed"
99 fi
100 ;;
101 curep)
8c73da68 102 if [ "$estat" = dbl ]; then
103 echo -en "${tag}\n${curep}\n" >>"$HOME/dc/autodl/done"
104 echo -en "${tag}\n$((${curep} + 1))\n" >>"$HOME/dc/autodl/done"
105 let curep+=2
106 else
107 echo -en "${tag}\n${curep}\n" >>"$HOME/dc/autodl/done"
108 let curep++
109 fi
aba77738
DC
110 echo "$curep" >"$d/.autodl/curep"
111 if [ -r "$d/.autodl/maxep" ]; then
112 if [ "$curep" -gt "`cat "$d/.autodl/maxep"`" ]; then
113 touch "$d/.autodl/disabled"
114 echo "$tag has reached max"
115 echo "$tag" >>"$HOME/dc/autodl/maxed"
116 fi
117 fi
118 ;;
119 esac
120 fi
121 fi
122 rm -f "$outfile"
123 rm -f "$HOME/dc/autodl/run/$tag"
124}
125
126for dir in $HOME/dc/autodl{,/cur,/run}; do
127 if [ -e "$dir" ]; then
128 if [ ! -d "$dir" ]; then
129 echo "$dir is not a directory, please remedy and restart" >&2
130 exit 1
131 fi
132 else
133 mkdir "$dir" || exit 1
134 fi
135done
136
137while [ $# -gt 0 ]; do
138 arg="$1"
139 shift
140 case "$arg" in
141 -k)
142 pid="$(cat "$HOME/dc/autodl/run/master")"
143 if [ -z "$pid" ]; then
144 echo "autodlctl: could not read a PID from $HOME/dc/autodl/run/master" >&2
145 exit 1
146 fi
147 kill "$pid"
148 exit 0
149 ;;
150 *)
151 echo "autodlctl: unrecognized option: \"$arg\"" >&2
152 exit 1
153 ;;
154 esac
155done
156
157lastget=0
158done=n
159trap "done=y" INT QUIT TERM
160echo $$ >"$HOME/dc/autodl/run/master"
161while [ "$done" != y ]; do
162 for pidfile in $HOME/dc/autodl/run/*; do
163 if [ "$pidfile" = "$HOME/dc/autodl/run/*" ]; then break; fi
164 pid="`cat "$pidfile"`"
165 if [ -d /proc/1 -a ! -d "/proc/$pid" ]; then
166 echo "removing stale pidfile $pidfile"
167 rm -f "$pidfile"
168 fi
169 done
170 for p in "${paths[@]}"; do
171 for d in "$p"/*; do
172 if [ -d "$d/.autodl" -a ! -e "$d/.autodl/disabled" ]; then
173 if [ -r "$d/.autodl/tag" ]; then
174 tag="`cat "$d/.autodl/tag"`"
175 else
176 tag="`basename "$d"`"
177 fi
07878795 178 start=y
aba77738
DC
179 if [ -e "$d/.autodl/disable" ]; then
180 echo "disabling $tag per user request"
07878795 181 start=n
aba77738
DC
182 touch "$d/.autodl/disabled"
183 rm -f "$d/.autodl/disable"
184 if [ -r "$HOME/dc/autodl/run/$tag" ]; then
185 pid="`cat "$HOME/dc/autodl/run/$tag"`"
186 echo "sending SIGUSR1 to $pid"
187 kill -USR1 "$pid"
188 else
189 echo "could not find pid for $tag"
190 fi
07878795 191 fi
192 if [ -e "$d/.autodl/restart" ]; then
193 echo "restarting $tag per user request"
194 rm -f "$d/.autodl/restart"
195 if [ -r "$HOME/dc/autodl/run/$tag" ]; then
196 pid="`cat "$HOME/dc/autodl/run/$tag"`"
197 echo "sending SIGUSR1 to $pid"
198 kill -USR1 "$pid"
ccc141a3 199 while [ -e "$HOME/dc/autodl/run/$tag" ]; do
200 echo "waiting for it to exit"
201 sleep 1
202 done
07878795 203 else
204 echo "could not find pid for $tag"
205 fi
206 fi
207 if [ $start = y ]; then
208 if [ ! -r "$d/.autodl/sexpr" ]; then
209 touch "$d/.autodl/disabled"
210 echo "$tag lacks sexpr" >&2
211 echo "$tag" >>"$HOME/dc/autodl/faulty"
212 else
213 if [ ! -e "$HOME/dc/autodl/run/$tag" ]; then
214 if [ $((`date +%s` - $lastget)) -gt 20 ]; then
215 getnext "$d" "$tag" &
216 lastget=`date +%s`
217 pid=$!
218 echo "$pid" >"$HOME/dc/autodl/run/$tag"
219 fi
aba77738
DC
220 fi
221 fi
222 fi
223 fi
224 done
225 done
226 sleep 10
227done
228
229for pidfile in $HOME/dc/autodl/run/*; do
230 if [ "$pidfile" = "$HOME/dc/autodl/run/*" ]; then break; fi
231 if [ "$(basename "$pidfile")" = master ]; then continue; fi
232 pid="`cat "$pidfile"`"
233 echo "sending SIGUSR1 to $pid for `basename "$pidfile"`"
234 kill -USR1 "$pid"
235done
236
237rm -f "$HOME/dc/autodl/run/master"