bootstrap

#!/bin/bash

BOOTSTRAP_STAGE="local"
BOOTSTRAP_DIR=$(dirname $0)
BOOTSTRAP_DOCKER_COMPOSE_FILE="$BOOTSTRAP_DIR/$BOOTSTRAP_STAGE/docker-compose.yml"

welcome() {
  echo "Welcome to the Arrêt bootstrap script!"
  echo "Stage: $BOOTSTRAP_STAGE"
  echo ""
}

setup_containers() {
  echo "Building and starting containers..."
  docker compose -f $BOOTSTRAP_DOCKER_COMPOSE_FILE build
  docker compose -f $BOOTSTRAP_DOCKER_COMPOSE_FILE up -d
}

shutdown_containers() {
  docker compose -f $BOOTSTRAP_DOCKER_COMPOSE_FILE down
}

test() {
  echo "Running tests..."
  echo ""

  # Clear redis
  redis-cli flushall > /dev/null

  # Run tests
  cargo test --features aio
}

benchmark() {
  echo "Running benchmarks..."
  echo ""

  # Clear redis
  redis-cli flushall > /dev/null

  # Run benchmarks
  cargo bench --features aio
}

usage() {
  echo "Usage: $0 [OPTIONS]"
  echo ""
  echo "Options:"
  echo "  up      Start the containers"
  echo "  down    Stop the containers"
  echo "  test    Run the tests"
  echo "  bench   Run the benchmarks"
  echo "  help    Show this help message"
  echo ""
}

main() {
  welcome

  if [ $1 == "up" ]; then
    setup_containers
  elif [ $1 == "down" ]; then
    shutdown_containers
  elif [ $1 == "test" ]; then
    test
  elif [ $1 == "bench" ]; then
    benchmark
  elif [ $1 == "help" ]; then
    usage
  else
    usage
  fi
}

main $@