#!/bin/sh

# fail on error
set -e

# example sources
EXAMPLES="format listdir range range_pipe signal"
EXAMPLES="${EXAMPLES} stream1 stream2 coroutine1 coroutine2"

# assembly sources
ASM_SOURCE_DIR="src/asm"
ASM_SOURCES="jump_all_gas make_all_gas ontop_all_gas"

# c++ sources
CXX_SOURCE_DIR="src"
CXX_SOURCES="context_stack"

# output lib
OSTD_LIB="libostd.a"

# compiler
if [ -z "$CXX" ]; then
    CXX="c++"
fi

# preprocessor
if [ -z "$CPP" ]; then
    CPP="cpp"
fi

# assembler
if [ -z "$AS" ]; then
    AS="as"
fi

# ar
if [ -z "$AR" ]; then
    AR="ar"
fi

# c++ standard
OSTD_CXXFLAGS="-std=c++1z"

# includes
OSTD_CXXFLAGS="${OSTD_CXXFLAGS} -I."

# optimization flags
OSTD_CXXFLAGS="${OSTD_CXXFLAGS} -O2"

# warnings
OSTD_CXXFLAGS="${OSTD_CXXFLAGS} -Wall -Wextra -Wshadow -Wold-style-cast"

# custom cxxflags
if [ ! -z "$CXXFLAGS" ]; then
    OSTD_CXXFLAGS="${OSTD_CXXFLAGS} ${CXXFLAGS}"
fi

# preprocessor flags
OSTD_CPPFLAGS=""

# custom cppflags
if [ ! -z "$CPPFLAGS" ]; then
    OSTD_CPPFLAGS="${OSTD_CPPFLAGS} ${CPPFLAGS}"
fi

# linker flags
OSTD_LDFLAGS=""

# custom linker flags
if [ ! -z "$LDFLAGS" ]; then
    OSTD_LDFLAGS="${OSTD_LDFLAGS} ${LDFLAGS}"
fi

# assembler flags
OSTD_ASFLAGS=""

# custom assembler flags
if [ ! -z "$LDFLAGS" ]; then
    OSTD_LDFLAGS="${OSTD_LDFLAGS} ${LDFLAGS}"
fi

#
# BUILD LOGIC
#

# clean everything
clean() {
    echo "Cleaning..."
    for ex in ${EXAMPLES}; do
        rm -f "examples/${ex}" "examples/${ex}.o"
    done
    for as in ${ASM_SOURCES}; do
        rm -f "${ASM_SOURCE_DIR}/${as}.o"
    done
    for cs in ${CXX_SOURCES}; do
        rm -f "${CXX_SOURCE_DIR}/${cs}.o"
    done
    rm -f "$OSTD_LIB"
    rm -f test_runner.o test_runner
}

# call_cxx input output
call_cxx() {
    echo "CXX: $1"
    eval "${CXX} ${OSTD_CPPFLAGS} ${OSTD_CXXFLAGS} -c -o \"${2}\" \"${1}\""
}

# call_as input output
call_as() {
    echo "AS: $1"
    eval "${CPP} -x assembler-with-cpp \"${1}\" | ${AS} -o \"${2}\""
}

# call_ld output file1 file2 ...
call_ld() {
    echo "LD: $1"
    eval "${CXX} ${OSTD_CPPFLAGS} ${OSTD_CXXFLAGS} ${OSTD_LDFLAGS} -o $@"
}

# call_ldlib output file1 file2 ...
call_ldlib() {
    echo "AR: $1"
    eval "${AR} rcs $@"
}

# build_example name
build_example() {
    call_cxx "examples/${1}.cc" "examples/${1}.o"
    call_ld "examples/${1}" "examples/${1}.o" "$OSTD_LIB"
    rm -f "examples/${1}.o"
}

# build test runner
build_test_runner() {
    call_cxx test_runner.cc test_runner.o
    call_ld test_runner test_runner.o "$OSTD_LIB"
    rm -f test_runner.o
}

# add_sfx_pfx str sfx pfx
add_sfx_pfx() {
    RET=""
    for it in $1; do
        RET="$RET ${3}${it}${2}"
    done
    echo $RET
}

# check if cleaning
if [ "$1" = "clean" ]; then
    clean
    exit 0
fi

# build assembly
echo "Building the library..."
for as in $ASM_SOURCES; do
    call_as "${ASM_SOURCE_DIR}/${as}.S" "${ASM_SOURCE_DIR}/${as}.o" &
done
for cs in $CXX_SOURCES; do
    call_cxx "${CXX_SOURCE_DIR}/${cs}.cc" "${CXX_SOURCE_DIR}/${cs}.o" &
done
wait
call_ldlib "$OSTD_LIB" \
    $(add_sfx_pfx "$ASM_SOURCES" ".o" "$ASM_SOURCE_DIR/") \
    $(add_sfx_pfx "$CXX_SOURCES" ".o" "$CXX_SOURCE_DIR/")

rm -f $(add_sfx_pfx "$ASM_SOURCES" ".o" "$ASM_SOURCE_DIR/")
rm -f $(add_sfx_pfx "$CXX_SOURCES" ".o" "$CXX_SOURCE_DIR/")

# build test runner
echo "Building test runner..."
build_test_runner &

# build examples
echo "Building examples..."
for ex in $EXAMPLES; do
    build_example "$ex" &
done
wait

# done
exit 0