#! /bin/sh
prefix=/usr
exec_prefix=${prefix}
BINDIR=${exec_prefix}/bin
LINK=g++
LIBDIR=${exec_prefix}/lib
LIBS="-lpthread -lffi -lgmp -lm -ldl -lstdc++ -lgcc_s -lgcc "
CFLAGS="-I/usr/lib/libffi-3.2.1/include -march=armv7-a -mfloat-abi=hard -mfpu=vfpv3-d16 -O2 -pipe -fstack-protector-strong -fno-plt"

DEFAULT_COMPILER="${BINDIR}/poly"
COMPILER=${DEFAULT_COMPILER}

# Extra options for Windows.  config.status sets these conditionals to either "" or "#".

EXTRALDFLAGS=""
##EXTRALDFLAGS="-Wl,-u,WinMain"
##EXTRALDFLAGS="-Wl,-u,WinMain"
##EXTRALDFLAGS="-Wl,-u,_WinMain@16 -Wl,--large-address-aware"
#EXTRALDFLAGS="-Wl,-u,_WinMain@16 -Wl,--large-address-aware"
#EXTRALDFLAGS+=" -mwindows"
##EXTRALDFLAGS+=" -mconsole"

#SUFFIX="obj"
SUFFIX="o"

# Msys passes the Windows TEMP in temp (lower case)
# On other systems allow TMPDIR to override /tmp.
#TEMPORARYDIR=${temp:-/tmp}
TEMPORARYDIR=${TMPDIR:-/tmp}

# Extra options for Mac OS X
#EXTRALDFLAGS="-Wl,-no_pie"

TMPOBJFILE=${TEMPORARYDIR}/polyobj.$$.$SUFFIX
trap 'rm -f "$TMPOBJFILE"' 0

compile()
{
    echo "val () = use (List.nth(CommandLine.arguments(), 2)); val () = PolyML.export(List.nth(CommandLine.arguments(), 3), main);" | ${COMPILER} -q --error-exit  "$1" "$2"
}

link()
{
    if [ X"$2" = "X" ]
    then
        ${LINK} ${EXTRALDFLAGS} ${CFLAGS} "$1" -L${LIBDIR} -Wl,-rpath,${LIBDIR} -lpolymain -lpolyml ${LIBS}
    else
        ${LINK} ${EXTRALDFLAGS} ${CFLAGS} "$1" -o "$2" -L${LIBDIR} -Wl,-rpath,${LIBDIR} -lpolymain -lpolyml ${LIBS}
    fi
}

printhelp()
{
    echo "Usage: polyc [OPTION]... [SOURCEFILE]"
    echo Compile and link a Standard ML source file with Poly/ML.
    echo
    echo "   -b poly      Use 'poly' as compiler instead of ${DEFAULT_COMPILER}"
    echo "   -c           Compile but do not link.  The object file is written to the source file with .$SUFFIX extension."
    echo "   -o output    Write the executable file to 'output'"
    echo "   --help       Write this text and exit"
    exit
}

usage()
{
    echo "$1"
    echo "Usage: polyc [OPTION]... [SOURCEFILE]"
    exit 1
}

checkml()
{
    extension="${1##*.}"
    case "$extension" in
        sml|ML)
             return 0 ;;
        o|obj)
             return 1;;
        *)
             test -r "$1" && file -b "$1" | grep -q text ;;
    esac
}

sourcefile=""
outputfile=""
compileonly="no"

while [ $# -gt 0 ]
do
    case "$1" in
        --help)
            printhelp ;;
        -b)
            shift
            [ $# -eq 0 ] && usage "Expected file name after -b"
            COMPILER="$1";;
        -c) compileonly="yes";;
        -o)
            shift
            [ $# -eq 0 ] && usage "Expected file name after -o"
            outputfile="$1";;
        *)
            [ X"$sourcefile" = "X" ] || usage "Only one source file name allowed"
            sourcefile="$1";;
    esac
    shift
done

[ X"$sourcefile" = "X" ] && usage "No input files"
[ -r "$sourcefile" ] || usage "Error: $sourcefile: No such file"

case "$compileonly" in
     yes)
	 if [ "x$outputfile" = "x" ]; then
	     basename=${sourcefile##*/}
             outputfile=${basename%.*}.o
	 fi
         compile "$sourcefile" "$outputfile"
         ;;
     no)
         if checkml "$sourcefile"
         then
             compile "$sourcefile" "$TMPOBJFILE" && link "$TMPOBJFILE" "$outputfile"
         else
             link "$sourcefile" "$outputfile"
         fi
         ;;
esac
