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

: "${SPARK_SSH_USER:?set SPARK_SSH_USER}"
: "${SPARK_SSH_HOST:?set SPARK_SSH_HOST}"

SPARK_LLAMA_PORT="${SPARK_LLAMA_PORT:-30000}"
LOCAL_LLAMA_PORT="${LOCAL_LLAMA_PORT:-18080}"
KNOWN_HOSTS="${KNOWN_HOSTS:-$HOME/.ssh/known_hosts}"

exec ssh -F none \
  -o UserKnownHostsFile="$KNOWN_HOSTS" \
  -o StrictHostKeyChecking=accept-new \
  -N \
  -L "${LOCAL_LLAMA_PORT}:127.0.0.1:${SPARK_LLAMA_PORT}" \
  "${SPARK_SSH_USER}@${SPARK_SSH_HOST}"