ufw-tools/ufw

307 lines
6.2 KiB
Plaintext
Raw Normal View History

2016-01-01 14:16:03 +00:00
#!/usr/bin/bash
2014-10-04 08:33:23 +00:00
# Copyright (c) 2012 fbt <fbt@fleshless.org>
2016-01-01 14:16:03 +00:00
# License: ISC
2014-10-04 08:33:23 +00:00
#
# About:
2016-01-01 14:16:03 +00:00
# A simple upload script for UFW (http://zerofiles.org)
2014-10-04 08:19:37 +00:00
2015-03-08 16:48:26 +00:00
_cat() {
while read; do
printf '%s\n' "$REPLY"
done
}
2015-10-13 10:52:26 +00:00
is_url() {
[[ "$1" =~ $cfg_url_regex ]]
}
2015-11-19 15:59:09 +00:00
msg() { printf '%s\n' "[ufw] $1"; }
err() { msg "(error) $1" >&2; }
2014-10-04 08:19:37 +00:00
2015-11-19 15:59:09 +00:00
usage() {
2015-03-08 16:48:26 +00:00
_cat <<- EOF
2016-01-01 14:16:03 +00:00
Usage: ufw [-RsF] [-D num] [file/url]"
2015-03-08 16:48:26 +00:00
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.
2015-06-23 20:25:48 +00:00
-d # Supply a dexcription.
2015-03-08 16:48:26 +00:00
-D <num> # Delay the shot by <num> seconds.
-p # Make the file public.
2015-06-21 13:05:12 +00:00
-u # Generate a shortlink from URL.
2015-11-13 02:14:26 +00:00
-m # Maximum filesize (takes K, M and G suffixes).
2016-03-03 16:00:39 +00:00
-a # Add the file to an album.
2015-03-08 16:48:26 +00:00
2015-11-19 15:59:09 +00:00
Config options (~/.config/ufw):
2015-03-08 16:48:26 +00:00
secret # Your personal token. Get it at https://zfh.so/settings_form
cfg_screenshot_ext # Screenshot file type, used by scrot.
2015-11-13 02:08:55 +00:00
cfg_max_filesize # Maximum filesize in megabytes.
2015-03-08 16:48:26 +00:00
# Others are self-explanatory:
cfg_url_regex
cfg_tmp_dir
2015-06-21 06:52:29 +00:00
cfg_service_url
2015-03-08 16:48:26 +00:00
EOF
2014-10-04 08:19:37 +00:00
}
2015-11-19 15:59:09 +00:00
ensure_env() {
2014-10-04 08:19:37 +00:00
for i in "$cfg_tmp_dir"; do
[[ -d "$i" ]] || { mkdir -p "$i"; }
done
}
2015-11-19 15:59:09 +00:00
get_file_hash() {
2015-06-21 06:52:29 +00:00
read file_hash _ < <( sha1sum "$1" )
}
2015-11-19 15:59:09 +00:00
get_max_filesize() {
2015-11-13 02:08:55 +00:00
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
2015-11-19 15:59:09 +00:00
err "Wrong cfg_max_filesize: $cfg_max_filesize"
2015-11-13 02:08:55 +00:00
return 1
fi
}
2015-11-19 15:59:09 +00:00
upload() {
2015-06-21 13:05:12 +00:00
if (( flag_scrot )); then
2015-10-13 10:57:56 +00:00
flag_rm=1
2015-11-19 15:59:09 +00:00
file=$(take_screenshot) || { return 1; }
2015-06-21 13:05:12 +00:00
else
(( $# )) || {
2015-11-19 15:59:09 +00:00
usage
2015-06-21 13:05:12 +00:00
return 1
}
target="$1"
2015-10-13 10:52:26 +00:00
if is_url "$target"; then
2015-10-13 10:57:56 +00:00
flag_rm=1
2015-11-19 15:59:09 +00:00
file=$(_mktemp "$cfg_tmp_dir")
2015-06-21 13:05:12 +00:00
2015-11-19 15:59:09 +00:00
get_max_filesize || { return 1; }
2015-11-13 02:08:55 +00:00
curl --max-filesize "$max_filesize_bytes" -skL "$target" > "$file"
curl_result=$?
if (( curl_result )); then
case "$curl_result" in
2015-11-19 15:59:09 +00:00
63) err "File exceeds cfg_max_filesize";;
*) err "Could not download file.";;
2015-11-13 02:08:55 +00:00
esac
2015-06-21 13:05:12 +00:00
return 1
2015-11-13 02:08:55 +00:00
fi
2015-06-21 13:05:12 +00:00
else
file="$target"
[[ -f "$file" ]] || {
2015-11-19 15:59:09 +00:00
err "No such file: ${file}"
2015-06-21 13:05:12 +00:00
return 1
}
fi
fi
2015-11-19 15:59:09 +00:00
get_file_hash "$file"
2015-06-21 06:52:29 +00:00
2016-03-03 18:49:49 +00:00
curl -sL \
-F file="@$file" \
-F upload_mode='api' \
-F flag_private="$flag_private" \
-F short="$flag_shortlink" \
-F secret="$secret" \
-F submit="" \
"${misc_curl_args[@]}" \
-A 'zerofiles.org upload script' \
"$cfg_service_url/upload"
fi
if (( $? )); then
2015-11-19 15:59:09 +00:00
err 'Something has gone wrong with the upload.'
return 7
else
if (( flag_rm )); then
2015-11-19 15:59:09 +00:00
msg "Removing file: $file"
rm "$file"
2015-06-21 13:05:12 +00:00
fi
fi
}
2015-11-19 15:59:09 +00:00
get_shortlink() {
2015-06-21 13:05:12 +00:00
declare url=$1
if ! curl -fsL "$cfg_service_url/api?mode=url_add&url=${url}&raw=1"; then
2015-11-19 15:59:09 +00:00
err 'Something went wrong!'
2015-06-21 06:52:29 +00:00
fi
2014-10-04 08:19:37 +00:00
}
2015-11-19 15:59:09 +00:00
take_screenshot() {
2015-02-10 13:51:20 +00:00
declare tmp_file
2014-10-04 08:19:37 +00:00
[[ "$scrot_exec" ]] || { scrot_exec=$(type -P scrot); }
2014-10-04 08:19:37 +00:00
[[ "$scrot_exec" ]] || {
2015-11-19 15:59:09 +00:00
err "Please install scrot to use this function"
2014-10-04 08:19:37 +00:00
return 1
}
2015-02-10 13:45:15 +00:00
(( "$flag_scrot_fullscreen" )) || { scrot_args+=( '-s' ); }
2014-10-04 08:19:37 +00:00
[[ "$cfg_scrot_delay" ]] && { scrot_args+=( "-d $cfg_scrot_delay" ); }
2015-11-19 15:59:09 +00:00
tmp_file="$(_mktemp "${cfg_tmp_dir}" ".${cfg_screenshot_ext}")"
2014-10-04 08:19:37 +00:00
2014-10-04 08:50:49 +00:00
scrot "${scrot_args[@]}" "$tmp_file" || {
2015-11-19 15:59:09 +00:00
err "Failed to take a screenshot."
2014-10-04 08:19:37 +00:00
return 1
}
printf '%s\n' "$tmp_file"
2014-10-04 08:19:37 +00:00
}
2015-11-19 15:59:09 +00:00
_mktemp() {
2015-02-10 13:51:20 +00:00
declare tmp_file_name tmp_file_name_extra="$2" tmp_dir="$1"
2014-10-04 08:45:31 +00:00
[[ -d "$tmp_dir" ]] || {
2015-11-19 15:59:09 +00:00
err "${tmp_dir} does not exist or is not a directory."
2014-10-04 08:45:31 +00:00
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}"
2014-10-04 08:45:31 +00:00
}
2015-11-19 15:59:09 +00:00
get_my_ip() {
2015-10-12 14:59:33 +00:00
my_ip=$( curl -fsL https://zfh.so/ip )
if ! (( $? )); then
printf '%s\n' $my_ip
fi
}
2015-11-19 15:59:09 +00:00
set_argv() {
2015-02-20 08:23:26 +00:00
declare arg opt c
declare -g argv
2015-02-01 14:33:10 +00:00
while (( $# )); do
2015-02-20 08:23:26 +00:00
unset -v arg opt c
2015-02-01 14:33:10 +00:00
case "$1" in
2015-02-20 08:23:26 +00:00
(--) argv+=( "$1" ); break;;
2015-02-01 14:33:10 +00:00
(--*)
IFS='=' read arg opt <<< "$1"
2015-02-20 08:23:26 +00:00
argv+=( "$arg" )
2015-02-01 14:33:10 +00:00
[[ "$opt" ]] && {
2015-02-20 08:23:26 +00:00
argv+=( "$opt" )
2015-02-01 14:33:10 +00:00
}
2014-10-04 09:17:06 +00:00
;;
2015-02-01 14:33:10 +00:00
(-*)
while read -n1 c
do
case "$c" in
-|'') :;;
2015-02-20 08:23:26 +00:00
*) argv+=( "-$c" );;
2015-02-01 14:33:10 +00:00
esac
done <<< "$1"
;;
2015-02-20 08:23:26 +00:00
(*) argv+=( "$1" );;
2014-10-04 09:17:06 +00:00
esac
2015-02-01 14:33:10 +00:00
shift
2014-10-04 09:17:06 +00:00
done
}
2014-10-04 08:19:37 +00:00
main() {
2015-02-10 13:45:15 +00:00
declare args file target flag_private flag_rm flag_scrot flag_scrot_fullscreen cfg_scrot_delay
declare flag_shortlink=0
2014-10-04 09:17:06 +00:00
2015-11-19 15:59:09 +00:00
# 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'
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
2014-10-04 08:19:37 +00:00
while [[ "$1" ]]; do
case "$1" in
-p|--private) flag_private='false';;
2014-10-04 08:19:37 +00:00
-R|--remove-file) flag_rm='1';;
-s|--screenshot) flag_scrot='1';;
-D|--screenshot-delay) cfg_scrot_delay="$2"; shift;;
2015-03-08 16:48:26 +00:00
-F|--fullscreen)
flag_scrot='1'
flag_scrot_fullscreen='1'
;;
2015-06-21 13:05:12 +00:00
2015-10-12 14:59:33 +00:00
-i|--my-ip) action='getmyip';;
-S|--short) flag_shortlink='1';;
2015-06-21 13:05:12 +00:00
-u|--url) action='url';;
2016-03-03 16:00:39 +00:00
-a|--album-id)
misc_curl_args+=( -F "album_id=$2" )
shift
;;
2016-03-03 18:49:49 +00:00
-A|--album-name)
misc_curl_args+=( -F "album_name=$2" )
shift
;;
2016-03-03 16:00:39 +00:00
-d|--description)
2016-03-03 18:49:49 +00:00
misc_curl_args+=( -F "description=$2" )
2016-03-03 16:00:39 +00:00
shift
;;
2014-10-04 08:33:23 +00:00
2015-11-19 15:59:09 +00:00
-h|--help|--usage) usage; return;;
2014-10-04 08:19:37 +00:00
2015-02-10 13:45:15 +00:00
--) shift; break;;
2014-10-04 08:19:37 +00:00
2015-02-10 13:45:15 +00:00
*) break;;
2014-10-04 08:19:37 +00:00
esac
shift
done
2015-11-19 15:59:09 +00:00
ensure_env
2014-10-04 08:19:37 +00:00
2015-06-21 13:05:12 +00:00
case "${action:-upload}" in
2015-11-19 15:59:09 +00:00
upload) upload "$1";;
url) get_shortlink "$1";;
getmyip) get_my_ip;;
2015-06-21 13:05:12 +00:00
esac
2014-10-04 08:19:37 +00:00
}
2015-11-19 15:59:09 +00:00
set_argv "$@"
2015-02-20 08:23:26 +00:00
main "${argv[@]}"