#!/usr/bin/env bash
set -euo pipefail

STACK_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
STACK_NAME="${STACK_NAME:-ai-router}"
cd "$STACK_DIR"

set -a
# shellcheck disable=SC1091
source .env
set +a

# shellcheck disable=SC1091
source "${STACK_DIR}/scripts/no-proxy-default.sh"
NO_PROXY_VAL="${NO_PROXY:-${NO_PROXY_DEFAULT}}"

echo "== Scale vless-proxy and vpn-watchdog to 0 =="
docker service scale "${STACK_NAME}_vless-proxy=0" "${STACK_NAME}_vpn-watchdog=0" || true

echo "== Clear litellm proxy env =="
docker service update \
  --env-rm HTTP_PROXY \
  --env-rm HTTPS_PROXY \
  --env-add "NO_PROXY=${NO_PROXY_VAL}" \
  "${STACK_NAME}_litellm" || true

echo "VPN disabled."