1
0
mirror of https://tildegit.org/tomasino/pb.git synced 2024-12-23 03:25:18 +01:00
tomasino-pb/pb
James Tomasino 79bda6971e Merge branch 'master' of tildeverse.org:tomasino/pb
* 'master' of tildeverse.org:tomasino/pb:
  change url to ttm.sh
  added -c to bash completion
  upped version number, added dodcumentation for -c
  without -c, remove extra labels
  color formatting added with -c
2018-08-16 12:49:24 -04:00

183 lines
3.8 KiB
Bash
Executable File

#!/bin/sh
version="v.2018.08.15"
ENDPOINT="https://ttm.sh"
flag_options="hvcfs::x"
flag_version=0
flag_help=0
flag_file=0
flag_shortlist=0
flag_colors=0
data=""
SUCCESS=$(tput setaf 190)
ERROR=$(tput setaf 196)
RESET=$(tput sgr0)
show_help() {
cat > /dev/stdout << END
pb [options] filename
or
(command-with-stdout) | pb
Uploads a file or data to the tilde 0x0 paste bin
OPTIONAL FLAGS:
-h Show this help
-v Show current version number
-f Explicitly interpret stdin as filename
-c Pretty color output
-s server_address Use alternative pastebin server address
END
}
die () {
msg="$1"
code="$2"
# exit code defaults to 1
if printf "%s" "${code}" | grep -q '^[0-9]+$'; then
code=1
fi
# output message to stdout or stderr based on code
if [ ! -z "${msg}" ]; then
if [ "${code}" -eq 0 ]; then
printf "%s\\n" "${msg}"
else
if [ ${flag_colors} -gt 0 ]; then
printf "%s%s%s\\n" "$ERROR" "${msg}" "$RESET" >&2
else
printf "%s\\n" "${msg}" >&2
fi
fi
fi
exit "${code}"
}
# is not interactive shell, use stdin
if [ -t 0 ]; then
flag_file=1
else
data="$(cat < /dev/stdin )"
fi
if ! parsed=$(getopt ${flag_options} "$@"); then
die "Invalid input" 2
fi
eval set -- "${parsed}"
while true; do
case "$1" in
-h)
flag_help=1
;;
-v)
flag_version=1
;;
-c)
flag_colors=1
shift
;;
-f)
flag_file=1
;;
-s)
shift
ENDPOINT="$1"
;;
-x)
flag_shortlist=1
;;
--)
shift
break
;;
*)
die "Internal error: $1" 3
;;
esac
shift
done
if [ -z "$data" ]; then
data="$*"
fi
if [ ${flag_version} -gt 0 ]; then
printf "%s\\n" "${version}"
die "" 0
fi
if [ ${flag_help} -gt 0 ]; then
show_help
die "" 0
fi
if [ ${flag_shortlist} -gt 0 ]; then
out="-f -v -h -s -c"
lsresults="$(ls)"
die "${out} ${lsresults}" 0
fi
if [ ${flag_file} -gt 0 ]; then
if [ -z "${data}" ]; then
if [ ${flag_colors} -gt 0 ]; then
printf "%sProvide data to upload%s\\n" "$ERROR" "$RESET"
else
printf "Provide data to upload\\n"
fi
elif [ ! -f "${data}" ]; then
if [ ${flag_colors} -gt 0 ]; then
printf "%s%s%s\\tFile not found.%s\\n" "$RESET" "${data}" "$ERROR" "$RESET"
else
printf "%s\\tFile not found.\\n" "${data}"
fi
# attempt to split data and upload each string as file
for f in ${data}
do
# if there's nothing to parse, skip this loop
if [ "$f" = "$data" ]; then
break;
fi
if [ ${flag_colors} -gt 0 ]; then
printf "%s%s\\t%s" "$RESET" "${f}" "$SUCCESS"
fi
if [ -f "${f}" ]; then
curl -F"file=@${f}" "${ENDPOINT}"
if [ ${flag_colors} -gt 0 ]; then
printf "%s" "$RESET"
fi
else
if [ ${flag_colors} -gt 0 ]; then
printf "%sFile not found.%s\\n" "$ERROR" "$RESET"
else
printf "File not found.\\n"
fi
fi
done
else
if [ ${flag_colors} -gt 0 ]; then
printf "%s${data}\\t%s" "$RESET" "$SUCCESS"
curl -F"file=@${data}" "${ENDPOINT}"
printf "%s" "$RESET"
else
curl -F"file=@${data}" "${ENDPOINT}"
fi
fi
else
if [ -z "${data}" ]; then
if [ ${flag_colors} -gt 0 ]; then
printf "%sNo data found for upload. Please try again.%s\\n" "$ERROR" "$RESET"
else
printf "No data found for upload. Please try again.\\n"
fi
else
if [ ${flag_colors} -gt 0 ]; then
printf "%s" "$SUCCESS"
printf "%s" "${data}" | curl -F"file=@-;filename=null.txt" "${ENDPOINT}"
printf "%s" "$RESET"
else
printf "%s" "${data}" | curl -F"file=@-;filename=null.txt" "${ENDPOINT}"
fi
fi
fi