80 lines
1.7 KiB
Bash
Executable File
80 lines
1.7 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
USE_TOR=false
|
|
DELAY=1
|
|
REQ_URL=""
|
|
|
|
function usage {
|
|
echo "./$(basename "$0") [-t] [-s]"
|
|
echo "Mass tagger for moebooru imageboards"
|
|
echo "Tags existing pictures inside a folder"
|
|
echo " -h shows this help message"
|
|
echo " -t downloads using tor (requires torsocks)"
|
|
echo " -s sets the delay after each request, defaults to 1"
|
|
echo " -k uses konachan API for tagging"
|
|
echo " -y uses yande.re API for tagging"
|
|
echo " -c uses (CUSTOM.URL) API for tagging"
|
|
}
|
|
|
|
# list of arguments expected in the input
|
|
optstring=":hts:kyc:"
|
|
|
|
while getopts ${optstring} arg; do
|
|
case ${arg} in
|
|
h)
|
|
usage
|
|
exit
|
|
;;
|
|
t)
|
|
USE_TOR=true
|
|
echo -n "Using Tor with IP: "
|
|
torsocks curl ip.me
|
|
;;
|
|
k)
|
|
REQ_URL="https://konachan.com/post.json?tags=md5:"
|
|
echo "Using konachan API for tagging"
|
|
;;
|
|
y)
|
|
REQ_URL="https://yande.re/post.json?tags=md5:"
|
|
echo "Using yande.re API for tagging"
|
|
;;
|
|
c)
|
|
REQ_URL="https://$OPTARG/post.json?tags=md5:"
|
|
;;
|
|
s)
|
|
DELAY="${OPTARG}"
|
|
;;
|
|
:)
|
|
echo "$0: Must supply an argument to -$OPTARG." >&2
|
|
exit 1
|
|
;;
|
|
?)
|
|
echo "Invalid option: -${OPTARG}."
|
|
exit 2
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [[ -z "$REQ_URL" ]]; then
|
|
echo "No imageboard selected. Exiting."
|
|
exit
|
|
fi
|
|
|
|
for FILE in *; do
|
|
echo "$FILE"
|
|
# GET MD5 HASH
|
|
FILE_MD5=$(md5sum "$FILE" | awk '{print $1}')
|
|
# DOWNLOAD JSON
|
|
if $USE_TOR; then
|
|
JSON=$(torsocks curl -s "$REQ_URL$FILE_MD5")
|
|
else
|
|
JSON=$(curl -s "$REQ_URL$FILE_MD5")
|
|
fi
|
|
# STORE TAGS INTO VARIABLES
|
|
FILE_TAGS=$(echo "$JSON" | jq -r '.[] | ."tags"' | sed 's/\ /,/g')
|
|
# ADD TAGS TO IMAGE
|
|
setfattr -n user.xdg.tags -v "$FILE_TAGS" "$FILE"
|
|
# DELAY BEFORE NEXT FETCH
|
|
sleep "$DELAY"
|
|
done
|