add snippets for search

Author: morganchen12

firestore/app/src/main/java/com/google/example/firestore/DocSnippets.java

@@ -62,11 +62,14 @@
 import com.google.firebase.firestore.pipeline.Expression;
 import com.google.firebase.firestore.pipeline.FindNearestStage;
 import com.google.firebase.firestore.pipeline.SampleStage;
+import com.google.firebase.firestore.pipeline.SearchStage;
 import com.google.firebase.firestore.pipeline.UnnestOptions;
 
 import static com.google.firebase.firestore.pipeline.Expression.field;
 import static com.google.firebase.firestore.pipeline.Expression.constant;
 import static com.google.firebase.firestore.pipeline.Expression.variable;
+import static com.google.firebase.firestore.pipeline.Expression.documentMatches;
+import static com.google.firebase.firestore.pipeline.Expression.score;
 import static com.google.firebase.firestore.pipeline.AggregateFunction.average;
 import static com.google.firebase.firestore.pipeline.AggregateFunction.countAll;
 
@@ -4048,4 +4051,41 @@ void toScalarExpressionStage() {
         // [END to_scalar_expression]
     }
 
+    void searchBasicQuery() {
+        // [START search_basic_query]
+        Pipeline pipeline = db.pipeline().collection("restaurants")
+                .search(SearchStage.withQuery(documentMatches("waffles")));
+        // [END search_basic_query]
+    }
+
+    void searchExactMatch() {
+        // [START search_exact_match]
+        Pipeline pipeline = db.pipeline().collection("restaurants")
+                .search(SearchStage.withQuery(documentMatches("\"belgian waffles\"")));
+        // [END search_exact_match]
+    }
+
+    void searchTwoTerms() {
+        // [START search_two_terms]
+        Pipeline pipeline = db.pipeline().collection("restaurants")
+                .search(SearchStage.withQuery(documentMatches("waffles eggs")));
+        // [END search_two_terms]
+    }
+
+    void searchExcludeTerm() {
+        // [START search_exclude_term]
+        Pipeline pipeline = db.pipeline().collection("restaurants")
+                .search(SearchStage.withQuery(documentMatches("coffee -waffles")));
+        // [END search_exclude_term]
+    }
+
+    void searchAddScore() {
+        // [START search_add_score]
+        Pipeline pipeline = db.pipeline().collection("restaurants")
+                .search(
+                        SearchStage.withQuery(documentMatches("menu:waffles"))
+                                .withAddFields(score().alias("score")));
+        // [END search_add_score]
+    }
+
 }

firestore/app/src/main/java/com/google/example/firestore/kotlin/DocSnippets.kt

@@ -32,11 +32,14 @@ import com.google.firebase.firestore.pipeline.AggregateFunction.Companion.countA
 import com.google.firebase.firestore.pipeline.Expression.Companion.constant
 import com.google.firebase.firestore.pipeline.Expression.Companion.field
 import com.google.firebase.firestore.pipeline.Expression.Companion.concat
+import com.google.firebase.firestore.pipeline.Expression.Companion.documentMatches
 import com.google.firebase.firestore.pipeline.Expression.Companion.length
+import com.google.firebase.firestore.pipeline.Expression.Companion.score
 import com.google.firebase.firestore.pipeline.Expression.Companion.type
 import com.google.firebase.firestore.pipeline.Expression.Companion.variable
 import com.google.firebase.firestore.pipeline.FindNearestStage
 import com.google.firebase.firestore.pipeline.SampleStage
+import com.google.firebase.firestore.pipeline.SearchStage
 import com.google.firebase.firestore.pipeline.UnnestOptions
 import com.google.firebase.firestore.toObject
 import java.util.Date
@@ -3656,4 +3659,42 @@ abstract class DocSnippets(val db: FirebaseFirestore) {
         // [END to_scalar_expression]
     }
 
+    fun searchBasicQuery() {
+        // [START search_basic_query]
+        val pipeline = db.pipeline().collection("restaurants")
+            .search(SearchStage.withQuery(documentMatches("waffles")))
+        // [END search_basic_query]
+    }
+
+    fun searchExactMatch() {
+        // [START search_exact_match]
+        val pipeline = db.pipeline().collection("restaurants")
+            .search(SearchStage.withQuery(documentMatches("\"belgian waffles\"")))
+        // [END search_exact_match]
+    }
+
+    fun searchTwoTerms() {
+        // [START search_two_terms]
+        val pipeline = db.pipeline().collection("restaurants")
+            .search(SearchStage.withQuery(documentMatches("waffles eggs")))
+        // [END search_two_terms]
+    }
+
+    fun searchExcludeTerm() {
+        // [START search_exclude_term]
+        val pipeline = db.pipeline().collection("restaurants")
+            .search(SearchStage.withQuery(documentMatches("coffee -waffles")))
+        // [END search_exclude_term]
+    }
+
+    fun searchAddScore() {
+        // [START search_add_score]
+        val pipeline = db.pipeline().collection("restaurants")
+            .search(
+                SearchStage.withQuery(documentMatches("menu:waffles"))
+                    .withAddFields(score().alias("score"))
+            )
+        // [END search_add_score]
+    }
+
 }