#include <algorithm>
#include <catch2/catch.hpp>
#include <chrono>
#include <filesystem>
#include <fstream>
#include <iomanip>
#include <iostream>

#include "daggy/Serialization.hpp"
#include "daggy/Utilities.hpp"

TEST_CASE("string_utilities", "[utilities_string]")
{
  std::string test = "/this/is/{{A}}/test/{{A}}";
  auto res         = daggy::globalSub(test, "{{A}}", "hello");
  REQUIRE(res == "/this/is/hello/test/hello");
}

TEST_CASE("string_expansion", "[utilities_parameter_expansion]")
{
  SECTION("Basic expansion")
  {
    std::string testParams{
        R"({"DATE": ["2021-05-06", "2021-05-07" ], "SOURCE": "name", "TYPE": ["a", "b", "c"]})"};
    auto params = daggy::configFromJSON(testParams);
    std::vector<std::string> cmd{"/usr/bin/echo", "{{DATE}}", "{{SOURCE}}",
                                 "{{TYPE}}"};
    auto allCommands = daggy::interpolateValues(cmd, params);

    REQUIRE(allCommands.size() == 6);
  }

  SECTION("Skip over unused parameters")
  {
    std::string testParams{
        R"({"DATE": ["2021-05-06", "2021-05-07" ], "SOURCE": "name", "TYPE": ["a", "b", "c"]})"};
    auto params = daggy::configFromJSON(testParams);
    std::vector<std::string> cmd{"/usr/bin/echo", "{{DATE}}", "{{SOURCE}}"};
    auto allCommands = daggy::interpolateValues(cmd, params);

    // TYPE isn't used, so it's just |DATE| * |SOURCE|
    REQUIRE(allCommands.size() == 2);
  }

  SECTION("Expand within a command part")
  {
    std::string testParams{
        R"({"DATE": ["2021-05-06", "2021-05-07" ], "SOURCE": ["A", "B"], "TYPE": ["a", "b", "c"]})"};
    auto params = daggy::configFromJSON(testParams);
    std::vector<std::string> cmd{"/usr/bin/touch", "{{DATE}}_{{SOURCE}}"};
    auto result = daggy::interpolateValues(cmd, params);

    // TYPE isn't used, so it's just |DATE| * |SOURCE|
    REQUIRE(result.size() == 4);
  }
}