#!/usr/bin/env bash
# Thin front door for the image. `docker run <image> <lean-bench subcommand> ...`
# works, `submit` is an alias for `ingest`, and anything else runs verbatim.
set -euo pipefail

if [ -f /run/secrets/llm_api_key ] && [ -z "${DASHSCOPE_API_KEY:-}" ]; then
  DASHSCOPE_API_KEY="$(cat /run/secrets/llm_api_key)"
  export DASHSCOPE_API_KEY
fi

mkdir -p "$(dirname "${LEANBENCH_DATABASE:-/data/leanbench.sqlite}")" 2>/dev/null || true

case "${1:-doctor}" in
  bash | sh)
    shift
    exec bash "$@"
    ;;
  submit)
    shift
    exec lean-bench ingest "$@"
    ;;
  doctor | ingest | detect | search | export | sources | stats | drop)
    exec lean-bench "$@"
    ;;
  lean-bench)
    shift
    exec lean-bench "$@"
    ;;
  *)
    exec "$@"
    ;;
esac