Cast Firestore

Author: MarkDuckworth

packages/firestore/src/api/pipeline.ts

@@ -6,6 +6,7 @@ import { Stage } from '../lite-api/stage';
 import { UserDataReader } from '../lite-api/user_data_reader';
 import { AbstractUserDataWriter } from '../lite-api/user_data_writer';
 import { DocumentKey } from '../model/document_key';
+import { cast } from '../util/input_validation';
 
 import { ensureFirestoreConfigured, Firestore } from './database';
 
@@ -23,7 +24,7 @@ export class Pipeline<
    * @param converter
    */
   constructor(
-    private db: Firestore,
+    db: Firestore,
     userDataReader: UserDataReader,
     userDataWriter: AbstractUserDataWriter,
     documentReferenceFactory: (id: DocumentKey) => DocumentReference,
@@ -74,7 +75,8 @@ export class Pipeline<
    * @return A Promise representing the asynchronous pipeline execution.
    */
   execute(): Promise<Array<PipelineResult<AppModelType>>> {
-    const client = ensureFirestoreConfigured(this.db);
+    const firestore = cast(this._db, Firestore);
+    const client = ensureFirestoreConfigured(firestore);
     return firestoreClientExecutePipeline(client, this).then(result => {
       const docs = result.map(
         element =>

packages/firestore/src/lite-api/pipeline.ts

@@ -124,15 +124,19 @@ export class Pipeline<AppModelType = DocumentData>
   /**
    * @internal
    * @private
-   * @param liteDb
+   * @param _db
    * @param userDataReader
    * @param userDataWriter
    * @param documentReferenceFactory
    * @param stages
    * @param converter
    */
   constructor(
-    private liteDb: Firestore,
+    /**
+     * @internal
+     * @private
+     */
+    public _db: Firestore,
     private userDataReader: UserDataReader,
     /**
      * @internal
@@ -184,7 +188,7 @@ export class Pipeline<AppModelType = DocumentData>
       )
     );
     return new Pipeline(
-      this.liteDb,
+      this._db,
       this.userDataReader,
       this.userDataWriter,
       this.documentReferenceFactory,
@@ -230,7 +234,7 @@ export class Pipeline<AppModelType = DocumentData>
     projections = this.readUserData('select', projections);
     copy.push(new Select(projections));
     return new Pipeline(
-      this.liteDb,
+      this._db,
       this.userDataReader,
       this.userDataWriter,
       this.documentReferenceFactory,
@@ -322,7 +326,7 @@ export class Pipeline<AppModelType = DocumentData>
     this.readUserData('where', condition);
     copy.push(new Where(condition));
     return new Pipeline(
-      this.liteDb,
+      this._db,
       this.userDataReader,
       this.userDataWriter,
       this.documentReferenceFactory,
@@ -355,7 +359,7 @@ export class Pipeline<AppModelType = DocumentData>
     const copy = this.stages.map(s => s);
     copy.push(new Offset(offset));
     return new Pipeline(
-      this.liteDb,
+      this._db,
       this.userDataReader,
       this.userDataWriter,
       this.documentReferenceFactory,
@@ -393,7 +397,7 @@ export class Pipeline<AppModelType = DocumentData>
     const copy = this.stages.map(s => s);
     copy.push(new Limit(limit));
     return new Pipeline(
-      this.liteDb,
+      this._db,
       this.userDataReader,
       this.userDataWriter,
       this.documentReferenceFactory,
@@ -438,7 +442,7 @@ export class Pipeline<AppModelType = DocumentData>
       )
     );
     return new Pipeline(
-      this.liteDb,
+      this._db,
       this.userDataReader,
       this.userDataWriter,
       this.documentReferenceFactory,
@@ -551,7 +555,7 @@ export class Pipeline<AppModelType = DocumentData>
       );
     }
     return new Pipeline(
-      this.liteDb,
+      this._db,
       this.userDataReader,
       this.userDataWriter,
       this.documentReferenceFactory,
@@ -579,7 +583,7 @@ export class Pipeline<AppModelType = DocumentData>
       )
     );
     return new Pipeline(
-      this.liteDb,
+      this._db,
       this.userDataReader,
       this.userDataWriter,
       this.documentReferenceFactory,
@@ -640,7 +644,7 @@ export class Pipeline<AppModelType = DocumentData>
     }
 
     return new Pipeline(
-      this.liteDb,
+      this._db,
       this.userDataReader,
       this.userDataWriter,
       this.documentReferenceFactory,
@@ -678,7 +682,7 @@ export class Pipeline<AppModelType = DocumentData>
     });
     copy.push(new GenericStage(name, params));
     return new Pipeline(
-      this.liteDb,
+      this._db,
       this.userDataReader,
       this.userDataWriter,
       this.documentReferenceFactory,
@@ -785,7 +789,7 @@ export class Pipeline<AppModelType = DocumentData>
    * @return A Promise representing the asynchronous pipeline execution.
    */
   execute(): Promise<Array<PipelineResult<AppModelType>>> {
-    const datastore = getDatastore(this.liteDb);
+    const datastore = getDatastore(this._db);
     return invokeExecutePipeline(datastore, this).then(result => {
       const docs = result
         // Currently ignore any response from ExecutePipeline that does