cannam@54: #!/bin/bash
cannam@54: 
cannam@54: # Disable shellcheck warnings for useless-use-of-cat. UUOC is good
cannam@54: # practice, not bad: clearer, safer, less error-prone.
cannam@54: # shellcheck disable=SC2002
cannam@54: 
cannam@54: sml="$VEXT_SML"
cannam@54: 
cannam@54: set -eu
cannam@54: 
cannam@54: mydir=$(dirname "$0")
cannam@54: program="$mydir/vext.sml"
cannam@54: 
cannam@54: hasher=
cannam@54: local_install=
cannam@54: if [ -w "$mydir" ]; then
cannam@54:     if echo | sha256sum >/dev/null 2>&1 ; then
cannam@54: 	hasher=sha256sum
cannam@54:         local_install=true
cannam@54:     elif echo | shasum >/dev/null 2>&1 ; then
cannam@54: 	hasher=shasum
cannam@54: 	local_install=true
cannam@54:     else
cannam@54:         echo "WARNING: sha256sum or shasum program not found" 1>&2
cannam@54:     fi
cannam@54: fi
cannam@54: 
cannam@54: if [ -n "$local_install" ]; then
cannam@54:     hash=$(echo "$sml" | cat "$program" - | $hasher | cut -c1-16)
cannam@54:     gen_sml=$mydir/.vext-$hash.sml
cannam@54:     gen_out=$mydir/.vext-$hash.bin
cannam@54:     trap 'rm -f $gen_sml' 0
cannam@54: else
cannam@54:     gen_sml=$(mktemp /tmp/vext-XXXXXXXX.sml)
cannam@54:     gen_out=$(mktemp /tmp/vext-XXXXXXXX.bin)
cannam@54:     trap 'rm -f $gen_sml $gen_out' 0
cannam@54: fi
cannam@54: 
cannam@54: if [ -x "$gen_out" ]; then
cannam@54:     exec "$gen_out" "$@"
cannam@54: fi
cannam@54: 
cannam@54: # We need one of Poly/ML, SML/NJ, or MLton. Since we're running a
cannam@54: # single-file SML program as if it were a script, our order of
cannam@54: # preference is based on startup speed, except in the local_install
cannam@54: # case where we retain a persistent binary.
cannam@54: 
cannam@54: if [ -z "$sml" ]; then
cannam@54:     if [ -n "$local_install" ] && mlton 2>&1 | grep -q 'MLton'; then
cannam@54: 	sml="mlton"
cannam@54:     elif sml -h 2>&1 | grep -q 'Standard ML of New Jersey'; then
cannam@54: 	sml="smlnj"
cannam@54:     # We would prefer Poly/ML to SML/NJ, except that Poly v5.7 has a
cannam@54:     # nasty bug that occasionally causes it to deadlock on startup.
cannam@54:     # That appears to be fixed in their repo, so we could promote it
cannam@54:     # up the order again at some point in future
cannam@54:     elif echo | poly -v 2>/dev/null | grep -q 'Poly/ML'; then
cannam@54: 	sml="poly"
cannam@54:     elif mlton 2>&1 | grep -q 'MLton'; then
cannam@54: 	sml="mlton"
cannam@54:     else cat 1>&2 <<EOF
cannam@54: 
cannam@54: ERROR: No supported SML compiler or interpreter found       
cannam@54: EOF
cannam@54: 	cat <<EOF
cannam@54: 
cannam@54:   The Vext external source code manager needs a Standard ML (SML)
cannam@54:   compiler or interpreter to run.
cannam@54: 
cannam@54:   Please ensure you have one of the following SML implementations
cannam@54:   installed and present in your PATH, and try again.
cannam@54: 
cannam@54:     1. Standard ML of New Jersey
cannam@54:        - often found in a distribution package called: smlnj
cannam@54:        - executable name: sml
cannam@54: 
cannam@54:     2. Poly/ML
cannam@54:        - often found in a distribution package called: polyml
cannam@54:        - executable name: poly
cannam@54: 
cannam@54:     3. MLton
cannam@54:        - often found in a distribution package called: mlton
cannam@54:        - executable name: mlton
cannam@54: 
cannam@54: EOF
cannam@54: 	exit 2
cannam@54:     fi
cannam@54: fi
cannam@54: 
cannam@54: arglist=""
cannam@54: for arg in "$@"; do
cannam@54:     if [ -n "$arglist" ]; then arglist="$arglist,"; fi
cannam@54:     if echo "$arg" | grep -q '[^a-z]' ; then
cannam@54: 	arglist="$arglist\"usage\""
cannam@54:     else
cannam@54: 	arglist="$arglist\"$arg\""
cannam@54:     fi
cannam@54: done
cannam@54: 
cannam@54: case "$sml" in
cannam@54:     poly)
cannam@54:         if [ -n "$local_install" ] && polyc --help >/dev/null 2>&1 ; then
cannam@54:             if [ ! -x "$gen_out" ]; then
cannam@54:                 polyc -o "$gen_out" "$program"
cannam@54:             fi
cannam@54: 	    "$gen_out" "$@"
cannam@54:         else
cannam@54:             echo 'use "'"$program"'"; vext ['"$arglist"'];' |
cannam@54:                 poly -q --error-exit
cannam@54:         fi ;;
cannam@54:     mlton)
cannam@54:         if [ ! -x "$gen_out" ]; then
cannam@54: 	    echo "val _ = main ()" | cat "$program" - > "$gen_sml"
cannam@54: 	    mlton -output "$gen_out" "$gen_sml"
cannam@54:         fi
cannam@54: 	"$gen_out" "$@" ;;
cannam@54:     smlnj)
cannam@54: 	cat "$program" | (
cannam@54: 	    cat <<EOF
cannam@54: val smlrun__cp = 
cannam@54:     let val x = !Control.Print.out in
cannam@54:         Control.Print.out := { say = fn _ => (), flush = fn () => () };
cannam@54:         x
cannam@54:     end;
cannam@54: val smlrun__prev = ref "";
cannam@54: Control.Print.out := { 
cannam@54:     say = fn s => 
cannam@54:         (if String.isSubstring " Error" s
cannam@54:          then (Control.Print.out := smlrun__cp;
cannam@54:                (#say smlrun__cp) (!smlrun__prev);
cannam@54:                (#say smlrun__cp) s)
cannam@54:          else (smlrun__prev := s; ())),
cannam@54:     flush = fn s => ()
cannam@54: };
cannam@54: EOF
cannam@54: 	    cat -
cannam@54: 	    cat <<EOF
cannam@54: val _ = vext [$arglist];
cannam@54: val _ = OS.Process.exit (OS.Process.success);
cannam@54: EOF
cannam@54:             ) > "$gen_sml"
cannam@54: 	CM_VERBOSE=false sml "$gen_sml" ;;
cannam@54:     *)
cannam@54: 	echo "Unknown SML implementation name: $sml";
cannam@54: 	exit 2 ;;
cannam@54: esac
cannam@54: