#!/bin/bash
set -e

VAST_HOST="${VAST_VOICE_HOST:-ssh4.vast.ai}"
VAST_PORT="${VAST_VOICE_PORT:-17850}"
LOCAL_PORT=3001
SSH_KEY="$HOME/.ssh/id_ed25519"

pgrep -f "ssh.*-L ${LOCAL_PORT}:localhost:${LOCAL_PORT}.*${VAST_HOST}" > /dev/null 2>&1 && {
  echo "[tunnel] Already running"
  exit 0
}

while true; do
  echo "[tunnel] Connecting to ${VAST_HOST}:${VAST_PORT} → localhost:${LOCAL_PORT}"
  ssh -N \
    -o StrictHostKeyChecking=no \
    -o ServerAliveInterval=30 \
    -o ServerAliveCountMax=3 \
    -o ExitOnForwardFailure=yes \
    -o ConnectTimeout=10 \
    -i "$SSH_KEY" \
    -p "$VAST_PORT" \
    -L "${LOCAL_PORT}:localhost:${LOCAL_PORT}" \
    "root@${VAST_HOST}" 2>&1 || true
  echo "[tunnel] Disconnected, retrying in 10s..."
  sleep 10
done