Implement bundle features in local store.

packages/firestore/src/core/bundle.ts

@@ -17,6 +17,12 @@
 
 import { Query } from './query';
 import { SnapshotVersion } from './snapshot_version';
+import { JsonProtoSerializer } from '../remote/serializer';
+import * as bundleProto from '../protos/firestore_bundle_proto';
+import * as api from '../protos/firestore_proto_api';
+import { DocumentKey } from '../model/document_key';
+import { MaybeDocument, NoDocument } from '../model/document';
+import { debugAssert } from '../util/assert';
 
 /**
  * Represents a Firestore bundle saved by the SDK in its local storage.
@@ -40,3 +46,43 @@ export interface NamedQuery {
   /** The time at which the results for this query were read. */
   readonly readTime: SnapshotVersion;
 }
+
+/**
+ * Represents the [metadata, document] pairs from the bundles in an array.
+ */
+export type BundledDocuments = Array<
+  [bundleProto.BundledDocumentMetadata, api.Document | undefined]
+>;
+
+/**
+ * Helper to convert objects from bundles to model objects in the SDK.
+ */
+export class BundleConverter {
+  constructor(private serializer: JsonProtoSerializer) {}
+
+  toDocumentKey(name: string): DocumentKey {
+    return this.serializer.fromName(name);
+  }
+
+  /**
+   * Converts a [metadata, document] pair to a MaybeDocument.
+   */
+  toMaybeDocument(
+    metadata: bundleProto.BundledDocumentMetadata,
+    doc: api.Document | undefined
+  ): MaybeDocument {
+    if (metadata.exists) {
+      debugAssert(!!doc, 'Document is undefined when metadata.exist is true.');
+      return this.serializer.fromDocument(doc!, false);
+    } else {
+      return new NoDocument(
+        this.toDocumentKey(metadata.name!),
+        this.toSnapshotVersion(metadata.readTime!)
+      );
+    }
+  }
+
+  toSnapshotVersion(time: api.Timestamp): SnapshotVersion {
+    return this.serializer.fromVersion(time);
+  }
+}

packages/firestore/src/core/component_provider.ts

@@ -42,6 +42,7 @@ import {
   MemoryEagerDelegate,
   MemoryPersistence
 } from '../local/memory_persistence';
+import { BundleConverter } from './bundle';
 
 const MEMORY_ONLY_PERSISTENCE_ERROR_MESSAGE =
   'You are using the memory-only build of Firestore. Persistence support is ' +
@@ -125,10 +126,12 @@ export class MemoryComponentProvider implements ComponentProvider {
   }
 
   createLocalStore(cfg: ComponentConfiguration): LocalStore {
+    const serializer = cfg.platform.newSerializer(cfg.databaseInfo.databaseId);
     return new LocalStore(
       this.persistence,
       new IndexFreeQueryEngine(),
-      cfg.initialUser
+      cfg.initialUser,
+      new BundleConverter(serializer)
     );
   }
 
@@ -209,10 +212,12 @@ export class IndexedDbComponentProvider extends MemoryComponentProvider {
   }
 
   createLocalStore(cfg: ComponentConfiguration): LocalStore {
+    const serializer = cfg.platform.newSerializer(cfg.databaseInfo.databaseId);
     return new MultiTabLocalStore(
       this.persistence,
       new IndexFreeQueryEngine(),
-      cfg.initialUser
+      cfg.initialUser,
+      new BundleConverter(serializer)
     );
   }
 

packages/firestore/src/local/indexeddb_remote_document_cache.ts

@@ -445,20 +445,20 @@ export class IndexedDbRemoteDocumentCache implements RemoteDocumentCache {
         primitiveComparator(l.canonicalString(), r.canonicalString())
       );
 
-      this.changes.forEach((key, maybeDocument) => {
+      this.changes.forEach((key, documentChange) => {
         const previousSize = this.documentSizes.get(key);
         debugAssert(
           previousSize !== undefined,
           `Cannot modify a document that wasn't read (for ${key})`
         );
-        if (maybeDocument) {
+        if (documentChange.maybeDoc) {
           debugAssert(
-            !this.readTime.isEqual(SnapshotVersion.min()),
+            !this.getReadTime(key).isEqual(SnapshotVersion.min()),
             'Cannot add a document with a read time of zero'
           );
           const doc = this.documentCache.serializer.toDbRemoteDocument(
-            maybeDocument,
-            this.readTime
+            documentChange.maybeDoc!,
+            this.getReadTime(key)
           );
           collectionParents = collectionParents.add(key.path.popLast());
 
@@ -474,7 +474,7 @@ export class IndexedDbRemoteDocumentCache implements RemoteDocumentCache {
             // preserved in `getNewDocumentChanges()`.
             const deletedDoc = this.documentCache.serializer.toDbRemoteDocument(
               new NoDocument(key, SnapshotVersion.min()),
-              this.readTime
+              this.getReadTime(key)
             );
             promises.push(
               this.documentCache.addEntry(transaction, key, deletedDoc)