path: root/fb.in

#!/bin/sh
#----------------------------------------------------
# Author:       Florian "Bluewind" Pritz <flo@xssn.at>
# Contributor:  Moritz Wilhelmy
#
# Licensed under GPLv3
#   (see COPYING for full license text)
#
#----------------------------------------------------
# Optional dependency: xclip
#----------------------------------------------------

VERSION="@VERSION@"

DELETE=
EXTENSION=""
GET=
ENCRYPT=
TAR=
COMPRESS=0
TAREXT=".tar"
TAROPTS=""
DISPLAYHISTORY=
PASTEBIN="http://paste.xinu.at"
FBHIST="$HOME/.fb-client.hist"
WARNSIZE=10485760
USERAGENT="fb-client/$VERSION"
CLIPBOARD=""
EXITCODE=0
LIBDIR="@LIBDIR@"

require_executable() {
	if ! type $1 >/dev/null; then
		echo "Error: $1 not found. Please install." >&2
		exit 1
	fi
}

do_upload() {
	local EXTRA=""
	file="$1"
	basefilename="`basename -- "$file"`"
	basedirname="`dirname -- "$file"`"

	if [ ! -e "$file" ]; then
		echo "Error: File \"$file\" doesn't exist" >&2
		EXITCODE=1
		return 1
	fi

	if [ -d "$file" ]; then
		cd "$basedirname"
		tar $TAROPTS -cf "$TMPDIR/$basefilename$TAREXT" -- "$basefilename"
		COMPRESS=0
		file="$TMPDIR/$basefilename$TAREXT"
	fi

	if [ "$COMPRESS" = "1" ]; then
		require_executable gzip
		gzip -c -- "$file" > "$TMPDIR/$basefilename.gz"
		file="$TMPDIR/$basefilename.gz"
	elif [ "$COMPRESS" = "2" ]; then
		require_executable xz
		xz -c -- "$file" > "$TMPDIR/$basefilename.xz"
		file="$TMPDIR/$basefilename.xz"
	fi

	basefilename="`basename -- "$file"`"

	if [ "$ENCRYPT" = "1" ]; then
		if ! gpg -o "$TMPDIR/$basefilename.gpg" -e "$file"; then
			EXITCODE=1
			return 1
		fi
		file="$TMPDIR/$basefilename.gpg"
	fi

	TMPFILE=`mktemp "$TMPDIR/data.XXXXXX"`
	if [ `stat -c %s -- "$file"` -gt "$WARNSIZE" ]; then
		WARNSIZE=`$LIBDIR/fb-helper d "$PASTEBIN/file/get_max_size"`
		if [ `stat -c %s -- "$file"` -gt "$WARNSIZE" ]; then
			echo "Warning: Your upload is too big and would be rejected. Maximum size is: $WARNSIZE bytes. Skipping..." >&2
			EXITCODE=1
			return 1
		fi
	fi

	if ! $LIBDIR/fb-helper u "$PASTEBIN/file/do_upload" "$file" > $TMPFILE; then
		EXITCODE=1
		return 1
	fi
	sed '$d' $TMPFILE >&2
	URL=`tail -1 $TMPFILE`"$EXTENSION"
	echo $URL
	CLIPBOARD="$CLIPBOARD $URL"
	touch "$FBHIST"
	echo "$URL :: $basefilename :: `date +%s` :: `stat -c %s -- "$file"`" >> "$FBHIST"
}

read_stdin() {
	if tty -s; then
		echo "^C to exit, ^D to send"
	fi
	cat > "$1"
}

id_from_arg() {
	if echo "$1" | grep -qE "^https?://"; then
		echo "$1" | sed -r 's/https?:\/\/[^\/]+\/([^\/]+).*/\1/'
	else
		echo "$1"
	fi
}

help() {
	cat <<!
fb-client version $VERSION
usage: [cat |] `basename "$0"` [switches] [options] [<file(s)|ID(s)|folder(s)>]

  Upload/nopaste file(s)/stdin to paste.xinu.at and copy URL(s) to clipboard.
  ~/.netrc: machine paste.xinu.at password PASSWORD

  Switches:
    -d <ID(s)>     delete the IDs
    -g <ID(s)>     download the IDs and output on stdout (use with care!)
    -h             this help
    -v             show the client version
    -H             display an upload history

  Options:
    -e <extension>   extension for default highlighting (e.g. "diff")
    -t               upload a tar file containing all files (and directories)
    -c               compress the file being uploaded with gz or xz if used 2 times
                     When used in conjunction with -g this decompresses the download
    -C               encrypt the file being uploaded
!
}

while getopts "e:gdhHtcCv" OPTION; do
	case $OPTION in
		e) EXTENSION="$OPTARG";;
		g) GET=1;;
		C) ENCRYPT=1; require_executable gpg;;
		c) COMPRESS=`expr $COMPRESS + 1`
			 if [ "$COMPRESS" == "1" ]; then
				 TAROPTS="-z"
				 TAREXT=".tar.gz"
			 elif [ "$COMPRESS" == "2" ]; then
				 TAROPTS="-J"
				 TAREXT=".tar.xz"
			 fi
			 ;;
		t) TAR=1;;
		d) DELETE=1;;
		H) DISPLAYHISTORY=1;;
		v) echo "$VERSION"; exit 0;;
		h|\?) help; exit 0;;
	esac
done

shift `expr $OPTIND - 1`


TMPDIR="`mktemp -d "/tmp/fb.XXXXXX"`"
trap "rm -rf '${TMPDIR}'" EXIT TERM

if [ "$DELETE" ] || [ "$GET" ]; then
	if [ $# -eq 0 ]; then
		echo "Error: no ID specified" >&2
		exit 1
	fi
	for i in "$@"; do
		i=$(id_from_arg "$i")
		if [ "$DELETE" ]; then
			$LIBDIR/fb-helper d "$PASTEBIN/file/delete/$i" || EXITCODE=1
		elif [ "$GET" ]; then
			if [ "$COMPRESS" = "1" ]; then
				require_executable zcat
				$LIBDIR/fb-helper d "$PASTEBIN/$i" | zcat || EXITCODE=1
			elif [ "$COMPRESS" = "2" ]; then
				require_executable xzcat
				$LIBDIR/fb-helper d "$PASTEBIN/$i" | xzcat || EXITCODE=1
			else
				$LIBDIR/fb-helper d "$PASTEBIN/$i" || EXITCODE=1
			fi
		fi
	done
elif [ "$DISPLAYHISTORY" ]; then
	maxage=`echo 10*24*60*60 | bc`
	minsize=`echo 10*1024| bc`

	now=`date +%s`
	oldest_time=`echo "$now - $maxage" | bc`

	awk -F" :: " '
		($4 < '$minsize' || $3 < '$oldest_time' ) { print $1" - "$2}
	' $FBHIST

elif [ $# -eq 0 ]; then
	if [ "$TAR" ]; then
		echo "Error: -t is not supported when operating on stdin" >&2
		exit 1
	fi
	read_stdin "$TMPDIR/stdin"
	do_upload "$TMPDIR/stdin"
else
	if [ "$TAR" ]; then
		tar $TAROPTS -cf "$TMPDIR/upload$TAREXT" -- "$@"
		COMPRESS=0
		do_upload "$TMPDIR/upload$TAREXT"
		break
	else
		for i in "$@"; do
			if echo "$i" | grep -qE "^(f|ht)tp(s)?://.+"; then
				cd $TMPDIR
				if ! $LIBDIR/fb-helper d "$i" > "`basename "$i"`"; then
					EXITCODE=1
					continue
				fi
				for f in *; do
					do_upload "$f" && rm -f -- "$f"
				done
			else
				do_upload "$i"
			fi
		done
	fi
fi

if [ "$CLIPBOARD" != "" ]; then
	[ "`type xclip 2>/dev/null`" ] && echo -n $CLIPBOARD | nohup xclip >/dev/null 2>&1
fi

exit $EXITCODE

#vim: set noet: