(SrSSKJr SSKJr SSKJr SSKJr SSKJr SSK J r SSK J r SS KJr SS KJr SS KJr S rS rSrSrSrSrSrSrSrSrSrSrSr Sr!Sr"S\\#4Sjr$S\\#4Sjr%S\\#4Sjr&g )!z'Utilities for Cloud Firestore commands.)absolute_import)division)unicode_literals)Optional)encoding) api_utils)apis) exceptions) arg_utils)textv1c8[R"S[5$)N firestore)r GetMessagesModuleFIRESTORE_INDEX_API_VERSION0lib/googlecloudsdk/command_lib/firestore/util.pyrr"s   -H IIrcg)zPython hook to get the value for the default database. Firestore currently only supports one database called '(default)'. Returns: The name of the default database. z (default)rrrrGetDatabaseFallthroughr&s rcg)aWPython hook to get the value for the '-' collection group. See details at: https://cloud.google.com/apis/design/design_patterns#list_sub-collections This allows us to describe or delete an index by specifying just its ID, without needing to know which collection group it belongs to. Returns: The value of the wildcard collection group. -rrrrGetCollectionGroupFallthroughr1s rcg)N __default__rrrr)GetDefaultFieldCollectionGroupFallthroughrAs rcg)N*rrrrGetDefaultFieldPathFallthroughrEs rc /nURH_nURnURnUR(aM.U(aU(dU(aMEU(aMNUR U5 Ma U(ax[ R "SSR[R"[U5S5[R"[U5S5SRSU55S95eU$)aPython hook to validate the field configuration of the given request. Note that this hook is only called after the request has been formed based on the spec. Thus, the validation of the user's choices for order and array-config, as well as the check for the required field-path attribute, have already been performed. As such the only remaining things to verify are that the user has specified at least 2 fields, and that exactly one of order or array-config was specified for each field. Args: unused_ref: The resource ref (unused). args: The parsed arg namespace. request: The request formed based on the spec. Returns: The original request assuming the field configuration is valid. Raises: InvalidArgumentException: If the field configuration is invalid. z--field-configzzExactly one of 'order' or 'array-config' must be specified for the {field_word} with the following {path_word}: [{paths}].fieldpathz, c3:# UHnURv M g7fN) fieldPath).0 field_configs r &ValidateFieldConfig..qsG0E )220Es) field_word path_wordpaths) r'order arrayConfig vectorConfigappendr InvalidArgumentExceptionformatr Pluralizelenjoin) unused_refargsrequestinvalid_field_configsr'r- array_configs rValidateFieldConfigr;Is&''l   E++L   ,ll""<0(  - - BBH&~~c*?&@'JnnS)>%?H))G0EGGCICH II .rc[5n[R"[R"UR5UR 5$)aPython hook to extract the operation metadata message. This is needed because apitools gives us a MetadataValue message for the operation metadata field, instead of the actual message that we want. Args: response: The response field in the operation returned by the API. unused_args: The parsed arg namespace (unused). Returns: The metadata field converted to a GoogleFirestoreAdminV1IndexOperationMetadata message. )rr DictToMessage MessageToDictmetadata,GoogleFirestoreAdminV1IndexOperationMetadata)response unused_argsmessagess rExtractOperationMetadatarDws< (   X../;; ==rcUR[5:wa/UR[5:Xa[R "SS5eU$)aPython hook to validate that the field reference is correctly specified. The user should be able to describe database-wide settings as well as collection-group wide settings; however it doesn't make sense to describe a particular field path's settings unless the collection group was also specified. The API will catch this but it's better to do it here for a clearer error message. Args: ref: The field resource reference. unused_args: The parsed arg namespace (unused). request: The field describe request. Returns: The original request assuming the field configuration is valid. Raises: InvalidArgumentException: If the field resource is invalid. FIELDzBCollection group must be provided if the field path was specified.)fieldsIdrcollectionGroupsIdrr r1)refrBr8s rValidateFieldArgrJsH$ ll466  I KK  - -L NN .rcURS5(dgURHcnURHPnURnURnU(aU(dU(aM2U(aM;[ R "SS5e Me g)zValidates the repeated --index arg. Args: args: The parsed arg namespace. Raises: InvalidArgumentException: If the provided indexes are incorrectly specified. indexNz--indexzZExactly one of 'order' or 'array-config' must be specified for each --index flag provided.) IsSpecifiedrLfieldsr-r.r r1)r7rLr!r-r:s rValidateFieldIndexArgsrOsp   ' " " zzekke&&l L%% 11  ./ / rc[5nSUlSnUR(aUR5nUR UR 5US9UlU$)a Update field request for TTL. Args: ref: The field resource reference(unused). args: The parsed arg namespace. request: The ttl field request. Raises: InvalidArgumentException: If the provided indexes are incorrectly specified. Returns: UpdateFieldRequest ttlConfigN)namerQ)r updateMask enable_ttlGoogleFirestoreAdminV1TtlConfigGoogleFirestoreAdminV1Field RelativeNamegoogleFirestoreAdminV1Field)rIr7r8rC ttl_configs rUpdateFieldRequestTtlsrZs] ("'* __99;J(0(L(L    )M)'% .rcx[5nURRRnUH nX#lM U$r$)rGoogleFirestoreAdminV1IndexQueryScopeValueValuesEnum COLLECTION queryScope)indexesrCscoperLs r AddQueryScoperbs;  (**DDOOe .rc[U5 U$r$)rO)r6r7reqs rValidateFieldUpdateRequestres *rc [5nUR(aUR/S9nO@URS5(a(UR(aURR nOSn[ R"USU5 U$)aUpdate patch request to include indexConfig. The mapping of index config message to API behavior is as follows: None - Clears the exemption indexes=[] - Disables all indexes indexes=[...] - Sets the index config to the indexes provided Args: unused_ref: The field resource reference. args: The parsed arg namespace. req: The auto-generated patch request. Returns: FirestoreProjectsDatabasesCollectionGroupsFieldsPatchRequest )r`rLNz'googleFirestoreAdminV1Field.indexConfig)rdisable_indexes!GoogleFirestoreAdminV1IndexConfigrMrX indexConfigr SetFieldInMessage)r6r7rdrC index_configs rAddIndexConfigToUpdateRequestrlst ( ==b=IL   S%D%D22>>LL cG*, *rc [R"5n[UR5nUS:Xa2[ UR 5 UR UR5S9$US:Xa2[ UR 5 UR UR5S9$US:Xa;[UR 5 UR URUR S9S9$Ub&[R"S RU5S 5eg) zParses the args and returns the encryption configuration, or none. Args: args: The parsed arg namespace. Returns: The encryption configuration, or none. zgoogle-default-encryption)googleDefaultEncryptionzuse-source-encryption)useSourceEncryptionzcustomer-managed-encryption) kmsKeyName)customerManagedEncryptionNzInvalid encryption type: {}zencryption-type)fs_api GetMessages_NormalizeStringencryption_type_ThrowIfKmsKeyNameSet kms_key_name&GoogleFirestoreAdminV1EncryptionConfig4GoogleFirestoreAdminV1GoogleDefaultEncryptionOptions-GoogleFirestoreAdminV1SourceEncryptionOptions_ThrowIfKmsKeyNameNotSet6GoogleFirestoreAdminV1CustomerManagedEncryptionOptionsr r1r2)r7rCrus rExtractEncryptionConfigr}s   !($T%9%9:/33$++,  : : ( ] ] _ ; 11$++,  : :$RRT ; 77T../  : :"*"a"a((#b#  ;  "  - -%,,_=  rvaluec2U(dgUR5$r$)casefold)r~s rrtrt%s    rrwc8Ub[R"SS5eg)N--kms-key-nameznA KMS Key cannot be set when using an --encryption-type of google-default-encryption or use-source-encryption.)r ConflictingArgumentsExceptionrws rrvrv+s(  2 2 ? rc8Uc[R"SS5eg)Nrz{The KMS Key Name is required when using customer-managed encryption (CMEK), please use --kms-key-name to specify this value)r RequiredArgumentExceptionrs rr{r{4s(  . . 0 rN)'__doc__ __future__rrrtypingrapitools.base.pyr googlecloudsdk.api_lib.firestorerrrgooglecloudsdk.api_lib.utilr googlecloudsdk.callioper $googlecloudsdk.command_lib.util.apisr googlecloudsdk.core.utilr rrrrrrr;rDrJrOrZrbrerlr}strrtrvr{rrrrs.&'%@,.:)#J   +\=&4/,4