#!/usr/bin/env bash

# A general compile script for my C/C++ projects
# This is what I use instead of makefiles
# It's just a simple script that works well for small projects and
# can be modified to suit your needs. The nice thing is that you only have to
# know one language, bash, not make or cmake or anything else, to modify it.
#
# To customize this script for your projects you will most likely only need to
# change the variables at the top of the file.
#    COMPILER, COMPILE_FLAGS, FILE_EXTENSION, SOURCE_LOCATIONS, OUTPUT_NAME
#
# Currently, this does not support multiple file extensions (ex. .c, .cpp, etc.)
# since that would involve a different compiler for each filetype. I might add it
# later if I need that functionality.
# It compiles the release and debug builds to the same location, only keeping the latest one.
#
# Usage: (commands implemented so far)
#    ./project [build] [release] (Builds the project, release version if specified)
#    ./project clean [release] (Removes all .o files in obj/debug, obj/release if specified)
#    ./project delete <filename> (Deletes **/filename.cpp in all directories that contain code and their corresponding object files)
#
# Examples:
#     ./project OR ./project build
#         compiles all object files to obj/debug/ then the binary to bin/
#     ./project clean
#         deletes all object files in obj/debug/
#     ./project release OR ./project build release
#         same as ./project but with obj/release/ for the object files
#     ./project delete ABC
#         deletes **/ABC.cpp, obj/debug/ABC.o, obj/release/ABC.o
#
# Recommended Use
#     ./project build && bin/outfile


COMPILER=clang++
COMPILE_FLAGS="--std=c++1z -Wall -fno-exceptions" # A -DDEBUG or -DRELEASE will be appened to the end of this before building

FILE_EXTENSION="cpp"
SOURCE_LOCATIONS="src" # any directories that contain source code (ex. "src src/extra_features lib")
OUTPUT_NAME=outfile

# ================================

GREEN='\033[0;32m'
YELLOW='\033[0;33m'
NO_COLOR='\033[0m'

echoColored() {
    echo -e "${GREEN}$@${NO_COLOR}"
}

setCompileFlags() {
    if [[ $1 == release ]]; then
        OPTIM="-O2"
        mode="release"
        MODE_FLAG="RELEASE"
    else
        OPTIM=""
        mode="debug"
        MODE_FLAG="DEBUG"
    fi
}

if [[ $1 == delete ]]; then
    filename=$2

    if [[ $filename != "" ]]; then
        DELETED_FILE=0
        if [[ -f obj/debug/${filename}.o ]]; then
            echo -e "${GREEN}Removing${NO_COLOR} obj/debug/${filename}.o"
            rm obj/debug/${filename}.o
            DELETED_FILE=1
        fi

        if [[ -f obj/release/${filename}.o ]]; then
            echo -e "${GREEN}Removing${NO_COLOR} obj/release/${filename}.o"
            rm obj/release/${filename}.o
            DELETED_FILE=1
        fi

        for dir in ${SOURCE_LOCATIONS}; do
            if [[ -f $dir/${filename}.${FILE_EXTENSION} ]]; then
                echo -e "${GREEN}Removing${NO_COLOR} $dir/${filename}.${FILE_EXTENSION}"
                rm $dir/${filename}.${FILE_EXTENSION}
                DELETED_FILE=1
            fi
        done

        if [[ $DELETED_FILE == 0 ]]; then
            echoColored "No files named ${filename}"
        fi
    else
        echoColored "No file specified for delete"
    fi
elif [[ $1 == clean ]]; then
    setCompileFlags $2

    if [[ $(ls obj/${mode}) != "" ]]; then
        echoColored "Cleaning obj/${mode}"
        rm obj/${mode}/*.o
    else
        echoColored "obj/${mode} is empty"
    fi
else
    # compile
    if [[ $1 == build ]]; then
        m=$2
    else
        m=$1
    fi

    setCompileFlags $m

    COMPILE_FLAGS="${COMPILE_FLAGS} -D${MODE_FLAG}"

    if ! [[ -d bin ]]; then
        mkdir bin
    fi

    if ! [[ -d obj/${mode} ]]; then
        mkdir -p obj/${mode}
    fi

    CHANGE=0 # were any files recompiled?
    for dir in ${SOURCE_LOCATIONS}; do
        for file in ${dir}/*.${FILE_EXTENSION}; do
            filename=$(basename -- $file)
            filename="${filename%.*}"
            # -nt = newer than
            if [[ $file -nt obj/${mode}/$filename.o ]]; then
                echo -e "${GREEN}Compiling${NO_COLOR} ${file}"
                ${COMPILER} ${COMPILE_FLAGS} ${OPTIM} -c $file -o obj/${mode}/$filename.o || exit 1
                CHANGE=1
            fi
        done
    done

    if [[ $CHANGE != 0 ]]; then
        echo -e "${YELLOW}---------------------------------${NO_COLOR}"
        echo -e "${GREEN}Compiling${NO_COLOR} ${OUTPUT_NAME}"

        ${COMPILER} ${OPTIM} obj/${mode}/*.o -o bin/${OUTPUT_NAME}
        echoColored "Done"
    else
        echoColored "Nothing to do"
    fi
fi