#!/bin/bash

set -e

export COMPOSE_DOCKER_CLI_BUILD=1
export DOCKER_BUILDKIT=1

if [ -f .env ]; then
  export `grep -v '^#' .env | xargs`
else
  export DEV_UID=$(id -u) DEV_GID=$(id -g) LOCAL_DEV_CHECK=1
fi

COMPOSE_FILE=docker-compose.yml

if [ -f docker-compose.override.yml ]; then
  COMPOSE_FILE="$COMPOSE_FILE -f docker-compose.override.yml"
fi

if [ $# -eq 0 ]; then
  echo "Usage: bin/compose <command> [args*]"
  echo
  echo "Commands:"
  echo "    setup - Has to be run once initially. Installs backend and frontend dependencies. "
  echo "    reset - Resets everything by removing all containers and deleting all volumes. You need to run \`setup\` again afterwards. "
  echo "    start - Starts both backend and frontend in the background. Access via http://localhost:3000/ by default."
  echo "    run   - Starts the frontend in the background and backend in the foreground. Useful for debugging using pry."
  echo "    restart [service] - Restarts one or all running services."
  echo "    rspec - Runs rspec inside the \`backend-test\` container which will be started if it's not running yet."
  echo "    *     - Everything else will be passed straight to \`docker-compose\`."
  echo

  exit 1
fi

if [ -f config/database.yml ]; then
  echo
  printf "\033[0;31mError\033[0m: Found local \`config/database.yml\` - The docker setup will not work with this file present."
  echo " You could delete it or rename it for the time being."

  exit 1
fi

if command -v docker-compose &> /dev/null
then
  DOCKER_COMPOSE="docker-compose"
else
  DOCKER_COMPOSE="docker compose"
fi

if [[ "$@" = "start" ]]; then
  # backend will be started automatically as a dependency of the frontend
  $DOCKER_COMPOSE -f $COMPOSE_FILE up -d frontend
elif [[ "$@" = "run" ]]; then
  $DOCKER_COMPOSE -f $COMPOSE_FILE up -d frontend
  $DOCKER_COMPOSE -f $COMPOSE_FILE stop backend
  $DOCKER_COMPOSE -f $COMPOSE_FILE run --rm backend rm -f tmp/pids/server.pid # delete if necessary so new server can come up
  $DOCKER_COMPOSE -f $COMPOSE_FILE run --rm -p ${PORT:-3000}:3000 --name rails backend # run backend in TTY so you can debug using pry for instance
elif [[ "$1" = "setup" ]]; then
  $DOCKER_COMPOSE -f $COMPOSE_FILE run backend setup
  yes no | $DOCKER_COMPOSE -f $COMPOSE_FILE run frontend npm install
elif [[ "$1" = "restart" ]]; then
  $DOCKER_COMPOSE -f $COMPOSE_FILE restart "${@:2}"
elif [[ "$1" = "reset" ]]; then
  $DOCKER_COMPOSE -f $COMPOSE_FILE down && docker volume rm `docker volume ls -q | grep ${PWD##*/}_`
elif [[ "$1" = "rspec" ]]; then
  function get-container-name() {
    name=`$DOCKER_COMPOSE ps backend-test | tail -n1 | cut -d ' ' -f1`

    if [ "$name" = 'NAME' ]; then
      return 1;
    else
      echo "$name"
    fi
  }

  if ! get-container-name > /dev/null; then
    echo "Test backend not running yet. Starting it..."

    $DOCKER_COMPOSE -f $COMPOSE_FILE up -d backend-test

    container=`get-container-name`

    while ! docker logs --since 1m $container 2>&1 | grep "Ready for tests" &> /dev/null; do
      sleep 1
      printf "."
    done

    echo "Ready for tests"
  fi

  $DOCKER_COMPOSE -f $COMPOSE_FILE exec backend-test bundle exec rspec "${@:2}"
else
  $DOCKER_COMPOSE -f $COMPOSE_FILE "$@"
fi