add subquery snippets

Author: morganchen12

firestore-next/package.json

@@ -6,7 +6,7 @@
   },
   "license": "Apache-2.0",
   "dependencies": {
-    "firebase": "eap-firestore-pipelines",
+    "firebase": "^12.12.0",
     "geofire-common": "^6.0.0"
   },
   "devDependencies": {

firestore-next/test.firestore.js

@@ -1352,7 +1352,9 @@ describe("firestore-pipelines", () => {
         conditional,
         concat,
         like,
-        execute
+        execute,
+        variable,
+        equal
     } = require("firebase/firestore/pipelines");
 
     let app;
@@ -3394,4 +3396,63 @@ describe("firestore-pipelines", () => {
         // [END vector_length]
         console.log(result);
     }
+
+    // https://firebase.google.com/docs/firestore/pipelines/perform-joins-with-sub-pipelines
+    async function defineStage() {
+        // [START define_example]
+        const result = await execute(db.pipeline().collection("authors")
+            .define(
+                field("id").as("currentAuthorId")
+            )
+            // ...
+        // [END define_example]
+            .addFields(
+                db.pipeline().collection("books")
+                    .where(field("author_id").equal(variable("currentAuthorId")))
+                    .aggregate(
+                        field("rating").average().as("avgRating")
+                    )
+                    .toScalarExpression()
+                    .as("averageBookRating")
+            ));
+        console.log(result);
+    }
+
+    // https://firebase.google.com/docs/firestore/pipelines/perform-joins-with-sub-pipelines
+    async function toArrayExpressionStage() {
+        // [START to_array_expression]
+        const projectTasks = await execute(db.pipeline().collection("projects")
+            .define(
+                field("id").as("parentId")
+            )
+            .addFields(
+                db.pipeline().collection("tasks")
+                    .where(field("project_id").equal(variable("parentId")))
+                    .select(field("title"))
+                    .toArrayExpression()
+                    .as("taskTitles")
+            ));
+        // [END to_array_expression]
+        console.log(projectTasks);
+    }
+
+    // https://firebase.google.com/docs/firestore/pipelines/perform-joins-with-sub-pipelines
+    async function toScalarExpressionStage() {
+        // [START to_scalar_expression]
+        const result = await execute(db.pipeline().collection("authors")
+            .define(
+                field("id").as("currentAuthorId")
+            )
+            .addFields(
+                db.pipeline().collection("books")
+                    .where(field("author_id").equal(variable("currentAuthorId")))
+                    .aggregate(
+                        field("rating").average().as("avgRating")
+                    )
+                    .toScalarExpression()
+                    .as("averageBookRating")
+            ));
+        // [END to_scalar_expression]
+        console.log(result);
+    }
 });

snippets/firestore-next/test-firestore/define_example.js

@@ -0,0 +1,13 @@
+// This snippet file was generated by processing the source file:
+// ./firestore-next/test.firestore.js
+//
+// To update the snippets in this file, edit the source and then run
+// 'npm run snippets'.
+
+// [START define_example_modular]
+const result = await execute(db.pipeline().collection("authors")
+    .define(
+        field("id").as("currentAuthorId")
+    )
+    // ...
+// [END define_example_modular]

snippets/firestore-next/test-firestore/to_array_expression.js

@@ -0,0 +1,19 @@
+// This snippet file was generated by processing the source file:
+// ./firestore-next/test.firestore.js
+//
+// To update the snippets in this file, edit the source and then run
+// 'npm run snippets'.
+
+// [START to_array_expression_modular]
+const projectTasks = await execute(db.pipeline().collection("projects")
+    .define(
+        field("id").as("parentId")
+    )
+    .addFields(
+        db.pipeline().collection("tasks")
+            .where(field("project_id").equal(variable("parentId")))
+            .select(field("title"))
+            .toArrayExpression()
+            .as("taskTitles")
+    ));
+// [END to_array_expression_modular]

snippets/firestore-next/test-firestore/to_scalar_expression.js

@@ -0,0 +1,21 @@
+// This snippet file was generated by processing the source file:
+// ./firestore-next/test.firestore.js
+//
+// To update the snippets in this file, edit the source and then run
+// 'npm run snippets'.
+
+// [START to_scalar_expression_modular]
+const result = await execute(db.pipeline().collection("authors")
+    .define(
+        field("id").as("currentAuthorId")
+    )
+    .addFields(
+        db.pipeline().collection("books")
+            .where(field("author_id").equal(variable("currentAuthorId")))
+            .aggregate(
+                field("rating").average().as("avgRating")
+            )
+            .toScalarExpression()
+            .as("averageBookRating")
+    ));
+// [END to_scalar_expression_modular]