booru-tools/e621dl.sh

105 lines
3.1 KiB
Bash
Raw Normal View History

#!/bin/bash
USE_TOR=false
DELAY=1
UA="Booru-Tools/0.1"
HAS_USERNAME=false
HAS_API_KEY=false
function usage {
echo "./$(basename $0) [-t] [-s]"
echo "Simply make a files.txt inside a folder and paste all your links, then run this script to download them all!"
echo " -h shows this help message"
echo " -t e621 BLOCKS TOR WITH CLOUDFLARE, DOES NOT WORK"
echo " -p downloads using proxy"
echo " -s sets the delay after each request, defaults to 1"
echo " -u e621.net username"
echo " -k e621.net API key"
}
# list of arguments expected in the input
optstring=":hts:u:k:"
while getopts ${optstring} arg; do
case ${arg} in
h)
usage
exit
;;
t)
USE_TOR=true
echo "e621 BLOCKS TOR WITH CLOUDFLARE"
echo "EXITING SINCE SCRIPT WILL NOT WORK"
exit
;;
s)
DELAY="${OPTARG}"
;;
u)
HAS_USERNAME=true
USERNAME="${OPTARG}"
;;
k)
HAS_API_KEY=true
API_KEY="${OPTARG}"
;;
:)
echo "$0: Must supply an argument to -$OPTARG." >&2
exit 1
;;
?)
echo "Invalid option: -${OPTARG}."
exit 2
;;
esac
done
while read f; do
echo "$f"
# MODIFY URL TO API CALL
JSON_URL=`echo $f | sed "s/\?q.*//g"`
JSON_URL="$JSON_URL.json"
# DOWNLOAD JSON
if $HAS_USERNAME && $HAS_API_KEY ; then
JSON_URL="$JSON_URL?login=$USERNAME&api_key=$API_KEY"
fi
echo $JSON_URL
if $USE_TOR; then
#JSON=`torsocks curl -A "$UA" -s "$JSON_URL"`
echo "e621 BLOCKS TOR WITH CLOUDFLARE"
echo "EXITING SINCE SCRIPT WILL NOT WORK"
exit
else
JSON=`curl -A "$UA" -s "$JSON_URL"`
fi
# STORE TAGS INTO VARIABLES (god what the hell)
FILE_URL=`echo $JSON | jq -r '.post | .file."url"'`
FILE_TAGS_GENERAL=`echo $JSON | jq -r '.post."tags"."general"' | sed 's/\"//g' | sed 's/\[//g' | sed 's/\]//g' | tr -d '[:space:]'`
FILE_TAGS_ARTIST=`echo $JSON | jq -r '.post."tags"."artist"' | sed 's/\"//g' | sed 's/\[//g' | sed 's/\]//g' | tr -d '[:space:]'`
FILE_TAGS_CHARACTER=`echo $JSON | jq -r '.post."tags"."character"' | sed 's/\"//g' | sed 's/\[//g' | sed 's/\]//g' | tr -d '[:space:]'`
FILE_TAGS_SPECIES=`echo $JSON | jq -r '.post."tags"."species"' | sed 's/\"//g' | sed 's/\[//g' | sed 's/\]//g' | tr -d '[:space:]'`
FILE_TAGS_META=`echo $JSON | jq -r '.post."tags"."meta"' | sed 's/\"//g' | sed 's/\[//g' | sed 's/\]//g' | tr -d '[:space:]'`
FILE_DESCRIPTION=`echo $JSON | jq -r '.post'.'description'`
FILE_MD5=`echo $JSON | jq -r '.post'.'file'.'md5'`
FILE_EXT=`echo $JSON | jq -r '.post'.'file'.'ext'`
FILE_TAGS=`echo "$FILE_TAGS_ARTIST,$FILE_TAGS_CHARACTER,$FILE_TAGS_GENERAL,$FILE_TAGS_SPECIES,$FILE_TAGS_META"`
#echo $FILE_TAGS
# DOWNLOAD FILE
if $USE_TOR; then
#torsocks curl -O -J $FILE_URL
echo "e621 BLOCKS TOR WITH CLOUDFLARE"
echo "EXITING SINCE SCRIPT WILL NOT WORK"
exit
else
curl -O -J $FILE_URL
fi
# ADD TAGS TO NEW IMAGE
FILE="$FILE_MD5.$FILE_EXT"
setfattr -n user.xdg.tags -v "$FILE_TAGS" "$FILE"
setfattr -n user.xdg.creator -v "$FILE_TAGS_ARTIST" "$FILE"
setfattr -n user.xdg.comment -v "$FILE_DESCRIPTION" "$FILE"
setfattr --name=user.checksum --value="$FILE_MD5" "$FILE"
# DELAY BEFORE NEXT FETCH
sleep $DELAY
done < files.txt