#!/bin/bash
###############################################################################
#
# nbfake - capture command output in a logfile
#
# nbfake behavior can be changed by setting the following env vars:
# NBFAKE_HOST SSH to $NBFAKE_HOST and run command
# NBFAKE_LOG Logfile for nbfake output
# NB_WASH_GROUPS comma-separated list of groups to wash into
# NB_WASH_ENABLED must be set in order to enable wash groups
#
###############################################################################
if [[ -z "$@" ]]; then
cat <<__EOF
nbfake usage:
nbfake <command to run>
nbfake behavior can be changed by setting the following env vars:
NBFAKE_HOST SSH to \$NBFAKE_HOST and run command
NBFAKE_LOG Logfile for nbfake output
NB_WASH_GROUPS comma-separated list of groups to wash into
NB_WASH_ENABLED must be set in order to enable wash groups
__EOF
exit
fi
#==============================================================================
# Setup
#==============================================================================
# Can't always trust $PWD
CURRWD=$(pwd)
# Make sure nbfake host and logfile are set. Fall back to old-style variable names
if [[ -z "$NBFAKE_HOST" && -n "$TARGETHOST" ]]; then
MY_NBFAKE_HOST=$TARGETHOST
unset TARGETHOST
else
MY_NBFAKE_HOST=$NBFAKE_HOST
unset NBFAKE_HOST
fi
if [[ -z "$NBFAKE_LOG" && -n "$TARGETHOST_LOGF" ]]; then
MY_NBFAKE_LOG=$TARGETHOST_LOGF
unset TARGETHOST_LOGF
else
MY_NBFAKE_LOG=$NBFAKE_LOG
unset NBFAKE_LOG
fi
# Set default nbfake log
if [[ -z "$MY_NBFAKE_LOG" ]]; then
MY_NBFAKE_LOG=NBFAKE-$(date +%GWW%V.%u_%T)
fi
# wash groups handling. Default is no action
WASHCMD=""
if [[ -n ${NB_WASH_ENABLED+1} && -n ${NB_WASH_GROUPS+1} ]]; then
grouplist=`echo $NB_WASH_GROUPS | tr ',' ' '`
WASHCMD="wash -q -n $grouplist -X"
fi
#==============================================================================
# Run and log
#==============================================================================
cat <<__EOF >&2
#======================================================================
# NBFAKE logging command to: $MY_NBFAKE_LOG
# $WASHCMD $*
#======================================================================
__EOF
if [[ -z "$MY_NBFAKE_HOST" ]]; then
# Run locally
sh -c "cd $CURRWD;export DISPLAY=$DISPLAY; export PATH=$PATH; nohup $WASHCMD $* >> $MY_NBFAKE_LOG 2>&1 &"
else
# run remotely
ssh -X -n -f $MY_NBFAKE_HOST "sh -c \"cd $CURRWD;export DISPLAY=$DISPLAY; export PATH=$PATH; nohup $WASHCMD $* >> $MY_NBFAKE_LOG 2>&1 &\""
fi