#!/usr/bin/env sh
# MIT License, Copyright (c) 2020 Marvin Borner

set -e

cd "$(dirname "$0")"

mode="${1}"
network="rtl8139"

qemu_with_flags() {
    SDL_VIDEO_X11_DGAMOUSE=0 qemu-system-i386 -no-reboot -vga std -serial stdio -rtc base=localtime -m 256M -net nic,model=${network},macaddr=42:42:42:42:42:42 -net user "$@"
}

make_cross() {
    if [ ! -d "./cross/" ]; then
        # Create directory
        mkdir -p cross
        cd cross
        DIR=$(pwd)

        # Get sources
        mkdir "${DIR}/src" && cd "${DIR}/src"
        echo "Downloading..."
        curl -sSL "https://ftp.gnu.org/gnu/binutils/binutils-2.34.tar.xz" | tar xJ
        curl -sSL "https://ftp.gnu.org/gnu/gcc/gcc-9.3.0/gcc-9.3.0.tar.xz" | tar xJ

        # Prepare compiling
        mkdir -p "${DIR}/opt/bin"
        export PREFIX="${DIR}/opt"
        export TARGET=i686-elf
        export PATH="$PREFIX/bin:$PATH"

        # Compile binutils
        mkdir "${DIR}/src/build-binutils" && cd "${DIR}/src/build-binutils"
        ../binutils-2.34/configure --target="$TARGET" --prefix="$PREFIX" --with-sysroot --disable-nls --disable-werror
        make
        make install

        # Compile GCC
        mkdir "${DIR}/src/build-gcc" && cd "${DIR}/src/build-gcc"
        ../gcc-9.3.0/configure --target="$TARGET" --prefix="$PREFIX" --disable-nls --enable-languages=c,c++ --without-headers
        make all-gcc
        make all-target-libgcc
        make install-gcc
        make install-target-libgcc

        cd "${DIR}/.."
    else
        cd cross
        DIR=$(pwd)
        export PREFIX="${DIR}/opt"
        export TARGET=i686-elf
        export PATH="$PREFIX/bin:$PATH"
        cd ..
    fi
}

make_disk() {
    rm -rf disk && mkdir -p disk/font/

    cd disk/font/
    VERSION="1.8.1"
    wget -q "https://github.com/fcambus/spleen/releases/download/$VERSION/spleen-$VERSION.tar.gz"
    tar xf "spleen-$VERSION.tar.gz"
    mv spleen-"$VERSION"/*.psfu .
    rm -rf "spleen-$VERSION"*

    cp /usr/share/kbd/consolefonts/ter-p32b.psf.gz . &&
        cp /usr/share/kbd/consolefonts/ter-p32n.psf.gz . &&
        gunzip ter-p32* || echo "Terminus font not found!"

    cd ../../
}

make_build() {
    if ! [ -d "disk/" ]; then
        echo "Creating disk..."
        make_disk
    fi

    mkdir -p build/
    rm -rf build/*

    echo "Building..."
    make

    # Create disk image
    dd if=/dev/zero of=build/disk.img bs=1k count=16k status=none
    sudo mke2fs -q build/disk.img
    dd if=build/boot.bin of=build/disk.img conv=notrunc status=none
    cp build/kernel.bin . # For nicer disk img
    ./ext2util/ext2util -x build/disk.img -wf kernel.bin -i 5 >/dev/null
    rm kernel.bin

    mkdir -p mnt/
    sudo mount build/disk.img mnt/
    sudo cp -r disk/* mnt/
    sudo cp build/test mnt/
    cat disk/**/* >/dev/null
    sudo umount mnt/
    rm -rf mnt/

    printf "Build finshed successfully!\n\n"
}

make_test() {
    qemu_with_flags -hdb build/disk.img
}

make_debug() {
    qemu_with_flags -hdb build/disk.img -s -S
}

make_disasm() {
    objdump -drwC -Mintel build/debug.o --visualize-jumps=color | less -R
    #hexdump -C build/kernel.bin | less -R
}

make_sync() {
    ctags -R --exclude=.git --exclude=build --exclude=iso --exclude=cross .

    make --always-make --dry-run |
        grep -wE 'gcc' |
        grep -w '\-c' |
        jq -nR '[inputs|{directory:"'"$(pwd)"'/src/", command:., file: match(" [^ ]+$").string[1:]}]' \
            >compile_commands.json
}

make_tidy() {
    shfmt -w ./run
    find src/ -type f -regex '.*\.\(c\|h\)' -exec clang-format -i {} \;
    # This may or may not work
    find src/ -type f -print0 | xargs -0 -l -i sh -c '[ -n "$(tail -c1 {})" ] && printf "\n" >> {}'
}

make_clean() {
    rm -rf build/
}

if [ "${mode}" = "cross" ]; then
    make_cross
elif [ "${mode}" = "build" ]; then
    make_cross
    make_clean
    make_build
elif [ "${mode}" = "clean" ]; then
    make_clean
elif [ "${mode}" = "test" ]; then
    make_cross
    make_clean
    make_build
    make_sync &
    make_test
elif [ "${mode}" = "debug" ]; then
    make_cross
    make_build
    make_sync &
    make_debug
elif [ "${mode}" = "again" ]; then
    make_test
elif [ "${mode}" = "disasm" ]; then
    make_cross
    make_build
    make_disasm
elif [ "${mode}" = "sync" ]; then
    make_sync
elif [ "${mode}" = "tidy" ]; then
    make_tidy
elif [ "${mode}" = "disk" ]; then
    make_disk
elif [ "${mode}" = "" ]; then # TODO: Prevent code duplication in build script via functions?
    make_cross
    make_clean
    make_build
    make_sync &
    make_test
else
    echo "Please use the following syntax:"
    echo "./run {cross | build | clean | test | disasm | sync | tidy} [-y]"
    echo "The default option is 'test'"
fi