cmkr/src/gen.cpp

#include "gen.h"

#include <filesystem>
#include <fstream>
#include <map>
#include <sstream>
#include <stdexcept>
#include <toml.hpp>

namespace fs = std::filesystem;

namespace cmkr::gen {

namespace detail {

inline std::string to_upper(const std::string &str) {
    std::string temp;
    temp.reserve(str.size());
    for (auto c : str) {
        temp.push_back(toupper(c));
    }
    return temp;
}

} // namespace detail

void generate_project(const char *str) {
    fs::create_directory("src");
    auto dir_name = fs::current_path().stem();
    if (!strcmp(str, "exe")) {
        std::ofstream ofs("src/main.cpp");
        if (ofs.is_open()) {
            ofs << "#include <iostream>\n\nint main() {\n\tstd::cout << \"Hello world!\" << "
                   "std::endl;\n}";
        }
        ofs.flush();
        ofs.close();

        std::ofstream ofs2("cmake.toml");
        if (ofs2.is_open()) {
            ofs2 << "[cmake]\nminimum_required = \"3.0\"\n\n[project]\nname = \""
                 << dir_name.string()
                 << "\"\nversion = "
                    "\"0.1.0\"\n\n[[exe]]\nname = \""
                 << dir_name.string() << "\"\nsources = [\"src/main.cpp\"]\n";
        }
        ofs2.flush();
        ofs2.close();
    } else if (!strcmp(str, "static") || !strcmp(str, "shared")) {
        std::ofstream ofs("src/lib.cpp");
        if (ofs.is_open()) {
            ofs << "int dll_main() {\n\treturn 0;\n}";
        }
        ofs.flush();
        ofs.close();

        std::ofstream ofs2("cmake.toml");
        if (ofs2.is_open()) {
            ofs2 << "[cmake]\nminimum_required = \"3.0\"\n\n[project]\nname = \""
                 << dir_name.string()
                 << "\"\nversion = "
                    "\"0.1.0\"\n\n[[lib]]\nname = \""
                 << dir_name.string() << "\"\nsources = [\"src/lib.cpp\"]\ntype = \"" << str
                 << "\"\n";
        }
        ofs2.flush();
        ofs2.close();
    } else {
        throw std::runtime_error("Unknown project type. Types are exe, shared, static!");
    }
}

void generate_cmake(const char *path) {
    std::stringstream ss;
    std::vector<std::string> subdirs;

    const auto toml = toml::parse(fs::path(path) / "cmake.toml");
    if (toml.contains("cmake")) {
        const auto &cmake = toml::find(toml, "cmake");
        const std::string cmake_min = toml::find(cmake, "minimum_required").as_string();
        ss << "cmake_minimum_required(VERSION " << cmake_min << ")\n\n"
           << "set(CMAKE_EXPORT_COMPILE_COMMANDS ON)\n\n";

        if (cmake.contains("cpp_flags")) {
            ss << "set(CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS}";
            const auto flags = toml::find(cmake, "cpp_flags").as_array();
            for (const auto &flag : flags) {
                ss << " " << flag;
            }
            ss << ")\n\n";
        }

        if (cmake.contains("c_flags")) {
            ss << "set(CMAKE_C_FLAGS ${CMAKE_C_FLAGS}";
            const auto flags = toml::find(cmake, "c_flags").as_array();
            for (const auto &flag : flags) {
                ss << " " << flag;
            }
            ss << ")\n\n";
        }

        if (cmake.contains("link_flags")) {
            ss << "set(CMAKE_EXE_LINKER_FLAGS ${CMAKE_EXE_LINKER_FLAGS}";
            const auto flags = toml::find(cmake, "link_flags").as_array();
            for (const auto &flag : flags) {
                ss << " " << flag;
            }
            ss << ")\n\n";
        }

        if (cmake.contains("subdirs")) {
            const auto dirs = toml::find(cmake, "subdirs").as_array();
            for (const auto &dir : dirs) {
                ss << "add_subdirectory(" << dir << ")\n";
                subdirs.push_back(dir.as_string());
            }
            ss << "\n\n";
        }
    }

    if (toml.contains("project")) {
        const auto &project = toml::find(toml, "project");
        const std::string proj_name = toml::find(project, "name").as_string();
        const std::string proj_version = toml::find(project, "version").as_string();

        ss << "project(" << proj_name << " VERSION " << proj_version << ")\n\n";
    }

    if (toml.contains("dependencies")) {
        std::map<std::string, std::string> deps =
            toml::find<std::map<std::string, std::string>>(toml, "dependencies");
        for (const auto &dep : deps) {
            ss << "find_package(" << dep.first;
            if (dep.second != "*") {
                ss << " " << dep.second << " CONFIG REQUIRED)\n";
            } else {
                ss << " CONFIG REQUIRED)\n";
            }
        }
    }

    ss << "\n";

    if (toml.contains("bin")) {
        const auto &bins = toml::find(toml, "bin").as_array();

        for (const auto &bin : bins) {
            const std::string bin_name = toml::find(bin, "name").as_string();
            const std::string type = toml::find(bin, "type").as_string();
            std::string bin_type;
            std::string add_command;
            if (type == "exe") {
                bin_type = "";
                add_command = "add_executable";
            } else if (type == "shared" || type == "static") {
                bin_type = detail::to_upper(type);
                add_command = "add_library";
            } else {
                throw std::runtime_error(
                    "Unknown binary type! Supported types are exe, shared and static");
            }

            const auto srcs = toml::find(bin, "sources").as_array();
            ss << "set(" << detail::to_upper(bin_name) << "_SOURCES\n";
            for (const auto &src : srcs) {
                ss << "\t" << src << "\n";
            }
            ss << "\t)\n\n"
               << add_command << "(" << bin_name << " " << bin_type << " ${"
               << detail::to_upper(bin_name) << "_SOURCES})\n\n";

            if (bin.contains("include_dirs")) {
                const auto includes = toml::find(bin, "include_dirs").as_array();
                ss << "target_include_directories(" << bin_name << " PUBLIC\n\t";
                for (const auto &inc : includes) {
                    ss << inc << "\n\t";
                }
                ss << ")\n\n";
            }

            if (bin.contains("link_libs")) {
                const auto libraries = toml::find(bin, "link_libs").as_array();
                ss << "target_link_libraries(" << bin_name << " PUBLIC\n\t";
                for (const auto &l : libraries) {
                    ss << l << "\n\t";
                }
                ss << ")\n\n";
            }

            if (bin.contains("features")) {
                const auto feats = toml::find(bin, "features").as_array();
                ss << "target_compile_features(" << bin_name << " PUBLIC\n\t";
                for (const auto &feat : feats) {
                    ss << feat << "\n\t";
                }
                ss << ")\n\n";
            }

            if (bin.contains("defines")) {
                const auto defs = toml::find(bin, "defines").as_array();
                ss << "target_add_definitions(" << bin_name << " PUBLIC\n\t";
                for (const auto &def : defs) {
                    ss << def << "\n\t";
                }
                ss << ")\n\n";
            }
        }
    }

    ss << "\n\n";
    // printf("%s\n", ss.str().c_str());
    std::ofstream ofs(fs::path(path) / "CMakeLists.txt");
    if (ofs.is_open()) {
        ofs << ss.rdbuf();
    }
    ofs.flush();
    ofs.close();
    for (const auto &sub : subdirs) {
        generate_cmake(sub.c_str());
    }
}
} // namespace cmkr::gen

void cmkr_gen_generate_project(const char *typ) {
    cmkr::gen::generate_project(typ);
}

void cmkr_gen_generate_cmake(const char *path) {
    cmkr::gen::generate_cmake(path);
}