Add type to DocumentReference/CollectionReference/Query

Author: schmidt-sebastian

packages/firestore/exp/index.d.ts

@@ -232,6 +232,7 @@ export type SetOptions =
 
 export class DocumentReference<T = DocumentData> {
   private constructor();
+  readonly type: 'document';
   readonly id: string;
   readonly firestore: FirebaseFirestore;
   readonly path: string;
@@ -272,6 +273,7 @@ export type WhereFilterOp =
 
 export class Query<T = DocumentData> {
   protected constructor();
+  readonly type: 'query' | 'collection';
   readonly firestore: FirebaseFirestore;
   where(
     fieldPath: string | FieldPath,
@@ -320,6 +322,7 @@ export interface DocumentChange<T = DocumentData> {
 
 export class CollectionReference<T = DocumentData> extends Query<T> {
   private constructor();
+  readonly type: 'collection';
   readonly id: string;
   readonly path: string;
   withConverter<U>(

packages/firestore/lite/index.d.ts

@@ -193,6 +193,7 @@ export type SetOptions =
 
 export class DocumentReference<T = DocumentData> {
   private constructor();
+  readonly type: 'document';
   readonly id: string;
   readonly firestore: FirebaseFirestore;
   readonly path: string;
@@ -227,6 +228,7 @@ export type WhereFilterOp =
 
 export class Query<T = DocumentData> {
   protected constructor();
+  readonly type: 'query' | 'collection';
   readonly firestore: FirebaseFirestore;
   where(
     fieldPath: string | FieldPath,
@@ -264,6 +266,7 @@ export class QuerySnapshot<T = DocumentData> {
 
 export class CollectionReference<T = DocumentData> extends Query<T> {
   private constructor();
+  readonly type: 'collection';
   readonly id: string;
   readonly path: string;
   withConverter<U>(

packages/firestore/lite/src/api/reference.ts

@@ -74,6 +74,8 @@ import { Code, FirestoreError } from '../../../src/util/error';
 export class DocumentReference<T = firestore.DocumentData>
   extends DocumentKeyReference<T>
   implements firestore.DocumentReference<T> {
+  readonly type = 'document';
+
   constructor(
     readonly firestore: Firestore,
     key: DocumentKey,
@@ -99,6 +101,8 @@ export class DocumentReference<T = firestore.DocumentData>
 
 export class Query<T = firestore.DocumentData> extends BaseQuery
   implements firestore.Query<T> {
+  readonly type: 'query' | 'collection' = 'query';
+
   // This is the lite version of the Query class in the main SDK.
   constructor(
     readonly firestore: Firestore,
@@ -256,6 +260,8 @@ export class Query<T = firestore.DocumentData> extends BaseQuery
 
 export class CollectionReference<T = firestore.DocumentData> extends Query<T>
   implements firestore.CollectionReference<T> {
+  readonly type = 'collection';
+
   constructor(
     readonly firestore: Firestore,
     readonly _path: ResourcePath,