459 lines
10 KiB
Bash
Executable File
459 lines
10 KiB
Bash
Executable File
#!/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.
|
|
-n|--no-notify # Don't send immediate notifications for this upload.
|
|
--notify # Force a notification for this upload.
|
|
-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.
|
|
-t|--tags "<tag1[, tag2,..]>" # Add tags to the file.
|
|
-q|--quiet # Be quiet.
|
|
-v|--void # Don't add the file to my files.
|
|
-c|--config <file> # Use an alternate config file
|
|
|
|
Config options (~/.config/ufw):
|
|
secret # Your personal token. Get it at https://zfh.so/settings_form
|
|
cfg_screenshot_ext # Screenshot file type, used by maim.
|
|
cfg_max_filesize # Maximum filesize (takes K, M and G suffixes).
|
|
# Others are self-explanatory:
|
|
cfg_url_regex
|
|
cfg_tmp_dir
|
|
cfg_service_url
|
|
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/albumctl.json?m=new&name=$1&secret=$secret" )
|
|
|
|
get_api_status <<< "$api_response"
|
|
|
|
if (( api_status == 200 )); then
|
|
album_id=$( jq -r '.data.album.id' <<< "$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_scn )); 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" -sL "$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 tags="$tags" \
|
|
-F notify="$flag_notify" \
|
|
-F void="${flag_void:-0}" \
|
|
-F submit="" \
|
|
"${misc_curl_args[@]}" \
|
|
-A 'zerofiles.org upload script' \
|
|
"$cfg_service_url/maw.json" )
|
|
|
|
if (( flag_shortlink )); then
|
|
if (( flag_directlink )); then
|
|
file_url_request='short_url'
|
|
else
|
|
file_url_request='page_short_url'
|
|
fi
|
|
else
|
|
file_url_request='url'
|
|
fi
|
|
|
|
get_api_status <<< "$api_response"
|
|
|
|
if (( api_status == 200 )); then
|
|
file_link=$( jq -r ".data.${file_url_request}" <<< "$api_response" )
|
|
|
|
(( flag_shortlink )) || {
|
|
(( flag_directlink )) || file_link="${file_link#*.}"
|
|
}
|
|
|
|
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/shrink.json?url=${url}&secret=$secret" )
|
|
get_api_status <<< "$api_response"
|
|
|
|
if (( api_status == 200 )); then
|
|
read -r short_url real_url < <(
|
|
jq -r '.data.short_url' <<< "$api_response"
|
|
)
|
|
|
|
printf '%s\n' "$short_url"
|
|
else
|
|
printf 'Error: %s %s\n' "$api_status" "$api_status_message"
|
|
fi
|
|
}
|
|
|
|
take_screenshot() {
|
|
declare tmp_file
|
|
|
|
[[ "$scn_exec" ]] || { scn_exec=$(type -P maim); }
|
|
[[ "$scn_exec" ]] || {
|
|
err "Please install maim to use this function"
|
|
return 1
|
|
}
|
|
|
|
(( "$flag_scn_fullscreen" )) || { scn_args+=( '-s' ); }
|
|
[[ "$cfg_scn_delay" ]] && { scn_args+=( "-d $cfg_scn_delay" ); }
|
|
|
|
tmp_file="$(_mktemp "${cfg_tmp_dir}" ".${cfg_screenshot_ext}")"
|
|
|
|
maim "${scn_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 < <(
|
|
jq -r '.status.code, .status.message'
|
|
)
|
|
|
|
if (( api_status >= 400 )); 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/ip.json" )
|
|
|
|
get_api_status <<< "$api_response"
|
|
|
|
if (( api_status == 200 )); then
|
|
my_ip=$( jq -r '.data.ip' <<< "$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/token_request.json?login=$login" )
|
|
|
|
if get_api_status <<< "$api_response"; then
|
|
printf 'Check your email.\n'
|
|
else
|
|
printf 'Error: %s %s\n' "$api_status" "$api_status_message"
|
|
fi
|
|
}
|
|
|
|
get_token() {
|
|
[[ $secret ]] && return 0
|
|
[[ -f $cfg_file ]] && return 1
|
|
|
|
declare api_response login password
|
|
|
|
printf "First-time setup...\n"
|
|
read -p 'Username: ' login
|
|
read -sp 'Password: ' password
|
|
printf '\n'
|
|
|
|
login_response=$(curl -c $XDG_CONFIG_DIR/ufw.cookiejar -d "login=$login" -d "password=$password" -sl "$cfg_service_url/login.json")
|
|
get_api_status <<< "$login_response" || {
|
|
jq -r '.status.message' <<< "$login_response"
|
|
return 1
|
|
}
|
|
|
|
api_response=$(curl -b $XDG_CONFIG_DIR/ufw.cookiejar -sl "$cfg_service_url/api_token.json")
|
|
if get_api_status <<< "$api_response"; then
|
|
secret=$(jq -r '.data.token' <<< "$api_response")
|
|
printf 'export secret="%s"' "$secret" >> "$cfg_file"
|
|
chmod 600 "$cfg_file"
|
|
printf 'Done\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_scn flag_scn_fullscreen cfg_scn_delay flag_notify
|
|
declare flag_shortlink
|
|
|
|
# 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+.-]+)?://.+'
|
|
|
|
[[ $1 ]] || { usage; return 1; }
|
|
|
|
while [[ "$1" ]]; do
|
|
case "$1" in
|
|
-h|--help|--usage) usage; return;;
|
|
-D|--screenshot-delay) cfg_scn_delay="$2"; shift;;
|
|
-m|--max-filesize) cfg_max_filesize=$2; shift;;
|
|
-A|--album-name) album_name=$2; shift;;
|
|
-t|--tags) tags=$2; shift;;
|
|
-c|--config) cfg_file=$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_scn='1';;
|
|
-S|--short-url) flag_shortlink=1;;
|
|
-P|--page-url) flag_directlink=0;;
|
|
-q|--quiet) flag_quiet=1;;
|
|
-v|--void) flag_void=1;;
|
|
|
|
--notify) flag_notify=1;;
|
|
--no-notify|-n) flag_notify=0;;
|
|
|
|
-F|--fullscreen)
|
|
flag_scn='1'
|
|
flag_scn_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
|
|
|
|
# Defaults
|
|
: ${cfg_file:="$XDG_CONFIG_DIR/ufw"}
|
|
: ${cfg_tmp_dir:="$XDG_RUNTIME_DIR/ufw"}
|
|
: ${cfg_service_url:='https://8fw.me'}
|
|
: ${cfg_screenshot_ext:='png'}
|
|
: ${cfg_max_filesize:='200M'}
|
|
: ${flag_shortlink:=0}
|
|
: ${flag_directlink:=1}
|
|
|
|
if [[ -f "$cfg_file" ]]; then
|
|
if ! source "$cfg_file"; then
|
|
printf 'Failed to source configuration file: %s\n' "$cfg_file"
|
|
return $?
|
|
fi
|
|
fi
|
|
|
|
TEMPDIR="$cfg_tmp_dir"
|
|
|
|
for i in "$cfg_tmp_dir"; do
|
|
[[ -d "$i" ]] || { mkdir -p "$i"; }
|
|
done
|
|
|
|
case "${action:-upload}" in
|
|
upload)
|
|
get_token || return $?
|
|
|
|
if (( $# )); then
|
|
for t in "$@"; do
|
|
upload "$t"
|
|
done
|
|
else
|
|
upload
|
|
fi
|
|
;;
|
|
|
|
url)
|
|
get_token || return $?
|
|
|
|
for t in "$@"; do
|
|
get_shortlink "$1"
|
|
done
|
|
;;
|
|
|
|
getmyip) get_my_ip;;
|
|
login) login;;
|
|
esac
|
|
}
|
|
|
|
set_argv "$@"
|
|
main "${argv[@]}"
|