#!/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

NO_PROXY_VAL="${NO_PROXY:-localhost,127.0.0.1,api.novita.ai,novita.ai}"

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."