#!/usr/bin/bash
# Copyright (c) 2012 fbt <fbt@fleshless.org>
# License: ISC
#
# About:
# A simple upload script for UFW (http://zerofiles.org)

_cat() {
	while read; do
		printf '%s\n' "$REPLY"
	done
}

is_url() {
	[[ "$1" =~ $cfg_url_regex ]]
}

msg() {
	if ! (( flag_quiet )); then
		printf '%s\n' "$1"
	fi
}

err() { printf '(error) %s\n' "$*" >&2;  }

usage() {
	_cat <<- EOF
		Usage: ufw [-RsF] [-D num] [file/url]"
		    Flags:
		        -R|--remove-file              # Remove the file after uploading.
		        -s|--screenshot               # Make a screenshot and upload it instead of a file.
		        -F|--fullscreen               # Make a fullscreen shot instead of prompting for a window/area. Implies -s.
		        -d|--description              # Supply a description.
		        -D|--screenshot-delay <num>   # Delay the shot by <num> seconds.
		        -p|--public                   # Make the file public.
		        -u|--shorten <URL>            # Generate a shortlink from URL.
		        -S|--short-url                # Get a shortlink when uploading a file.
		        -P|--page-url                 # Get a link to the file page instead of a direct one.
		        -m|--max-filesize <num>[suf]  # Maximum filesize (takes K, M and G suffixes).
		        -a|--album-id                 # Add the file to an album.
		        -A|--album-name               # Add the file to an album by name. The album will be created, if necessary.
		
		    Config options (~/.config/ufw):
		        secret               # Your personal token. Get it at https://zfh.so/settings_form
		        cfg_screenshot_ext   # Screenshot file type, used by scrot.
		        cfg_max_filesize     # Maximum filesize (takes K, M and G suffixes).
		        # Others are self-explanatory:
		        cfg_url_regex
		        cfg_tmp_dir
		        cfg_service_url
		        cfg_api_path
	EOF
}

get_file_hash() {
	read file_hash _ < <( sha1sum "$1" )
}

get_max_filesize() {
	if [[ $cfg_max_filesize =~ ^[0-9]+[BKMG]?$ ]]; then
		max_filesize_base="${cfg_max_filesize//[BKMG]/}"

		case "$cfg_max_filesize" in
			*K) max_filesize_bytes=$(( max_filesize_base * 1024 ));;
			*M) max_filesize_bytes=$(( max_filesize_base * 1024 * 1024 ));;
			*G) max_filesize_bytes=$(( max_filesize_base * 1024 * 1024 * 1024 ));;
			*) max_filesize_bytes=$max_filesize_base;;
		esac
	else
		err "Wrong cfg_max_filesize: $cfg_max_filesize"
		return 1
	fi
}

get_album_id() {
	declare api_response api_status api_status_message

	api_response=$( curl -fsL "$cfg_service_url/$cfg_api_path/album?a=new&name=$1&format=json&secret=$secret" )

	get_api_status <<< "$api_response"

	if (( api_status == 200 )); then
		album_id=$( jshon -e data -e album -e id -u <<< "$api_response" )
		misc_curl_args+=( -F album_id="$album_id" )
	else
		printf 'Error [album]: %s %s\n' "$api_status" "$api_status_message"
		return 1
	fi
}

upload() {
	declare api_response api_status api_status_message

	if (( flag_scrot )); then
		flag_rm=1
		file=$(take_screenshot) || { return 1; }
	else
		(( $# )) || {
			usage
			return 1
		}

		target="$1"

		if is_url "$target"; then
			flag_rm=1
			file=$(_mktemp "$cfg_tmp_dir")

			get_max_filesize || { return 1; }

			curl --max-filesize "$max_filesize_bytes" -skL "$target" > "$file"
			curl_result=$?

			if (( curl_result )); then
				case "$curl_result" in
					63) err "File exceeds cfg_max_filesize";;
					*) err "Could not download file.";;
				esac

				return 1
			fi
		else
			file="$target"

			[[ -f "$file" ]] || {
				err "No such file: ${file}"
				return 1
			}
		fi
	fi

	get_file_hash "$file"

	if [[ "$album_name" ]]; then
		get_album_id "$album_name"
	fi

	api_response=$( curl -sL \
		-F file="@$file" \
		-F api_format='json' \
		-F flag_private="$flag_private" \
		-F secret="$secret" \
		-F submit="" \
		"${misc_curl_args[@]}" \
		-A 'zerofiles.org upload script' \
		"$cfg_service_url/$cfg_api_path/upload" )

	if (( flag_shortlink )); then
		if (( flag_directlink )); then
			file_url_request='short_url_direct'
		else
			file_url_request='short_url'
		fi
	else
		if (( flag_directlink )); then
			file_url_request='direct_url'
		else
			file_url_request='page_url'
		fi
	fi

	get_api_status <<< "$api_response"

	if (( api_status == 200 )); then
		file_link=$( jshon -e data -e file -e "$file_url_request" -u <<< "$api_response" )

		printf '%s\n' "$file_link"

		if (( flag_rm )); then
			msg "Removing file: $file"
			rm "$file"
		fi
	else
		printf 'Error: %s %s\n' "$api_status" "$api_status_message"
		return 1
	fi
}

get_shortlink() {
	declare url=$1 api_response api_status api_status_message

	api_response=$( curl -sL "$cfg_service_url/$cfg_api_path/url?a=add&url=${url}&format=json&secret=$secret" )

	get_api_status <<< "$api_response"

	if (( api_status == 200 )); then
		read -r short_url real_url < <(
			jshon -e data -e URL -e shorturl -u <<< "$api_response"
		)

		printf '%s\n' "$short_url"
	else
		printf 'Error: %s %s\n' "$api_status" "$api_status_message"
	fi
}

take_screenshot() {
	declare tmp_file

	[[ "$scrot_exec" ]] || { scrot_exec=$(type -P scrot); }
	[[ "$scrot_exec" ]] || {
		err "Please install scrot to use this function"
		return 1
	}

	(( "$flag_scrot_fullscreen" )) || { scrot_args+=( '-s' ); }
	[[ "$cfg_scrot_delay" ]] && { scrot_args+=( "-d $cfg_scrot_delay" ); }

	tmp_file="$(_mktemp "${cfg_tmp_dir}" ".${cfg_screenshot_ext}")"

	scrot "${scrot_args[@]}" "$tmp_file" || {
		err "Failed to take a screenshot."
		return 1
	}

	printf '%s\n' "$tmp_file"
}

get_api_status() {
	read -d '' -r api_status api_status_message < <(
		jshon -e status -e code -u -p -e message -u
	)

	if ! (( api_status == 200 )); then
		return 1
	fi
}

_mktemp() {
	declare tmp_file_name tmp_file_name_extra="$2" tmp_dir="$1" 

	[[ -d "$tmp_dir" ]] || {
		err "${tmp_dir} does not exist or is not a directory."
		return 1
	}

	until [[ ! -e "${tmp_dir}/${tmp_file_name}" ]]; do
		tmp_file_name="${RANDOM}${RANDOM}${tmp_file_name_extra}"
	done

	printf '%s\n' "${tmp_dir}/${tmp_file_name}"
}

get_my_ip() {
	declare api_response api_status api_status_message my_ip

	api_response=$( curl -sl "$cfg_service_url/$cfg_api_path/ip?format=json" )

	get_api_status <<< "$api_response"

	if (( api_status == 200 )); then
		my_ip=$( jshon -e data -e ip -u <<< "$api_response" )
		printf '%s\n' "$my_ip"
	else
		printf 'Error: %s %s\n' "$api_status" "$api_status_message"
	fi
}

login() {
	declare api_response

	api_response=$( curl -sl "$cfg_service_url/$cfg_api_path/get_login_token?login=$login&format=json" )

	if get_api_status <<< "$api_response"; then
		printf 'Check your email.\n'
	else
		printf 'Error: %s %s\n' "$api_status" "$api_status_message"
	fi
}

set_argv() {
	declare arg opt c
	declare -g argv

	while (( $# )); do
		unset -v arg opt c

		case "$1" in
			(--) argv+=( "$1" ); break;;

			(--*)
				IFS='=' read arg opt <<< "$1"
				argv+=( "$arg" )

				[[ "$opt" ]] && {
					argv+=( "$opt" )
				}
			;;

			(-*)
				while read -n1 c
				do
					case "$c" in
						-|'') :;;
						*) argv+=( "-$c" );;
					esac
				done <<< "$1"
			;;

			(*) argv+=( "$1" );;
		esac
		shift
	done
}

main() {
	declare args file target flag_private flag_rm flag_scrot flag_scrot_fullscreen cfg_scrot_delay
	declare flag_shortlink=0

	# Defaults for XDG
	if ! [[ "$XDG_RUNTIME_DIR" ]]; then
		XDG_RUNTIME_DIR="/run/user/$UID"
	fi

	if ! [[ "$XDG_CONFIG_DIR" ]]; then
		XDG_CONFIG_DIR="$HOME/.config"
	fi

	cfg_url_regex='^[A-Za-z]([A-Za-z0-9+.-]+)?://.+'
	cfg_tmp_dir="$XDG_RUNTIME_DIR/ufw"; TEMPDIR="$cfg_tmp_dir"
	cfg_service_url='https://8fw.me'
	cfg_screenshot_ext='png'
	cfg_max_filesize='200M'
	cfg_api_path='/api/2'

	flag_shortlink=0
	flag_directlink=1

	if [[ -f "$HOME/.suprc" ]]; then
		printf 'Found legacy configfile, moving...\n' >&2
		mv "$HOME/.suprc" "$XDG_CONFIG_DIR/ufw"
	fi

	if [[ -f "$XDG_CONFIG_DIR/ufw" ]]; then
		if ! source "$XDG_CONFIG_DIR/ufw"; then
			printf 'Failed to source configuration file: %s\n' "$XDG_CONFIG_DIR/ufw"
			return $?
		fi
	fi

	while [[ "$1" ]]; do
		case "$1" in
			-h|--help|--usage) usage; return;;
			-D|--screenshot-delay) cfg_scrot_delay="$2"; shift;;
			-m|--max-filesize) cfg_max_filesize=$2; shift;;
			-A|--album-name) album_name=$2; shift;;

			-i|--my-ip) action='getmyip';;
			-u|--shorten) action='url';;

			-p|--public) flag_private='false';;
			-R|--remove-file) flag_rm='1';;
			-s|--screenshot) flag_scrot='1';;
			-S|--short-url) flag_shortlink=1;;
			-P|--page-url) flag_directlink=0;;
			-q|--quiet) flag_quiet=1;;

			-F|--fullscreen)
				flag_scrot='1'
				flag_scrot_fullscreen='1'
			;;

			-a|--album-id)
				misc_curl_args+=( -F "album_id=$2" )
				shift
			;;

			-d|--description)
				misc_curl_args+=( -F "description=$2" )
				shift
			;;

			--) shift; break;;

			-*)
				err "Unknown flag: $1"
				usage
				return 1
			;;

			*) break;;
		esac
		shift
	done

	for i in "$cfg_tmp_dir"; do
		[[ -d "$i" ]] || { mkdir -p "$i"; }
	done

	case "${action:-upload}" in
		upload) upload "$1";;
		url) get_shortlink "$1";;
		getmyip) get_my_ip;;
		login) login;;
	esac
}

set_argv "$@"
main "${argv[@]}"