#!/bin/bash
# Copyright 2006-2017, Matthew Welland.
#
# This file is part of Megatest.
#
# Megatest is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Megatest is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Megatest. If not, see <http://www.gnu.org/licenses/>.
###############################################################################
#
# 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