#!/bin/zsh

# to make the script fail safely

set -euo pipefail

out=${1:-output}

echo -e "" > $out

echo `date` | tee -a $out

echo "" >> $out

echo "Below is the feedback and provisional marks for your CW 5 submission." >> $out

echo "Please note all marks are provisional until ratified by the" >> $out

echo "assessment board -- this is not an official results transcript." >> $out

echo "" >> $out

echo "The feedback for your submission for cw05.sc" >> $out

echo "" >> $out

# marks for CW 5

marks=$(( 0.0 ))

# compilation tests

function scala_compile {

  (ulimit -t 30; JAVA_OPTS="-Xmx1g" amm -s "$1" 2> c$out 1> c$out)

}

# functional tests

setopt no_nomatch

function scala_assert {

  rm -rf /Users/cu/.ammonite/cache/3.0.0-M2/* 1> /dev/null 2> /dev/null ;  

  (ulimit -t 300; JAVA_OPTS="-Xmx1g -XX:MaxJavaStackTraceDepth=10" amm -s "$1" 1>> $out 2>> $out)

}

function scala_fail {

  rm -rf /Users/cu/.ammonite/cache/3.0.0-M2/* 1> /dev/null 2> /dev/null ;  

  (ulimit -t 300; JAVA_OPTS="-Xmx1g -XX:MaxJavaStackTraceDepth=10" amm -s "$1" | tee -a $out)

}

### compilation test

echo -e "0) cw05.sc compiles?" | tee -a $out

if (scala_compile "c.sc")

then

    echo -e "  --> yes cw05.sc compiles\n" | tee -a $out

    tsts=$(( 0 ))

else

    echo -e "  --> AMM DID NOT compile cw05.sc\n" | tee -a $out

    cat c$out | tee -a $out

    tsts=$(( 1 ))

fi

### tokenise test

if [ $tsts -eq 0 ]

then

  echo -e "1) tokenise test\n" | tee -a $out

  cat cw05.sc c1_add.sc > cw05_add.sc

  if (scala_assert "c1.sc")     

  then

      echo -e "  --> success (+ 2.0 Marks)\n" | tee -a $out

      #cat $out | tee -a $out

      marks=$(( marks + 2.0 ))

  else

      echo -e "  --> FAILED\n" | tee -a $out

      cat $out | tee -a $out

  fi

fi

### parsing test

if [ $tsts -eq 0 ]

then

  echo -e "2) parsing test\n" | tee -a $out

  cat cw05.sc c2_add.sc > cw05_add.sc

  START=$(date +%s)

  if (scala_assert "c2.sc")     

  then

      END=$(date +%s)

      DIFF=$(( $END - $START ))

      echo -e "  This test ran for $DIFF seconds" | tee -a $out

      echo -e "  --> success (+ 3.0 Marks)\n" | tee -a $out

      #cat $out | tee -a $out

      marks=$(( marks + 3.0 ))

  else

      END=$(date +%s)

      DIFF=$(( $END - $START ))

      echo -e "  This test ran for $DIFF seconds" | tee -a $out

      echo -e "  --> FAILED\n" | tee -a $out

      cat $out | tee -a $out

  fi

fi

### mand test

echo >> $out

#echo -e "\n\n\n" >> $out

tstsB=$(( 1 ))

if [ $tsts -eq 0 ]

then

    echo -e "3.1) Generating ll-file for mand.fun:\n" | tee -a $out

    echo >> $out

    cat cw05.sc c3_add.sc > cw051_add.sc

    if (scala_assert "c3.sc")

    then

        echo -e "  --> success" | tee -a $out

        tstsB=$(( 0 ))

    else

        echo -e "  --> FAIL\n" | tee -a $out

        cat c$out >> $out

        tail -25 $out | pr -to10

        tstsB=$(( 1 ))

    fi

fi

if [ $tstsB -eq 0 ]

then

    echo -e "3.2) Running mand.ll and check its output" | tee -a $out

    if (scala_fail "c4.sc")

    then

        echo -e "  --> success (+ 12.0 Marks)\n" | tee -a $out

        marks=$(( marks + 12.0 ))

    else

        echo -e "  --> FAIL\n" | tee -a $out

	echo -e "  generated:" | tee -a $out

        cat "mand.ll" | pr -to10

        #cat c$out >> $out

    fi

fi

### mand test

echo >> $out

#echo -e "\n\n\n" >> $out

if [ $tsts -eq 0 ]

then

    echo -e "4.1) Generating ll-file for sqr.fun:\n" | tee -a $out

    echo >> $out

    cat cw05.sc c3_add.sc > cw051_add.sc

    if (scala_assert "c5.sc")

    then

        echo -e "  --> success" | tee -a $out

        tstsB=$(( 0 ))

    else

        echo -e "  --> FAIL\n" | tee -a $out

        cat c$out >> $out

        tail -25 $out | pr -to10

        tstsB=$(( 1 ))

    fi

fi

if [ $tstsB -eq 0 ]

then

    echo -e "4.2) Running sqr.ll and check its output" | tee -a $out

    if (scala_fail "c6.sc")

    then

        echo -e "  --> success (+ 6.0 Marks)\n" | tee -a $out

        marks=$(( marks + 6.0 ))

    else

        echo -e "  --> FAIL\n" | tee -a $out

	echo -e "  generated:" | tee -a $out

        cat "sqr.ll" | pr -to10

        #cat c$out >> $out

    fi

fi

if [ $tsts -eq 0 ]

then

    echo -e "5) Does main return an integer in mand.ll (define i32 @main() {...})? " | tee -a $out

    if (grep "define i32 @main()" "mand.ll" 1> /dev/null 2> /dev/null)

    then

        echo -e "  --> success (+ 2.0 Marks)\n" | tee -a $out

        marks=$(( marks + 2.0 ))

    else

        echo -e "  --> FAIL\n" | tee -a $out

        #tail -25 $out | pr -to10

        #cat c$out >> $out

    fi

fi

## final marks

echo >> $out

echo "Overall mark for CW 5" | tee -a $out

printf " %0.2f\n" $marks | tee -a $out