azurehpc/libexec/azhpc-status.sh

#!/bin/bash
azhpc_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )/.." && pwd )"
source "$azhpc_dir/libexec/common.sh"

config_file="config.json"

DEBUG_ON=0
COLOR_ON=1

function usage() {
    echo "Command:"
    echo "    $0 [options]"
    echo
    echo "Arguments"
    echo "    -h --help  : diplay this help"
    echo "    -c --config: config file to use"
    echo "                 default: config.json"
    echo
}

while true; do
    case $1 in
        -h|--help)
        usage
        exit 0
        ;;
        -c|--config)
        config_path="$2"
        shift
        shift
        ;;
        *)
        break
    esac
done

read_value location ".location"
read_value resource_group ".resource_group"
read_value vnet_name ".vnet.name"
read_value address_prefix ".vnet.address_prefix"
read_value admin_user ".admin_user"
read_value install_node ".install_from"

ssh_private_key=${admin_user}_id_rsa
ssh_public_key=${admin_user}_id_rsa.pub
if [ ! -e "$ssh_private_key" ]; then
    error "keys not found"
fi

fqdn=$(
    az network public-ip show \
        --resource-group $resource_group \
        --name ${install_node}pip --query dnsSettings.fqdn \
        --output tsv \
)

if [ "$fqdn" = "" ]; then
    status "The install node does not have a public IP.  Using hostname - $install_node - and must be on this node must be on the same vnet"
fi

if [ "$COLOR_ON" -ne "1" ]; then
    printf "%%-20s %s\n" Hostname Status
else
    printf "\e[1m%-20s %s\e[0m\n" Hostname Status
fi

nodes=()

for resource_name in $(jq -r ".resources | keys | @tsv" $config_file); do
    read_value resource_type ".resources.$resource_name.type"

    read_value password_required ".resources.$resource_name.password" "<no-password>"

    if [ "$password_required" != "<no-password>" ]; then
        continue
    fi

    if [ "$resource_type" = "vm" ]; then

        read_value resource_instances ".resources.$resource_name.instances" 1
        
        if [ "$resource_instances" = 1 ]; then
            nodes+=($resource_name)
        else
            for i in $(seq -w $resource_instances); do
                nodes+=(${resource_name}${i})
            done
        fi

    elif [ "$resource_type" = "vmss" ]; then
        nodes+=($(az vmss list-instances \
            --resource-group $resource_group \
            --name $resource_name \
            --query [].osProfile.computerName \
            --output tsv))

    else
        error "unknown resource type ($resource_type) for $resource_name"
    fi

done

ssh $SSH_ARGS -q -i $ssh_private_key $admin_user@$fqdn 'pssh -H "'${nodes[@]}'" -i "printf \"%-20s%s\n\" \"\$(hostname)\" \"\$(uptime)\""' | grep -v SUCCESS