#!/usr/bin/env bash -l

set -e

DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"

VAGRANT_STARTED=false

green="\x1B[0;32m" # '\e[1;32m' is too bright for white bg.

blue="\x1B[1;34m"

endColor="\x1B[0m"

function echoblue() {

    echo -e "${blue}$1${endColor}"

}

function install() {

    pushd "$DIR"

    echoblue "---> checking vagrant"

    if vagrant status | grep -q -v "running"; then

        echoblue "---> starting vagrant"

        # if which virtualenvwrapper.sh > /dev/null 2>&1 ; then

        #     echoblue "---> activating ansible"

        #     source `which virtualenvwrapper.sh`

        #     workon ansible

        # fi

        vagrant up

        VAGRANT_STARTED=true

        # if type -t deactivate > /dev/null 2>&1 ; then

        #     deactivate

        # fi

    fi

    echoblue "---> done"

    popd > /dev/null

    return 0

}

function usage() {

    cat <<EOF

Usage: $0 [-I] [-h]

    -I : do not run install

    -h : print this message

EOF

}

environment=""

do_install=true

while getopts "Ih" opt; do

  case $opt in

    I) do_install=false ;;

    h) usage; exit 0 ;;

    \?) usage >&2; exit 1 ;;

  esac

done

shift $((OPTIND-1))

echo "do_install: $do_install"

[[ "$do_install" == true ]] && echoblue "DO INSTALL" && install;

pushd "$DIR"

echoblue "---> cleaning build folder"

rm -fr root-puppet

echoblue "---> creating build folder"

mkdir -p root-puppet/var/lib/puppet/provision

mkdir -p root-puppet/etc/puppet/hiera

echoblue "---> copying provision files"

rsync --exclude='.git' --exclude='.hg*' --exclude='custom.yaml' -aviuPz ../dev/provisioning/ root-puppet/var/lib/puppet/provision/

echoblue "---> copying provision files done"

echoblue "---> copying template file"

cp -a ../dev/provisioning/custom.yaml.tmpl root-puppet/etc/puppet/hiera

echoblue "---> copying template files done"

#  :datadir: "."

echoblue "---> add path in root-puppet/var/lib/puppet/provision/hiera.yaml"

sed -i "" -E 's/^([[:space:]]+\:datadir\:).*$/\1 \"\/etc\/puppet\/hiera\"/' root-puppet/var/lib/puppet/provision/hiera.yaml

echoblue "---> add path in root-puppet/var/lib/puppet/provision/hiera.yaml done"

echoblue "---> building package"

vagrant ssh -c "/vagrant/build_rpm_puppet.sh"

echoblue "---> building package done"

if [ "$VAGRANT_STARTED" = true ]; then

    echoblue "---> Stopping vagrant"

    vagrant halt

    echoblue "---> Stopping vagrant done"

fi

popd > /dev/null

echoblue "---> done"