#!/bin/bash source $(dirname $0)/vars ## Helper funcs pinfo() { echo -e "\e[32m$@\e[39m" } perror() { echo -e "\e[31m$@\e[39m" exit 1 } sshcmd() { ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null $* } dls() { path=$1 shift if [ -z "$DLS_ADM_TOKEN" ]; then test -f $tknfile && DLS_ADM_TOKEN=$(cat $tknfile) fi curl -s -H 'Content-Type: application/json' -H 'Authorization: '$DLS_ADM_TOKEN http://127.0.0.1:7606${path} "$@" } kctl() { kubectl --kubeconfig $ctxdir/kubeconfig "$@" } extract_var() { where=$1 what=$2 grep -rh " $what: " $ctxdir/$where/*.yaml|awk '{print $2}' } ## Run funcs check_root() { [ $UID != 0 ] && perror This program needs to be run as root. Aborting... } prereqs() { set -e cd "$(dirname $0)/.." ctxdir="$PWD" source scripts/vars } check_conf() { all_clusters=$(ls $ctxdir/clusters|wc -l) if [ "$all_clusters" != "1" ]; then perror "Those helper scripts are not capable of running several clusters at once, check your configuration. Aborting..." fi } fresh_start() { rm -rf $ctxdir/secrets } unlock_store() { # Unlock DLS store after checking if online # if [ -f $tknfile ]; then DLS_ADM_TOKEN="$(cat $tknfile)" fi store_state=$(curl -H 'Content-Type: application/json' -H "Authorization: $DLS_ADM_TOKEN" -sw %{http_code} localhost:7606/hosts -o /dev/null) if [ "$store_state" == "000" ] ; then perror "Direktil Local Server seems not up, please fix. Aborting." elif [ "$store_state" == "200" ] ; then pinfo "Direktil Local Server store already unlocked" else pinfo "Unlocking the DLS store ..." DLS_ADM_TOKEN=$(dls /public/unlock-store -d "\"${DLS_UNLOCK_TOKEN}\""|tr -d \") pinfo "Admin access token is $DLS_ADM_TOKEN" echo $DLS_ADM_TOKEN > $tknfile chmod 444 $tknfile fi } get_hosts() { cd $ctxdir/hosts hosts_files=$(ls *.yaml|sed 's/.yaml//') for h in ${hosts_files}; do ip=$(grep ip: $ctxdir/hosts/${h}.yaml|awk '{print $2}') hosts[$h]="$ip" done cd - &>/dev/null } get_parts() { for host in ${!hosts[*]}; do mkdir -p $ctxdir/data/$host for part in kernel initrd-v2 do partfile=$ctxdir/data/$host/$part test -f $partfile || { pinfo "Downloading $part for host $host" && dls /hosts/$host/$part -o $partfile; } done diskfile=$ctxdir/data/$host/disk test -f $diskfile || { pinfo "Creating disk $diskfile" && truncate -s ${QEMU_DISK_SIZE:-30G} $diskfile; } done } create_kubeconfig() { if test -f $ctxdir/kubeconfig; then pinfo "kubeconfig file detected in config dir, won't overwrite... remove it for an update." return fi adm_token=$(dls /clusters/base/tokens/admin) ca_cert=$(dls /clusters/base/CAs/cluster/certificate|base64 -w0) vip=$(extract_var clusters public_vip) vip_api_port=$(extract_var clusters api_port) pinfo "Writing new kubeconfig conf in $ctxdir directory, you may want to move it to \~/.kube/ directory for usability" sed -e "s/__VIP_IP__/$vip/" \ -e "s/__VIP_API_PORT__/$vip_api_port/" \ -e "s/__CA_CERT__/$ca_cert/" \ -e "s/__ADM_TOKEN__/$adm_token/" \ scripts/.template.kubeconfig > $ctxdir/kubeconfig chmod 444 $ctxdir/kubeconfig } destroyvms() { for host in ${!hosts[*]}; do if test -f $ctxdir/data/$host/pid ; then pid=$(cat $ctxdir/data/$host/pid) if ! test -d /proc/$pid ; then pinfo "VM $host seems not running" else pinfo "Cleaning VM $host with PID $pid..." kill $pid && sleep 1 fi fi done }