script.sh

#!/bin/bash
#                                   _                 _
#                                  | |               | |
#   _____      _____    _   _ _ __ | | ___   __ _  __| | ___ _ __
#  / _ \ \ /\ / / _ \  | | | | '_ \| |/ _ \ / _` |/ _` |/ _ \ '__|
#  |(_) \ V  V / (_)|  | |_| | |_) | | (_) | (_| | (_| |  __/ |
#  \___/ \_/\_/ \___/   \__,_| .__/|_|\___/ \__,_|\__,_|\___|_|
#                            | |
#                            |_|
#
# OWOUPLOADER.SH SCRIPT.
# ----------------------
#
# This script allows for native support to upload to the image server
# and url shortener component of whats-th.is. Through this you can do
# plethora of actions.
#
# A big thank you to jomo/imgur-screenshot to which I've edited parts
# of his script into my own.

if [ ! $(id -u) -ne 0 ]; then
	echo "ERROR : This script cannot be run as sudo."
	echo "ERROR : You need to remove the sudo from \"sudo ./setup.sh\"."
	exit 1
fi

owodir="$HOME/.config/owo"

current_version="v0.0.17"

##################################


if [ ! -d $owodir ]; then
	echo "INFO  : Could not find config directory. Please run setup.sh"
	exit 1
fi

if [ ! -d $path ]; then
	mkdir -p $path
fi
source $owodir/conf.cfg

key=$userkey >&2
output_url=$finished_url >&2

directoryname=$scr_directory >&2
filename=$scr_filename >&2
path=$scr_path >&2

print_debug=$debug >&2

##################################

function is_mac() {
	uname | grep -q "Darwin"
}

function check_key() {
	if [ -z "$key" ]; then
		echo "INFO  : \$key not found, please set \$userkey in your config file."
		echo "INFO  : You can find the key in $owodir/conf.cfg"
		exit 1
	fi
}

function notify() {
	if is_mac; then
		/usr/local/bin/terminal-notifier -title owo.whats-th.is -message "${1}" -appIcon $owodir/icon.icns
	else
		notify-send owo.whats-th.is "${1}" -i $owodir/icon.png
	fi
}

function delete_scr() {
	if [ "$keep_scr" != "true" ]; then
		rm "$path$filename"
	fi
}

function clipboard() {
	if is_mac; then
		echo "${1}" | tr -d "\n\r" | pbcopy
	else
		echo "${1}" | tr -d "\n\r" | xclip -i -sel c -f | xclip -i -sel p
	fi
}

function shorten() {
	check_key

	if [ -z "${2}" ]; then
		notify "Please enter the URL you wish to shorten."
		echo "INFO  : Please enter the URL you wish to shorten."

		read url
	else
		url=${2}
	fi
	#Check if the URL entered is valid.
	regex='(https?|ftp|file)://[-A-Za-z0-9\+&@#/%?=~_|!:,.;]*[-A-Za-z0-9\+&@#/%=~_|]'
	if [[ $url =~ $regex ]]; then
		result=$(curl -s "https://api.awau.moe/shorten/polr?action=shorten&key=$key&url=$url")

		#Check if the URL got sucessfully shortened.
		if grep -q "https://" <<< "${result}"; then
			d=$1
			if [ "$d" = "true" ]; then
					clipboard $result
					echo $result
					notify "Copied the link to the keyboard."
			else
					echo $result
			fi
		else
			notify "Shortening failed!"
		fi
	else
		notify "URL is not valid!"
	fi
}

function screenshot() {
	check_key

	# Alert the user that the upload has begun.
	notify "Select an area to begin the upload."

	# Begin our screen capture.
	if is_mac; then
		screencapture -o -i $path$filename
	else
		maim -s $path$filename
	fi

	# Make a directory for our user if it doesnt already exsist.
	mkdir -p $path

	# Open our new entry to use it!
	entry=$path$filename
	upload=$(curl -s -F "files[]=@"$entry";type=image/png" https://api.awau.moe/upload/pomf?key="$key")

	if [ "$print_debug" = true ] ; then
		echo $upload
	fi

	if egrep -q '"success":\s*true' <<< "${upload}"; then
		item="$(egrep -o '"url":\s*"[^"]+"' <<<"${upload}" | cut -d "\"" -f 4)"
		d=$1
		if [ "$d" = true ]; then
			if [ "$scr_copy" = true ]; then
				clipboard "https://$output_url/$item"
				notify "Upload complete! Copied the link to your clipboard."
			else
				echo "https://$output_url/$item"
			fi
		else
			output="https://$output_url/$item"
		fi
	else
		notify "Upload failed! Please check your logs ($owodir/log.txt) for details."
		echo "UPLOAD FAILED" > $owodir/log.txt
		echo "The server left the following response" >> $owodir/log.txt
		echo "--------------------------------------" >> $owodir/log.txt
		echo " " >> $owodir/log.txt
		echo "    " $upload >> $owodir/log.txt
	fi
	delete_scr
}

function upload() {

	check_key

	entry=$1
	mimetype=$(file -b --mime-type $entry)

		filesize=$(wc -c <"$entry")
		if [[ $filesize -le 83886081 ]]; then
			upload=$(curl -s -F "files[]=@"$entry";type=$mimetype" https://api.awau.moe/upload/pomf?key="$key")
			item="$(egrep -o '"url":\s*"[^"]+"' <<<"${upload}" | cut -d "\"" -f 4)"
		else
			echo "ERROR : File size too large or another error occured!"
			exit 1
		fi

	if egrep -q '"success":\s*true' <<< "${upload}"; then
		d=$2
		if [ "$d" = true ]; then
				clipboard "https://$output_url/$item"
			echo "https://$output_url/$item"
		else
		output="https://$output_url/$item"
		fi
	else
		notify "Upload failed! Please check your logs ($owodir/log.txt) for details."
		echo "UPLOAD FAILED" > $owodir/log.txt
		echo "The server left the following response" >> $owodir/log.txt
		echo "--------------------------------------" >> $owodir/log.txt
		echo " " >> $owodir/log.txt
		echo "    " $upload >> $owodir/log.txt
	fi
}

function runupdate() {
	cp $owodir/conf.cfg $owodir/conf_backup_$current_version.cfg

	if [ ! -d $owodir/.git ]; then
		git -C $owodir init
		git -C $owodir remote add origin https://github.com/whats-this/owo.sh.git
		git -C $owodir fetch --all
		git -C $owodir checkout -t origin/stable
		git -C $owodir reset --hard origin/stable
	else
		git -C $owodir pull origin stable
        fi
}
##################################

if [ "${1}" = "-h" ] || [ "${1}" = "--help" ]; then
	echo "usage: ${0} [-h | --check | -v]"
	echo ""
	echo "   -h --help                  Show this help screen to you."
	echo "   -v --version               Show current application version."
	echo "   -c --check                 Checks if dependencies are installed."
	echo "      --update                Checks if theres an update available."
	echo "   -l --shorten               Begins the url shortening process."
	echo "   -s --screenshot            Begins the screenshot uploading process."
	echo "   -sl                        Takes a screenshot and shortens the URL."
	echo "   -ul                        Uploads file and shortens URL."
	echo ""
	exit 0
fi

##################################

if [ "${1}" = "-v" ] || [ "${1}" = "--version" ]; then
	echo "INFO  : You are on version $current_version"
	exit 0
fi

##################################

if [ "${1}" = "-c" ] || [ "${1}" = "--check" ]; then
	if is_mac; then
		(which terminal-notifier &>/dev/null && echo "FOUND : found terminal-notifier") || echo "ERROR : terminal-notifier not found"
		(which screencapture &>/dev/null && echo "FOUND : found screencapture") || echo "ERROR : screencapture not found"
		(which pbcopy &>/dev/null && echo "FOUND : found pbcopy") || echo "ERROR : pbcopy not found"
	else
		(which notify-send &>/dev/null && echo "FOUND : found notify-send") || echo "ERROR : notify-send (from libnotify-bin) not found"
		(which maim &>/dev/null && echo "FOUND : found maim") || echo "ERROR : maim not found"
		(which xclip &>/dev/null && echo "FOUND : found xclip") || echo "ERROR : xclip not found"
	fi
	(which curl &>/dev/null && echo "FOUND : found curl") || echo "ERROR : curl not found"
	(which grep &>/dev/null && echo "FOUND : found grep") || echo "ERROR : grep not found"
	exit 0
fi

##################################

if [ "${1}" = "--update" ]; then
	if [ "${1}" = "-C" ]; then
		owodir="${4}"
	fi	
	remote_version="$(curl --compressed -fsSL --stderr - "https://api.github.com/repos/whats-this/owo.sh/releases" | egrep -m 1 --color 'tag_name":\s*".*"' | cut -d '"' -f 4)"
	if [ "${?}" -eq "0" ]; then
		if [ ! "${current_version}" = "${remote_version}" ] && [ ! -z "${current_version}" ] && [ ! -z "${remote_version}" ]; then
			echo "INFO  : Update found!"
			echo "INFO  : Version ${remote_version} is available (You have ${current_version})"

			echo "ALERT : You already have a configuration file in $owodir"
			echo "ALERT : Updating might break this config, are you sure you want to update?"

			read -p "INFO  : Continue anyway? (Y/N)" choice
			case "$choice" in
				y|Y ) runupdate;;
n|N ) exit 0;;
* ) echo "ERROR : That is an invalid response, (Y)es/(N)o.";;
esac


elif [ -z "${current_version}" ] || [ -z "${remote_version}" ]; then
	echo "ERROR : Version string is invalid."
	echo "INFO  : Current (local) version: '${current_version}'"
	echo "INFO  : Latest (remote) version: '${remote_version}'"
else
	echo "INFO  : Version ${current_version} is up to date."
fi
else
	echo "ERROR : Failed to check for latest version: ${remote_version}"
fi

exit 0
fi

##################################

if [ "${1}" = "-l" ] || [ "${1}" = "--shorten" ]; then
	shorten true
	exit 0
fi

##################################

if [ "${1}" = "-s" ] || [ "${1}" = "--screenshot" ]; then
	screenshot true
	exit 0
fi

##################################

if [ "${1}" = "-sl" ]; then
	screenshot false
	shorten true $output
	exit 0
fi

##################################

if [ "${1}" = "-ul" ]; then
	if [ -z "$2" ]; then
		echo "ERROR : Sorry, but thats incorrect syntax."
		echo "ERROR : Please use \"owo file.png\""
		exit 1
	fi
	upload ${2} false
	shorten true $output
	echo $result

	if is_mac; then
	    echo $result | pbcopy
	else
	    echo $result | xclip -i -sel c -f | xclip -i -sel p
        fi
	notify "Copied link to keyboard."
	exit 0
fi

upload ${1} true
notify "Copied link to keyboard."