#include <iostream>
#include <string>

#include "query/plan_interface.hpp"
#include "storage/edge_accessor.hpp"
#include "storage/vertex_accessor.hpp"
#include "using.hpp"

using std::cout;
using std::endl;

// Query: MATCH (p:profile {profile_id: 112, partner_id: 77}) RETURN p

class CPUPlan : public PlanInterface<Stream> {
 public:
  bool run(GraphDbAccessor &db_accessor, const TypedValueStore<> &args,
           Stream &stream) {
    stream.write_field("p");
    for (auto vertex : db_accessor.vertices()) {
      if (vertex.has_label(db_accessor.label("profile"))) {
        auto prop = vertex.PropsAt(db_accessor.property("profile_id"));
        if (prop.type_ == TypedValue::Type::Null) continue;
        auto cmp = prop == args.at(0);
        if (cmp.type_ != TypedValue::Type::Bool) continue;
        if (cmp.Value<bool>() != true) continue;

        auto prop2 = vertex.PropsAt(db_accessor.property("partner_id"));
        if (prop2.type_ == TypedValue::Type::Null) continue;
        auto cmp2 = prop2 == args.at(1);
        if (cmp2.type_ != TypedValue::Type::Bool) continue;
        if (cmp2.Value<bool>() != true) continue;
        stream.write_vertex_record(vertex);
      }
    }
    stream.write_meta("r");
    db_accessor.transaction_.commit();
    return true;
  }
  ~CPUPlan() {}
};

extern "C" PlanInterface<Stream> *produce() { return new CPUPlan(); }

extern "C" void destruct(PlanInterface<Stream> *p) { delete p; }