#!/bin/bash
# hg.sh - Made for Puppi

# All variables are exported
set -a 

# Sources common header for Puppi scripts
. $(dirname $0)/header || exit 10

# Show help
showhelp () {
    echo "This script performs the hg operations required by puppi::project::hg"
    echo "It integrates and uses variables provided by other core Puppi scripts"
    echo "It has the following options:"
    echo "-a <action> (Optional) What action to perform. Available options: deploy (default), rollback"
    echo "-s <source> (Required) Git source repo to use"
    echo "-d <destination> (Required) Directory where files are deployed"
    echo "-u <user> (Optional) User that performs the deploy operations. Default root"
    echo "-t <tag> (Optional) Tag to deploy"
    echo "-b <branch> (Optional) Branch to deploy"
    echo "-c <commit> (Optional) Commit to deploy"
    echo "-v <true|false> (Optional) If verbose"
    echo "-k <true|false> (Optional) If .hg dir is kept on deploy_root"
    echo 
    echo "Examples:"
    echo "hg.sh -a deploy -s $source -d $deploy_root -u $user -t $tag -b $branch -c $commit -v $bool_verbose -k $bool_keep_hgdata"
}

verbose="true"

# Check Arguments
while [ $# -gt 0 ]; do
  case "$1" in
    -a)
      case $2 in
          rollback)
          action="rollback"
          ;;
          *)
          action="install"
          ;;
      esac 
      shift 2 ;;
    -s)
      if [ $source ] ; then
        source=$source
      else
        source=$2
      fi
      shift 2 ;;
    -d)
      if [ $deploy_root ] ; then
        deploy_root=$deploy_root
      else
        deploy_root=$2
      fi
      shift 2 ;;
    -u)
      if [ $user ] ; then
        deploy_user=$user
      else
        deploy_user=$2
      fi
      shift 2 ;;
    -t)
      if [ $hg_tag ] ; then
        hg_tag=$hg_tag
      else
        hg_tag=$2
      fi
      shift 2 ;;
    -b)
      if [ $branch ] ; then
        branch=$branch
      else
        branch=$2
      fi
      shift 2 ;;
    -c)
      if [ $commit ] ; then
        commit=$commit
      else
        commit=$2
      fi
      shift 2 ;;
    -v)
      if [ $verbose ] ; then
        verbose=$verbose
      else
        verbose=$2
      fi
      shift 2 ;;
    -k)
      if [ $keep_hgdata ] ; then
        keep_hgdata=$keep_hgdata
      else
        keep_hgdata=$2
      fi
      shift 2 ;;
    *)
      showhelp
      exit ;;
  esac
done

if [ "x$verbose" == "xtrue" ] ; then
  verbosity="-v"
else
  verbosity=""
fi

cd /

hgdir=$deploy_root
if [ "x$keep_hgdata" != "xtrue" ] ; then
  if [ ! -d $archivedir/$project-hg ] ; then
    mkdir $archivedir/$project-hg
    chown -R $deploy_user:$deploy_user $archivedir/$project-hg
  fi
  hgdir=$archivedir/$project-hg/hgrepo
fi

do_install () {
  if [ -d $hgdir/.hg ] ; then
    cd $hgdir
    hg pull $verbosity origin $branch
    hg update $verbosity $branch
    if [ "x$?" != "x0" ] ; then
      hg update $verbosity $branch
    fi
  else
    hg clone $verbosity --branch $branch $source $hgdir
    cd $hgdir
  fi

  if [ "x$hg_tag" != "xundefined" ] ; then
    hg update $verbosity $hg_tag
  fi

  if [ "x$commit" != "xundefined" ] ; then
    hg update $verbosity $commit
  fi

  if [ "x$hgdir" == "x$archivedir/$project-hg" ] ; then
    rsync -a --exclude=".hg" $hgdir/$hgsubdir $deploy_root/
  fi

}

do_rollback () {

  echo "Rollback not yet supported"
}

# Action!
case "$action" in
    install) export -f do_install ; su $deploy_user -c do_install ;;
    rollback) do_rollback ;;
esac