#!/usr/bin/env bash
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
WSL_DIR="$(cd "$SCRIPT_DIR/.." && pwd)"
AUTOMATION_DIR="$(cd "$WSL_DIR/.." && pwd)"
DOCKER_ENV_FILE="$WSL_DIR/.env"
COMPOSE_FILE="$WSL_DIR/docker-compose.yml"
LOGS_DIR="$AUTOMATION_DIR/wsl_runtime/logs"

follow="${1:-all}"

compose_cmd() {
  docker compose --env-file "$DOCKER_ENV_FILE" -f "$COMPOSE_FILE" "$@"
}

case "$follow" in
  docker)
    compose_cmd logs -f
    ;;
  queue)
    tail -f "$LOGS_DIR/queue-runner.log"
    ;;
  all)
    compose_cmd logs -f &
    docker_pid=$!
    if [[ -f "$LOGS_DIR/queue-runner.log" ]]; then
      tail -f "$LOGS_DIR/queue-runner.log" &
      tail_pid=$!
      wait "$docker_pid" "$tail_pid"
    else
      wait "$docker_pid"
    fi
    ;;
  *)
    echo "Usage: $0 [all|docker|queue]"
    exit 1
    ;;
esac