Add ArrayContainsFilter

Author: wilhuff

Firestore/core/src/firebase/firestore/core/CMakeLists.txt

@@ -15,6 +15,8 @@
 cc_library(
   firebase_firestore_core
   SOURCES
+    array_contains_filter.cc
+    array_contains_filter.h
     database_info.cc
     database_info.h
     direction.cc

Firestore/core/src/firebase/firestore/core/array_contains_filter.cc

@@ -0,0 +1,50 @@
+/*
+ * Copyright 2019 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/array_contains_filter.h"
+
+#include <utility>
+
+#include "absl/algorithm/container.h"
+
+namespace firebase {
+namespace firestore {
+namespace core {
+
+using model::Document;
+using model::FieldPath;
+using model::FieldValue;
+
+using Operator = Filter::Operator;
+
+ArrayContainsFilter::ArrayContainsFilter(FieldPath field, FieldValue value)
+    : FieldFilter(std::move(field), Operator::ArrayContains, std::move(value)) {
+}
+
+bool ArrayContainsFilter::Matches(const Document& doc) const {
+  absl::optional<FieldValue> maybe_lhs = doc.field(field());
+  if (!maybe_lhs) return false;
+
+  const FieldValue& lhs = *maybe_lhs;
+  if (lhs.type() != FieldValue::Type::Array) return false;
+
+  const FieldValue::Array& contents = lhs.array_value();
+  return absl::c_linear_search(contents, value());
+}
+
+}  // namespace core
+}  // namespace firestore
+}  // namespace firebase

Firestore/core/src/firebase/firestore/core/array_contains_filter.h

@@ -0,0 +1,45 @@
+/*
+ * Copyright 2019 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_ARRAY_CONTAINS_FILTER_H_
+#define FIRESTORE_CORE_SRC_FIREBASE_FIRESTORE_CORE_ARRAY_CONTAINS_FILTER_H_
+
+#include <string>
+
+#include "Firestore/core/src/firebase/firestore/core/field_filter.h"
+#include "Firestore/core/src/firebase/firestore/model/field_path.h"
+
+namespace firebase {
+namespace firestore {
+namespace core {
+
+/**
+ * A Filter that implements the array-contains operator.
+ */
+class ArrayContainsFilter : public FieldFilter {
+ public:
+  ArrayContainsFilter() = default;
+
+  ArrayContainsFilter(model::FieldPath field, model::FieldValue value);
+
+  bool Matches(const model::Document& doc) const override;
+};
+
+}  // namespace core
+}  // namespace firestore
+}  // namespace firebase
+
+#endif  // FIRESTORE_CORE_SRC_FIREBASE_FIRESTORE_CORE_ARRAY_CONTAINS_FILTER_H_