#include "gen.hpp"
#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;
}
void handle_bins(std::stringstream &s, const std::string &bin_type) {
}
} // namespace detail

void generate_project(const std::string &str) {
    fs::create_directory("src");
    auto dir_name = fs::current_path().stem();
    if (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 (str == "static" || 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() {
    std::stringstream ss;

    const auto toml = toml::parse("cmake.toml");
    const auto &cmake = toml::find(toml, "cmake");
    const std::string cmake_min = toml::find(cmake, "minimum_required").as_string();
    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 << "cmake_minimum_required(VERSION " << cmake_min << ")\n\n"
       << "project(" << proj_name << " VERSION " << proj_version << ")\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("linker_flags")) {
        ss << "set(CMAKE_EXE_LINKER_FLAGS ${CMAKE_EXE_LINKER_FLAGS}";
        const auto flags = toml::find(cmake, "linker_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";
        }
        ss << "\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_directories")) {
                const auto includes = toml::find(bin, "include_directories").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_libraries")) {
                const auto libraries = toml::find(bin, "link_libraries").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("compile_features")) {
                const auto feats = toml::find(bin, "compile_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("definitions")) {
                const auto defs = toml::find(bin, "definitions").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";

    std::ofstream ofs("CMakeLists.txt");
    if (ofs.is_open()) {
        ofs << ss.rdbuf();
    }
    ofs.flush();
    ofs.close();
}
} // namespace cmkr::gen