108 lines
2.7 KiB
Bash
108 lines
2.7 KiB
Bash
#!/bin/bash
|
|
|
|
## Helper funcs
|
|
|
|
|
|
pinfo() {
|
|
echo -e "\e[32m$@\e[39m"
|
|
}
|
|
|
|
perror() {
|
|
echo -e "\e[31m$@\e[39m"
|
|
exit 1
|
|
}
|
|
|
|
dls() {
|
|
path=$1
|
|
shift
|
|
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
|
|
|
|
prereqs() {
|
|
[ $UID != 0 ] && perror This program needs to be run as root. Aborting...
|
|
set -ex
|
|
cd ..
|
|
ctxdir=$PWD
|
|
}
|
|
|
|
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
|
|
#
|
|
store_state=$(curl -sw %{http_code} localhost:7606/hosts -o /dev/null)
|
|
[ "$?" != 0 ] && perror "Direktil Local Server seems not up, please fix. Aborting."
|
|
|
|
if [ "$store_state" == "401" ]; then
|
|
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 > $ctxdir/secrets/.dls_adm_token
|
|
chmod 400 > $ctxdir/secrets/.dls_adm_token
|
|
fi
|
|
}
|
|
|
|
|
|
get_hosts() {
|
|
hosts_files=$(basename $ctxdir/hosts/*.yaml|sed 's/.yaml//')
|
|
for h in ${hosts_files}; do
|
|
ip=$(grep ip: $ctxdir/hosts/${h}.yaml|awk '{print $2}')
|
|
hosts[$h]="$ip"
|
|
done
|
|
}
|
|
|
|
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 || dls /hosts/$host/$part -o $partfile
|
|
done
|
|
diskfile=$ctxdir/data/$host/disk.raw
|
|
test -f $diskfile || truncate -s ${QEMU_DISK_SIZE:-30G} $diskfile
|
|
done
|
|
}
|
|
|
|
|
|
create_kubeconfig() {
|
|
adm_token=$(dls /clusters/base/tokens/admin)
|
|
ca_cert=$(dls clusters/base/CAs/cluster/certificate|base64 -w0)
|
|
vip=$(extract_var clusters vip)
|
|
vip=$(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 > kubeconfig
|
|
}
|
|
|
|
clean() {
|
|
sudo iptables -t nat -D POSTROUTING -j MASQUERADE -s $QEMU_BR_IP \! -o $QEMU_BR_NAME &>/dev/null
|
|
sudo iptables -D FORWARD -o $QEMU_BR_NAME -m conntrack --ctstate RELATED,ESTABLISHED -j ACCEPT &>/dev/null
|
|
sudo iptables -D FORWARD -j ACCEPT -i $QEMU_BR_NAME &>/dev/null
|
|
}
|
|
|