#ifndef RFL_VISIT_HPP_
#define RFL_VISIT_HPP_

#include <variant>

#include "Literal.hpp"
#include "TaggedUnion.hpp"
#include "internal/StringLiteral.hpp"
#include "internal/VisitTree.hpp"
#include "internal/VisitorWrapper.hpp"

namespace rfl {

  /// Implements the visitor pattern for Literals.
  template <class Visitor, internal::StringLiteral... _fields, class... Args>
  inline auto
  visit(const Visitor& _visitor, const Literal<_fields...> _literal, const Args&... _args) {
    constexpr int size = sizeof...(_fields);
    using WrapperType  = internal::VisitorWrapper<Visitor, _fields...>;
    const auto wrapper = WrapperType(&_visitor);
    return internal::VisitTree::visit<0, size, WrapperType>(wrapper, _literal.value(), _args...);
  }

  /// Implements the visitor pattern for TaggedUnions.
  template <class Visitor, internal::StringLiteral _discriminator, class... Args>
  inline auto
  visit(const Visitor& _visitor, const TaggedUnion<_discriminator, Args...>& _tagged_union) {
    return std::visit(_visitor, _tagged_union.variant());
  }

} // namespace rfl

#endif // RFL_VISIT_HPP_