pub_machine

#!/bin/bash

## Source this to add the functions to your environment

function choice {
    file=${1}
    lc=$(wc -l < $file)

    if [ ${lc} -gt 0 ]; then
        sed -n "$((1 + $RANDOM % ${lc})) p" ${file}
    fi
}

function check_machine {
    local ulist=$(ssh $LAST_PUBLIC "users");
    for u in ${ulist}; do
        if [ ${u} != "$(whoami)" ]; then
            LAST_PUBLIC=""
        fi
    done
}

function new_machine {
    export LAST_PUBLIC=""
    pub_machine
}

function pub_machine {
    local DOMAIN=${DOMAIN:-hosts.txt}
    if [ -f ${HOME}/.${DOMAIN} ]; then
        if [ -n "${LAST_PUBLIC}" ]; then
            check_machine
        fi

        while [ -z "${LAST_PUBLIC}" ]; do
            local mach=$(choice ${HOME}/.${DOMAIN})
            if [ ${mach:0:1} == "#" ]; then
                continue
            fi
            echo "Trying ${mach}..."
            LAST_PUBLIC="${mach}.${DOMAIN}"
            check_machine
        done
    fi
    export LAST_PUBLIC
}

function pub_connect {
    local DOMAIN=${DOMAIN:-hosts.txt}
    if [ -f ${HOME}/.${DOMAIN} ]; then
        pub_machine
        ssh $(whoami)@${LAST_PUBLIC} ${@}
    fi
}