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

usage() {
  cat <<'EOF'
Usage:
  start-relay-main.sh -t <token>
  start-relay-main.sh -u <token>
  start-relay-main.sh -tu <tcp_token> [udp_token]

Examples:
  start-relay-main.sh -t relay_token
  start-relay-main.sh -u relay_token
  start-relay-main.sh -tu relay_token
  start-relay-main.sh -tu tcp_token udp_token
EOF
}

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
COMMANDS_DIR="$SCRIPT_DIR/commands"

if [[ $# -lt 2 ]]; then
  usage
  exit 1
fi

MODE="$1"
shift

case "$MODE" in
  -t)
    TOKEN="$1"
    bash "$COMMANDS_DIR/start-relay-tcp.sh" "$TOKEN"
    ;;
  -u)
    TOKEN="$1"
    bash "$COMMANDS_DIR/start-relay-udp.sh" "$TOKEN"
    ;;
  -tu)
    TCP_TOKEN="$1"
    UDP_TOKEN="${2:-$TCP_TOKEN}"
    bash "$COMMANDS_DIR/start-relay-tcp.sh" "$TCP_TOKEN"
    bash "$COMMANDS_DIR/start-relay-udp.sh" "$UDP_TOKEN"
    ;;
  *)
    usage
    exit 1
    ;;
esac