#include "config/config.h"
#include "puml/class_diagram_generator.h"
#include "puml/sequence_diagram_generator.h"
#include "uml/class_diagram_model.h"
#include "uml/class_diagram_visitor.h"
#include "cx/compilation_database.h"
#include "uml/sequence_diagram_visitor.h"

#include <cli11/CLI11.hpp>
#include <glob/glob.hpp>
#include <spdlog/spdlog.h>

#include <filesystem>
#include <fstream>
#include <iostream>
#include <limits.h>
#include <malloc.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

using namespace clanguml;
using config::config;
using cx::compilation_database;

int main(int argc, const char *argv[])
{
    spdlog::set_pattern("[%l] %v");

    CLI::App app{"Clang-based PlantUML generator from C++ sources"};

    std::string config_path{".clanguml"};
    std::string compilation_database_dir{'.'};
    bool verbose{false};

    app.add_option(
        "-c,--config", config_path, "Location of configuration file");
    app.add_option("-d,--compile-database", compilation_database_dir,
        "Location of configuration file");
    app.add_flag("-v,--verbose", verbose, "Verbose logging");

    CLI11_PARSE(app, argc, argv);

    if (verbose)
        spdlog::set_level(spdlog::level::debug);

    spdlog::info("Loading clang-uml config from {}", config_path);

    auto config = config::load(config_path);

    spdlog::info("Loading compilation database from {} directory",
        config.compilation_database_dir);

    auto db =
        compilation_database::from_directory(config.compilation_database_dir);

    for (const auto &[name, diagram] : config.diagrams) {
        using config::class_diagram;
        using config::sequence_diagram;

        if (std::dynamic_pointer_cast<class_diagram>(diagram)) {
            generators::class_diagram::generate(
                db, name, dynamic_cast<class_diagram &>(*diagram));
        }
        else if (std::dynamic_pointer_cast<sequence_diagram>(diagram)) {
            generators::sequence_diagram::generate(
                db, name, dynamic_cast<sequence_diagram &>(*diagram));
        }
    }
    return 0;
}