|
|
@ -6,18 +6,7 @@ |
|
|
|
|
|
|
|
_self="${0##*/}" |
|
|
|
|
|
|
|
# Defaults for XDG |
|
|
|
if ! [[ "$XDG_RUNTIME_DIR" ]]; then |
|
|
|
XDG_RUNTIME_DIR="/run/user/$UID" |
|
|
|
fi |
|
|
|
|
|
|
|
cfg_url_regex='^[A-Za-z]([A-Za-z0-9+.-]+)?://.+' |
|
|
|
cfg_tmp_dir="$XDG_RUNTIME_DIR/sup"; TEMPDIR="$cfg_tmp_dir" |
|
|
|
cfg_service_url='https://8fw.me' |
|
|
|
cfg_screenshot_ext='png' |
|
|
|
cfg_max_filesize='200M' |
|
|
|
|
|
|
|
[[ -f $HOME/.suprc ]] && { source "$HOME/.suprc"; } |
|
|
|
|
|
|
|
_cat() { |
|
|
|
while read; do |
|
|
@ -29,10 +18,10 @@ is_url() { |
|
|
|
[[ "$1" =~ $cfg_url_regex ]] |
|
|
|
} |
|
|
|
|
|
|
|
sup.msg() { printf '%s\n' "[sup] $1"; } |
|
|
|
sup.err() { sup.msg "(error) $1" >&2; } |
|
|
|
msg() { printf '%s\n' "[ufw] $1"; } |
|
|
|
err() { msg "(error) $1" >&2; } |
|
|
|
|
|
|
|
sup.usage() { |
|
|
|
usage() { |
|
|
|
_cat <<- EOF |
|
|
|
Usage: ${_self} [-RsF] [-D num] [file/url]" |
|
|
|
Flags: |
|
|
@ -45,7 +34,7 @@ sup.usage() { |
|
|
|
-u # Generate a shortlink from URL. |
|
|
|
-m # Maximum filesize (takes K, M and G suffixes). |
|
|
|
|
|
|
|
Config options (~/.suprc): |
|
|
|
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 in megabytes. |
|
|
@ -56,17 +45,17 @@ sup.usage() { |
|
|
|
EOF |
|
|
|
} |
|
|
|
|
|
|
|
sup.env() { |
|
|
|
ensure_env() { |
|
|
|
for i in "$cfg_tmp_dir"; do |
|
|
|
[[ -d "$i" ]] || { mkdir -p "$i"; } |
|
|
|
done |
|
|
|
} |
|
|
|
|
|
|
|
sup.get_hash() { |
|
|
|
get_file_hash() { |
|
|
|
read file_hash _ < <( sha1sum "$1" ) |
|
|
|
} |
|
|
|
|
|
|
|
sup.get_max_filesize() { |
|
|
|
get_max_filesize() { |
|
|
|
if [[ $cfg_max_filesize =~ ^[0-9]+[BKMG]?$ ]]; then |
|
|
|
max_filesize_base="${cfg_max_filesize//[BKMG]/}" |
|
|
|
|
|
|
@ -77,18 +66,18 @@ sup.get_max_filesize() { |
|
|
|
*) max_filesize_bytes=$max_filesize_base;; |
|
|
|
esac |
|
|
|
else |
|
|
|
sup.err "Wrong cfg_max_filesize: $cfg_max_filesize" |
|
|
|
err "Wrong cfg_max_filesize: $cfg_max_filesize" |
|
|
|
return 1 |
|
|
|
fi |
|
|
|
} |
|
|
|
|
|
|
|
sup.upload() { |
|
|
|
upload() { |
|
|
|
if (( flag_scrot )); then |
|
|
|
flag_rm=1 |
|
|
|
file=$(sup.scrot) || { return 1; } |
|
|
|
file=$(take_screenshot) || { return 1; } |
|
|
|
else |
|
|
|
(( $# )) || { |
|
|
|
sup.usage |
|
|
|
usage |
|
|
|
return 1 |
|
|
|
} |
|
|
|
|
|
|
@ -96,17 +85,17 @@ sup.upload() { |
|
|
|
|
|
|
|
if is_url "$target"; then |
|
|
|
flag_rm=1 |
|
|
|
file=$(sup.mktemp "$cfg_tmp_dir") |
|
|
|
file=$(_mktemp "$cfg_tmp_dir") |
|
|
|
|
|
|
|
sup.get_max_filesize || { return 1; } |
|
|
|
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) sup.err "File exceeds cfg_max_filesize";; |
|
|
|
*) sup.err "Could not download file.";; |
|
|
|
63) err "File exceeds cfg_max_filesize";; |
|
|
|
*) err "Could not download file.";; |
|
|
|
esac |
|
|
|
|
|
|
|
return 1 |
|
|
@ -115,13 +104,13 @@ sup.upload() { |
|
|
|
file="$target" |
|
|
|
|
|
|
|
[[ -f "$file" ]] || { |
|
|
|
sup.err "No such file: ${file}" |
|
|
|
err "No such file: ${file}" |
|
|
|
return 1 |
|
|
|
} |
|
|
|
fi |
|
|
|
fi |
|
|
|
|
|
|
|
sup.get_hash "$file" |
|
|
|
get_file_hash "$file" |
|
|
|
|
|
|
|
file_url=$( curl -fsL "${cfg_service_url}/api?mode=file&file=${file_hash}&direct=1&short=${flag_shortlink}" ) |
|
|
|
if ! (( $? )); then |
|
|
@ -140,60 +129,51 @@ sup.upload() { |
|
|
|
fi |
|
|
|
|
|
|
|
if (( $? )); then |
|
|
|
sup.err 'Something has gone wrong with the upload.' |
|
|
|
err 'Something has gone wrong with the upload.' |
|
|
|
return 7 |
|
|
|
else |
|
|
|
if (( flag_rm )); then |
|
|
|
sup.msg "Removing file: $file" |
|
|
|
msg "Removing file: $file" |
|
|
|
rm "$file" |
|
|
|
fi |
|
|
|
fi |
|
|
|
} |
|
|
|
|
|
|
|
sup.url() { |
|
|
|
get_shortlink() { |
|
|
|
declare url=$1 |
|
|
|
|
|
|
|
if ! curl -fsL "$cfg_service_url/api?mode=url_add&url=${url}&raw=1"; then |
|
|
|
sup.err 'Something went wrong!' |
|
|
|
err 'Something went wrong!' |
|
|
|
fi |
|
|
|
} |
|
|
|
|
|
|
|
sup.mktemp() { |
|
|
|
>"$1" || { |
|
|
|
sup.err "Cannot create $1!" |
|
|
|
return 1 |
|
|
|
} |
|
|
|
|
|
|
|
printf '%s\n' "$1" |
|
|
|
} |
|
|
|
|
|
|
|
sup.scrot() { |
|
|
|
take_screenshot() { |
|
|
|
declare tmp_file |
|
|
|
|
|
|
|
[[ "$scrot_exec" ]] || { scrot_exec=$(type -P scrot); } |
|
|
|
[[ "$scrot_exec" ]] || { |
|
|
|
sup.err "Please install scrot to use this function" |
|
|
|
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="$(sup.mktemp "${cfg_tmp_dir}" ".${cfg_screenshot_ext}")" |
|
|
|
tmp_file="$(_mktemp "${cfg_tmp_dir}" ".${cfg_screenshot_ext}")" |
|
|
|
|
|
|
|
scrot "${scrot_args[@]}" "$tmp_file" || { |
|
|
|
sup.err "Failed to take a screenshot." |
|
|
|
err "Failed to take a screenshot." |
|
|
|
return 1 |
|
|
|
} |
|
|
|
|
|
|
|
printf '%s\n' "$tmp_file" |
|
|
|
} |
|
|
|
|
|
|
|
sup.mktemp() { |
|
|
|
_mktemp() { |
|
|
|
declare tmp_file_name tmp_file_name_extra="$2" tmp_dir="$1" |
|
|
|
|
|
|
|
[[ -d "$tmp_dir" ]] || { |
|
|
|
sup.err "${tmp_dir} does not exist or is not a directory." |
|
|
|
err "${tmp_dir} does not exist or is not a directory." |
|
|
|
return 1 |
|
|
|
} |
|
|
|
|
|
|
@ -204,7 +184,7 @@ sup.mktemp() { |
|
|
|
printf '%s\n' "${tmp_dir}/${tmp_file_name}" |
|
|
|
} |
|
|
|
|
|
|
|
sup.get_my_ip() { |
|
|
|
get_my_ip() { |
|
|
|
my_ip=$( curl -fsL https://zfh.so/ip ) |
|
|
|
|
|
|
|
if ! (( $? )); then |
|
|
@ -212,7 +192,7 @@ sup.get_my_ip() { |
|
|
|
fi |
|
|
|
} |
|
|
|
|
|
|
|
sup.set_argv() { |
|
|
|
set_argv() { |
|
|
|
declare arg opt c |
|
|
|
declare -g argv |
|
|
|
|
|
|
@ -251,6 +231,33 @@ 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' |
|
|
|
|
|
|
|
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 |
|
|
|
-p|--private) flag_private='true';; |
|
|
@ -269,7 +276,7 @@ main() { |
|
|
|
-S|--short) flag_shortlink='1';; |
|
|
|
-u|--url) action='url';; |
|
|
|
|
|
|
|
-h|--help|--usage) sup.usage; return;; |
|
|
|
-h|--help|--usage) usage; return;; |
|
|
|
|
|
|
|
--) shift; break;; |
|
|
|
|
|
|
@ -278,14 +285,14 @@ main() { |
|
|
|
shift |
|
|
|
done |
|
|
|
|
|
|
|
sup.env |
|
|
|
ensure_env |
|
|
|
|
|
|
|
case "${action:-upload}" in |
|
|
|
upload) sup.upload "$1";; |
|
|
|
url) sup.url "$1";; |
|
|
|
getmyip) sup.get_my_ip;; |
|
|
|
upload) upload "$1";; |
|
|
|
url) get_shortlink "$1";; |
|
|
|
getmyip) get_my_ip;; |
|
|
|
esac |
|
|
|
} |
|
|
|
|
|
|
|
sup.set_argv "$@" |
|
|
|
set_argv "$@" |
|
|
|
main "${argv[@]}" |
|
|
|