#include "cmake.hpp"

#include <filesystem>
#include <toml.hpp>

namespace fs = std::filesystem;

namespace cmkr::cmake {

namespace detail {
std::vector<std::string> to_string_vec(
    const std::vector<toml::basic_value<toml::discard_comments, std::unordered_map, std::vector>>
        vals) {
    std::vector<std::string> temp;
    for (const auto &val : vals)
        temp.push_back(val.as_string());
    return temp;
}
} // namespace detail

CMake::CMake(const std::string &path, bool build) {
    if (fs::exists(fs::path(path) / "cmake.toml")) {
        if (build) {
            const auto toml = toml::parse("cmake.toml");
            if (toml.contains("cmake")) {
                const auto &cmake = toml::find(toml, "cmake");

                if (cmake.contains("bin-dir")) {
                    bin_dir = toml::find(cmake, "bin-dir").as_string();
                }

                if (cmake.contains("generator")) {
                    generator = toml::find(cmake, "generator").as_string();
                }

                if (cmake.contains("arguments")) {
                    gen_args = detail::to_string_vec(toml::find(cmake, "arguments").as_array());
                }
            }
        } else {
            const auto toml = toml::parse((fs::path(path) / "cmake.toml").string());
            if (toml.contains("cmake")) {
                const auto &cmake = toml::find(toml, "cmake");
                cmake_version = toml::find(cmake, "minimum").as_string();

                if (cmake.contains("cpp-flags")) {
                    cppflags = detail::to_string_vec(toml::find(cmake, "cpp-flags").as_array());
                }

                if (cmake.contains("c-flags")) {
                    cflags = detail::to_string_vec(toml::find(cmake, "c-flags").as_array());
                }

                if (cmake.contains("link-flags")) {
                    linkflags = detail::to_string_vec(toml::find(cmake, "link-flags").as_array());
                }

                if (cmake.contains("subdirs")) {
                    subdirs = detail::to_string_vec(toml::find(cmake, "subdirs").as_array());
                }
            }

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

            if (toml.contains("find-package")) {
                using pkg_map = std::map<std::string, std::string>;
                packages = toml::find<pkg_map>(toml, "find-package");
            }

            if (toml.contains("fetch-content")) {
                using content_map = std::map<std::string, std::map<std::string, std::string>>;
                contents = toml::find<content_map>(toml, "fetch-content");
            }

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

                for (const auto &bin : bins) {
                    Bin b;
                    b.name = toml::find(bin, "name").as_string();
                    b.type = toml::find(bin, "type").as_string();

                    b.sources = detail::to_string_vec(toml::find(bin, "sources").as_array());

                    if (bin.contains("include-dirs")) {
                        b.include_dirs =
                            detail::to_string_vec(toml::find(bin, "include-dirs").as_array());
                    }

                    if (bin.contains("link-libs")) {
                        b.link_libs =
                            detail::to_string_vec(toml::find(bin, "link-libs").as_array());
                    }

                    if (bin.contains("features")) {
                        b.features = detail::to_string_vec(toml::find(bin, "features").as_array());
                    }

                    if (bin.contains("defines")) {
                        b.defines = detail::to_string_vec(toml::find(bin, "defines").as_array());
                    }
                    binaries.push_back(b);
                }
            }
        }
    }
}
} // namespace cmkr::cmake