1 files changed, 213 insertions, 0 deletions
diff --git a/Firestore/Source/Local/FSTLocalSerializer.mm b/Firestore/Source/Local/FSTLocalSerializer.mm
new file mode 100644
index 0000000..c531c77
--- /dev/null
+++ b/Firestore/Source/Local/FSTLocalSerializer.mm
@@ -0,0 +1,213 @@
+/*
+ * Copyright 2017 Google
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#import "Firestore/Source/Local/FSTLocalSerializer.h"
+
+#include <inttypes.h>
+
+#import "Firestore/Protos/objc/firestore/local/MaybeDocument.pbobjc.h"
+#import "Firestore/Protos/objc/firestore/local/Mutation.pbobjc.h"
+#import "Firestore/Protos/objc/firestore/local/Target.pbobjc.h"
+#import "Firestore/Protos/objc/google/firestore/v1beta1/Document.pbobjc.h"
+#import "Firestore/Source/Core/FSTQuery.h"
+#import "Firestore/Source/Local/FSTQueryData.h"
+#import "Firestore/Source/Model/FSTDocument.h"
+#import "Firestore/Source/Model/FSTFieldValue.h"
+#import "Firestore/Source/Model/FSTMutationBatch.h"
+#import "Firestore/Source/Remote/FSTSerializerBeta.h"
+#import "Firestore/Source/Util/FSTAssert.h"
+
+@interface FSTLocalSerializer ()
+
+@property(nonatomic, strong, readonly) FSTSerializerBeta *remoteSerializer;
+
+@end
+
+/** Serializer for values stored in the LocalStore. */
+@implementation FSTLocalSerializer
+
+- (instancetype)initWithRemoteSerializer:(FSTSerializerBeta *)remoteSerializer {
+  self = [super init];
+  if (self) {
+    _remoteSerializer = remoteSerializer;
+  }
+  return self;
+}
+
+- (FSTPBMaybeDocument *)encodedMaybeDocument:(FSTMaybeDocument *)document {
+  FSTPBMaybeDocument *proto = [FSTPBMaybeDocument message];
+
+  if ([document isKindOfClass:[FSTDeletedDocument class]]) {
+    proto.noDocument = [self encodedDeletedDocument:(FSTDeletedDocument *)document];
+  } else if ([document isKindOfClass:[FSTDocument class]]) {
+    proto.document = [self encodedDocument:(FSTDocument *)document];
+  } else {
+    FSTFail(@"Unknown document type %@", NSStringFromClass([document class]));
+  }
+
+  return proto;
+}
+
+- (FSTMaybeDocument *)decodedMaybeDocument:(FSTPBMaybeDocument *)proto {
+  switch (proto.documentTypeOneOfCase) {
+    case FSTPBMaybeDocument_DocumentType_OneOfCase_Document:
+      return [self decodedDocument:proto.document];
+
+    case FSTPBMaybeDocument_DocumentType_OneOfCase_NoDocument:
+      return [self decodedDeletedDocument:proto.noDocument];
+
+    default:
+      FSTFail(@"Unknown MaybeDocument %@", proto);
+  }
+}
+
+/**
+ * Encodes a Document for local storage. This differs from the v1beta1 RPC serializer for
+ * Documents in that it preserves the updateTime, which is considered an output only value by the
+ * server.
+ */
+- (GCFSDocument *)encodedDocument:(FSTDocument *)document {
+  FSTSerializerBeta *remoteSerializer = self.remoteSerializer;
+
+  GCFSDocument *proto = [GCFSDocument message];
+  proto.name = [remoteSerializer encodedDocumentKey:document.key];
+  proto.fields = [remoteSerializer encodedFields:document.data];
+  proto.updateTime = [remoteSerializer encodedVersion:document.version];
+
+  return proto;
+}
+
+/** Decodes a Document proto to the equivalent model. */
+- (FSTDocument *)decodedDocument:(GCFSDocument *)document {
+  FSTSerializerBeta *remoteSerializer = self.remoteSerializer;
+
+  FSTObjectValue *data = [remoteSerializer decodedFields:document.fields];
+  FSTDocumentKey *key = [remoteSerializer decodedDocumentKey:document.name];
+  FSTSnapshotVersion *version = [remoteSerializer decodedVersion:document.updateTime];
+  return [FSTDocument documentWithData:data key:key version:version hasLocalMutations:NO];
+}
+
+/** Encodes a NoDocument value to the equivalent proto. */
+- (FSTPBNoDocument *)encodedDeletedDocument:(FSTDeletedDocument *)document {
+  FSTSerializerBeta *remoteSerializer = self.remoteSerializer;
+
+  FSTPBNoDocument *proto = [FSTPBNoDocument message];
+  proto.name = [remoteSerializer encodedDocumentKey:document.key];
+  proto.readTime = [remoteSerializer encodedVersion:document.version];
+  return proto;
+}
+
+/** Decodes a NoDocument proto to the equivalent model. */
+- (FSTDeletedDocument *)decodedDeletedDocument:(FSTPBNoDocument *)proto {
+  FSTSerializerBeta *remoteSerializer = self.remoteSerializer;
+
+  FSTDocumentKey *key = [remoteSerializer decodedDocumentKey:proto.name];
+  FSTSnapshotVersion *version = [remoteSerializer decodedVersion:proto.readTime];
+  return [FSTDeletedDocument documentWithKey:key version:version];
+}
+
+- (FSTPBWriteBatch *)encodedMutationBatch:(FSTMutationBatch *)batch {
+  FSTSerializerBeta *remoteSerializer = self.remoteSerializer;
+
+  FSTPBWriteBatch *proto = [FSTPBWriteBatch message];
+  proto.batchId = batch.batchID;
+  proto.localWriteTime = [remoteSerializer encodedTimestamp:batch.localWriteTime];
+
+  NSMutableArray<GCFSWrite *> *writes = proto.writesArray;
+  for (FSTMutation *mutation in batch.mutations) {
+    [writes addObject:[remoteSerializer encodedMutation:mutation]];
+  }
+  return proto;
+}
+
+- (FSTMutationBatch *)decodedMutationBatch:(FSTPBWriteBatch *)batch {
+  FSTSerializerBeta *remoteSerializer = self.remoteSerializer;
+
+  int batchID = batch.batchId;
+  NSMutableArray<FSTMutation *> *mutations = [NSMutableArray array];
+  for (GCFSWrite *write in batch.writesArray) {
+    [mutations addObject:[remoteSerializer decodedMutation:write]];
+  }
+
+  FSTTimestamp *localWriteTime = [remoteSerializer decodedTimestamp:batch.localWriteTime];
+
+  return [[FSTMutationBatch alloc] initWithBatchID:batchID
+                                    localWriteTime:localWriteTime
+                                         mutations:mutations];
+}
+
+- (FSTPBTarget *)encodedQueryData:(FSTQueryData *)queryData {
+  FSTSerializerBeta *remoteSerializer = self.remoteSerializer;
+
+  FSTAssert(queryData.purpose == FSTQueryPurposeListen,
+            @"only queries with purpose %lu may be stored, got %lu",
+            (unsigned long)FSTQueryPurposeListen, (unsigned long)queryData.purpose);
+
+  FSTPBTarget *proto = [FSTPBTarget message];
+  proto.targetId = queryData.targetID;
+  proto.lastListenSequenceNumber = queryData.sequenceNumber;
+  proto.snapshotVersion = [remoteSerializer encodedVersion:queryData.snapshotVersion];
+  proto.resumeToken = queryData.resumeToken;
+
+  FSTQuery *query = queryData.query;
+  if ([query isDocumentQuery]) {
+    proto.documents = [remoteSerializer encodedDocumentsTarget:query];
+  } else {
+    proto.query = [remoteSerializer encodedQueryTarget:query];
+  }
+
+  return proto;
+}
+
+- (FSTQueryData *)decodedQueryData:(FSTPBTarget *)target {
+  FSTSerializerBeta *remoteSerializer = self.remoteSerializer;
+
+  FSTTargetID targetID = target.targetId;
+  FSTListenSequenceNumber sequenceNumber = target.lastListenSequenceNumber;
+  FSTSnapshotVersion *version = [remoteSerializer decodedVersion:target.snapshotVersion];
+  NSData *resumeToken = target.resumeToken;
+
+  FSTQuery *query;
+  switch (target.targetTypeOneOfCase) {
+    case FSTPBTarget_TargetType_OneOfCase_Documents:
+      query = [remoteSerializer decodedQueryFromDocumentsTarget:target.documents];
+      break;
+
+    case FSTPBTarget_TargetType_OneOfCase_Query:
+      query = [remoteSerializer decodedQueryFromQueryTarget:target.query];
+      break;
+
+    default:
+      FSTFail(@"Unknown Target.targetType %" PRId32, target.targetTypeOneOfCase);
+  }
+
+  return [[FSTQueryData alloc] initWithQuery:query
+                                    targetID:targetID
+                        listenSequenceNumber:sequenceNumber
+                                     purpose:FSTQueryPurposeListen
+                             snapshotVersion:version
+                                 resumeToken:resumeToken];
+}
+
+- (GPBTimestamp *)encodedVersion:(FSTSnapshotVersion *)version {
+  return [self.remoteSerializer encodedVersion:version];
+}
+
+- (FSTSnapshotVersion *)decodedVersion:(GPBTimestamp *)version {
+  return [self.remoteSerializer decodedVersion:version];
+}
+
+@end