LibreHub/include/binom/utils/node_visitor_base.h

#ifndef NODE_VISITOR_BASE_H
#define NODE_VISITOR_BASE_H

#include "../variables/variable.h"
#include "../utils/path.h"
#include "../utils/query.h"
#include <memory>
#include <vector>
#include <functional>
#include <optional>

namespace binom {

enum class VisitorType : ui8 {
  ram_storage_visitor = 0,
  file_storage_visitor = 1
};

class NodeVisitorBase {
public:
  static constexpr ui64 find_all = ui64(-1);
  virtual ~NodeVisitorBase() = default;

  virtual VarType getType() const = 0;
  virtual VisitorType getVisitorType() const = 0;
  VarTypeClass getTypeClass() const {return toTypeClass(getType());}
  virtual bool isNull() const = 0;
  bool isInvalid() const {return getTypeClass() == VarTypeClass::invalid_type;}
  bool isPrimitive() const {return getTypeClass() == VarTypeClass::primitive;}
  bool isBufferArray() const {return getTypeClass() == VarTypeClass::buffer_array;}
  bool isArray() const {return getTypeClass() == VarTypeClass::array;}
  bool isObject() const {return getTypeClass() == VarTypeClass::object;}
  bool isIterable() const  {return !isPrimitive();}
  virtual bool isValueRef() const = 0;
  virtual ui64 getElementCount() const = 0;

  virtual NodeVisitorBase& stepInside(ui64 index) = 0;
  virtual NodeVisitorBase& stepInside(BufferArray name) = 0;
  virtual NodeVisitorBase& stepInside(Path path) = 0;

  virtual bool contains(ui64 index) = 0;
  virtual bool contains(BufferArray name) = 0;
  virtual bool contains(Path path) = 0;

  virtual void setVariable(Variable var) = 0;
  virtual void setVariable(ui64 index, Variable var) = 0;
  virtual void setVariable(BufferArray name, Variable var) = 0;
  virtual void setVariable(Path path, Variable var) = 0;
  virtual void pushBack(Variable var) = 0;
  virtual void pushFront(Variable var) = 0;
  virtual void insert(ui64 index, Variable var) = 0;
  virtual void insert(BufferArray name, Variable var) = 0;
  virtual void remove(ui64 index, ui64 count = 1) = 0;
  virtual void remove(BufferArray name) = 0;
  virtual void remove(Path path) = 0;

  virtual NodeVisitorBase& operator()(ui64 index) = 0;
  virtual NodeVisitorBase& operator()(BufferArray name) = 0;
  virtual NodeVisitorBase& operator()(Path path) = 0;

  Variable getVariableClone();

  NodeVisitor& toRAMVisitor() {
    if(getVisitorType() != VisitorType::ram_storage_visitor)
      throw Exception(ErrCode::binom_invalid_visitor);
    return *reinterpret_cast<NodeVisitor*>(this);
  }

  FileNodeVisitor& toFileVisitor() {
    if(getVisitorType() != VisitorType::file_storage_visitor)
      throw Exception(ErrCode::binom_invalid_visitor);
    return *reinterpret_cast<FileNodeVisitor*>(this);
  }

 virtual operator Variable() = 0;

  NodeVisitorBase& upCast() {return *reinterpret_cast<NodeVisitorBase*>(this);}
};

typedef std::vector<std::unique_ptr<NodeVisitorBase>> NodeVector;

class TestExpressionFrame {
  QRel last_rel = QRel::AND;
  bool last_value = true;

  void invalidTest();
  void testUNumber(ui64 node_number, Query::QueryEpression& field);
  void testNumber(i64 node_number, Query::QueryEpression& field);;
  void testString(BufferArray node_string, Query::QueryEpression& field);;
  bool getRelationResult(bool last, QRel rel, bool current);

public:
  void operator()(Query::QueryEpression expr, FileNodeVisitor node, const ui64 index);
  void operator()(Query::QueryEpression expr, NodeVisitor node, const ui64 index);
  bool getValue() const {return last_value;}
};

}





#endif // NODE_VISITOR_BASE_H