#!/bin/bash

SUDO='sudo - root -c '
DEBUG='yes'

base_name=$(basename $0 .sh)
base_dir=$(cd `dirname $0` ; pwd)
local_modules_folder="$base_dir/puppet/modules"

log_file=/tmp/$base_name.$$.log

echo "log_file is $log_file"
stderr_file=/tmp/$base_name.$$.err

touch $log_file


print_to_log(){
  severity=$1 ; shift

  #Skip the debug messages if not in debug mode:
  [[ $severity = 'DEBUG' ]] && [[ $DEBUG != 'yes' ]] && return 0

  formated_date=$(date '+%Y-%m-%d_%H:%M:%S')
  echo "[$formated_date] [$base_name] [$severity] [$*]" | /usr/bin/tee -a $log_file
  return 0
}

sudo_run(){
  #Executed as root and log commands. exit the script if failed.
  sudo -n "$@" 2>$stderr_file
  if [[ $? -ne 0 ]]; then
    print_to_log "ERROR" "Failed to run $*"
    print_to_log "DEBUG" "stderr was: `cat $stderr_file`"
    exit 1
  else
    print_to_log "INFO" "Executed successfully: $*"
    return 0
  fi
}

sudo_test(){
  #Try to execute as root, and return with the execution exit code
  sudo -n "$@" 2>&1
  rc=$?

  if [[ $rc -ne 0 ]]; then
    print_to_log "WARNING" "Executed with with warnigns: $*"
  else
    print_to_log "INFO" "Executed successfully: $*"
  fi

  return $rc
}

download_puppet_cep_module(){
  git_base_url="https://raw.githubusercontent.com/ishkin/Proton/master/puppet/modules/cep"

  local_cep_folder="$local_modules_folder/cep"
  sudo_run mkdir -p "$local_cep_folder"
  sudo_run mkdir -p "$local_cep_folder/manifests"
  sudo_run mkdir -p "$local_cep_folder/files"
  sudo_run mkdir -p "$local_cep_folder/templates"

  for FILE in \
    "files/download_artifacts.ksh" \
    "manifests/init.pp" \
    "manifests/pkgs.pp" \
    "manifests/deploy.pp" \
    "manifests/download.pp" \
    "manifests/config.pp" \
    "manifests/validation.pp" \
    "manifests/install.pp" \
    ; do
        sudo_run /usr/bin/curl --silent --show-error --fail --tlsv1 --connect-timeout 30 --location  "$git_base_url/$FILE" --output "$local_cep_folder/$FILE"
        if [[ $FILE = *'.ksh' ]] || [[ $FILE = *'.sh' ]] ; then
          new_mode=755
        else
          new_mode=644
        fi
        sudo_run /bin/chmod $new_mode "$local_cep_folder/$FILE"
  done

  return 0
}

handle_hostname_changes(){
  #This fix will instruce cloud-init to control the /etc/hosts file
  #to ensure we can resolve the hostname using this file.
  if [[ -d /etc/cloud/cloud.cfg.d ]]; then
    sudo_run sh -c 'echo "manage_etc_hosts: True" > /etc/cloud/cloud.cfg.d/06_manage_etc_hosts.cfg'
  fi
}


####################################################
####                                            ####
####                    MAIN                    ####
####                                            ####
####################################################

# Update Ubuntu
sudo_run apt-get update

#Install puppet and git
sudo_run apt-get -y install puppet git

#Download our puppet module from github
download_puppet_cep_module

#install the apt and stdlib puppet modules from the puppetlabs github
sudo_run git clone "https://github.com/puppetlabs/puppetlabs-apt.git" "$local_modules_folder/apt/"
sudo_run git clone "https://github.com/puppetlabs/puppetlabs-stdlib.git" "$local_modules_folder/stdlib/"

handle_hostname_changes

#Install the cep module using puppet
sudo_run /usr/bin/puppet apply  --parser future --modulepath=$local_modules_folder:/etc/puppet/modules:/usr/share/puppet/modules -e 'include cep'

print_to_log 'INFO' "The log file is saved as: $log_file"

print_to_log 'INFO' 'Done!'

#In case we are preparing a Docker file - remove unnecessary files to reduce the image size:
if [[ $1 = "--docker"  ]]; then
  sudo_run /bin/rm -rf /var/lib/tomcat7/webapps/AuthoringTool /var/lib/tomcat7/webapps/ProtonOnWebServerAdmin/ /puppet/modules/cep/files/*.war
fi

#Remove the temporary stderr file:
/bin/rm -f $stderr_file

exit 0