#!/bin/bash

PID=$$

profile=
version_stamp=
subarch=
stage1_seed=
snapshot=
email_from="catalyst@$(hostname -f)"
email_to="root@$(hostname -f)"
config_file="/etc/catalyst/catalyst.conf"
verbose=0

usage() {
  msg=$1

  if [ -n "${msg}" ]; then
    echo -e "${msg}\n";
  fi

  cat <<EOH
Usage:
  stage_build [-p|--profile <profile>] [-v|--version-stamp <stamp>]
              [-a|--arch <arch>] [-s|--stage1-seed <seed>] [--verbose]
              [-f|--email-from <from>] [-t|--email-to <to>] [-h|--help]
              [-c|--config <config file>]

Options:
  -p|--profile        Sets the portage profile (required)
  -v|--version-stamp  Sets the version stamp (required)
  -a|--arch           Sets the 'subarch' in the spec (required)
  -s|--stage1-seed    Sets the seed for the stage1 (required)
  -S|--snapshot       Sets the snapshot name (if not given defaults to today's
                      date)
  --verbose           Send output of commands to console as well as log
  -f|--email-from     Sets the 'From' on emails sent from this script (defaults
                      to catalyst@localhost)
  -t|--email-to       Sets the 'To' on emails sent from this script (defaults
                      to root@localhost)
  -c|--config         Sets the config file that catalyst should use (defaults
                      to /etc/catalyst/catalyst.conf)
  -h|--help           Show this message and quit

Example:
  stage_build -p default-linux/x86/2006.1 -v 2007.0_pre -a i686 -s default/stage3-i686-2006.1
EOH
}

send_email() {
  subject="[${subarch}] $1"
  body=$2

  echo -e "From: ${email_from}\r\nTo: ${email_to}\r\nSubject: ${subject}\r\n\r\nArch: ${subarch}\r\nProfile: ${profile}\r\nVersion stamp: ${version_stamp}\r\nStage1 seed: ${stage1_seed}\r\nSnapshot: ${snapshot}\r\n\r\n${body}\r\n" | sendmail -f ${email_from} ${email_to}
}

run_cmd() {
  cmd=$1
  logfile=$2

  if [ $verbose = 1 ]; then
    eval "${cmd}" 2>&1 | tee ${logfile}
  else
    eval "${cmd}" &> ${logfile}
  fi
}

# Parse args
params=${#}
while [ ${#} -gt 0 ]
do
  a=${1}
  shift
  case "${a}" in
    -h|--help)
      usage
      exit 0
      ;;
    -p|--profile)
      profile=$1
      shift
      ;;
    -v|--version-stamp)
      version_stamp=$1
      shift
      ;;
    -a|--arch)
      subarch=$1
      shift
      ;;
    -f|--email-from)
      email_from=$1
      shift
      ;;
    -t|--email-to)
      email_to=$1
      shift
      ;;
    -s|--stage1-seed)
      stage1_seed=$1
      shift
      ;;
    -S|--snapshot)
      snapshot=$1
      shift
      ;;
    -c|--config)
      config_file=$1
      shift
      ;;
    --verbose)
      verbose=1
      ;;
    -*)
      echo "You have specified an invalid option: ${a}"
      usage
      exit 1
      ;;
    esac
done

# Make sure all required values were specified
if [ -z "${profile}" ]; then
  usage "You must specify a profile."
  exit 1
fi
if [ -z "${version_stamp}" ]; then
  usage "You must specify a version stamp."
  exit 1
fi
if [ -z "${subarch}" ]; then
  usage "You must specify an arch."
  exit 1
fi
if [ -z "${stage1_seed}" ]; then
  usage "You must specify a stage1 seed."
  exit 1
fi

tmpdir="$(mktemp -d)"
cd ${tmpdir}

if [ -z "${snapshot}" ]; then
  snapshot=`date +%Y%m%d`
  run_cmd "catalyst -c '${config_file}' -s '${snapshot}'" "${tmpdir}/catalyst_build_snapshot.${PID}.log"
  if [ $? != 0 ]; then
    send_email "Catalyst build error - snapshot" "$(<${tmpdir}/catalyst_build_snapshot.${PID}.log)"
    exit 1
  fi
fi

for i in 1 2 3; do
  echo -e "subarch: ${subarch}\ntarget: stage${i}\nversion_stamp: ${version_stamp}\nrel_type: default\nprofile: ${profile}\nsnapshot: ${snapshot}" > stage${i}.spec
  if [ ${i} = 1 ]; then
    echo "source_subpath: ${stage1_seed}" >> stage${i}.spec
  else 
    echo "source_subpath: default/stage$(expr ${i} - 1)-${subarch}-${version_stamp}" >> stage${i}.spec
  fi
  run_cmd "catalyst -c '${config_file}' -a -p -f stage${i}.spec" "${tmpdir}/catalyst_build_stage${i}.${PID}.log"
  if [ $? != 0 ]; then
    send_email "Catalyst build error - stage${i}" "$(tail -n 200 ${tmpdir}/catalyst_build_stage${i}.${PID}.log)\r\n\r\nFull build log at ${tmpdir}/catalyst_build_stage${i}.${PID}.log"
    exit 1
  fi
done

send_email "Catalyst build success" "Everything finished successfully."