entrypoint.sh

#!/bin/bash
# Entrypoint script for Heliosail-RX Docker container

set -e

# Function to display usage
usage() {
    cat << EOF
Usage: docker run heliosail-rx [COMMAND] [OPTIONS]

Commands:
  demo              Run architecture demonstration (default)
  mission FILE      Run mission from YAML config file
  test              Run all tests
  test-io           Run I/O infrastructure tests
  test-wrappers     Run physics module wrapper tests
  shell             Start interactive Python shell
  bash              Start bash shell

Examples:
  docker run heliosail-rx demo
  docker run -v /home/user/missions:/app/missions heliosail-rx mission /app/missions/my_mission.yaml
  docker run heliosail-rx test

Environment Variables:
  OUTPUT_DIR        Output directory for results (default: /app/outputs)
  LOG_LEVEL         Logging level (default: INFO)

EOF
}

# Default to demo if no command provided
COMMAND="${1:-demo}"

case "$COMMAND" in
  demo)
    echo "Starting Heliosail-RX Architecture Demo..."
    exec python architecture_demo.py
    ;;
  
  mission)
    if [ -z "$2" ]; then
      echo "Error: Mission file not specified"
      echo "Usage: docker run heliosail-rx mission <config_file>"
      exit 1
    fi
    echo "Running mission from: $2"
    exec python -c "
from io.config_loader import load_config
from api import Mission
from models.wrappers import TwoBodyModule, SolarRadiationPressureModule
from io import export_all

config = load_config('$2')
mission = Mission(config)
mission.add_physics_module('orbital_mechanics', TwoBodyModule())
mission.add_physics_module('sail_srp', SolarRadiationPressureModule())
result = mission.run()
export_all(result, output_dir='${OUTPUT_DIR:-/app/outputs}')
print('Mission complete!')
"
    ;;
  
  test)
    echo "Running all tests..."
    exec python -m pytest test_*.py -v
    ;;
  
  test-io)
    echo "Running I/O infrastructure tests..."
    exec python test_io_simple.py
    ;;
  
  test-wrappers)
    echo "Running physics module wrapper tests..."
    exec python test_wrappers_simple.py
    ;;
  
  shell)
    echo "Starting Python interactive shell..."
    exec python
    ;;
  
  bash)
    echo "Starting bash shell..."
    exec /bin/bash
    ;;
  
  --help|-h)
    usage
    exit 0
    ;;
  
  *)
    echo "Unknown command: $COMMAND"
    usage
    exit 1
    ;;
esac