#!/bin/bash
ENV_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )/.." &> /dev/null && pwd )
source $ENV_DIR/bin/messages

function dcom() {
  if [ $# -lt 1 ]; then
    errormsg "Usage: dcom [system] <docker-compose command>"
    return 1
  fi

  local SYSTEMS_CONFIG="$ENV_DIR/.systems"

  SYSTEM_IDENTIFIER="$1"
  SYSTEM_PATH=$(grep "^$SYSTEM_IDENTIFIER" "$SYSTEMS_CONFIG" | cut -d' ' -f2)

  if [ -z "$SYSTEM_PATH" ]; then # First Parameter is not a valid system identifier
    SYSTEMS_DIRECTORY=$(realpath "$ENV_DIR/systems")
    CURRENT_DIRECTORY=$(realpath $(pwd))

    # We currently are in a subdirectory of dev-env/systems
    if [[ "$CURRENT_DIRECTORY" == "$SYSTEMS_DIRECTORY"* ]]; then
      ASSUMED_SYSTEM_PATH=$(realpath --relative-to="$SYSTEMS_DIRECTORY" "$CURRENT_DIRECTORY")
      SYSTEM_PATH=$(grep "$ASSUMED_SYSTEM_PATH" "$SYSTEMS_CONFIG" | cut -d' ' -f2)

      if [ -z "$SYSTEM_PATH" ]; then # You are NOT in a configured system directory (though in a subdirectory of dev-env/systems).
        errormsg "$CURRENT_DIRECTORY is not a configured dev-env system path.\nPlease register the system in $ENV_DIR/.systems, navigate to a registered system, or provide a system identifer as the first parameter of dcom."
        return 1
      else # You are in a configured system directory
        SYSTEM_IDENTIFIER=$(grep "$ASSUMED_SYSTEM_PATH" "$SYSTEMS_CONFIG" | cut -d' ' -f1)
      fi
    else
      errormsg "$CURRENT_DIRECTORY is not a subdirectory of $ENV_DIR/systems.\nPlease navigate to a configured system, or provide a system identifer as the first parameter of dcom."
      return 1
    fi

  else # First Parameter is a valid system identifier
    shift # Remove first Parameter
  fi
  
  SYSTEM_PATH="$ENV_DIR/systems/$SYSTEM_PATH"
  SYSTEM_ENV_FILE="$SYSTEM_PATH/.env"
  DOCKER_COMPOSE_COMMAND="$@"
  DOCKER_COMPOSE_FILE="$SYSTEM_PATH/docker/docker-compose.yml"
  
  # Set .env variables if a file exist
  if [[ ! -f "$DOCKER_COMPOSE_FILE" ]]; then
    errormsg "$DOCKER_COMPOSE_FILE was not found. Have you run init yet?"
    return 1
  fi

  infomsg "[$SYSTEM_IDENTIFIER]: $DOCKER_COMPOSE_COMMAND"

  if [[ -f "$SYSTEM_ENV_FILE" ]] ; then
    export $(grep -v '^#' "$SYSTEM_ENV_FILE" | xargs)
  fi

  docker compose -f "$DOCKER_COMPOSE_FILE" $DOCKER_COMPOSE_COMMAND

  # Unset .env variables if a file exist
  if [[ -f "$SYSTEM_ENV_FILE" ]] ; then
    unset $(grep -v '^#' "$SYSTEM_ENV_FILE" | sed -E 's/(.*)=.*/\1/' | xargs)
  fi
}

function _dcom_completion() {
  local SYSTEMS_CONFIG="$ENV_DIR/.systems"
  local cur prev opts
  COMPREPLY=()
  cur="${COMP_WORDS[COMP_CWORD]}"
  prev="${COMP_WORDS[COMP_CWORD-1]}"
  opts="$(awk '{print $1}' $SYSTEMS_CONFIG | paste -sd' ' -)"

  COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
  return 0
}

complete -F _dcom_completion dcom