#pragma once

#include <map>
#include <ostream>
#include <unordered_set>
#include <vector>

#include <jvalidate/detail/pointer.h>
#include <jvalidate/forward.h>

namespace jvalidate {
class ValidationResult {
public:
  template <Adapter A, RegexEngine RE> friend class ValidationVisitor;

private:
  std::map<detail::Pointer, std::map<detail::Pointer, std::string>> annotations_;

public:
  friend std::ostream & operator<<(std::ostream & os, ValidationResult const & result) {
    for (auto const & [where, annotations] : result.annotations_) {
      if (annotations.size() == 1) {
        auto const & [schema_path, message] = *annotations.begin();
        std::cout << where << ": " << schema_path << ": " << message << "\n";
      } else {
        std::cout << where << "\n";
        for (auto const & [schema_path, message] : annotations) {
          std::cout << "  " << schema_path << ": " << message << "\n";
        }
      }
    }
    return os;
  }

private:
  void annotate(ValidationResult && result) {
    for (auto const & [where, errors] : result.annotations_) {
      for (auto const & [schema_path, message] : errors) {
        annotations_[where][schema_path] += message;
      }
    }
  }

  void annotate(detail::Pointer const & where, detail::Pointer const & schema_path,
                std::string const & message) {
    annotations_[where][schema_path] += message;
  }
};
}