#include "common/logging/backend.h"
#include "common/logging/text_formatter.h"
#include "common/scope_exit.h"
#include "core/core.h"
#include "core/mem_map.h"
#include "core/loader/loader.h"
#include "codegen.h"
#include <llvm/Support/CommandLine.h>

namespace cl = llvm::cl;

cl::opt<std::string> InputFilename(cl::Positional, cl::Required, cl::desc("<input rom filename>"));
cl::opt<std::string> OutputFilename(cl::Positional, cl::Required, cl::desc("<output object filename>"));
cl::opt<std::string> DebugFilename(cl::Positional, cl::Optional, cl::desc("<debug filename>"));

int main(int argc, const char *const *argv)
{
    // Remove all llvm options
    llvm::StringMap<cl::Option *> options;
    cl::getRegisteredOptions(options);
    for (auto i = options.begin(); i != options.end(); ++i)
    {
        if (i->getValue() != &InputFilename && i->getValue() != &OutputFilename && i->getValue() != &DebugFilename)
        {
            i->getValue()->setHiddenFlag(cl::Hidden);
        }
    }
    cl::ParseCommandLineOptions(argc, argv);

	std::shared_ptr<Log::Logger> logger = Log::InitGlobalLogger();
	Log::Filter log_filter(Log::Level::Debug);
	Log::SetFilter(&log_filter);
	std::thread logging_thread(Log::TextLoggingLoop, logger);
	SCOPE_EXIT({
		logger->Close();
		logging_thread.join();
	});

    auto input_rom = InputFilename.c_str();
    auto output_object = OutputFilename.c_str();
    auto output_debug = DebugFilename.getNumOccurrences() ? DebugFilename.c_str() : nullptr;

	Core::Init();
	Memory::Init();

	auto load_result = Loader::LoadFile(input_rom);
	if (Loader::ResultStatus::Success != load_result)
	{
        LOG_CRITICAL(BinaryTranslator, "Failed to load ROM (Error %i)!", load_result);
		return -1;
	}

	CodeGen code_generator(output_object, output_debug);
	code_generator.Run();
}