#!/bin/bash
getScriptDir() {
     SOURCE="${BASH_SOURCE[0]}"
     while [ -h "$SOURCE" ] ; do SOURCE="$(readlink "$SOURCE")"; done
     echo $( cd $( dirname $SOURCE) && pwd )
}

if [ "$SSDT_HOME" == "" ]
then
   export SSDT_SCRIPTS=$(getScriptDir)
   export SSDT_HOME=$(dirname $SSDT_SCRIPTS)
fi

if [ "$(which docker-compose)" ==  "" ]
then
   PATH="${PATH}:/usr/local/bin"
fi

export SSDT_DOCKER_VERSION=$(docker version --format '{{.Server.Version}}')

# deletes psql data from specified service
resetDbVolume() {
#   Works for named and unnamed volumes and custom $PGDATA.
#   Container must be stopped
   service=${1?Must supply db service to reset}
   container=$(composeGetContainer $service)
   if [ "$container" != "" ]
   then
      echo "deleting db data from $container"
      docker-compose run --no-deps --rm $service bash -c "rm -rf \${PGDATA?Missing PGDATA env}"
   fi
}

# finds docker container for specified compose service 
composeGetContainer() {
  service=${1?"Must supply service from current project"}
  container=$( docker-compose ps | cut -d " " -f 1 -s | grep _${service}_ )
  echo $container
}

executeSQL() {
   service=${1?Must supply db service to execute against}
   container=$(composeGetContainer $service)

   echo "$2" | docker exec -i $container  sh -c "gosu postgres psql \$DB_NAME"
}

composeGetProject() {  
  project=$( docker-compose ps | tail -n1 | cut -d "_" -f 1 -s )
  echo $project
}