danac.sh

#!/bin/bash
set -euo pipefail
IFS=$'\n\t'

# Get the directory where this script resides
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"

# Path to the compiler and the dana binary
DANAC="$SCRIPT_DIR/danac"
DANART="$SCRIPT_DIR/libdanart.a"

# Default compiler options
OUTPUT_FILE="a.out"
OPT_LEVEL=0
MODE=0 # 0=normal compile/link, 1=stdin->imm, 2=stdin->asm

# Function to show usage
usage() {
    echo "Usage:"
    echo "  $0 <input-file.dana> [-o output-file] [-O0|-O1|-O2|-O3]"
    echo "  $0 -i [-O0|-O1|-O2|-O3]    # interactive stdin, print imm to stdout"
    echo "  $0 -f [-O0|-O1|-O2|-O3]    # interactive stdin, print asm to stdout"
    echo "  $0 -h | --help             # show this help message"
    exit 1
}

# Parse arguments
if [ "$#" -lt 1 ]; then
    usage
fi

INPUT_FILE=""
while [[ $# -gt 0 ]]; do
    case "$1" in
        -h|--help)
            usage
            ;;
        -o)
            shift
            if [ -z "$1" ]; then
                echo "Error: Missing output file name after -o."
                usage
            fi
            OUTPUT_FILE="$1"
            shift
            ;;
        -O[0-9]*)
          case "$1" in
            -O[0-3]) OPT_LEVEL="${1:2}" ;;
            *) echo "Error: optimization must be -O0, -O1, -O2, or -O3"; exit 1 ;;
          esac
          shift
          ;;
        -i)
            MODE=1
            shift
            ;;
        -f)
            MODE=2
            shift
            ;;
        *)
            if [ -z "$INPUT_FILE" ]; then
                INPUT_FILE="$1"
                shift
            else
                echo "Error: Multiple input files provided."
                usage
            fi
            ;;
    esac
done

# Handle stdin modes (-i and -f)
if [[ $MODE -eq 1 || $MODE -eq 2 ]]; then
    if [[ -n "$INPUT_FILE" ]]; then
        echo "Error: No input file allowed with -i or -f."
        exit 1
    fi
    TMPFILE=$(mktemp /tmp/dana-stdin-XXXX.dana)
    cat > "$TMPFILE"
    "$DANAC" "$OPT_LEVEL" "$MODE" < "$TMPFILE"
    STATUS=$?
    rm -f "$TMPFILE"
    exit $STATUS
fi

# Check if input file was provided
if [ -z "$INPUT_FILE" ]; then
    usage
fi

# Convert to absolute paths
INPUT_FILE=$(realpath "$INPUT_FILE")
OUTPUT_FILE=$(realpath "$OUTPUT_FILE" 2>/dev/null || echo "$OUTPUT_FILE")

# Check if the file exists
if [ ! -f "$INPUT_FILE" ]; then
    echo "Error: File '$INPUT_FILE' does not exist."
    exit 1
fi

# Check if the file has a .dana extension
if [[ "$INPUT_FILE" != *.dana ]]; then
    echo "Error: File must have a .dana extension."
    exit 1
fi

# Extract path and base name
INPUT_DIR=$(dirname "$INPUT_FILE")
INPUT_BASE=$(basename "$INPUT_FILE" .dana)

# Run the compiler
"$DANAC" "$OPT_LEVEL" < "$INPUT_FILE"

# Check if compilation produced output.o
if [ ! -f output.o ]; then
    echo "Error: Compilation failed, output.o not found."
    exit 1
fi

# Move and rename files to match input file as the assignment says
mv output_opt.asm "$INPUT_DIR/$INPUT_BASE.asm"
mv output_opt.imm "$INPUT_DIR/$INPUT_BASE.imm"
mv output.o "$INPUT_DIR/$INPUT_BASE.o"

# Link with clang++
clang++ "$INPUT_DIR/$INPUT_BASE.o" "$DANART" -o "$OUTPUT_FILE"

# Check if linking failed
if [ $? -ne 0 ]; then
    echo "Linking failed."
    exit 1
fi