first commit

gelbtag.sh

@@ -0,0 +1,58 @@
+#!/bin/bash
+
+USE_TOR=false
+DELAY=1
+
+function usage {
+		echo "./$(basename $0) [-t] [-s]"
+		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"
+}
+
+# list of arguments expected in the input
+optstring=":hts:"
+
+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
+			;;
+		s)
+			DELAY="${OPTARG}"
+			;;
+		:)
+			echo "$0: Must supply an argument to -$OPTARG." >&2
+			exit 1
+			;;
+		?)
+			echo "Invalid option: -${OPTARG}."
+			exit 2
+			;;
+	esac
+done
+
+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 "https://gelbooru.com/index.php?page=dapi&s=post&q=index&json=1&tags=md5:$FILE_MD5" | jq .`
+	else
+		JSON=`curl -s "https://gelbooru.com/index.php?page=dapi&s=post&q=index&json=1&tags=md5:$FILE_MD5" | jq .`
+	fi
+	# STORE TAGS INTO VARIABLES
+	FILE_TAGS=`echo $JSON | jq -r '.post | .[] | ."tags"' | sed 's/\ /,/g'`
+	# ADD TAGS TO IMAGE
+	setfattr -n user.xdg.tags -v "$FILE_TAGS" "$FILE"
+	# DELAY BEFORE NEXT FETCH
+	sleep $DELAY
+done