#!/usr/bin/env bash

BASHTARD_PLAYBOOK_VARS[$BASHTARD_PLAYBOOK.dc]="required"
BASHTARD_PLAYBOOK_VARS[$BASHTARD_PLAYBOOK.ip]="required"
BASHTARD_PLAYBOOK_VARS[$BASHTARD_PLAYBOOK.rack]="required"

playbook_add() {
	local buffer

	# Make sure seaweedfs is installed
	"$BASHTARD_BIN" add seaweedfs || true

	# Generate systemd unit files
	case "${BASHTARD_PLATFORM[init]}" in
		systemd)
			cat <<-EOF > "$(config "fs.etcdir")/systemd/system/seaweedfs-volume@.service"
			[Unit]
			Description=SeaweedFS Volume
			After=network.target

			[Service]
			Type=simple
			User=root
			Group=root

			ExecStart=$(config "fs.bindir")/weed volume -options="$(config "fs.etcdir")/seaweedfs/volume-%i.conf"
			WorkingDirectory=/var/lib/seaweedfs/volume/%i
			SyslogIdentifier=seaweedfs

			[Install]
			WantedBy=multi-user.target
			EOF
			;;
		*) die "NYI"
	esac

	# Perform initial configuration
	playbook_sync

	# Start services
	while read -r volume
	do
		systemctl enable --now "seaweedfs-volume@$volume"
	done < <(config_subkeys "$BASHTARD_PLAYBOOK.volumes")
}

playbook_sync() {
	local buffer
	local conf
	local peers=()
	local hash

	buffer="$(tmpfile)"

	# Add all registered seaweedfs-master nodes as peers
	while read -r host
	do
		peers+=("[$(config_for "$(basename "$host")" "seaweedfs-master.ip")]:$(config_for "$(basename "$host")" "seaweedfs-master.port" "9333")")
	done < <(grep -FHl "seaweedfs-master" "$BASHTARD_ETCDIR/registry.d"/*)

	while read -r volume
	do
		conf="$(config "fs.etcdir")/seaweedfs/volume-$volume.conf"
		hash="$(file_hash "$conf")"

		info "$BASHTARD_PLAYBOOK/sync/$volume" "Updating $conf"

		# Generate config file
		file_template "volume.conf" \
			dc="$(config "$BASHTARD_PLAYBOOK.dc")" \
			dir="/var/lib/seaweedfs/volume/$volume" \
			disk="$(config "$BASHTARD_PLAYBOOK.volume.$volume.disk" "hdd")" \
			max="$(config "$BASHTARD_PLAYBOOK.volume.$volume.max" "0")" \
			free_space="$(config "$BASHTARD_PLAYBOOK.free-space" "10GiB")" \
			filesize_limit="$(config "$BASHTARD_PLAYBOOK.filesize-limit" "256")" \
			ip="$(config "$BASHTARD_PLAYBOOK.ip")" \
			mserver="$(join_args "${peers[@]}")" \
			port="$(config "$BASHTARD_PLAYBOOK.volume.$volume.port" "8080")" \
			rack="$(config "$BASHTARD_PLAYBOOK.rack")" \
			> "$buffer"

		# Write config file
		[[ "$(file_hash "$buffer")" == "$hash" ]] && continue
		info "$BASHTARD_PLAYBOOK/sync/$volume" "Configuration file changed"
		mv -- "$buffer" "$conf"

		# Restart volume server
		info "$BASHTARD_PLAYBOOK/sync/$volume" "Restarting volume $volume"
		[[ "$BASHTARD_ACTION" == "sync" ]] && systemctl restart "seaweedfs-volume@$volume"
	done < <(config_subkeys "$BASHTARD_PLAYBOOK.volumes")
}

playbook_del() {
	# Stop services
	while read -r volume
	do
		systemctl disable --now "seaweedfs-volume@$volume"
	done < <(config_subkeys "$BASHTARD_PLAYBOOK.volumes")

	# Remove systemd unit file
	rm -fr -- "$(config "fs.etcdir")/systemd/system/seaweedfs-volume@.service"

	# Remove configuration files
	rm -fr -- "$(config "fs.etcdir")/seaweedfs/volume.conf"
}