#!/bin/bash

print_help() {
    echo "Usage:"
    echo "   cproject <type> <name>"
    echo "Types:"
    echo "   bin, bench"
}

die() {
    echo "Unexpected error"
    exit 2
}

# Ninja does not work with backward-cpp
new_bin() {
    local name="$1"
    mkdir "$name" || die
    cd "$name" || die
    git clone 'https://github.com/fmtlib/fmt.git' third_party/fmt || die
    git clone 'https://github.com/bombela/backward-cpp.git' third_party/backward-cpp || die
    git clone 'https://github.com/gabime/spdlog.git' third_party/spdlog || die
    cat << EOF > CMakeLists.txt
cmake_minimum_required(VERSION 3.11)
project(${name})

if(NOT CMAKE_BUILD_TYPE)
    set(CMAKE_BUILD_TYPE Debug)
endif()

set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
set(CMAKE_C_STANDARD 11)
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
set(CMAKE_CXX_FLAGS "\${CMAKE_CXX_FLAGS} -Wall -Wextra")
set(CMAKE_CXX_FLAGS_DEBUG "\${CMAKE_CXX_FLAGS_DEBUG} -ggdb -O0")
set(CMAKE_EXE_LINKER_FLAGS "\${CMAKE_EXE_LINKER_FLAGS} -rdynamic")

add_subdirectory(third_party/fmt)
add_subdirectory(third_party/spdlog)
add_subdirectory(third_party/backward-cpp)

add_executable(main)
target_sources(main PRIVATE main.cpp \${BACKWARD_ENABLE})
target_link_libraries(main PRIVATE fmt spdlog -ldw)
EOF

    cat << 'EOF' > main.cpp
#include <iostream>
#include <string>
#include "fmt/format.h"
#include "spdlog/spdlog.h"

#define CRASH(...)                      \
    { ::spdlog::critical(__VA_ARGS__);  \
      int _ [[maybe_unused]] =          \
         *reinterpret_cast<int*>(0); }

inline int get_return_code() {
    CRASH("unimplemented function: {}", __func__);
    return 0;
}

int main(int argc, char* argv[]) {
    fmt::print("{}, {}\n\n", "hello", "world");

    spdlog::set_level(spdlog::level::debug);
    spdlog::debug("debug message");
    spdlog::info("info message");
    spdlog::warn("warning message");
    spdlog::error("Some error message with arg: {}", 1);
    return get_return_code();
}
EOF
    echo
    echo "Skeleton generated. Run the command and you shall see " \
         "the stacktrace demo."
    echo "  cd ${name} && cmake -B build && " \
         "make -C build -j main && build/main"
}

new_bench() {
    local name="$1"
    mkdir "$name" || die
    cd "$name" || die
    git clone 'https://github.com/fmtlib/fmt.git' third_party/fmt || die
    git clone 'https://github.com/bombela/backward-cpp.git' third_party/backward-cpp || die
    git clone 'https://github.com/gabime/spdlog.git' third_party/spdlog || die
    git clone 'https://github.com/google/benchmark.git' third_party/benchmark || die
    git clone 'https://github.com/google/googletest.git' third_party/benchmark/googletest || die
    cat << EOF > CMakeLists.txt
cmake_minimum_required(VERSION 3.11)
project(${name})

if(NOT CMAKE_BUILD_TYPE)
    set(CMAKE_BUILD_TYPE Release)
endif()

set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
set(CMAKE_C_STANDARD 11)
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
set(CMAKE_CXX_FLAGS "\${CMAKE_CXX_FLAGS} -Wall -Wextra -g")
set(CMAKE_EXE_LINKER_FLAGS "\${CMAKE_EXE_LINKER_FLAGS} -rdynamic")

set(BENCHMARK_ENABLE_LTO true)

add_subdirectory(third_party/fmt)
add_subdirectory(third_party/spdlog)
add_subdirectory(third_party/backward-cpp)
add_subdirectory(third_party/benchmark)

add_executable(bench)
target_sources(bench PRIVATE bench.cpp \${BACKWARD_ENABLE})
target_link_libraries(bench PRIVATE fmt spdlog benchmark::benchmark_main -ldw)
EOF

    cat << 'EOF' > bench.cpp
#include "benchmark/benchmark.h"
#include <cinttypes>
#include <cstdlib>

constexpr size_t ARRAY_LEN = 1024*64;
static void BM_SeqArrayAccess(benchmark::State& state) {
    int stride = state.range(0);
    uint64_t arr[ARRAY_LEN];
    for (auto _ : state) {
        for (size_t idx = 0; idx < ARRAY_LEN; idx += stride) {
            arr[idx] = random();
        }
    }
}
BENCHMARK(BM_SeqArrayAccess)
    ->Arg(1)
    ->Arg(2)
    ->Arg(3)
    ->Arg(4)
    ->Arg(5)
    ->Arg(6)
    ->Arg(7)
    ->Arg(8);
EOF
    echo
    echo "Skeleton generated."
    echo "  cd ${name} && cmake -B build && " \
         "make -C build -j bench && build/bench"
    echo "You may also need to change your CPU scheduler:"
    echo "  sudo cpupower frequency-set -g performance"
}

if [[ "$2" == "" ]]; then
    echo "Missing project name"
    echo
    print_help
    exit 1
fi

if [[ -e "$2" ]]; then
    echo "File \"$2\" already exists"
    exit 1
fi

if [[ "$1" == "bin" ]]; then
    new_bin "$2"
elif [[ "$1" == "bench" ]]; then
    new_bench "$2"
else
    echo "Unknown type: $1"
    echo
    print_help
    exit 1
fi