123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628 |
- /*
- * 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/API/FSTUserDataReader.h"
- #include <memory>
- #include <set>
- #include <string>
- #include <utility>
- #include <vector>
- #import "FIRGeoPoint.h"
- #import "FIRTimestamp.h"
- #import "Firestore/Source/API/FIRDocumentReference+Internal.h"
- #import "Firestore/Source/API/FIRFieldPath+Internal.h"
- #import "Firestore/Source/API/FIRFieldValue+Internal.h"
- #import "Firestore/Source/API/FIRFirestore+Internal.h"
- #import "Firestore/Source/API/FIRGeoPoint+Internal.h"
- #import "Firestore/Source/API/converters.h"
- #import "Firestore/core/include/firebase/firestore/geo_point.h"
- #include "Firestore/core/src/core/user_data.h"
- #include "Firestore/core/src/model/database_id.h"
- #include "Firestore/core/src/model/document_key.h"
- #include "Firestore/core/src/model/field_mask.h"
- #include "Firestore/core/src/model/field_path.h"
- #include "Firestore/core/src/model/field_transform.h"
- #include "Firestore/core/src/model/object_value.h"
- #include "Firestore/core/src/model/precondition.h"
- #include "Firestore/core/src/model/resource_path.h"
- #include "Firestore/core/src/model/transform_operation.h"
- #include "Firestore/core/src/model/value_util.h"
- #include "Firestore/core/src/nanopb/nanopb_util.h"
- #include "Firestore/core/src/nanopb/reader.h"
- #include "Firestore/core/src/remote/serializer.h"
- #include "Firestore/core/src/timestamp_internal.h"
- #include "Firestore/core/src/util/exception.h"
- #include "Firestore/core/src/util/hard_assert.h"
- #include "Firestore/core/src/util/read_context.h"
- #include "Firestore/core/src/util/string_apple.h"
- #include "absl/memory/memory.h"
- #include "absl/strings/match.h"
- #include "absl/types/optional.h"
- namespace nanopb = firebase::firestore::nanopb;
- using firebase::Timestamp;
- using firebase::TimestampInternal;
- using firebase::firestore::GeoPoint;
- using firebase::firestore::google_firestore_v1_ArrayValue;
- using firebase::firestore::google_firestore_v1_MapValue;
- using firebase::firestore::google_firestore_v1_MapValue_FieldsEntry;
- using firebase::firestore::google_firestore_v1_Value;
- using firebase::firestore::google_protobuf_NullValue_NULL_VALUE;
- using firebase::firestore::google_protobuf_Timestamp;
- using firebase::firestore::google_type_LatLng;
- using firebase::firestore::core::ParseAccumulator;
- using firebase::firestore::core::ParseContext;
- using firebase::firestore::core::ParsedSetData;
- using firebase::firestore::core::ParsedUpdateData;
- using firebase::firestore::core::UserDataSource;
- using firebase::firestore::model::ArrayTransform;
- using firebase::firestore::model::DatabaseId;
- using firebase::firestore::model::DeepClone;
- using firebase::firestore::model::DocumentKey;
- using firebase::firestore::model::FieldMask;
- using firebase::firestore::model::FieldPath;
- using firebase::firestore::model::FieldTransform;
- using firebase::firestore::model::NullValue;
- using firebase::firestore::model::NumericIncrementTransform;
- using firebase::firestore::model::ObjectValue;
- using firebase::firestore::model::ResourcePath;
- using firebase::firestore::model::ServerTimestampTransform;
- using firebase::firestore::model::TransformOperation;
- using firebase::firestore::nanopb::CheckedSize;
- using firebase::firestore::nanopb::Message;
- using firebase::firestore::remote::Serializer;
- using firebase::firestore::util::MakeString;
- using firebase::firestore::util::ReadContext;
- using firebase::firestore::util::ThrowInvalidArgument;
- using nanopb::StringReader;
- NS_ASSUME_NONNULL_BEGIN
- #pragma mark - FSTDocumentKeyReference
- @implementation FSTDocumentKeyReference {
- DocumentKey _key;
- DatabaseId _databaseID;
- }
- - (instancetype)initWithKey:(DocumentKey)key databaseID:(DatabaseId)databaseID {
- self = [super init];
- if (self) {
- _key = std::move(key);
- _databaseID = std::move(databaseID);
- }
- return self;
- }
- - (const model::DocumentKey &)key {
- return _key;
- }
- - (const model::DatabaseId &)databaseID {
- return _databaseID;
- }
- @end
- #pragma mark - FSTUserDataReader
- @interface FSTUserDataReader ()
- @property(strong, nonatomic, readonly) FSTPreConverterBlock preConverter;
- @end
- @implementation FSTUserDataReader {
- DatabaseId _databaseID;
- }
- - (instancetype)initWithDatabaseID:(DatabaseId)databaseID
- preConverter:(FSTPreConverterBlock)preConverter {
- self = [super init];
- if (self) {
- _databaseID = std::move(databaseID);
- _preConverter = preConverter;
- }
- return self;
- }
- - (ParsedSetData)parsedSetData:(id)input {
- // NOTE: The public API is typed as NSDictionary but we type 'input' as 'id' since we can't trust
- // Obj-C to verify the type for us.
- if (![input isKindOfClass:[NSDictionary class]]) {
- ThrowInvalidArgument("Data to be written must be an NSDictionary.");
- }
- ParseAccumulator accumulator{UserDataSource::Set};
- auto updateData = [self parseData:input context:accumulator.RootContext()];
- HARD_ASSERT(updateData.has_value(), "Parsed data should not be nil.");
- return std::move(accumulator).SetData(ObjectValue{std::move(*updateData)});
- }
- - (ParsedSetData)parsedMergeData:(id)input fieldMask:(nullable NSArray<id> *)fieldMask {
- // NOTE: The public API is typed as NSDictionary but we type 'input' as 'id' since we can't trust
- // Obj-C to verify the type for us.
- if (![input isKindOfClass:[NSDictionary class]]) {
- ThrowInvalidArgument("Data to be written must be an NSDictionary.");
- }
- ParseAccumulator accumulator{UserDataSource::MergeSet};
- auto updateData = [self parseData:input context:accumulator.RootContext()];
- HARD_ASSERT(updateData.has_value(), "Parsed data should not be nil.");
- ObjectValue updateObject{std::move(*updateData)};
- if (fieldMask) {
- std::set<FieldPath> validatedFieldPaths;
- for (id fieldPath in fieldMask) {
- FieldPath path;
- if ([fieldPath isKindOfClass:[NSString class]]) {
- path = FieldPath::FromDotSeparatedString(MakeString(fieldPath));
- } else if ([fieldPath isKindOfClass:[FIRFieldPath class]]) {
- path = static_cast<FIRFieldPath *>(fieldPath).internalValue;
- } else {
- ThrowInvalidArgument("All elements in mergeFields: must be NSStrings or FIRFieldPaths.");
- }
- // Verify that all elements specified in the field mask are part of the parsed context.
- if (!accumulator.Contains(path)) {
- ThrowInvalidArgument(
- "Field '%s' is specified in your field mask but missing from your input data.",
- path.CanonicalString());
- }
- validatedFieldPaths.insert(path);
- }
- return std::move(accumulator)
- .MergeData(std::move(updateObject), FieldMask{std::move(validatedFieldPaths)});
- } else {
- return std::move(accumulator).MergeData(std::move(updateObject));
- }
- }
- - (ParsedUpdateData)parsedUpdateData:(id)input {
- // NOTE: The public API is typed as NSDictionary but we type 'input' as 'id' since we can't trust
- // Obj-C to verify the type for us.
- if (![input isKindOfClass:[NSDictionary class]]) {
- ThrowInvalidArgument("Data to be written must be an NSDictionary.");
- }
- NSDictionary *dict = input;
- ParseAccumulator accumulator{UserDataSource::Update};
- __block ParseContext context = accumulator.RootContext();
- __block ObjectValue updateData;
- [dict enumerateKeysAndObjectsUsingBlock:^(id key, id value, BOOL *) {
- FieldPath path;
- if ([key isKindOfClass:[NSString class]]) {
- path = FieldPath::FromDotSeparatedString(MakeString(key));
- } else if ([key isKindOfClass:[FIRFieldPath class]]) {
- path = ((FIRFieldPath *)key).internalValue;
- } else {
- ThrowInvalidArgument("Dictionary keys in updateData: must be NSStrings or FIRFieldPaths.");
- }
- value = self.preConverter(value);
- if ([value isKindOfClass:[FSTDeleteFieldValue class]]) {
- // Add it to the field mask, but don't add anything to updateData.
- context.AddToFieldMask(std::move(path));
- } else {
- auto parsedValue = [self parseData:value context:context.ChildContext(path)];
- if (parsedValue) {
- context.AddToFieldMask(path);
- updateData.Set(path, std::move(*parsedValue));
- }
- }
- }];
- return std::move(accumulator).UpdateData(std::move(updateData));
- }
- - (Message<google_firestore_v1_Value>)parsedQueryValue:(id)input {
- return [self parsedQueryValue:input allowArrays:false];
- }
- - (Message<google_firestore_v1_Value>)parsedQueryValue:(id)input allowArrays:(bool)allowArrays {
- ParseAccumulator accumulator{allowArrays ? UserDataSource::ArrayArgument
- : UserDataSource::Argument};
- auto parsed = [self parseData:input context:accumulator.RootContext()];
- HARD_ASSERT(parsed, "Parsed data should not be nil.");
- HARD_ASSERT(accumulator.field_transforms().empty(),
- "Field transforms should have been disallowed.");
- return std::move(*parsed);
- }
- /**
- * Internal helper for parsing user data.
- *
- * @param input Data to be parsed.
- * @param context A context object representing the current path being parsed, the source of the
- * data being parsed, etc.
- *
- * @return The parsed value, or nil if the value was a FieldValue sentinel that should not be
- * included in the resulting parsed data.
- */
- - (absl::optional<Message<google_firestore_v1_Value>>)parseData:(id)input
- context:(ParseContext &&)context {
- input = self.preConverter(input);
- if ([input isKindOfClass:[NSDictionary class]]) {
- return [self parseDictionary:(NSDictionary *)input context:std::move(context)];
- } else if ([input isKindOfClass:[FIRFieldValue class]]) {
- // FieldValues usually parse into transforms (except FieldValue.delete()) in which case we
- // do not want to include this field in our parsed data (as doing so will overwrite the field
- // directly prior to the transform trying to transform it). So we don't call appendToFieldMask
- // and we return nil as our parsing result.
- [self parseSentinelFieldValue:(FIRFieldValue *)input context:std::move(context)];
- return absl::nullopt;
- } else {
- // If context path is unset we are already inside an array and we don't support field mask paths
- // more granular than the top-level array.
- if (context.path()) {
- context.AddToFieldMask(*context.path());
- }
- if ([input isKindOfClass:[NSArray class]]) {
- // TODO(b/34871131): Include the path containing the array in the error message.
- // In the case of IN queries, the parsed data is an array (representing the set of values to
- // be included for the IN query) that may directly contain additional arrays (each
- // representing an individual field value), so we disable this validation.
- if (context.array_element() && context.data_source() != UserDataSource::ArrayArgument) {
- ThrowInvalidArgument("Nested arrays are not supported");
- }
- return [self parseArray:(NSArray *)input context:std::move(context)];
- } else {
- return [self parseScalarValue:input context:std::move(context)];
- }
- }
- }
- - (Message<google_firestore_v1_Value>)parseDictionary:(NSDictionary<NSString *, id> *)dict
- context:(ParseContext &&)context {
- __block Message<google_firestore_v1_Value> result;
- result->which_value_type = google_firestore_v1_Value_map_value_tag;
- result->map_value = {};
- if (dict.count == 0) {
- const FieldPath *path = context.path();
- if (path && !path->empty()) {
- context.AddToFieldMask(*path);
- }
- } else {
- // Compute the final size of the fields array, which contains an entry for
- // all fields that are not FieldValue sentinels
- __block pb_size_t count = 0;
- [dict enumerateKeysAndObjectsUsingBlock:^(NSString *, id value, BOOL *) {
- if (![value isKindOfClass:[FIRFieldValue class]]) {
- ++count;
- }
- }];
- result->map_value.fields_count = count;
- result->map_value.fields = nanopb::MakeArray<google_firestore_v1_MapValue_FieldsEntry>(count);
- __block pb_size_t index = 0;
- [dict enumerateKeysAndObjectsUsingBlock:^(NSString *key, id value, BOOL *) {
- auto parsedValue = [self parseData:value context:context.ChildContext(MakeString(key))];
- if (parsedValue) {
- result->map_value.fields[index].key = nanopb::MakeBytesArray(MakeString(key));
- result->map_value.fields[index].value = *parsedValue->release();
- ++index;
- }
- }];
- }
- return std::move(result);
- }
- - (Message<google_firestore_v1_Value>)parseArray:(NSArray<id> *)array
- context:(ParseContext &&)context {
- __block Message<google_firestore_v1_Value> result;
- result->which_value_type = google_firestore_v1_Value_array_value_tag;
- result->array_value.values_count = CheckedSize([array count]);
- result->array_value.values =
- nanopb::MakeArray<google_firestore_v1_Value>(result->array_value.values_count);
- [array enumerateObjectsUsingBlock:^(id entry, NSUInteger idx, BOOL *) {
- auto parsedEntry = [self parseData:entry context:context.ChildContext(idx)];
- if (!parsedEntry) {
- // Just include nulls in the array for fields being replaced with a sentinel.
- parsedEntry.emplace(DeepClone(NullValue()));
- }
- result->array_value.values[idx] = *parsedEntry->release();
- }];
- return std::move(result);
- }
- /**
- * "Parses" the provided FIRFieldValue, adding any necessary transforms to
- * context.fieldTransforms.
- */
- - (void)parseSentinelFieldValue:(FIRFieldValue *)fieldValue context:(ParseContext &&)context {
- // Sentinels are only supported with writes, and not within arrays.
- if (!context.write()) {
- ThrowInvalidArgument("%s can only be used with updateData() and setData()%s",
- fieldValue.methodName, context.FieldDescription());
- }
- if (!context.path()) {
- ThrowInvalidArgument("%s is not currently supported inside arrays", fieldValue.methodName);
- }
- if ([fieldValue isKindOfClass:[FSTDeleteFieldValue class]]) {
- if (context.data_source() == UserDataSource::MergeSet) {
- // No transform to add for a delete, but we need to add it to our fieldMask so it gets
- // deleted.
- context.AddToFieldMask(*context.path());
- } else if (context.data_source() == UserDataSource::Update) {
- HARD_ASSERT(!context.path()->empty(),
- "FieldValue.delete() at the top level should have already been handled.");
- ThrowInvalidArgument("FieldValue.delete() can only appear at the top level of your "
- "update data%s",
- context.FieldDescription());
- } else {
- // We shouldn't encounter delete sentinels for queries or non-merge setData calls.
- ThrowInvalidArgument(
- "FieldValue.delete() can only be used with updateData() and setData() with merge:true%s",
- context.FieldDescription());
- }
- } else if ([fieldValue isKindOfClass:[FSTServerTimestampFieldValue class]]) {
- context.AddToFieldTransforms(*context.path(), ServerTimestampTransform());
- } else if ([fieldValue isKindOfClass:[FSTArrayUnionFieldValue class]]) {
- auto parsedElements =
- [self parseArrayTransformElements:((FSTArrayUnionFieldValue *)fieldValue).elements];
- ArrayTransform arrayUnion(TransformOperation::Type::ArrayUnion, std::move(parsedElements));
- context.AddToFieldTransforms(*context.path(), std::move(arrayUnion));
- } else if ([fieldValue isKindOfClass:[FSTArrayRemoveFieldValue class]]) {
- auto parsedElements =
- [self parseArrayTransformElements:((FSTArrayRemoveFieldValue *)fieldValue).elements];
- ArrayTransform arrayRemove(TransformOperation::Type::ArrayRemove, std::move(parsedElements));
- context.AddToFieldTransforms(*context.path(), std::move(arrayRemove));
- } else if ([fieldValue isKindOfClass:[FSTNumericIncrementFieldValue class]]) {
- auto *numericIncrementFieldValue = (FSTNumericIncrementFieldValue *)fieldValue;
- auto operand = [self parsedQueryValue:numericIncrementFieldValue.operand];
- NumericIncrementTransform numeric_increment(std::move(operand));
- context.AddToFieldTransforms(*context.path(), std::move(numeric_increment));
- } else {
- HARD_FAIL("Unknown FIRFieldValue type: %s", NSStringFromClass([fieldValue class]));
- }
- }
- /**
- * Helper to parse a scalar value (i.e. not an NSDictionary, NSArray, or FIRFieldValue).
- *
- * Note that it handles all NSNumber values that are encodable as int64_t or doubles
- * (depending on the underlying type of the NSNumber). Unsigned integer values are handled though
- * any value outside what is representable by int64_t (a signed 64-bit value) will throw an
- * exception.
- *
- * @return The parsed value.
- */
- - (Message<google_firestore_v1_Value>)parseScalarValue:(nullable id)input
- context:(ParseContext &&)context {
- if (!input || [input isMemberOfClass:[NSNull class]]) {
- return DeepClone(NullValue());
- } else if ([input isKindOfClass:[NSNumber class]]) {
- // Recover the underlying type of the number, using the method described here:
- // http://stackoverflow.com/questions/2518761/get-type-of-nsnumber
- const char *cType = [input objCType];
- // Type Encoding values taken from
- // https://developer.apple.com/library/mac/documentation/Cocoa/Conceptual/ObjCRuntimeGuide/
- // Articles/ocrtTypeEncodings.html
- switch (cType[0]) {
- case 'q':
- return [self encodeInteger:[input longLongValue]];
- case 'i': // Falls through.
- case 's': // Falls through.
- case 'l': // Falls through.
- case 'I': // Falls through.
- case 'S':
- // Coerce integer values that aren't long long. Allow unsigned integer types that are
- // guaranteed small enough to skip a length check.
- return [self encodeInteger:[input longLongValue]];
- case 'L': // Falls through.
- case 'Q':
- // Unsigned integers that could be too large. Note that the 'L' (long) case is handled here
- // because when compiled for LP64, unsigned long is 64 bits and could overflow int64_t.
- {
- unsigned long long extended = [input unsignedLongLongValue];
- if (extended > LLONG_MAX) {
- ThrowInvalidArgument("NSNumber (%s) is too large%s", [input unsignedLongLongValue],
- context.FieldDescription());
- } else {
- return [self encodeInteger:static_cast<int64_t>(extended)];
- }
- }
- case 'f':
- return [self encodeDouble:[input doubleValue]];
- case 'd':
- // Double values are already the right type, so just reuse the existing boxed double.
- //
- // Note that NSNumber already performs NaN normalization to a single shared instance
- // so there's no need to treat NaN specially here.
- return [self encodeDouble:[input doubleValue]];
- case 'B': // Falls through.
- case 'c': // Falls through.
- case 'C':
- // Boolean values are weird.
- //
- // On arm64, objCType of a BOOL-valued NSNumber will be "c", even though @encode(BOOL)
- // returns "B". "c" is the same as @encode(signed char). Unfortunately this means that
- // legitimate usage of signed chars is impossible, but this should be rare.
- //
- // Additionally, for consistency, map unsigned chars to bools in the same way.
- return [self encodeBoolean:[input boolValue]];
- default:
- // All documented codes should be handled above, so this shouldn't happen.
- HARD_FAIL("Unknown NSNumber objCType %s on %s", cType, input);
- }
- } else if ([input isKindOfClass:[NSString class]]) {
- std::string inputString = MakeString(input);
- return [self encodeStringValue:inputString];
- } else if ([input isKindOfClass:[NSDate class]]) {
- NSDate *inputDate = input;
- return [self encodeTimestampValue:api::MakeTimestamp(inputDate)];
- } else if ([input isKindOfClass:[FIRTimestamp class]]) {
- FIRTimestamp *inputTimestamp = input;
- Timestamp timestamp = TimestampInternal::Truncate(api::MakeTimestamp(inputTimestamp));
- return [self encodeTimestampValue:timestamp];
- } else if ([input isKindOfClass:[FIRGeoPoint class]]) {
- return [self encodeGeoPoint:api::MakeGeoPoint(input)];
- } else if ([input isKindOfClass:[NSData class]]) {
- NSData *inputData = input;
- return [self encodeBlob:(nanopb::MakeByteString(inputData))];
- } else if ([input isKindOfClass:[FSTDocumentKeyReference class]]) {
- FSTDocumentKeyReference *reference = input;
- if (reference.databaseID != _databaseID) {
- const DatabaseId &other = reference.databaseID;
- ThrowInvalidArgument(
- "Document Reference is for database %s/%s but should be for database %s/%s%s",
- other.project_id(), other.database_id(), _databaseID.project_id(),
- _databaseID.database_id(), context.FieldDescription());
- }
- return [self encodeReference:_databaseID key:reference.key];
- } else {
- ThrowInvalidArgument("Unsupported type: %s%s", NSStringFromClass([input class]),
- context.FieldDescription());
- }
- }
- - (Message<google_firestore_v1_Value>)encodeBoolean:(bool)value {
- Message<google_firestore_v1_Value> result;
- result->which_value_type = google_firestore_v1_Value_boolean_value_tag;
- result->boolean_value = value;
- return result;
- }
- - (Message<google_firestore_v1_Value>)encodeInteger:(int64_t)value {
- Message<google_firestore_v1_Value> result;
- result->which_value_type = google_firestore_v1_Value_integer_value_tag;
- result->integer_value = value;
- return result;
- }
- - (Message<google_firestore_v1_Value>)encodeDouble:(double)value {
- Message<google_firestore_v1_Value> result;
- result->which_value_type = google_firestore_v1_Value_double_value_tag;
- result->double_value = value;
- return result;
- }
- - (Message<google_firestore_v1_Value>)encodeTimestampValue:(Timestamp)value {
- Message<google_firestore_v1_Value> result;
- result->which_value_type = google_firestore_v1_Value_timestamp_value_tag;
- result->timestamp_value.seconds = value.seconds();
- result->timestamp_value.nanos = value.nanoseconds();
- return result;
- }
- - (Message<google_firestore_v1_Value>)encodeStringValue:(const std::string &)value {
- Message<google_firestore_v1_Value> result;
- result->which_value_type = google_firestore_v1_Value_string_value_tag;
- result->string_value = nanopb::MakeBytesArray(value);
- return result;
- }
- - (Message<google_firestore_v1_Value>)encodeBlob:(const nanopb::ByteString &)value {
- Message<google_firestore_v1_Value> result;
- result->which_value_type = google_firestore_v1_Value_bytes_value_tag;
- // Copy the blob so that pb_release can do the right thing.
- result->bytes_value = nanopb::CopyBytesArray(value.get());
- return result;
- }
- - (Message<google_firestore_v1_Value>)encodeReference:(const DatabaseId &)databaseId
- key:(const DocumentKey &)key {
- HARD_ASSERT(_databaseID == databaseId, "Database %s cannot encode reference from %s",
- _databaseID.ToString(), databaseId.ToString());
- std::string referenceName = ResourcePath({"projects", databaseId.project_id(), "databases",
- databaseId.database_id(), "documents", key.ToString()})
- .CanonicalString();
- Message<google_firestore_v1_Value> result;
- result->which_value_type = google_firestore_v1_Value_reference_value_tag;
- result->reference_value = nanopb::MakeBytesArray(referenceName);
- return result;
- }
- - (Message<google_firestore_v1_Value>)encodeGeoPoint:(const GeoPoint &)value {
- Message<google_firestore_v1_Value> result;
- result->which_value_type = google_firestore_v1_Value_geo_point_value_tag;
- result->geo_point_value.latitude = value.latitude();
- result->geo_point_value.longitude = value.longitude();
- return result;
- }
- - (Message<google_firestore_v1_ArrayValue>)parseArrayTransformElements:(NSArray<id> *)elements {
- ParseAccumulator accumulator{UserDataSource::Argument};
- Message<google_firestore_v1_ArrayValue> array_value;
- array_value->values_count = CheckedSize(elements.count);
- array_value->values = nanopb::MakeArray<google_firestore_v1_Value>(array_value->values_count);
- for (NSUInteger i = 0; i < elements.count; i++) {
- id element = elements[i];
- // Although array transforms are used with writes, the actual elements being unioned or removed
- // are not considered writes since they cannot contain any FieldValue sentinels, etc.
- ParseContext context = accumulator.RootContext();
- auto parsedElement = [self parseData:element context:context.ChildContext(i)];
- HARD_ASSERT(parsedElement && accumulator.field_transforms().empty(),
- "Failed to properly parse array transform element: %s", element);
- array_value->values[i] = *parsedElement->release();
- }
- return array_value;
- }
- @end
- NS_ASSUME_NONNULL_END
|