113 lines
2.7 KiB
Bash
113 lines
2.7 KiB
Bash
#!/bin/bash
|
|
module_load ssh
|
|
remote_script () {
|
|
# usage: remote_script script -- <ssh setup options> host <ssh options>
|
|
# see ssh function
|
|
#
|
|
local sshargs;local user;local supass;local fn;local args;local file
|
|
|
|
[[ "$*" =~ "--" ]] && sshargs=$(sed 's/.*--\(.*\)/\1/' <<< "$@")
|
|
[[ ! $sshargs ]] && { echo missing remote machine, must provide at least a hostname, -- hostname; return 3; }
|
|
# reset arguments to just those before --
|
|
set -- $(sed 's/\(.*\)--.*/\1/' <<< "$@")
|
|
|
|
local OPTION
|
|
local OPTARG
|
|
local OPTIND
|
|
while getopts 'u:s:f:p:' OPTION; do
|
|
# echo OPTION $OPTION ARG $OPTARG INDX $OPTIND
|
|
case "$OPTION" in
|
|
s)
|
|
# script filename to run
|
|
file=$OPTARG
|
|
;;
|
|
f)
|
|
# run a function within the script
|
|
fn=$OPTARG
|
|
;;
|
|
u)
|
|
# run command as another user
|
|
user=$OPTARG
|
|
;;
|
|
p)
|
|
# password of sudo account for running as root or other user
|
|
supass=$OPTARG
|
|
;;
|
|
*)
|
|
echo unknown remote script option -$OPTARG
|
|
return 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
shift $((OPTIND - 1))
|
|
|
|
[[ ! $file ]] && { file=$1;shift 1; }
|
|
if [[ ! -f $file ]]; then
|
|
file=$(module_find $file)
|
|
[[ ! $file ]] && { echo no module or script at $file to run; return 1; }
|
|
fi
|
|
|
|
# remaining script arguments
|
|
args=$*
|
|
# echo arguments for script: $args
|
|
|
|
script=$file
|
|
# if script loads a module then use bundler
|
|
if [[ $(cat $script | grep module_load) ]]; then
|
|
# echo bundling modules with script
|
|
local temp
|
|
module_load bundle
|
|
script=$( mktemp -t TEMP_FILE_script.XXXXXXXX )
|
|
bundle $file $script
|
|
temp=true
|
|
fi
|
|
|
|
local _sudo
|
|
if [[ $user ]]; then
|
|
[[ ! $supass ]] && { echo password must be supplied for connecting sudo account. use -s;return 7; }
|
|
# echo running script remotely as user $user
|
|
_sudo="echo '${supass}' | sudo --stdin -u ${user} 2>/dev/null"
|
|
else
|
|
if [[ $supass ]]; then
|
|
_sudo="echo '${supass}' | sudo --stdin 2>/dev/null"
|
|
# echo running script remotely as root
|
|
fi
|
|
fi
|
|
|
|
src="$(cat $script);"
|
|
if [[ $fn ]]; then
|
|
# echo running function
|
|
cmd="bash -c '$src $fn ${args[*]}'"
|
|
else
|
|
# echo running as script
|
|
local sargs
|
|
[[ ${args} ]] && sargs="$(printf '%s\n' "set -- ${args}"); "
|
|
cmd="bash -c '$sargs $src'"
|
|
fi
|
|
|
|
# echo ssh $sshargs "$_sudo" "$cmd"
|
|
ssh $sshargs "$_sudo" "$cmd"
|
|
|
|
[[ $temp ]]&& rm -f $script
|
|
|
|
}
|
|
|
|
|
|
# can be used with a file or HEREDOC
|
|
# todo test with < $file or with a pipe
|
|
remote_stdin () {
|
|
local file
|
|
# echo args $@
|
|
file=$( mktemp -t TEMP_FILE_script.XXXXXXXX )
|
|
cat > $file
|
|
echo remote_script -s "$file" "$@"
|
|
#remote_script -s "$file" "$@"
|
|
rm $file
|
|
|
|
}
|
|
|
|
# # if script was executed then call the function
|
|
(return 0 2>/dev/null) ||remote_script $@
|
|
|