Mercurial > ec-dotfiles
annotate moefetch.sh @ 228:5d3a0645b504
- Restructured some things.
- Removed -- from commands. Replaced with safe_path command.
- Updated help.
- All global variables should be initally set at init_globals command.
author | edhoprima@gmail.com <edhoprima@gmail.com> |
---|---|
date | Tue, 15 Dec 2009 04:39:46 +0000 |
parents | 8b1f6f6b6a3b |
children | b03fef57b465 |
rev | line source |
---|---|
221
e891b563b797
wrong rule caused mass headache
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
220
diff
changeset
|
1 #!/bin/sh |
148 | 2 |
3 # Copyright (c) 2009, edogawaconan <me@myconan.net> | |
4 # | |
5 # Permission to use, copy, modify, and/or distribute this software for any | |
6 # purpose with or without fee is hereby granted, provided that the above | |
7 # copyright notice and this permission notice appear in all copies. | |
8 # | |
9 # THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES | |
10 # WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF | |
11 # MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR | |
12 # ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES | |
13 # WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN | |
14 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF | |
15 # OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. | |
159 | 16 # |
17 # Lots of bugs here. Use with care | |
148 | 18 # USE WITH CARE |
159 | 19 # |
20 # what it does: fetch every picture that has the specified TAGS. | |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
21 # requirement: wget, libxslt, openssl |
148 | 22 |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
23 # program additional paths for: cut, sed, wc, openssl, wget, xsltproc, grep |
159 | 24 ADDITIONAL_PATH= |
148 | 25 |
159 | 26 # default server address. Danbooru only! I do not take responsibility of stupidity. |
27 DEFAULT_SITE="moe.imouto.org" | |
148 | 28 |
29 # base directory. make sure it's writeable. I do not take responsibility if you don't own the folder and files as no check is done for this one. | |
159 | 30 # Structure is ${BASE_DIR}/<TAGS> |
31 # Absolute path only. | |
32 # Leave empty to use whatever folder you're running this at | |
193 | 33 BASE_DIR= |
148 | 34 |
35 # not user modifiable from here | |
36 | |
193 | 37 ### TODO: |
211 | 38 ### - sanity validator(?) |
193 | 39 ### - unified repository to save bandwidth |
40 ### - bug stomping | |
41 ### - sanity checking | |
227 | 42 ### - MOAR comments |
193 | 43 ### WILL BE FOR 0.3 |
44 | |
159 | 45 # useless welcome message. Also version |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
46 msg_welcome() { |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
47 echo "moefetch ${_version} |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
48 Copyright (c) 2009 edogawaconan <me@myconan.net> |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
49 " |
159 | 50 } |
51 | |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
52 # Sanitize path. Totally safe. Usage: cmd "$(safe_path "${filename}")" |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
53 safe_path() |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
54 { |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
55 # It all depends on the first character. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
56 __start=$(printf "%s" "$*" | cut -c 1) |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
57 __path= |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
58 case "${__start}" in |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
59 .|/) __path="$*";; # . and / is safe. No change. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
60 *) __path="./$*";; # Anything else must be prefixed with ./ |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
61 esac |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
62 printf "%s" "${__path}" # Return. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
63 } |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
64 |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
65 # Checks md5. OpenSSL should be available on anything usable. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
66 get_md5() { cat "$(safe_path "${1}")" | openssl dgst -md5; } |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
67 |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
68 # Safely get basename. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
69 get_basename() { basename "$(safe_path "${1}")"; } |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
70 |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
71 # Safely get filename (basename without the extension). |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
72 get_filename() { get_basename "${1%.*}"; } |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
73 |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
74 # Transformation for tag url. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
75 get_cleantags() { printf "%s " "$*" | sed -e 's/\&/%26/g;s/=/%3D/g'; } |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
76 |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
77 # Returns something if not an md5 value. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
78 is_not_md5() { get_filename "$1" | sed -e 's/\([0-9a-f]\{32\}\)//g'; } |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
79 |
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
80 |
159 | 81 # fatal error handler |
82 Err_Fatal() { | |
214
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
83 echo " |
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
84 Fatal error: ${1}" |
159 | 85 exit 1 |
86 } | |
87 | |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
88 Err_Impossible() { |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
89 echo " |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
90 Impossible error. Or you modified content of the working directories when the script is running. |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
91 Please report to moefetch.googlecode.com if you see this message (complete with entire run log)" |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
92 exit 1 |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
93 } |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
94 |
159 | 95 # help message |
96 Err_Help() { | |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
97 echo "moefetch.sh COMMAND [-n] [-p PASSWORD] [-s SITE_URL] [-u USERNAME] TAGS |
174
0948e76a57a1
added help. Bump to 0.1-beta2
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
173
diff
changeset
|
98 |
176
3d2ae9417273
even more improvement
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
175
diff
changeset
|
99 COMMAND: |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
100 (quick)fetch: |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
101 Do a complete update. Add prefix quick to skip file checking |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
102 check: |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
103 Get list of new files, clean up local folder and print total new files |
175
5b7a154dbd21
cosmetics fix for help message
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
174
diff
changeset
|
104 |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
105 OPTIONS: |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
106 -n: |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
107 Skip checking repository directory. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
108 -p PASSWORD: |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
109 Specifies password for login. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
110 -s SITE_URL: |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
111 Specify URL of the Danbooru powered site you want to leech from. Default is ${DEFAULT_SITE}. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
112 -u USERNAME: |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
113 Specifies username for login. |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
114 TAGS: |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
115 Tags you want to download. Separated by spaces. Tag name follows standard Danbooru tagging scheme." |
193 | 116 exit 2 |
159 | 117 } |
118 | |
119 # generate link by transforming xml | |
120 Generate_Link() { | |
214
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
121 echo " |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
122 Fetching XML file" |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
123 __tempnum=1000 |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
124 __iternum=1 |
195
652d9e268cee
test migration to printf
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
194
diff
changeset
|
125 > "${TEMP_PREFIX}-list" |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
126 while [ "${__tempnum}" -ge 1000 ]; do |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
127 __url="http://${SITE}/post/index.xml?tags=$(get_cleantags "${TAGS}")&offset=0&limit=1000&page=${__iternum}" |
227 | 128 [ ${_use_login} -eq 1 ] && __url="${__url}&login=${LOGIN_USER}&password_hash=${LOGIN_PASS}" |
129 wget "${__url}" -O "${TEMP_PREFIX}-xml" -e continue=off || Err_Fatal "Failed download catalog file" | |
214
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
130 printf "Processing XML file... " |
213
dd95cf01602c
working around limit
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
212
diff
changeset
|
131 # xslt evilry |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
132 xsltproc - "${TEMP_PREFIX}-xml" <<EOF | sed 's/.*\(http.*\)\(\/[a-f0-9]\{32\}\).*\.\([^\.]*\)/\1\2.\3/g' | grep ^http > "${TEMP_PREFIX}-templist" |
148 | 133 <xsl:stylesheet xmlns:xsl="http://www.w3.org/1999/XSL/Transform" version="1.0"> |
134 <xsl:output method="xml" indent="yes"/> | |
135 <xsl:template match="post"> | |
136 <xsl:value-of select="@file_url" /> | |
137 </xsl:template> | |
138 </xsl:stylesheet> | |
139 EOF | |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
140 __tempnum=$(echo $(wc -l < "${TEMP_PREFIX}-templist")) |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
141 __iternum=$((__iternum + 1)) |
213
dd95cf01602c
working around limit
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
212
diff
changeset
|
142 cat "${TEMP_PREFIX}-templist" >> "${TEMP_PREFIX}-list" |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
143 echo "${__tempnum} file(s) available" |
213
dd95cf01602c
working around limit
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
212
diff
changeset
|
144 done |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
145 numfiles=$(echo $(wc -l < "${TEMP_PREFIX}-list")) |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
146 echo "${numfiles} file(s) available on server" |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
147 [ "${numfiles}" -gt 0 ] || Err_Fatal "Error in processing list or no files can be found with specified tag(s) or site." |
214
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
148 } |
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
149 |
148 | 150 |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
151 progress_init() { |
205
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
152 _last="-" |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
153 printf "${_last}" |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
154 } |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
155 |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
156 progress_anim() { |
205
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
157 case "${_last}" in |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
158 /) _last="-";; |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
159 -) _last=\\;; |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
160 \\) _last=\|;; |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
161 \|) _last="/";; |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
162 esac |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
163 printf "\b${_last}" |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
164 } |
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
165 |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
166 progress_done() { printf "\bdone\n"; } |
205
2e866999c042
now with useless animation
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
204
diff
changeset
|
167 |
200 | 168 # getting rid of ls (as per suggestion) |
169 Count_Files() { | |
224 | 170 numfiles=0 |
171 for dircontent in "${*}/"* "${*}/".*; do | |
172 if [ "${dircontent}" != "${*}/*" ] || [ -e "${dircontent}" ]; then | |
173 numfiles=$((numfiles + 1)) | |
200 | 174 fi |
175 done | |
224 | 176 echo $((numfiles - 2)) |
200 | 177 } |
178 | |
159 | 179 # check tools availability |
180 Check_Tools() { | |
181 # verify all programs required do indeed exist | |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
182 commands="cut sed wc wget xsltproc xargs rm mkdir chown comm grep date openssl" |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
183 for cmd in ${commands} |
159 | 184 do |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
185 [ "$(command -v "${cmd}")" ] || Err_Fatal "${cmd} doesn't exist in ${PATH}" |
159 | 186 done |
187 } | |
188 | |
189 # verify required folders exist and writeable | |
190 Check_Folders(){ | |
223 | 191 [ -O "${BASE_DIR}" ] || Err_Fatal "You don't own ${BASE_DIR}. Please fix ${BASE_DIR} or run this script in your own directory." |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
192 for directory in temp trash deleted "${SITE_DIR}/${TARGET_DIR}"; do |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
193 if [ ! -d "${BASE_DIR}/${directory}" ]; then |
216
a869987c4646
did I say 'mess up'?
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
215
diff
changeset
|
194 mkdir -p "${BASE_DIR}/${directory}" || Err_Impossible |
159 | 195 fi |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
196 if [ ! -O "${BASE_DIR}/${directory}" ]; then |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
197 echo "You don't own the ${BASE_DIR}/${directory}, applying globally writeable permission on it" |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
198 chmod -R u=rwX,g=rwX,o=rwX "${BASE_DIR}/${directory}" || Err_Impossible |
159 | 199 fi |
200 done | |
223 | 201 [ "$(Count_Files "${BASE_DIR}/${SITE_DIR}/${TARGET_DIR}")" -eq 0 ] && ISNEW=1 |
201
30d2fb656029
scrapping grep -vf
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
200
diff
changeset
|
202 for i in error ok list newlist templist; do |
196 | 203 touch "${TEMP_PREFIX}-${i}" || Fatal_Err "Error creating ${TEMP_PREFIX}-${i}. This shouldn't happen" |
159 | 204 done |
205 # | |
206 } | |
207 | |
187
efd957294c8c
refactoring. cleanup. etc.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
186
diff
changeset
|
208 # Do some cleanup |
efd957294c8c
refactoring. cleanup. etc.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
186
diff
changeset
|
209 Cleanup_Repository() { |
efd957294c8c
refactoring. cleanup. etc.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
186
diff
changeset
|
210 # current dir: ${BASE_DIR}/${SITE_DIR}/${TARGET_DIR} |
207
17d816a63b4c
final progress version
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
206
diff
changeset
|
211 printf "Cleaning up repository folder... " |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
212 progress_init |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
213 trash_dir="${BASE_DIR}/trash/${trash_dir}/$(date -u "+${SITE_DIR}-${TARGET_DIR}-%Y%m%d-%H.%M")" |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
214 trashes="These files have been moved to ${trash_dir}:" |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
215 has_trash= |
223 | 216 if [ ! -d "${trash_dir}" ]; then |
216
a869987c4646
did I say 'mess up'?
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
215
diff
changeset
|
217 mkdir -p "${trash_dir}" || Err_Impossible |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
218 else |
223 | 219 if [ ! -O "${trash_dir}" ]; then |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
220 chmod -R u=rwX,g=rwX,o=rwX "${BASE_DIR}/${directory}" || Err_Impossible |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
221 fi |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
222 fi |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
223 for trash in "${BASE_DIR}/${SITE_DIR}/${TARGET_DIR}/"* |
187
efd957294c8c
refactoring. cleanup. etc.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
186
diff
changeset
|
224 do |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
225 is_trash= |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
226 if [ -d "${trash}" ] || [ -n "$(is_not_md5 "${trash}")" ] || [ -z "$(grep "$(get_basename "${trash}")" "${TEMP_PREFIX}-list")" ]; then |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
227 is_trash=1 |
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
228 has_trash=1 |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
229 mv -f -- "${trash}" "${trash_dir}" || Err_Impossible |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
230 trashes="${trashes} |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
231 $(get_basename "${trash}")" |
187
efd957294c8c
refactoring. cleanup. etc.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
186
diff
changeset
|
232 fi |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
233 progress_anim |
187
efd957294c8c
refactoring. cleanup. etc.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
186
diff
changeset
|
234 done |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
235 rmdir "${trash_dir}" 2>/dev/null |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
236 progress_done |
223 | 237 [ -n "${has_trash}" ] && echo "${trashes}" |
187
efd957294c8c
refactoring. cleanup. etc.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
186
diff
changeset
|
238 } |
efd957294c8c
refactoring. cleanup. etc.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
186
diff
changeset
|
239 |
159 | 240 # check files correctness |
241 Check_Files() { | |
223 | 242 if [ ! -n "${ISNEW}" ]; then |
243 [ -z "${NOCLEAN}" ] && Cleanup_Repository | |
207
17d816a63b4c
final progress version
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
206
diff
changeset
|
244 printf "Checking for errors... " |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
245 progress_init |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
246 __files_error="These files do not match its md5:" |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
247 __files_notdanbooru="These files are not checked:" |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
248 __has_err_filename= |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
249 __has_err_md5= |
196 | 250 > "${TEMP_PREFIX}-error" |
214
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
251 > "${TEMP_PREFIX}-ok" |
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
252 for file in "${BASE_DIR}/${SITE_DIR}/${TARGET_DIR}/"* |
159 | 253 do |
223 | 254 if [ "${file}" != "${BASE_DIR}/${SITE_DIR}/${TARGET_DIR}/*" ]; then |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
255 if [ -n "$(is_not_md5 "${file}")" ] || [ -d "${file}" ]; then |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
256 __files_notdanbooru="${__files_notdanbooru} |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
257 $(get_basename "${file}")" |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
258 __has_err_filename=1 |
214
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
259 else |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
260 if [ "$(get_md5 "${file}")" = "$(get_filename "${file}")" ]; then |
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
261 echo "$(get_basename "${file}")" >> "${TEMP_PREFIX}-ok" |
217 | 262 else |
263 rm "${file}" || Err_Fatal "Error removing ${file}" | |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
264 echo "$(get_basename "${file}")" >> "${TEMP_PREFIX}-error" |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
265 __files_error="${__files_error} |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
266 $(get_basename "${file}")" |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
267 __has_err_md5=1 |
217 | 268 fi |
187
efd957294c8c
refactoring. cleanup. etc.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
186
diff
changeset
|
269 fi |
148 | 270 fi |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
271 progress_anim |
148 | 272 done |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
273 progress_done |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
274 if [ ! -n "${__has_err_md5}" ] && [ ! -n "${__has_err_filename}" ]; then |
203 | 275 echo "All files OK" |
170 | 276 else |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
277 if [ -n "${__has_err_md5}" ]; then |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
278 echo "${__files_error}" |
214
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
279 echo "$(echo $(wc -l < "${TEMP_PREFIX}-error")) file(s) removed" |
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
280 fi |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
281 [ -n "${__has_err_filename}" ] && echo "${__files_notdanbooru}" |
170 | 282 fi |
214
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
283 echo "$(echo $(wc -l < "${TEMP_PREFIX}-ok")) file(s) available locally" |
148 | 284 |
207
17d816a63b4c
final progress version
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
206
diff
changeset
|
285 printf "Generating list of new files... " |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
286 progress_init |
218 | 287 cp -f "${TEMP_PREFIX}-list" "${TEMP_PREFIX}-templist" |
214
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
288 while read -r is_ok; do |
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
289 grep -v "${is_ok}" "${TEMP_PREFIX}-templist" > "${TEMP_PREFIX}-newlist" |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
290 cp -f "${TEMP_PREFIX}-newlist" "${TEMP_PREFIX}-templist" || Err_Impossible |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
291 progress_anim |
203 | 292 done < "${TEMP_PREFIX}-ok" |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
293 progress_done |
214
a6624fb9b317
major cleanup. tweaking.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
213
diff
changeset
|
294 echo "$(echo $(wc -l < "${TEMP_PREFIX}-newlist")) file(s) to be downloaded" |
148 | 295 else |
223 | 296 if [ -n "${ISQUICK}" ]; then |
207
17d816a63b4c
final progress version
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
206
diff
changeset
|
297 echo "Quick mode selected. Skipping check" |
152 | 298 else |
299 echo "Empty local repository" | |
300 fi | |
200 | 301 cat "${TEMP_PREFIX}-list" > "${TEMP_PREFIX}-newlist" |
148 | 302 fi |
303 } | |
304 | |
159 | 305 # start downloading the images |
306 Fetch_Images() { | |
223 | 307 if [ "$(echo $(wc -l < "${TEMP_PREFIX}-newlist"))" -eq 0 ]; then |
148 | 308 echo "No new file" |
309 else | |
207
17d816a63b4c
final progress version
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
206
diff
changeset
|
310 printf "Starting wget... " |
160
68227a30d0b3
forgot to fix Fetch_Images to reflect new folder naming scheme
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
159
diff
changeset
|
311 cd "${BASE_DIR}/${SITE_DIR}/${TARGET_DIR}" |
196 | 312 wget -e continue=on -bi "${TEMP_PREFIX}-newlist" -o "${TEMP_PREFIX}.log" |
148 | 313 fi |
314 } | |
315 | |
159 | 316 # initialize base variables and initial command check |
227 | 317 init() |
318 { | |
159 | 319 # path initialization |
227 | 320 # check if additional path is specified |
321 if [ -n "${ADDITIONAL_PATH}" ] | |
322 then | |
323 # insert the additional path | |
324 PATH="${ADDITIONAL_PATH}:${PATH}" | |
325 export PATH | |
326 fi | |
158
cba73f6a96bb
grep check. OpenSolaris' default grep doesn't support -f
edhoprima
parents:
157
diff
changeset
|
327 |
159 | 328 # misc variables |
166 | 329 ISQUICK= |
330 ISNEW= | |
215
710082ce6788
major cleanup part2.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
214
diff
changeset
|
331 |
227 | 332 # minimum number of arguments: 2 (command and tag). If less than two, exit and print help message |
159 | 333 [ $# -lt 2 ] && Err_Help |
334 case "$1" in | |
174
0948e76a57a1
added help. Bump to 0.1-beta2
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
173
diff
changeset
|
335 check|fetch|quickfetch) |
159 | 336 echo "Starting..." |
337 JOB="$1" | |
338 ;; | |
339 *) | |
340 Err_Help | |
341 ;; | |
342 esac | |
343 shift | |
344 SITE= | |
185
6d926d4b3c5a
initial clean system support
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
184
diff
changeset
|
345 TAGS= |
227 | 346 __has_pass=0 |
347 __has_user=0 | |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
348 while getopts "s:(site)n(noclean)u:(user)p:(password)" opt |
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
349 do |
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
350 case "$opt" in |
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
351 s) SITE="$OPTARG";; |
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
352 n) NOCLEAN=1;; |
227 | 353 p) |
354 LOGIN_PASS=$(printf "%s" "$OPTARG" | openssl dgst -sha1) | |
355 __has_pass=1 | |
356 ;; | |
357 u) | |
358 LOGIN_USER="$OPTARG" | |
359 __has_user=1 | |
360 ;; | |
185
6d926d4b3c5a
initial clean system support
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
184
diff
changeset
|
361 esac |
6d926d4b3c5a
initial clean system support
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
184
diff
changeset
|
362 done |
225
265a9ca47a19
- Replaced md5(sum) with openssl. Less platform dependent because the tool is same across platforms
edhoprima
parents:
224
diff
changeset
|
363 TAGS="$@" |
223 | 364 [ -n "${SITE}" ] || SITE=${DEFAULT_SITE} |
365 [ -n "${TAGS}" ] || Err_Fatal "No tag specified" | |
181
d3b7927bdb2b
restructuring and add check if the xml is processed properly
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
180
diff
changeset
|
366 # Get base folder - default, current folder or fallback to ${HOME} |
223 | 367 [ -n "${BASE_DIR}" ] || BASE_DIR=${PWD} |
368 [ -n "${BASE_DIR}" ] || BASE_DIR=${HOME} | |
227 | 369 [ -n "$(echo "${BASE_DIR}" | cut -c1 | grep \/)" ] || BASE_DIR="/${BASE_DIR}" |
370 # see if both pass and use are set. If they're set, switch _use_login variable content to 1. | |
371 [ ${__has_pass} -eq 1 -a ${__has_user} -eq 1 ] && _use_login=1 | |
181
d3b7927bdb2b
restructuring and add check if the xml is processed properly
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
180
diff
changeset
|
372 |
159 | 373 echo "Tags: ${TAGS}" |
374 # slash is not wanted for folder name | |
193 | 375 TARGET_DIR=$(echo "${TAGS}" | sed -e 's/\//_/g') |
376 SITE_DIR=$(echo "${SITE}" | sed -e 's/\/$//g;s/\//_/g') | |
195
652d9e268cee
test migration to printf
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
194
diff
changeset
|
377 TEMP_PREFIX="${BASE_DIR}/temp/${SITE_DIR}-${TARGET_DIR}" |
159 | 378 } |
148 | 379 |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
380 # global variables goes here |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
381 init_globals() |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
382 { |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
383 _version="0.3-beta3" # version of this script |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
384 _use_login=0 # variable to check whether a login is used or not |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
385 } |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
386 |
227 | 387 main() |
388 { | |
228
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
389 #initialize global variables |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
390 init_globals |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
391 #print welcome message |
5d3a0645b504
- Restructured some things.
edhoprima@gmail.com <edhoprima@gmail.com>
parents:
227
diff
changeset
|
392 msg_welcome |
227 | 393 # initialization |
394 init "$@" | |
395 Check_Tools | |
396 Check_Folders | |
158
cba73f6a96bb
grep check. OpenSolaris' default grep doesn't support -f
edhoprima
parents:
157
diff
changeset
|
397 |
148 | 398 |
227 | 399 # let's do the job! |
400 case "${JOB}" in | |
401 check) | |
402 Generate_Link | |
403 Check_Files | |
404 ;; | |
405 fetch) | |
406 Generate_Link | |
407 Check_Files | |
408 Fetch_Images | |
409 ;; | |
410 quickfetch) | |
411 ISNEW=1 | |
412 ISQUICK=1 | |
413 Generate_Link | |
414 Check_Files | |
415 Fetch_Images | |
416 ;; | |
417 esac | |
418 } | |
419 | |
420 # call the main routine! | |
421 main "$@" | |
422 |