README

LICENSE

@@ -1,4 +1,4 @@
-Copyright (c) 2014, Jack L. Frost <fbt@fleshless.org>
+Copyright (c) 2012-2014, Jack L. Frost <fbt@fleshless.org>
 
 
 Permission to use, copy, modify, and/or distribute this software for any purpose with or without

README.md

@@ -20,4 +20,18 @@ sup
 ---
 A very simple tool to upload files to the zfh.so file hosting
 
-	Usage: sup [-RsF] [-D num] [file/url]
+	Usage: sup [-RsF] [-D num] [file/url]"
+	Flags:
+	    -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 <num>             # Delay the shot by <num> seconds.
+	    -p                   # Make the file private. Requires $secret to be set in the config.
+
+	Config options (~/.suprc):
+	    secret               # Your personal token. Get it at https://zfh.so/settings_form
+	    cfg_screenshot_ext   # Screenshot file type, used by scrot.
+	    # Others are self-explanatory:
+	    cfg_url_regex
+	    cfg_tmp_dir
+	    cfg_script_url

sup

@@ -4,18 +4,42 @@
 # About:
 # A simple upload script for ZFH (http://zerofiles.org)
 
+_self="${0##*/}"
+
 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_screenshot_ext='png'
 
-[[ -f $HOME/.suprc ]] && { source $HOME/.suprc; }
+[[ -f $HOME/.suprc ]] && { source "$HOME/.suprc"; }
 
-sup.msg() { echo "[sup] $1"; }
+_cat() {
+	while read; do
+		printf '%s\n' "$REPLY"
+	done
+}
+
+sup.msg() { printf '%s\n' "[sup] $1"; }
 sup.err() { sup.msg "(error) $1" >&2; }
 
 sup.usage() {
-	echo "Usage: `basename $0` [-RsF] [-D num] [file/url]"
+	_cat <<- EOF
+		Usage: ${_self} [-RsF] [-D num] [file/url]"
+		    Flags:
+		        -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 <num>             # Delay the shot by <num> seconds.
+		        -p                   # Make the file private. Requires \$secret to be set in the config.
+		
+		    Config options (~/.suprc):
+		        secret               # Your personal token. Get it at https://zfh.so/settings_form
+		        cfg_screenshot_ext   # Screenshot file type, used by scrot.
+		        # Others are self-explanatory:
+		        cfg_url_regex
+		        cfg_tmp_dir
+		        cfg_script_url
+	EOF
 }
 
 sup.env() {
@@ -32,7 +56,7 @@ sup.upload() {
 		-F submit="" \
 		"$cfg_script_url" -sL -A 'zerofiles.org upload script' || { return 1; }
 
-	[[ "$flag_rm" ]] && { rm "$file"; }
+	(( "$flag_rm" )) && { rm "$file"; }
 
 	return 0
 }
@@ -43,19 +67,19 @@ sup.mktemp() {
 		return 1
 	}
 
-	echo "$1"
+	printf '%s\n' "$1"
 }
 
 sup.scrot() {
-	local tmp_file
+	declare tmp_file
 
-	[[ "$scrot_exec" ]] || { scrot_exec=`which scrot`; }
+	[[ "$scrot_exec" ]] || { scrot_exec=$(type -P scrot); }
 	[[ "$scrot_exec" ]] || {
 		sup.err "Please install scrot to use this function"
 		return 1
 	}
 
-	[[ "$flag_scrot_fullscreen" ]] || { scrot_args+=( '-s' ); }
+	(( "$flag_scrot_fullscreen" )) || { scrot_args+=( '-s' ); }
 	[[ "$cfg_scrot_delay" ]] && { scrot_args+=( "-d $cfg_scrot_delay" ); }
 
 	tmp_file="$(sup.mktemp "${cfg_tmp_dir}" ".${cfg_screenshot_ext}")"
@@ -65,18 +89,13 @@ sup.scrot() {
 		return 1
 	}
 
-	echo "$tmp_file"
+	printf '%s\n' "$tmp_file"
 }
 
-sup.exclude() {
-	[[ "$2" ]] && { echo "$1"; return 1; }
-	return 0
-}
-
-sup.if_url() { echo "$1" | grep -oE "$cfg_url_regex" &>/dev/null; }
+sup.if_url() { printf '%s\n' "$1" | grep -oE "$cfg_url_regex" &>/dev/null; }
 
 sup.mktemp() {
-	local tmp_file_name tmp_file_name_extra="$2" tmp_dir="$1" 
+	declare tmp_file_name tmp_file_name_extra="$2" tmp_dir="$1" 
 
 	[[ -d "$tmp_dir" ]] || {
 		sup.err "${tmp_dir} does not exist or is not a directory."
@@ -87,35 +106,46 @@ sup.mktemp() {
 		tmp_file_name="${RANDOM}${RANDOM}${tmp_file_name_extra}"
 	done
 
-	echo "${tmp_dir}/${tmp_file_name}"
+	printf '%s\n' "${tmp_dir}/${tmp_file_name}"
 }
 
-sup.getopt() {
-	local keys
+sup.set_argv() {
+	declare arg opt c
+	declare -g argv
 
-	for i in "$@"; do
-		case "$i" in
-			--*) echo "$i";;
+	while (( $# )); do
+		unset -v arg opt c
 
-			-*)
-				keys=( `echo "${i##*-}" | fold -w1` )
+		case "$1" in
+			(--) argv+=( "$1" ); break;;
 
-				[[ "${#keys[@]}" -gt 1 ]] && {
-					for k in ${keys[@]}; do
-						echo "-${k}"
-					done
-				} || {
-					echo "$i"
+			(--*)
+				IFS='=' read arg opt <<< "$1"
+				argv+=( "$arg" )
+
+				[[ "$opt" ]] && {
+					argv+=( "$opt" )
 				}
 			;;
 
-			*) echo "$i";;
+			(-*)
+				while read -n1 c
+				do
+					case "$c" in
+						-|'') :;;
+						*) argv+=( "-$c" );;
+					esac
+				done <<< "$1"
+			;;
+
+			(*) argv+=( "$1" );;
 		esac
+		shift
 	done
 }
 
 main() {
-	local args file target flag_private flag_rm flag_scrot flag_scrot_fullscreen cfg_scrot_delay
+	declare args file target flag_private flag_rm flag_scrot flag_scrot_fullscreen cfg_scrot_delay
 
 	while [[ "$1" ]]; do
 		case "$1" in
@@ -123,38 +153,52 @@ main() {
 			-R|--remove-file) flag_rm='1';;
 			
 			-s|--screenshot) flag_scrot='1';;
-			-F|--fullscreen) flag_scrot_fullscreen='1';;
 			-D|--screenshot-delay) cfg_scrot_delay="$2"; shift;;
+			-F|--fullscreen)
+				flag_scrot='1'
+				flag_scrot_fullscreen='1'
+			;;
 			
 			-h|--help|--usage) sup.usage; return;;
 
-			--) break;;
+			--) shift; break;;
 
-			*) args+=( "$1" );;
+			*) break;;
 		esac
 		shift
 	done
 
 	sup.env
 
-	if [[ "$flag_scrot" ]]; then
-		file="$(sup.scrot)" || { return 1; }
+	if (( flag_scrot )); then
+		file=$(sup.scrot) || { return 1; }
 	else
-		target="${args[0]}"; [[ "$target" ]] || { sup.usage; exit 1; }
+		(( $# )) || {
+			sup.usage
+			return 1
+		}
+
+		target="$1"
 
 		if sup.if_url "$target"; then
-			file=`sup.mktemp "$cfg_tmp_dir"`
+			file=$(sup.mktemp "$cfg_tmp_dir")
 
-			curl -s "$target" > "$file" || {
+			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.upload
 }
 
-main $(sup.getopt "$@")
+sup.set_argv "$@"
+main "${argv[@]}"