fastfix

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

ufw

@@ -30,6 +30,7 @@ usage() {
 		        -p                   # Make the file public.
 		        -u                   # Generate a shortlink from URL.
 		        -m                   # Maximum filesize (takes K, M and G suffixes).
+		        -a                   # Add the file to an album.
 		
 		    Config options (~/.config/ufw):
 		        secret               # Your personal token. Get it at https://zfh.so/settings_form
@@ -109,21 +110,16 @@ upload() {
 
 	get_file_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="" \
+		"${misc_curl_args[@]}" \
 		-A 'zerofiles.org upload script' \
 		"$cfg_service_url/upload"
-	fi
 
 	if (( $? )); then
 		err 'Something has gone wrong with the upload.'
@@ -261,7 +257,6 @@ 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'
@@ -273,6 +268,21 @@ main() {
 			-S|--short) flag_shortlink='1';;
 			-u|--url) action='url';;
 
+			-a|--album-id)
+				misc_curl_args+=( -F "album_id=$2" )
+				shift
+			;;
+
+			-A|--album-name)
+				misc_curl_args+=( -F "album_name=$2" )
+				shift
+			;;
+
+			-d|--description)
+				misc_curl_args+=( -F "description=$2" )
+				shift
+			;;
+			
 			-h|--help|--usage) usage; return;;
 
 			--) shift; break;;