Initial (incomplete) implementation of f:f::core::Query (#1471)

3 files changed, 130 insertions, 0 deletions

diff --git a/Firestore/core/src/firebase/firestore/core/CMakeLists.txt b/Firestore/core/src/firebase/firestore/core/CMakeLists.txt
index cf3cafe..1aab09a 100644
--- a/Firestore/core/src/firebase/firestore/core/CMakeLists.txt
+++ b/Firestore/core/src/firebase/firestore/core/CMakeLists.txt
@@ -19,6 +19,8 @@ cc_library(
     database_info.h
     target_id_generator.cc
     target_id_generator.h
+    query.cc
+    query.h
   DEPENDS
     absl_strings
     firebase_firestore_model
diff --git a/Firestore/core/src/firebase/firestore/core/query.cc b/Firestore/core/src/firebase/firestore/core/query.cc
new file mode 100644
index 0000000..ad0f7a3
--- /dev/null
+++ b/Firestore/core/src/firebase/firestore/core/query.cc
@@ -0,0 +1,61 @@
+/*
+ * Copyright 2018 Google
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "Firestore/core/src/firebase/firestore/core/query.h"
+
+#include "Firestore/core/src/firebase/firestore/model/document_key.h"
+#include "Firestore/core/src/firebase/firestore/model/resource_path.h"
+
+namespace firebase {
+namespace firestore {
+namespace core {
+
+using model::Document;
+using model::DocumentKey;
+using model::ResourcePath;
+
+bool Query::Matches(const Document& doc) const {
+  return MatchesPath(doc) && MatchesOrderBy(doc) && MatchesFilters(doc) &&
+         MatchesBounds(doc);
+}
+
+bool Query::MatchesPath(const Document& doc) const {
+  ResourcePath doc_path = doc.key().path();
+  if (DocumentKey::IsDocumentKey(path_)) {
+    return path_ == doc_path;
+  } else {
+    return path_.IsPrefixOf(doc_path) && path_.size() == doc_path.size() - 1;
+  }
+}
+
+bool Query::MatchesFilters(const Document&) const {
+  // TODO(rsgowman): Implement this correctly.
+  return true;
+}
+
+bool Query::MatchesOrderBy(const Document&) const {
+  // TODO(rsgowman): Implement this correctly.
+  return true;
+}
+
+bool Query::MatchesBounds(const Document&) const {
+  // TODO(rsgowman): Implement this correctly.
+  return true;
+}
+
+}  // namespace core
+}  // namespace firestore
+}  // namespace firebase
diff --git a/Firestore/core/src/firebase/firestore/core/query.h b/Firestore/core/src/firebase/firestore/core/query.h
new file mode 100644
index 0000000..84d645d
--- /dev/null
+++ b/Firestore/core/src/firebase/firestore/core/query.h
@@ -0,0 +1,67 @@
+/*
+ * Copyright 2018 Google
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef FIRESTORE_CORE_SRC_FIREBASE_FIRESTORE_CORE_QUERY_H_
+#define FIRESTORE_CORE_SRC_FIREBASE_FIRESTORE_CORE_QUERY_H_
+
+#include <utility>
+
+#include "Firestore/core/src/firebase/firestore/model/document.h"
+#include "Firestore/core/src/firebase/firestore/model/resource_path.h"
+
+namespace firebase {
+namespace firestore {
+namespace core {
+
+/** Represents the internal structure of a Firestore Query. */
+class Query {
+ public:
+  /**
+   * Creates and returns a new Query.
+   *
+   * @param path The path to the collection to be queried over.
+   * @return A new instance of Query.
+   */
+  static Query AtPath(model::ResourcePath path) {
+    return Query(std::move(path));
+  }
+
+  /** Initializes a query with all of its components directly. */
+  explicit Query(model::ResourcePath path /* TODO(rsgowman): other params */)
+      : path_(std::move(path)) {
+  }
+
+  const model::ResourcePath& path() const {
+    return path_;
+  }
+
+  /** Returns true if the document matches the constraints of this query. */
+  bool Matches(const model::Document& doc) const;
+
+ private:
+  bool MatchesPath(const model::Document& doc) const;
+  bool MatchesFilters(const model::Document& doc) const;
+  bool MatchesOrderBy(const model::Document& doc) const;
+  bool MatchesBounds(const model::Document& doc) const;
+
+  const model::ResourcePath path_;
+};
+
+}  // namespace core
+}  // namespace firestore
+}  // namespace firebase
+
+#endif  // FIRESTORE_CORE_SRC_FIREBASE_FIRESTORE_CORE_QUERY_H_