#!/usr/bin/env bash

#

# Author: Makarius

#

# Poly/ML 5.0 startup script.

export -n INFILE OUTFILE COPYDB COMPRESS MLTEXT TERMINATE NOWRITE

## diagnostics

function fail_out()

{

  echo "Unable to create output heap file: \"$OUTFILE\"" >&2

  exit 2

}

function check_file()

{

  if [ ! -f "$1" ]; then

    echo "Unable to locate $1" >&2

    echo "Please check your ML system settings!" >&2

    exit 2

  fi

}

## compiler executables and libraries

POLY="$ML_HOME/poly"

check_file "$POLY"

if [ "$(basename "$ML_HOME")" = bin ]; then

  POLYLIB="$(cd "$ML_HOME"; cd "$(pwd -P)"; cd ../lib; pwd)"

else

  POLYLIB="$ML_HOME"

fi

export LD_LIBRARY_PATH="$POLYLIB:$LD_LIBRARY_PATH"

export DYLD_LIBRARY_PATH="$POLYLIB:$DYLD_LIBRARY_PATH"

## prepare databases

if [ -z "$INFILE" ]; then

  PRG="$POLY"

  EXIT="fun exit 0 = (OS.Process.exit OS.Process.success): unit | exit _ = OS.Process.exit OS.Process.failure;"

else

  check_file "$INFILE"

  PRG="$INFILE"

  EXIT=""

fi

if [ -z "$OUTFILE" ]; then

  COMMIT='fun commit () = (TextIO.output (TextIO.stdErr, "Error - Database is not opened for writing.\n"); false);'

else

  if [ -z "$COMPRESS" ]; then

    COMMIT="fun commit () = (TextIO.output (TextIO.stdOut, \"Exporting $OUTFILE\n\"); PolyML.export (\"$OUTFILE\", PolyML.rootFunction); true);"

  else

    COMMIT="fun commit () = (PolyML.shareCommonData PolyML.rootFunction; TextIO.output (TextIO.stdOut, \"Exporting $OUTFILE\n\"); PolyML.export (\"$OUTFILE\", PolyML.rootFunction); true);"

  fi

  [ -f "$OUTFILE" ] && { chmod +w "$OUTFILE" || fail_out; }

  rm -f "${OUTFILE}.o" || fail_out

fi

## run it!

MLTEXT="$EXIT $COMMIT $MLTEXT"

MLEXIT="commit();"

if [ -z "$TERMINATE" ]; then

  FEEDER_OPTS=""

else

  FEEDER_OPTS="-q"

fi

"$ISABELLE_HOME/lib/scripts/feeder" -p -h "$MLTEXT" -t "$MLEXIT" $FEEDER_OPTS | \

  { read FPID; "$PRG" -q $ML_OPTIONS; RC="$?"; kill -HUP "$FPID"; exit "$RC"; }

RC="$?"

if [ -n "$OUTFILE" ]; then

  if [ -e "${OUTFILE}.o" ]; then

    cc -o "$OUTFILE" "${OUTFILE}.o" -L"$POLYLIB" -lpolymain -lpolyml $POLY_LINK_OPTIONS || fail_out

    rm -f "${OUTFILE}.o"

    [ -e "${OUTFILE}.exe" ] && mv "${OUTFILE}.exe" "$OUTFILE"

  fi

  [ -f "$OUTFILE" -a -n "$NOWRITE" ] && chmod -w "$OUTFILE"

fi

exit "$RC"