#include <gtest/gtest.h>
#include <iostream>
#include <filesystem>
#include <vector>
#include <cstdio>

static const std::string arcNewCmd = "python3 ../tests/arcilator-header-cpp.py";
static const std::string arcOldCmd = "python3 ../tests/ahc_old.py";
static const std::string viewDepth = "--view-depth 1";

std::string executeCommand(const std::string &command) {
    std::array<char, 256> buffer{};
    std::string result{};

    std::unique_ptr<FILE, decltype(&pclose)> pipe(popen(command.c_str(), "r"), pclose);
    if (!pipe) {
        throw std::runtime_error("popen() failed!");
    }

    while (fgets(buffer.data(), buffer.size(), pipe.get()) != nullptr) {
        result += std::string{buffer.data()};
    }
    return result;
}

std::pair<std::string, std::string> doNewOldTest(const std::string &newCmd, const std::string &oldCmd) {
    auto retNew = executeCommand(newCmd);
    auto retOld = executeCommand(oldCmd);

    return {retNew, retOld};
}

void doTest(const std::string &testFile) {
    auto res = doNewOldTest(arcNewCmd + " " + testFile + " " + viewDepth, arcOldCmd + " " + testFile + " " + viewDepth);
    EXPECT_EQ(res.first, res.second);
}

TEST(ArcilatorHeaderCppTests, MuxTestHeader) {
    const auto testFile = "../tests/mlir/mux_test.json";
    doTest(testFile);
}

TEST(ArcilatorHeaderCppTests, ArcilatorOfficialTestHeader) {
    const auto testFile = "../tests/mlir/arcilator.json";
    doTest(testFile);
}

TEST(ArcilatorHeaderCppTests, DecoupledGCDHeader) {
    const auto testFile = "../tests/mlir/decoupledGCD.json";
    doTest(testFile);
}

TEST(ArcilatorHeaderCppTests, RISCVMiniHeader) {
    const auto testFile = "../tests/mlir/riscv-mini.json";
    doTest(testFile);
}