up() {
  # Ensure the main network exists
  docker network inspect invariant-explorer-web >/dev/null 2>&1 || \
    docker network create invariant-explorer-web

  # Start your local docker-compose services
  docker compose -f docker-compose.local.yml up -d

  echo "Proxy started at http://localhost/api/v1/proxy/"
  echo "See http://localhost/api/v1/proxy/docs for API documentation"
}

build() {
  # Build local services
  docker compose -f docker-compose.local.yml build
}

down() {
  # Bring down local services
  docker compose -f docker-compose.local.yml down
  docker compose -f tests/docker-compose.test.yml down
}


tests() {
  echo "Setting up test environment..."

  # Ensure test network exists
  docker network inspect invariant-proxy-web-test >/dev/null 2>&1 || \
    docker network create invariant-proxy-web-test

  # Setup the explorer.test.yml file
  CONFIG_DIR="/tmp/invariant-proxy-test/configs"
  FILE="$CONFIG_DIR/explorer.test.yml"
  mkdir -p "$CONFIG_DIR"
  # Download the file
  curl -L -o "$FILE" https://raw.githubusercontent.com/invariantlabs-ai/explorer/main/configs/explorer.test.yml
  # Verify if the file exists
  if [ ! -f "$FILE" ]; then
    echo "Error: File $FILE not found. Issue with download."
    exit 1
  fi
  echo "File successfully downloaded: $FILE"

  # Start containers
  docker compose -f tests/docker-compose.test.yml down
  docker compose -f tests/docker-compose.test.yml build
  docker compose -f tests/docker-compose.test.yml up -d

  until [ "$(docker inspect -f '{{.State.Health.Status}}' explorer-proxy-test-app-api)" = "healthy" ]; do
    echo "explorer-proxy-test-app-api container starting..."
    sleep 2
  done

  until [ "$(docker inspect -f '{{.State.Health.Status}}' explorer-proxy-test)" = "healthy" ]; do
    echo "explorer-proxy-test container starting..."
    sleep 2
  done

  echo "app-api and proxy are available. Running tests..."

  # Make call to signup endpoint
  curl -k -X POST http://127.0.0.1/api/v1/user/signup

  docker build -t 'explorer-proxy-test' -f ./tests/Dockerfile.test ./tests

  docker run \
    --mount type=bind,source=./tests,target=/tests \
    --network invariant-proxy-web-test \
    -e OPENAI_API_KEY="$OPENAI_API_KEY" \
    --env-file ./tests/.env.test \
    explorer-proxy-test $@
}

# -----------------------------
# Command dispatcher
# -----------------------------
case "$1" in
  "up")
    up
    ;;
  "build")
    build
    ;;
  "down")
    down
    ;;
  "logs")
    docker compose -f docker-compose.local.yml logs -f
    ;;
  "tests")
    shift
    tests $@
    ;;
  *)
    echo "Usage: $0 [up|build|down|logs|tests]"
    exit 1
    ;;
esac