pass keys to ufw

Signed-off-by: fbt <fbt@fleshless.org>

ufw

@@ -1,12 +1,9 @@
-#!/usr/bin/env bash
+#!/usr/bin/bash
 # Copyright (c) 2012 fbt <fbt@fleshless.org>
+# License: ISC
 #
 # About:
-# A simple upload script for ZFH (http://zerofiles.org)
-
-_self="${0##*/}"
-
-
+# A simple upload script for UFW (http://zerofiles.org)
 
 _cat() {
 	while read; do
@@ -23,14 +20,14 @@ err() { msg "(error) $1" >&2; }
 
 usage() {
 	_cat <<- EOF
-		Usage: ${_self} [-RsF] [-D num] [file/url]"
+		Usage: ufw [-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                   # Supply a dexcription.
 		        -D <num>             # Delay the shot by <num> seconds.
-		        -p                   # Make the file private. Requires \$secret to be set in the config.
+		        -p                   # Make the file public.
 		        -u                   # Generate a shortlink from URL.
 		        -m                   # Maximum filesize (takes K, M and G suffixes).
 		
@@ -260,7 +257,7 @@ main() {
 
 	while [[ "$1" ]]; do
 		case "$1" in
-			-p|--private) flag_private='true';;
+			-p|--private) flag_private='false';;
 			-R|--remove-file) flag_rm='1';;
 			
 			-s|--screenshot) flag_scrot='1';;

ufwd

@@ -19,9 +19,12 @@ main() {
 	while (( $# )); do
 		case "$1" in
 			(--help|-h) usage; return 0;;
+
 			(--workdir|-d) cfg_workdir="$2"; shift;;
 			(--scan-delay|-D) cfg_scan_delay="$2"; shift;;
 
+			(--notify|-n) flag_enable_notifications=1;;
+
 			(--) shift; break;;
 			(-*)
 				err "Unknown key: $1"
@@ -34,6 +37,15 @@ main() {
 		shift
 	done
 
+	if (( flag_enable_notifications )); then
+		if type -P notify-send &>/dev/null; then
+			msg "Found notify-send."
+		else
+			err "notify-send not found in PATH, disabling notifications."
+			flag_enable_notifications=0
+		fi
+	fi
+
 	mkdir -p "$cfg_workdir" || {
 		return 1
 	}
@@ -44,7 +56,18 @@ main() {
 
 	while sleep "$cfg_scan_delay"; do
 		for i in *; do
-			ufw -R "$i"
+			upload_output=$( ufw "$@" -R "$i" )
+			upload_return=$?
+
+			if (( flag_enable_notifications )); then
+				if (( $upload_return )); then
+					notify-send 'ufwd' "File upload failed: $upload_output"
+				else
+					notify-send 'ufwd' "File uploaded: $upload_output"
+				fi
+			fi
+
+			printf '%s\n' "$upload_output"
 		done
 	done
 }