#!/usr/bin/env zsh

# I'll just leave this here
__setup_clang_ubuntu() {
    local site="http://apt.llvm.org"
    local name="$(lsb_release -c | cut -f2)"
    local version=4.0
    local priority=$(( int(version * 100 + 0.5) ))
    # TODO: use https? this is sketchy
    echo wget -O - "$site/llvm-snapshot.gpg.key" \| apt-key add -
    echo echo -n \""\
deb $site/$name/ llvm-toolchain-$name main\n\
# deb-src $site/$name/ llvm-toolchain-$name main\n\
# $version\n\
deb $site/$name/ llvm-toolchain-$name-$version main\n\
# deb-src $site/$name/ llvm-toolchain-$name-$version main\n\
"\" \> "/etc/apt/sources.list.d/llvm-toolchain-$name.list"
    echo apt-get update
    echo apt-get install clang-$version
    echo apt-get install lld-$version
    echo update-alternatives --install /usr/bin/clang clang /usr/bin/clang-$version $priority
    echo update-alternatives --install /usr/bin/clang++ clang++ /usr/bin/clang++-$version $priority
    echo update-alternatives --install /usr/bin/llvm-symbolizer llvm-symbolizer /usr/bin/llvm-symbolizer-$version $priority
}

compile() {
    if [ $# -lt 2 ]; then
        echo "usage: compile [clang|gcc] {debug|release} [flags...] {source file}" >&2
        return 1
    fi

    local sepples=0
    local CC=gcc
    local CXX=g++
    local our_flags=(-I.)

    maybe_include() {
        [ -d "$1" ] && our_flags+=("-I$1")
    }
    maybe_include "-I$HOME/opt/local/include"
    maybe_include "-I$HOME/src/ustl"

    local debug_flags=(-O1 -g -D_DEBUG);
    local release_flags=(-Ofast -march=native -g0 -fomit-frame-pointer -s -DNDEBUG)
    local dr_flags=(-Ofast -march=native -g -fomit-frame-pointer -DNDEBUG)
    local hardened_flags=(-O3 -march=native -g0 -s
                          -DNDEBUG -D_FORTIFY_SOURCE=2
                          -Wformat -Wformat-security -Werror=format-security)

    if [ -z $MSYSTEM ]; then
        hardened_flags+=(-fPIE -pie)
        hardened_flags+=(-Wl,-O1,--sort-common,--as-needed,-z,relro,-z,now)
    fi

    if [ -e /usr/bin/pprof ]; then
        local malloc=(-ltcmalloc -fno-builtin-malloc -fno-builtin-calloc -fno-builtin-realloc -fno-builtin-free)
        debug_flags+=($malloc[@])
        derelease_flags+=($malloc[@])
        derelease_flags+=(-lprofiler)
    fi

    local file=${@[-1]}
    [ "${file##*.}" = "c" ] || [ "${file##*.}" = "h" ] || sepples=1

    { [ $1 = clang ] && CC="clang" && CXX="clang++" && shift } || \
    { [ $1 = gcc ] && CC="gcc" && CXX="g++" && shift }

    if [ $CC = clang ]; then
        debug_flags+=(-ftrapv)
        if [ -z $MSYSTEM ]; then # only available on linux
            debug_flags+=(-fsanitize=undefined) # this SHOULD work with mingw,
                                                # but it fails to link.
            debug_flags+=(-fsanitize=address)
            debug_flags+=(-fvisibility=hidden -fuse-ld=gold -flto -fsanitize=cfi)

            hardened_flags+=(-fsanitize=safe-stack)
            hardened_flags+=(-fstack-protector-strong)
            hardened_flags+=(-fvisibility=hidden -fuse-ld=gold -flto -fsanitize=cfi)
        else
            our_flags+=(-fansi-escape-codes)
        fi
    fi

    { [ $1 = debug ] && our_flags+=($debug_flags) && shift } || \
    { [ $1 = release ] && our_flags+=($release_flags) && shift } || \
    { [ $1 = derelease ] && our_flags+=($dr_flags) && shift } || \
    { [ $1 = hardened ] && our_flags+=($hardened_flags) && shift } || \
    { echo "please specify either debug or (de)release or hardened" >&2; return 1 }

    # TODO add static option

    local compiler=
    if [ $sepples -eq 1 ]; then
        compiler=$CXX
        std="-std=gnu++1z"
    else
        compiler=$CC
        std="-std=gnu11"
    fi

    local flags=(${@[1,-2]})

    local out=/tmp/${${file%%.*}##*/}
    if [ -n $MSYSTEM ]; then
        # explicitly output to .exe to avoid weirdness
        out="$out.exe"
    fi

    # TODO: naive idea:
    # allow multiple source files (using the firstmost to determine the program name)
    # by generating a file that #includes each given file.

    echo $compiler $std ${our_flags[@]} ${flags[@]} $file -o $out >&2
    $compiler $std ${our_flags[@]} ${flags[@]} $file -o $out && echo "$out"
}

compile "$@"