#!/usr/bin/env bash

BASHTARD_PLAYBOOK_VARS[$BASHTARD_PLAYBOOK.entry.host]="required"
BASHTARD_PLAYBOOK_VARS[$BASHTARD_PLAYBOOK.entry.token]="required"

playbook_add() {
	local role

	role="$(config "$BASHTARD_PLAYBOOK.role" "agent")"

	pkg install curl

	info "$BASHTARD_PLAYBOOK" "Writing config.yaml for k3s"
	mkdir -pv -- /etc/rancher/k3s
	cat <<-EOF > /etc/rancher/k3s/config.yaml
	node-name: "${BASHTARD_PLATFORM[fqdn]}"
	node-ip: "$(config "$BASHTARD_PLAYBOOK.internal-ip" "127.0.0.1")"
	server: "https://$(config "$BASHTARD_PLAYBOOK.entry.host"):$(config "$BASHTARD_PLAYBOOK.entry.port" "6443")"
	token: "$(config "$BASHTARD_PLAYBOOK.entry.token")"
	EOF

	if [[ -n "$(config "$BASHTARD_PLAYBOOK.external-ip" "")" ]]
	then
		printf "%s: %s\n" "node-external-ip" "$(config "$BASHTARD_PLAYBOOK.external-ip" "")"\
			>> /etc/rancher/k3s/config.yaml
	fi

	if [[ "$role" == "server" ]]
	then
		cat <<-EOF >> /etc/rancher/k3s/config.yaml
		cluster-cidr: "$(config "$BASHTARD_PLAYBOOK.cluster-cidr" "172.19.0.0/16")"
		cluster-domain: "$(config "$BASHTARD_PLAYBOOK.cluster-domain" "cluster.local")"
		service-cidr: "$(config "$BASHTARD_PLAYBOOK.service-cidr" "172.20.0.0/16")"
		service-node-port-range: "$(config "$BASHTARD_PLAYBOOK.service-node-port-min" "30000")-$(config "$BASHTARD_PLAYBOOK.service-node-port-max" "32767")"
		flannel-ipv6-masq: true
		disable:
		- traefik
		EOF
	fi

	info "$BASHTARD_PLAYBOOK" "Installing k3s"
	curl -sfL https://get.k3s.io | sh -s - "$role"

	notice "$BASHTARD_PLAYBOOK" "Waiting for node to become available"
	{ grep -q -m 1 "${BASHTARD_PLATFORM[fqdn]}[[:space:]]\+Ready"; kill $!; } < <(k3s kubectl get node -w)

	true
}

playbook_sync() {
	:;
}

playbook_del() {
	local role

	role="$(config "$BASHTARD_PLAYBOOK.role" "agent")"

	case "$role" in
		server)
			/usr/local/bin/k3s-uninstall.sh
			;;
		agent)
			/usr/local/bin/k3s-agent-uninstall.sh
			;;
	esac
}