json-validator/include/jvalidate/detail/pointer.h

#pragma once

#include <cstdint>
#include <iostream>
#include <string>
#include <string_view>
#include <variant>
#include <vector>

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

namespace jvalidate::detail {

class Pointer {
public:
  Pointer() = default;
  Pointer(std::vector<std::variant<std::string, size_t>> const & tokens) : tokens_(tokens) {}
  Pointer(std::string_view path) {
    auto append_with_parse = [this](std::string in) {
      if (in.find_first_not_of("0123456789") == std::string::npos) {
        tokens_.push_back(std::stoull(in));
      } else {
        tokens_.push_back(std::move(in));
      }
    };

    path.remove_prefix(1);
    size_t p = path.find('/');
    for (; p != std::string::npos; path.remove_prefix(p + 1), p = path.find('/')) {
      append_with_parse(std::string(path.substr(0, p)));
    }

    if (not path.empty()) {
      append_with_parse(std::string(path));
    }
  }

  template <Adapter A> A walk(A document) const {
    for (auto const & token : tokens_) {
      document = std::visit([&document](auto const & next) { return document[next]; }, token);
    }
    return document;
  }

  Pointer parent() const { return Pointer({tokens_.begin(), tokens_.end() - 1}); }

  Pointer & operator/=(Pointer const & relative) {
    tokens_.insert(tokens_.end(), relative.tokens_.begin(), relative.tokens_.end());
    return *this;
  }

  Pointer operator/(Pointer const & relative) const { return Pointer(*this) /= relative; }

  Pointer & operator/=(std::string_view key) {
    tokens_.emplace_back(std::string(key));
    return *this;
  }

  Pointer operator/(std::string_view key) const { return Pointer(*this) /= key; }

  Pointer & operator/=(size_t index) {
    tokens_.emplace_back(index);
    return *this;
  }

  Pointer operator/(size_t index) const { return Pointer(*this) /= index; }

  friend std::ostream & operator<<(std::ostream & os, Pointer const & self) {
    for (auto const & elem : self.tokens_) {
      std::visit([&os](auto const & v) { os << '/' << v; }, elem);
    }
    return os;
  }
  auto operator<=>(Pointer const &) const = default;

private:
  std::vector<std::variant<std::string, size_t>> tokens_{};
};
}