#!/bin/bash
# 2008-04-10
AUTHOR="tyler@tolaris.com"
NAME="smount"
UNAME="sumount"
VERSION="1.2"
DESC="Mount/unmount remote fuse.sshfs filesystems in /tmp/"
# Warning: sanitises username/host input before running sshfs, 
#    but will allow invalid input.
# Warning: assumes /tmp/HOSTNAME is empty or doesn't exist
# Warning: is not multiuser friendly

IFS=' 
	'

die() {
	printf "$@\n"
	exit 1
}

VersionAndExit() {
	echo "$NAME version $VERSION, by $AUTHOR"
	exit 0
}

UsageAndExit() {
   if [ -n "$2" ] ; then echo -e "$2\n" ; fi
   printf "\
Usage: $NAME [-h|-v] [user@]host ...\n\
       <$NAME -u|$UNAME> host ...\n\
       <$NAME|$UNAME> -a\n\
\n\
$DESC\n\
\n\
  -l	list mounted fuse.sshfs filesystems\n\
  -u	unmount host (same as \"$UNAME host\")\n\
  -a	unmount all fuse.sshfs filesystems\n\
  -h	display help\n\
  -v	display version\n\
"
   exit $1
}

# Check prerequisites
which sshfs > /dev/null || die "sshfs not found"
which fusermount > /dev/null || die "fusermount not found"

# defaults
TEMP=/tmp
RETCODE=0

while getopts ":ahluv" options; do
   case $options in
      h ) UsageAndExit 0;;
      v ) VersionAndExit 0;;
      a ) ALLUMOUNT="1" ; UMOUNTMODE="1";;
      l ) STATUSMODE="1";;
      u ) UMOUNTMODE="1";;
      * ) UsageAndExit 1 "Option unknown: -$OPTARG";;
   esac
done

# Clear $@ of switches and grab target list
shift $(($OPTIND - 1))
TARGETLIST=$@

# Unmount all mode?
if [ -n "$ALLUMOUNT" ] ; then
   MOUNTPOINTS=`mount -l -t fuse.sshfs | cut -f 3 -d ' '`
   for i in $MOUNTPOINTS ; do
      fusermount -u $i
      if [ "$?" -ne 0 ] ; then
         let $((RETCODE++))
		else
         rmdir $i
      fi
   done
	exit $RETCODE
fi

# Status mode?
if [ -n "$STATUSMODE" -o -z "$TARGETLIST" ] ; then
   mount -l -t fuse.sshfs
   exit 0
fi

# Implicit unmount mode?
if [ "$UNAME" = `basename "$0"` ] ; then
   UMOUNTMODE="1"
fi

# Loop on targets, mounting or unmounting
for TARGET in $TARGETLIST ; do
   
   HOST=`echo "$TARGET" | sed 's/^.*@//;s/[^a-zA-Z0-9.-].*$//'`
   REMOTE_USER=`echo "$TARGET" | grep -o '^.*@' | sed 's/@//'`
   
   if [ -z "$REMOTE_USER" ] ; then
      REMOTE_USER=root
   fi

   # Unmount?
   if [ -n "$UMOUNTMODE" ] ; then
      fusermount -u $TEMP/$HOST
      if [ "$?" -ne 0 ] ; then
         let $((RETCODE++))
		else
         rmdir $TEMP/$HOST
      fi

   # Or mount?
   else
      if [ ! -d $TEMP/$HOST ] ; then mkdir $TEMP/$HOST ; fi
      sshfs $REMOTE_USER@$HOST:/ $TEMP/$HOST/ 2> /dev/null
      # did sshfs fail?
      if [ "$?" -ne 0 ] ; then
         let $((RETCODE++))
         rmdir $TEMP/$HOST
      fi
   fi
done

exit $RETCODE