#!/usr/bin/env bash
set -e

# Default to run_tests if no arguments provided
if [ $# -eq 0 ]; then
    echo "==> FlyGuard container started without arguments. Running verification tests..."
    exec python3 tests/run_tests.py
fi

# Subcommand shortcuts
case "$1" in
    test|tests|run-tests)
        shift
        exec python3 tests/run_tests.py "$@"
        ;;
    pytest)
        shift
        exec pytest tests "$@"
        ;;
    pipeline|run-pipeline)
        shift
        exec python3 tools/run_pipeline.py "$@"
        ;;
    benchmark|make-benchmark)
        shift
        exec python3 tools/make_benchmark.py "$@"
        ;;
    evaluate)
        shift
        exec python3 tools/evaluate.py "$@"
        ;;
    train-mbon)
        shift
        exec python3 tools/train_mbon.py "$@"
        ;;
    train-track)
        shift
        exec python3 tools/train_track.py "$@"
        ;;
    plot|plot-benchmark)
        shift
        exec python3 tools/plot_benchmark.py "$@"
        ;;
    bash|sh)
        exec /bin/bash "$@"
        ;;
    *)
        exec "$@"
        ;;
esac