set -euo pipefail

stop_daemon() {
	echo "Killing warp-svc process..."
	sudo pkill -x warp-svc
	echo "warp-svc is not running."
}

disconnect_client() {
	echo "Disconnecting from WARP..."
	if warp-cli status 2>&1 | grep -qi "Connected"; then
		sudo warp-cli disconnect
		echo "Disconnected."
	else
		echo "WARP is already disconnected."
	fi
}

main() {
	if [[ -t 2 ]]; then
		echo -e "\e[31mERROR:\e[0m Must be run as root" >&2
	else
		echo "ERROR: Must be run as root" >&2
	fi

	disconnect_client
	stop_daemon

	echo "Final status:"
	warp-cli status || true
	sudo systemd-resolve --flush-caches
	sudo systemctl restart systemd-resolved
}

main "$@"