#include <doctest/doctest.h>

#include <fstream>
#include <string>

#include "secsgem/config/loader.hpp"
#include "secsgem/secs2/item.hpp"

using namespace secsgem;
namespace gem = secsgem::gem;
namespace s2  = secsgem::secs2;

#ifndef SECSGEM_DATA_DIR
#error "SECSGEM_DATA_DIR not defined; see CMakeLists.txt"
#endif

TEST_CASE("control_state.yaml parses into the same shape as default_table()") {
  auto cfg = config::load_control_state(SECSGEM_DATA_DIR "/control_state.yaml");
  auto baseline = gem::ControlTransitionTable::default_table();

  CHECK(cfg.initial == gem::ControlState::HostOffline);
  // Same row count; same (from, on) pairs map to the same destination.
  CHECK(cfg.table.size() == baseline.size());

  for (const auto& r : baseline.rows()) {
    const auto* loaded = cfg.table.find(r.from, r.on);
    REQUIRE(loaded != nullptr);
    CHECK(loaded->to == r.to);
    CHECK(loaded->then == r.then);
    CHECK(loaded->ack_code == r.ack_code);
  }
}

TEST_CASE("equipment.yaml populates SVIDs, ECIDs, CEIDs, alarms, recipes, commands") {
  gem::EquipmentDataModel m;
  auto desc = config::load_equipment(SECSGEM_DATA_DIR "/equipment.yaml", m);

  CHECK(desc.model_name == "SECSGEM-SIM");
  CHECK(desc.software_rev == "0.1.0");
  REQUIRE(desc.emit_on_control_change.has_value());
  CHECK(*desc.emit_on_control_change == 100);

  CHECK(m.svids.all().size() == 3);
  CHECK(m.svids.get(1)->name == "ControlState");
  CHECK(m.svids.get(3)->value == s2::Item::boolean(true));

  CHECK(m.ecids.all().size() == 2);
  CHECK(m.ecids.get(10)->value == s2::Item::u4(uint32_t{1}));

  CHECK(m.events.all_events().size() == 3);
  CHECK(m.events.has_event(100));
  CHECK(m.events.has_event(300));

  CHECK(m.alarms.all().size() == 2);
  CHECK(m.alarms.get(1)->text == "Chiller Temp High");
  CHECK(m.alarms.get(1)->severity_category == 4);

  CHECK(m.recipes.list().size() == 2);
  CHECK(m.recipes.get("RECIPE-A").value().find("CHAMBER ARGON") != std::string::npos);

  CHECK(m.commands.has("START"));
  auto start = m.commands.dispatch("START", {});
  CHECK(start.ack == gem::HostCmdAck::Accept);
  CHECK(start.emit_ceid.value_or(0) == 300);
  auto fault = m.commands.dispatch("FAULT", {});
  CHECK(fault.set_alarm.value_or(0) == 1);

  // Spool config loaded from YAML.
  CHECK(m.spool.is_stream_spoolable(5));
  CHECK(m.spool.is_stream_spoolable(6));
  CHECK_FALSE(m.spool.is_stream_spoolable(1));
  CHECK(m.spool.max_size() == 100);

  // force_spool flag plumbed from YAML through CommandSpec.
  auto on  = m.commands.dispatch("SPOOL_ON", {});
  auto off = m.commands.dispatch("SPOOL_OFF", {});
  REQUIRE(on.force_spool.has_value());
  CHECK(*on.force_spool == true);
  REQUIRE(off.force_spool.has_value());
  CHECK(*off.force_spool == false);

  // Capabilities + equipment_type loaded for S1F20.
  CHECK(desc.equipment_type == "EQUIPMENT");
  REQUIRE(desc.capabilities.size() >= 5);
  bool saw_spooling = false;
  for (const auto& [code, name] : desc.capabilities) {
    if (code == 14) saw_spooling = true;
  }
  CHECK(saw_spooling);

  // DVIDs loaded for S1F22.
  CHECK(m.dvids.all().size() == 2);
  CHECK(m.dvids.get(100)->name == "WaferCounter");
  CHECK(m.dvids.get(101)->units == "Torr");
}

TEST_CASE("loader surfaces YAML errors with file path") {
  CHECK_THROWS_AS(config::load_equipment("/tmp/does-not-exist.yaml", *new gem::EquipmentDataModel),
                  config::ConfigError);
}