#!/usr/bin/env bash

set -e

if [ -z "$1" ]; then
    echo "No argument supplied"
    exit 1
fi

declare -a nac3args
while [ $# -ge 1 ]; do
  case "$1" in
    --help)
      echo "Usage: run_demo.sh [--help] [--out OUTFILE] [--lli] [--debug] -- [NAC3ARGS...]"
      exit
      ;;
    --out)
      shift
      outfile="$1"
      ;;
    --lli)
      use_lli=1
      ;;
    --debug)
      debug=1
      ;;
    --)
      shift
      break
      ;;
    *)
      break
      ;;
  esac
  shift
done

while [ $# -ge 1 ]; do
  nac3args+=("$1")
  shift
done

if [ -n "$debug" ] && [ -e ../../target/debug/nac3standalone ]; then
    nac3standalone=../../target/debug/nac3standalone
    externfns=../../target/debug/deps/liblinalg_externfns.so
elif [ -e ../../target/release/nac3standalone ]; then
    nac3standalone=../../target/release/nac3standalone
    externfns=../../target/release/deps/liblinalg_externfns.so
else
    # used by Nix builds
    nac3standalone=../../target/x86_64-unknown-linux-gnu/release/nac3standalone
    externfns=../../target/x86_64-unknown-linux-gnu/release/deps/liblinalg_externfns.so
fi

rm -f ./*.o ./*.bc demo
if [ -z "$use_lli" ]; then
  $nac3standalone "${nac3args[@]}"

  clang -c -std=gnu11 -Wall -Wextra -O3 -o demo.o demo.c
  clang -lm -o demo module.o demo.o $externfns

  if [ -z "$outfile" ]; then
    ./demo
  else
    ./demo > "$outfile"
  fi
else
  $nac3standalone --emit-llvm "${nac3args[@]}"

  clang -c -std=gnu11 -Wall -Wextra -O3 -emit-llvm -o demo.bc demo.c

  shopt -s nullglob
  llvm-link -o nac3out.bc module*.bc main.bc
  shopt -u nullglob

  if [ -z "$outfile" ]; then
    lli -load=$externfns --extra-module demo.bc --extra-module irrt.bc nac3out.bc
  else
    lli -load=$externfns --extra-module demo.bc --extra-module irrt.bc nac3out.bc > "$outfile"
  fi
fi