driver.sh

#!/bin/sh

# According to the book, if no option is specified, the
# code should be compiled. This script sets the mode to
# --compile if none of the ones in the book are used.
MODE="--compile"
if [ "$1" = "--lex" ]; then
  MODE="--lex"
  shift
elif [ "$1" = "--parse" ]; then
  MODE="--parse"
  shift
elif [ "$1" = "--codegen" ]; then
  MODE="--codegen"
  shift
elif [ "$1" = "--tacky" ]; then
  MODE="--tacky"
  shift
elif [ "$1" = "-S" ]; then
  MODE="-S"
  shift
elif [ "$1" = "--compile" ]; then
  MODE="--compile"
  shift
fi
SOURCE_FILE=$1

# Find out what operating system this is
UNAME=`uname`

# Extract the filename from the source file path
BASE=${SOURCE_FILE%.*}

# PREPROCESSED is what we call the file after it has been
# run through the C preprocessor
PREPROCESSED=$BASE".i"

# ASSEMBLED is what we call the assembly file if there is one
ASSEMBLED=$BASE".s"

# ROOT_DIR is the directory where driver.sh is executing from, this
# is handy for figuring out where your compiler executable is
ROOT_DIR=`dirname "$0"`

# Run the C preprocessor on the file
# If you want CPP to include file/line markers so you can track
# the original source file and line, remove the -P option
gcc -P -E $SOURCE_FILE -o $PREPROCESSED

# Exit if there is an error
ERR=$?
if [ $ERR -ne 0 ]; then
  rm $PREPROCESSED
  exit $ERR
fi

# Execute my C compiler, change this line to execute yours
# Here is an example of one where the Ocaml compiler puts
# the executable main.exe in _build/default:
# $ROOT_DIR/_build/default/bin/main.exe $MODE $PREPROCESSED

# Exit if there was an error
ERR=$?
rm $PREPROCESSED
if [ $ERR -ne 0 ]; then
  exit $ERR
fi

if [ "$MODE" = "--compile" ]; then
# If this is running on an Arm Mac, then it is probably using
# clang, so we need to modify the generated asm a little bit
  if [ "$UNAME" = "Darwin" ]; then
    mv $ASSEMBLED "__astemp"
    sed -e "/@progbits/d" < __astemp | sed -e "s/.globl main/.globl _main/" | sed -e "s/^main:/_main:/" > $ASSEMBLED
  fi
  gcc $ASSEMBLED -o $BASE
  ERR=$?
  rm $ASSEMBLED
  exit $ERR
fi