yaml.cxx

module;
#include <filesystem>
#include <ostream>
#include <vector>
#define RYML_SINGLE_HDR_DEFINE_NOW
#include "rapidyaml.hxx"
export module maud_:yaml;
import :filesystem;

namespace c4::yml {
export std::string to_string(ConstNodeRef n) { return {n.val().data(), n.val().size()}; }
export std::string_view to_view(ConstNodeRef n) {
  return {n.val().data(), n.val().size()};
}
export void PrintTo(ConstNodeRef p, std::ostream *os) { *os << p; }
}  // namespace c4::yml

export using c4::yml::ConstNodeRef;

export using c4::yml::operator<<;
export using c4::yml::as_json;
export using c4::yml::parse_in_place;

export decltype(auto) set_map(c4::yml::NodeRef node, auto key) {
  node[key] |= c4::yml::MAP;
  return node[key];
}

/// A helper for unit tests parameterized using a YAML file.
export struct Parameter : c4::yml::ConstNodeRef {
  /// Each test parameter is identified by name (its key in the top level mapping)
  std::string name() const { return {key().data(), key().size()}; }

  friend void PrintTo(Parameter p, std::ostream *os) { *os << p.name(); }

  /// Read a file containing a YAML mapping into a vector<Parameter>
  static auto read_file(std::filesystem::path const &path) {
    struct : std::vector<Parameter> {
      Padded<> yaml;
      c4::yml::Tree tree;
    } set;

    set.yaml = read(path);
    set.tree = parse_in_place(set.yaml.data());
    set.resize(set.tree.rootref().num_children());

    for (auto parameter = set.begin(); auto n : set.tree.rootref()) {
      *parameter++ = {n};
    }
    return set;
  }
};