container.training/prepare-vms/lib/pssh.sh

# This file can be sourced in order to directly run commands on
# a group of VMs whose IPs are located in ips.txt of the directory in which
# the command is run.

pssh() {
    if [ -z "$TAG" ]; then
        >/dev/stderr echo "Variable \$TAG is not set."
        return
    fi

    HOSTFILE="tags/$TAG/ips.txt"

    [ -f $HOSTFILE ] || {
        >/dev/stderr echo "Hostfile $HOSTFILE not found."
        return
    }

    echo "[parallel-ssh] $@"
    export PSSH=$(which pssh || which parallel-ssh)

    case "$INFRACLASS" in
        hetzner) LOGIN=root ;;
        linode)  LOGIN=root ;;
        *)       LOGIN=ubuntu ;;
    esac

    if [ -f "tags/$TAG/id_rsa" ]; then
        KEYFLAG="-O IdentityFile=tags/$TAG/id_rsa"
    else
        KEYFLAG=""
    fi

    $PSSH $KEYFLAG -h $HOSTFILE -l $LOGIN \
        --par 100 \
        --timeout 300 \
        -O LogLevel=ERROR \
        -O UserKnownHostsFile=/dev/null \
        -O StrictHostKeyChecking=no \
        -O ForwardAgent=yes \
        "$@"
}