#!/bin/sh
set -eu

repo=${UNIDESK_TRAN_REPO_ROOT:-/root/unidesk}
if [ ! -f "$repo/scripts/cli.ts" ]; then
  self_dir=$(CDPATH= cd -- "$(dirname -- "$0")" && pwd)
  repo=$(CDPATH= cd -- "$self_dir/.." && pwd)
fi

host=${UNIDESK_MAIN_SERVER_IP:-${UNIDESK_MAIN_SERVER_HOST:-${CODE_QUEUE_DEV_CONTAINER_MASTER_HOST:-}}}
runner_env=0
if [ -n "${CODE_QUEUE_SERVICE_ROLE:-}" ] || [ -n "${CODE_QUEUE_INSTANCE_ID:-}" ] || [ -n "${KUBERNETES_SERVICE_HOST:-}" ]; then
  runner_env=1
fi

if [ "$runner_env" = 1 ] && [ -n "$host" ] && [ "${UNIDESK_TRAN_LOCAL:-}" != "1" ]; then
  exec bun "$repo/scripts/cli.ts" --main-server-ip "$host" ssh "$@"
fi

exec bun "$repo/scripts/cli.ts" ssh "$@"