#!/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
}