runner-for-client

#!/bin/bash

# Copyright 2022 CNRS-LAAS
# 
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# 
#     http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# 
# Author: David Gauchard

root=$(cd ${0%/*}; pwd)
name=${0##*/}
vnc=/opt/TurboVNC/bin/vncviewer

help ()
{
    if [ -d ${root}/__gputainer ]; then
        cat << EOF

This script must be started on your desktop host with:

    scp ${LOGNAME}@$(hostname):${root}/${name} . && ./${0##*/}  ${LOGNAME}@$(hostname)  ${root}  <application-name>

where application name is one of:

$(cd ${root}; for d in *; do [ -d ${d} ] && echo ${d}; done | grep -v ^__)

EOF
    else
        cat << EOF
usage: ./${name} ${LOGNAME}@$(hostname) <path> <application-name>
EOF
    fi
    exit 1
}

server=${1}
path=${2}
app=${3}
[ -z "${app}" -o -d "${root}/__gputainer" ] && help ${0}
shift 3

if [ ! -x ${vnc} ]; then
    echo "TurboVNC must be installed ('${vnc}')"
    exit 1
fi

set -e

running=false
ssh -t ${server} ${path}/__gputainer/runner-on-server ${app} "${@}" | while read line; do
    if echo "${line}" | grep ${vnc} > /dev/null; then
        if ! ${running} ; then
            running=true
            echo "Starting VNC client"
            echo "  ${line}"
            ( ${line} 2>&1 & vncpid=$!; echo $! >&3 ) 3>vncpid | while read vncline; do
                echo -e "vncclient($(<vncpid)) -- ${vncline}\r"
                if echo "${vncline}" | grep "Connection reset by peer"; then
                    echo Killing VNC client $(<vncpid)
                    kill $(<vncpid)
                    break
                fi
            done
        fi
    fi
done

echo "Remote runner stopped"