#!/bin/bash

OPM_DATA_ROOT=$1

# Copy results from a test run to refence dir
# $1 = source directory to copy data from
# $2 = destination directory to copy data to
# $3 = base file name for files to copy
# $4...$@ = file types to copy
copyToReferenceDir () {
  SRC_DIR=$1
  DST_DIR=$2;
  STEM=$3;
  FILETYPES=${@:4};

  for filetype in $FILETYPES; do
    cp "$WORKSPACE/$SRC_DIR$STEM.$filetype" $DST_DIR
  done
}

tests=${@:2}
test -z "$tests" && tests="spe11 spe12 spe3 spe9 norne_init"
if grep -q -i "norne " <<< $ghprbCommentBody
then
  if test -d $WORKSPACE/deps/opm-data/norne/flow
  then
    tests="$tests norne_full"
  fi
fi

echo $tests

for test_name in ${tests}; do
  if grep -q "spe11" <<< $test_name
  then
    copyToReferenceDir \
      $configuration/build-opm-simulators/tests/results/flow_sequential+spe1/ \
      $OPM_DATA_ROOT/spe1/opm-simulation-reference/ \
      SPE1CASE1 \
      EGRID INIT SMSPEC UNRST UNSMRY
  fi

  if grep -q "spe12" <<< $test_name
  then
    copyToReferenceDir \
      $configuration/build-opm-simulators/tests/results/flow+spe1/ \
      $OPM_DATA_ROOT/spe1/opm-simulation-reference/ \
      SPE1CASE2 \
      EGRID INIT SMSPEC UNRST UNSMRY
  fi

  if grep -q "spe3" <<< $test_name
  then
    copyToReferenceDir \
      $configuration/build-opm-simulators/tests/results/flow+spe3/ \
      $OPM_DATA_ROOT/spe3/opm-simulation-reference/ \
      SPE3CASE1 \
      EGRID INIT PRT SMSPEC UNRST UNSMRY
  fi

  if grep -q "spe9" <<< $test_name
  then
    copyToReferenceDir \
      $configuration/build-opm-simulators/tests/results/flow+spe9/ \
      $OPM_DATA_ROOT/spe9/opm-simulation-reference/ \
      SPE9_CP_SHORT \
      EGRID INIT PRT SMSPEC UNRST UNSMRY
  fi

  if grep -q "norne_init" <<< $test_name
  then
    copyToReferenceDir \
      $configuration/build-opm-simulators/tests/results/init/flow+norne/ \
      $OPM_DATA_ROOT/norne/opm-simulation-reference/ \
      NORNE_ATW2013 \
      EGRID INIT
  fi

  if grep -q "norne_full" <<< $test_name
  then
    copyToReferenceDir \
      deps/opm-data/norne/flow/ \
      $OPM_DATA_ROOT/norne/opm-simulation-reference/ \
      NORNE_ATW2013 \
      UNSMRY
  fi
done

if [ -z "${@:2}" ]
then
  # User did not specify tests to update, probe
  pushd $OPM_DATA_ROOT > /dev/null
  tests=""
  git status | grep "SPE1CASE1" && tests="spe11"
  git status | grep "SPE1CASE2" && tests="$tests spe12"
  git status | grep "SPE3CASE1" && tests="$tests spe3"
  git status | grep "SPE9_CP" && tests="$tests spe9"
  git status | grep "NORNE_ATW2013.INIT" && tests="$tests norne_init"
  git status | grep "NORNE_ATW2013.UNSMRY" && tests="$tests norne_full"
  popd > /dev/null
fi

echo -e "update reference data for $tests\n" > /tmp/cmsg
if [ -z "$REASON" ]
then
  echo -e "Reason: fill in this\n" >> /tmp/cmsg
else
  echo -e "Reason: $REASON\n" >> /tmp/cmsg
fi
for dep in ert opm-common opm-core opm-grid opm-material opm-parser opm-output ewoms
do
  pushd $WORKSPACE/deps/$dep > /dev/null
  name=`printf "%-14s" $dep`
  rev=`git rev-parse HEAD`
  echo -e "$name = $rev" >> /tmp/cmsg
  popd > /dev/null
done
echo -e "opm-simulators = `git rev-parse HEAD`" >> /tmp/cmsg

cd $OPM_DATA_ROOT
if [ -n "$BRANCH_NAME" ]
then
  git checkout -b $BRANCH_NAME origin/master
fi

if [ -z "$REASON" ]
then
  git commit -a -t /tmp/cmsg
else
  git commit -a -F /tmp/cmsg
fi