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

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

SPARK_OLLAMA_PORT="${SPARK_OLLAMA_PORT:-11434}"
LOCAL_OLLAMA_PORT="${LOCAL_OLLAMA_PORT:-11435}"
KNOWN_HOSTS="${KNOWN_HOSTS:-$HOME/.ssh/known_hosts}"

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