#!/usr/bin/env bash
# Shared parts of the local demos, sourced by dashboard-ce/tools/demo.sh and by the demos of workspaces that extend
# these packages: a MongoDB replica set (a container, or DASHBOARD_DEMO_MONGO_URI), 30 days of generated data kept live,
# and the backend and frontend development servers of dev-lib.sh. The caller sets DATABASE and STATS_MODE and runs
# from the root of its workspace.
#   DASHBOARD_DEMO_MONGO_URI    MongoDB to use instead of the container
#   DASHBOARD_DEMO_MONGO_PORT   host port of the container (default 27018)
#   DASHBOARD_DEMO_KEEP_DATA    true reuses the demo database instead of generating it again

TOOLS_DIR="$(dirname "${BASH_SOURCE[0]}")"

# shellcheck source=dashboard-ce/tools/dev-lib.sh
source "$TOOLS_DIR/dev-lib.sh"

MONGO_IMAGE="${DASHBOARD_TEST_MONGO_IMAGE:-openvidu/mongodb:8.0.29-r1}"
MONGO_CONTAINER=openvidu-dashboard-demo-mongo
MONGO_PORT="${DASHBOARD_DEMO_MONGO_PORT:-27018}"

start_mongo() {
	if [ "$(docker inspect -f '{{.State.Running}}' "$MONGO_CONTAINER" 2>/dev/null)" != true ]; then
		docker rm -f "$MONGO_CONTAINER" >/dev/null 2>&1 || true
		echo "Starting MongoDB ($MONGO_CONTAINER, port $MONGO_PORT)"
		# Docker's default soft limit of 1024 open files crashes mongod once a few databases have been generated.
		docker run -d --name "$MONGO_CONTAINER" -p "127.0.0.1:$MONGO_PORT:27017" --ulimit nofile=64000:64000 \
			--entrypoint /opt/bitnami/mongodb/bin/mongod "$MONGO_IMAGE" \
			--replSet rs0 --bind_ip_all --port 27017 --dbpath /tmp --wiredTigerCacheSizeGB 1 >/dev/null
	fi

	for _ in $(seq 1 60); do
		docker exec "$MONGO_CONTAINER" /opt/bitnami/mongodb/bin/mongosh --quiet --eval \
			'try { rs.status().ok } catch (e) { rs.initiate({ _id: "rs0", members: [{ _id: 0, host: "localhost:27017" }] }).ok }' \
			2>/dev/null | grep -q 1 && return 0
		sleep 1
	done

	echo "MongoDB did not become ready" >&2
	exit 1
}

# prepare_data: MongoDB, the generated database (unless kept) and the keep-alive of its active sessions; sets MONGO_URI.
prepare_data() {
	if [ -n "${DASHBOARD_DEMO_MONGO_URI:-}" ]; then
		MONGO_URI="$DASHBOARD_DEMO_MONGO_URI"
	else
		docker info >/dev/null 2>&1 || {
			echo "Docker is required: MongoDB runs in a container (or set DASHBOARD_DEMO_MONGO_URI)." >&2
			exit 1
		}
		start_mongo
		MONGO_URI="mongodb://127.0.0.1:$MONGO_PORT/?directConnection=true"
	fi

	if [ "${DASHBOARD_DEMO_KEEP_DATA:-false}" != true ]; then
		echo "Generating $DATABASE (scenario S, 30 days, $STATS_MODE stats)"
		pnpm --silent --dir "$TOOLS_DIR/data-generator" run generate --uri "$MONGO_URI" --db "$DATABASE" \
			--scenario S --days 30 --stats-mode "$STATS_MODE" --drop --five-minute-rollups --node-rollups --room-sums \
			--active-window 60 --min-active-rooms 10 | tail -2
	fi

	pnpm --silent --dir "$TOOLS_DIR/data-generator" run keep-alive --uri "$MONGO_URI" --db "$DATABASE" &
	PIDS+=("$!")
}

# start_backend <package directory> <port> [VAR=value...]: the backend development server on the demo database, in
# the background.
start_backend() {
	local directory="$1"
	local port="$2"

	shift 2
	start_dev_backend "$directory" "$port" DASHBOARD_BASE_PATH=/ ADMIN_USERNAME=admin ADMIN_PASSWORD=admin \
		DATABASE_URL="$MONGO_URI" DATABASE_NAME="$DATABASE" "$@"
}