Finish FieldValue migration

Author: schmidt-sebastian

packages/firestore/src/api/database.ts

@@ -17,6 +17,8 @@
 
 import * as firestore from '@firebase/firestore-types';
 
+import * as api from '../protos/firestore_proto_api';
+
 import { FirebaseApp } from '@firebase/app-types';
 import { FirebaseService, _FirebaseApp } from '@firebase/app-types/private';
 import { DatabaseId, DatabaseInfo } from '../core/database_info';
@@ -38,12 +40,6 @@ import { MemoryPersistenceProvider } from '../local/memory_persistence';
 import { PersistenceProvider } from '../local/persistence';
 import { Document, MaybeDocument, NoDocument } from '../model/document';
 import { DocumentKey } from '../model/document_key';
-import {
-  ArrayValue,
-  FieldValue,
-  RefValue,
-  ServerTimestampValue
-} from '../model/field_value';
 import { DeleteMutation, Mutation, Precondition } from '../model/mutation';
 import { FieldPath, ResourcePath } from '../model/path';
 import { PlatformSupport } from '../platform/platform';
@@ -101,6 +97,8 @@ import {
 import { UserDataWriter } from './user_data_writer';
 import { FirebaseAuthInternalName } from '@firebase/auth-interop-types';
 import { Provider } from '@firebase/component';
+import { isServerTimestamp } from '../model/server_timestamps';
+import { JsonProtoSerializer } from '../remote/serializer';
 
 // settings() defaults:
 const DEFAULT_HOST = 'firestore.googleapis.com';
@@ -555,7 +553,12 @@ export class Firestore implements firestore.FirebaseFirestore, FirebaseService {
         return value;
       }
     };
-    return new UserDataReader(preConverter);
+    return new UserDataReader(
+      new JsonProtoSerializer(databaseId, {
+        useProto3Json: PlatformSupport.getPlatform().useProto3Json
+      }),
+      preConverter
+    );
   }
 
   private static databaseIdFromApp(app: FirebaseApp): DatabaseId {
@@ -1357,14 +1360,20 @@ export interface SnapshotOptions extends firestore.SnapshotOptions {}
 
 export class DocumentSnapshot<T = firestore.DocumentData>
   implements firestore.DocumentSnapshot<T> {
+  private readonly _serializer: JsonProtoSerializer;
+
   constructor(
     private _firestore: Firestore,
     private _key: DocumentKey,
     public _document: Document | null,
     private _fromCache: boolean,
     private _hasPendingWrites: boolean,
     private readonly _converter?: firestore.FirestoreDataConverter<T>
-  ) {}
+  ) {
+    this._serializer = new JsonProtoSerializer(this._firestore._databaseId, {
+      useProto3Json: PlatformSupport.getPlatform().useProto3Json
+    });
+  }
 
   data(options?: firestore.SnapshotOptions): T | undefined {
     validateBetweenNumberOfArgs('DocumentSnapshot.data', arguments, 0, 1);
@@ -1386,11 +1395,13 @@ export class DocumentSnapshot<T = firestore.DocumentData>
       } else {
         const userDataWriter = new UserDataWriter(
           this._firestore,
+          this._serializer,
           this._firestore._areTimestampsInSnapshotsEnabled(),
           options.serverTimestamps,
           /* converter= */ undefined
         );
-        return userDataWriter.convertValue(this._document.data()) as T;
+        const proto = this._document.data().toProto();
+        return userDataWriter.convertValue(proto) as T;
       }
     }
   }
@@ -1408,6 +1419,7 @@ export class DocumentSnapshot<T = firestore.DocumentData>
       if (value !== null) {
         const userDataWriter = new UserDataWriter(
           this._firestore,
+          this._serializer,
           this._firestore._areTimestampsInSnapshotsEnabled(),
           options.serverTimestamps,
           this._converter
@@ -1495,7 +1507,7 @@ export class Query<T = firestore.DocumentData> implements firestore.Query<T> {
     ];
     validateStringEnum('Query.where', whereFilterOpEnums, 2, opStr);
 
-    let fieldValue: FieldValue;
+    let fieldValue: api.Value;
     const fieldPath = fieldPathFromArgument('Query.where', field);
     const operator = Operator.fromString(opStr);
     if (fieldPath.isKeyField()) {
@@ -1510,11 +1522,11 @@ export class Query<T = firestore.DocumentData> implements firestore.Query<T> {
         );
       } else if (operator === Operator.IN) {
         this.validateDisjunctiveFilterElements(value, operator);
-        const referenceList: FieldValue[] = [];
-        for (const arrayValue of value as FieldValue[]) {
+        const referenceList: api.Value[] = [];
+        for (const arrayValue of value as api.Value[]) {
           referenceList.push(this.parseDocumentIdValue(arrayValue));
         }
-        fieldValue = new ArrayValue(referenceList);
+        fieldValue = { arrayValue: { values: referenceList } };
       } else {
         fieldValue = this.parseDocumentIdValue(value);
       }
@@ -1720,7 +1732,7 @@ export class Query<T = firestore.DocumentData> implements firestore.Query<T> {
             `${methodName}().`
         );
       }
-      return this.boundFromDocument(methodName, snap._document!, before);
+      return this.boundFromDocument(snap._document!, before);
     } else {
       const allFields = [docOrField].concat(fields);
       return this.boundFromFields(methodName, allFields, before);
@@ -1738,12 +1750,8 @@ export class Query<T = firestore.DocumentData> implements firestore.Query<T> {
    * of the query or if any of the fields in the order by are an uncommitted
    * server timestamp.
    */
-  private boundFromDocument(
-    methodName: string,
-    doc: Document,
-    before: boolean
-  ): Bound {
-    const components: FieldValue[] = [];
+  private boundFromDocument(doc: Document, before: boolean): Bound {
+    const components: api.Value[] = [];
 
     // Because people expect to continue/end a query at the exact document
     // provided, we need to use the implicit sort order rather than the explicit
@@ -1754,10 +1762,12 @@ export class Query<T = firestore.DocumentData> implements firestore.Query<T> {
     // results.
     for (const orderBy of this._query.orderBy) {
       if (orderBy.field.isKeyField()) {
-        components.push(new RefValue(this.firestore._databaseId, doc.key));
+        components.push({
+          referenceValue: doc.key.toName(this.firestore._databaseId)
+        });
       } else {
         const value = doc.field(orderBy.field);
-        if (value instanceof ServerTimestampValue) {
+        if (isServerTimestamp(value)) {
           throw new FirestoreError(
             Code.INVALID_ARGUMENT,
             'Invalid query. You are trying to start or end a query using a ' +
@@ -1801,7 +1811,7 @@ export class Query<T = firestore.DocumentData> implements firestore.Query<T> {
       );
     }
 
-    const components: FieldValue[] = [];
+    const components: api.Value[] = [];
     for (let i = 0; i < values.length; i++) {
       const rawValue = values[i];
       const orderByComponent = orderBy[i];
@@ -1835,7 +1845,9 @@ export class Query<T = firestore.DocumentData> implements firestore.Query<T> {
           );
         }
         const key = new DocumentKey(path);
-        components.push(new RefValue(this.firestore._databaseId, key));
+        components.push({
+          referenceValue: key.toName(this.firestore._databaseId)
+        });
       } else {
         const wrapped = this.firestore._dataReader.parseQueryValue(
           methodName,
@@ -2034,7 +2046,7 @@ export class Query<T = firestore.DocumentData> implements firestore.Query<T> {
    * appropriate errors if the value is anything other than a DocumentReference
    * or String, or if the string is malformed.
    */
-  private parseDocumentIdValue(documentIdValue: unknown): RefValue {
+  private parseDocumentIdValue(documentIdValue: unknown): api.Value {
     if (typeof documentIdValue === 'string') {
       if (documentIdValue === '') {
         throw new FirestoreError(
@@ -2065,10 +2077,12 @@ export class Query<T = firestore.DocumentData> implements firestore.Query<T> {
             `but '${path}' is not because it has an odd number of segments (${path.length}).`
         );
       }
-      return new RefValue(this.firestore._databaseId, new DocumentKey(path));
+      return {
+        referenceValue: new DocumentKey(path).toName(this.firestore._databaseId)
+      };
     } else if (documentIdValue instanceof DocumentReference) {
       const ref = documentIdValue as DocumentReference<T>;
-      return new RefValue(this.firestore._databaseId, ref._key);
+      return { referenceValue: ref._key.toName(this.firestore._databaseId) };
     } else {
       throw new FirestoreError(
         Code.INVALID_ARGUMENT,