Another default domain change

sup

@@ -6,9 +6,14 @@
 
 _self="${0##*/}"
 
+# Defaults for XDG
+if ! [[ "$XDG_RUNTIME_DIR" ]]; then
+	XDG_RUNTIME_DIR="/run/user/$UID"
+fi
+
 cfg_url_regex='^[A-Za-z]([A-Za-z0-9+.-]+)?://.+'
-cfg_tmp_dir="/tmp/$USER/sup"; TEMPDIR="$cfg_tmp_dir"
-cfg_script_url='https://zfh.so/upload'
+cfg_tmp_dir="$XDG_RUNTIME_DIR/sup"; TEMPDIR="$cfg_tmp_dir"
+cfg_service_url='https://8fw.me'
 cfg_screenshot_ext='png'
 
 [[ -f $HOME/.suprc ]] && { source "$HOME/.suprc"; }
@@ -19,6 +24,10 @@ _cat() {
 	done
 }
 
+is_url() {
+	[[ "$1" =~ $cfg_url_regex ]]
+}
+
 sup.msg() { printf '%s\n' "[sup] $1"; }
 sup.err() { sup.msg "(error) $1" >&2; }
 
@@ -29,8 +38,10 @@ sup.usage() {
 		        -R                   # Remove the file after uploading.
 		        -s                   # Make a screenshot and upload it instead of a file.
 		        -F                   # Make a fullscreen shot instead of prompting for a window/area. Implies -s.
+		        -d                   # Supply a dexcription.
 		        -D <num>             # Delay the shot by <num> seconds.
 		        -p                   # Make the file private. Requires \$secret to be set in the config.
+		        -u                   # Generate a shortlink from URL.
 		
 		    Config options (~/.suprc):
 		        secret               # Your personal token. Get it at https://zfh.so/settings_form
@@ -38,7 +49,7 @@ sup.usage() {
 		        # Others are self-explanatory:
 		        cfg_url_regex
 		        cfg_tmp_dir
-		        cfg_script_url
+		        cfg_service_url
 	EOF
 }
 
@@ -48,15 +59,75 @@ sup.env() {
 	done
 }
 
+sup.get_hash() {
+	read file_hash _ < <( sha1sum "$1" )
+}
+
 sup.upload() {
-	curl -fsL \
+	if (( flag_scrot )); then
+		flag_rm=1
+		file=$(sup.scrot) || { return 1; }
+	else
+		(( $# )) || {
+			sup.usage
+			return 1
+		}
+
+		target="$1"
+
+		if is_url "$target"; then
+			flag_rm=1
+			file=$(sup.mktemp "$cfg_tmp_dir")
+
+			curl -skL "$target" > "$file" || {
+				sup.err "Could not download file."
+				return 1
+			}
+		else
+			file="$target"
+
+			[[ -f "$file" ]] || {
+				sup.err "No such file: ${file}"
+				return 1
+			}
+		fi
+	fi
+
+	sup.get_hash "$file"
+
+	file_url=$( curl -fsL "${cfg_service_url}/api?mode=file&file=${file_hash}&direct=1&short=${flag_shortlink}" )
+	if ! (( $? )); then
+		printf '%s\n' "$file_url"
+	else
+		curl -sL \
 			-F file="@$file" \
 			-F upload_mode='api' \
 			-F flag_private="$flag_private" \
+			-F short="$flag_shortlink" \
+			-F description="$description" \
 			-F secret="$secret" \
 			-F submit="" \
 			-A 'zerofiles.org upload script' \
-		"$cfg_script_url"
+			"$cfg_service_url/upload"
+	fi
+
+	if (( $? )); then
+		sup.err 'Something has gone wrong with the upload.'
+		return 7
+	else
+		if (( flag_rm )); then
+			sup.msg "Removing file: $file"
+			rm "$file"
+		fi
+	fi
+}
+
+sup.url() {
+	declare url=$1
+
+	if ! curl -fsL "$cfg_service_url/api?mode=url_add&url=${url}&raw=1"; then
+		sup.err 'Something went wrong!'
+	fi
 }
 
 sup.mktemp() {
@@ -90,8 +161,6 @@ sup.scrot() {
 	printf '%s\n' "$tmp_file"
 }
 
-sup.if_url() { printf '%s\n' "$1" | grep -oE "$cfg_url_regex" &>/dev/null; }
-
 sup.mktemp() {
 	declare tmp_file_name tmp_file_name_extra="$2" tmp_dir="$1" 
 
@@ -107,6 +176,14 @@ sup.mktemp() {
 	printf '%s\n' "${tmp_dir}/${tmp_file_name}"
 }
 
+sup.get_my_ip() {
+	my_ip=$( curl -fsL https://zfh.so/ip )
+
+	if ! (( $? )); then
+		printf '%s\n' $my_ip
+	fi
+}
+
 sup.set_argv() {
 	declare arg opt c
 	declare -g argv
@@ -144,6 +221,7 @@ sup.set_argv() {
 
 main() {
 	declare args file target flag_private flag_rm flag_scrot flag_scrot_fullscreen cfg_scrot_delay
+	declare flag_shortlink=0
 
 	while [[ "$1" ]]; do
 		case "$1" in
@@ -151,12 +229,18 @@ main() {
 			-R|--remove-file) flag_rm='1';;
 			
 			-s|--screenshot) flag_scrot='1';;
+			-d|--description) description="$2"; shift;;
 			-D|--screenshot-delay) cfg_scrot_delay="$2"; shift;;
 			-F|--fullscreen)
 				flag_scrot='1'
 				flag_scrot_fullscreen='1'
 			;;
 
+			-i|--my-ip) action='getmyip';;
+
+			-S|--short) flag_shortlink='1';;
+			-u|--url) action='url';;
+			
 			-h|--help|--usage) sup.usage; return;;
 
 			--) shift; break;;
@@ -168,41 +252,11 @@ main() {
 
 	sup.env
 
-	if (( flag_scrot )); then
-		file=$(sup.scrot) || { return 1; }
-	else
-		(( $# )) || {
-			sup.usage
-			return 1
-		}
-
-		target="$1"
-
-		if sup.if_url "$target"; then
-			file=$(sup.mktemp "$cfg_tmp_dir")
-
-			curl -skL "$target" > "$file" || {
-				sup.err "Could not download file."
-				return 1
-			}
-		else
-			file="$target"
-
-			[[ -f "$file" ]] || {
-				sup.err "No such file: ${file}"
-				return 1
-			}
-		fi
-	fi
-
-	if sup.upload; then
-		if (( "$flag_rm" )); then
-			rm "$file"
-		fi
-	else
-		sup.err 'Something has gone wrong with the upload.'
-		return 7
-	fi
+	case "${action:-upload}" in
+		upload) sup.upload "$1";;
+		url) sup.url "$1";;
+		getmyip) sup.get_my_ip;;
+	esac
 }
 
 sup.set_argv "$@"

supd

@@ -1,8 +1,13 @@
 #!/usr/bin/env bash
 shopt -s nullglob
 
+# Defaults for XDG
+if ! [[ "$XDG_RUNTIME_DIR" ]]; then
+	XDG_RUNTIME_DIR="/run/user/$UID"
+fi
+
 # Defaults
-cfg_workdir="/tmp/$USER/supd"
+cfg_workdir="$XDG_RUNTIME_DIR/supd"
 cfg_scan_delay='30'
 
 msg() { printf '%s\n' "$*"; }