o tBhF@sxdZddlZddlmZddlmZddlZddlZddlmZddlmZddlm Z ddlm Z d Z gd Z e j Z e jZe jZe jZe jZe jZe jZe jZe jZe jZe jZejZejZe jZe jZe jZe jZe jZe jZe j Z e j!Z!e j"Z"d d Z#d"ddZ$d"ddZ%d"ddZ&d#ddZ'd#ddZ(GdddZ)GdddZ*GdddZ+GdddZ,Gd d!d!Z-dS)$z Firebase Cloud Messaging module.N)http)_auth) _http_client)_messaging_encoder)_messaging_utils)_utils _messaging) AndroidConfigAndroidFCMOptionsAndroidNotification APNSConfigAPNSFCMOptions APNSPayloadApsApsAlert BatchResponse CriticalSound ErrorInfo FCMOptions LightSettingsMessageMulticastMessage NotificationQuotaExceededErrorSenderIdMismatchError SendResponseThirdPartyAuthErrorTopicManagementResponseUnregisteredError WebpushConfigWebpushFCMOptionsWebpushNotificationWebpushNotificationActionsendsend_allsend_multicastsubscribe_to_topicunsubscribe_from_topiccCst|ttSN)rget_app_service_MESSAGING_ATTRIBUTE_MessagingService)appr-o/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/firebase_admin/messaging.py_get_messaging_service^sr/FcCt|||S)aSends the given message via Firebase Cloud Messaging (FCM). If the ``dry_run`` mode is enabled, the message will not be actually delivered to the recipients. Instead FCM performs all the usual validations, and emulates the send operation. Args: message: An instance of ``messaging.Message``. dry_run: A boolean indicating whether to run the operation in dry run mode (optional). app: An App instance (optional). Returns: string: A message ID string that uniquely identifies the sent message. Raises: FirebaseError: If an error occurs while sending the message to the FCM service. ValueError: If the input arguments are invalid. )r/r#)messagedry_runr,r-r-r.r#ar#cCr0)aSends the given list of messages via Firebase Cloud Messaging as a single batch. If the ``dry_run`` mode is enabled, the message will not be actually delivered to the recipients. Instead FCM performs all the usual validations, and emulates the send operation. Args: messages: A list of ``messaging.Message`` instances. dry_run: A boolean indicating whether to run the operation in dry run mode (optional). app: An App instance (optional). Returns: BatchResponse: A ``messaging.BatchResponse`` instance. Raises: FirebaseError: If an error occurs while sending the message to the FCM service. ValueError: If the input arguments are invalid. )r/r$)messagesr2r,r-r-r.r$ur3r$cs6tts tdfddjD}t|||S)aSends the given mutlicast message to all tokens via Firebase Cloud Messaging (FCM). If the ``dry_run`` mode is enabled, the message will not be actually delivered to the recipients. Instead FCM performs all the usual validations, and emulates the send operation. Args: multicast_message: An instance of ``messaging.MulticastMessage``. dry_run: A boolean indicating whether to run the operation in dry run mode (optional). app: An App instance (optional). Returns: BatchResponse: A ``messaging.BatchResponse`` instance. Raises: FirebaseError: If an error occurs while sending the message to the FCM service. ValueError: If the input arguments are invalid. z@Message must be an instance of messaging.MulticastMessage class.c s.g|]}tjjjjjj|dqS))data notificationandroidwebpushapns fcm_optionstoken)rr5r6r7r8r9r:).0r;multicast_messager-r. s z"send_multicast..) isinstancer ValueErrortokensr/r$)r>r2r,r4r-r=r.r%s   r%cCt|||dS)a4Subscribes a list of registration tokens to an FCM topic. Args: tokens: A non-empty list of device registration tokens. List may not have more than 1000 elements. topic: Name of the topic to subscribe to. May contain the ``/topics/`` prefix. app: An App instance (optional). Returns: TopicManagementResponse: A ``TopicManagementResponse`` instance. Raises: FirebaseError: If an error occurs while communicating with instance ID service. ValueError: If the input arguments are invalid. ziid/v1:batchAddr/make_topic_management_requestrBtopicr,r-r-r.r&r&cCrC)a<Unsubscribes a list of registration tokens from an FCM topic. Args: tokens: A non-empty list of device registration tokens. List may not have more than 1000 elements. topic: Name of the topic to unsubscribe from. May contain the ``/topics/`` prefix. app: An App instance (optional). Returns: TopicManagementResponse: A ``TopicManagementResponse`` instance. Raises: FirebaseError: If an error occurs while communicating with instance ID service. ValueError: If the input arguments are invalid. ziid/v1:batchRemoverDrFr-r-r.r'rHr'c@s0eZdZdZddZeddZeddZdS) rzBAn error encountered when performing a topic management operation.cCs||_||_dSr()_index_reason)selfindexreasonr-r-r.__init__s zErrorInfo.__init__cC|jS)zBIndex of the registration token to which this error is related to.)rIrKr-r-r.rLzErrorInfo.indexcCrO)z*String describing the nature of the error.)rJrPr-r-r.rMrQzErrorInfo.reasonN)__name__ __module__ __qualname____doc__rNpropertyrLrMr-r-r-r.rs rc@<eZdZdZddZeddZeddZedd Zd S) rz8The response received from a topic management operation.cCst|tr d|vrtd|d|_d|_g|_t|dD]"\}}d|vr:|jd7_|jt ||dq|jd7_qdS)Nresultsz*Unexpected topic management response: {0}.rerror) r@dictrAformat_success_count_failure_count_errors enumerateappendr)rKresprLresultr-r-r.rNsz TopicManagementResponse.__init__cCrO)zCNumber of tokens that were successfully subscribed or unsubscribed.r]rPr-r-r. success_countrQz%TopicManagementResponse.success_countcCrO)zLNumber of tokens that could not be subscribed or unsubscribed due to errors.)r^rPr-r-r. failure_countrQz%TopicManagementResponse.failure_countcCrO)z;A list of ``messaging.ErrorInfo`` objects (possibly empty).)r_rPr-r-r.errorsrQzTopicManagementResponse.errorsN) rRrSrTrUrNrVrerfrgr-r-r-r.rs  rc@rW) rz:The response received from a batch request to the FCM API.cCs||_tdd|D|_dS)NcSsg|]}|jr|qSr-)success)r<rbr-r-r.r?sz*BatchResponse.__init__..) _responseslenr])rK responsesr-r-r.rNszBatchResponse.__init__cCrO)z>A list of ``messaging.SendResponse`` objects (possibly empty).)rirPr-r-r.rkrQzBatchResponse.responsescCrOr(rdrPr-r-r.re szBatchResponse.success_countcCst|j|jSr()rjrkrerPr-r-r.rfszBatchResponse.failure_countN) rRrSrTrUrNrVrkrerfr-r-r-r.rs  rc@rW) rzHThe response received from an individual batched request to the FCM API.cCs&||_d|_|r|dd|_dSdS)Nname) _exception _message_idget)rKrb exceptionr-r-r.rNs zSendResponse.__init__cCrO)z9A message ID string that uniquely identifies the message.)rnrPr-r-r. message_idrQzSendResponse.message_idcCs|jduo|j S)z3A boolean indicating if the request was successful.N)rnrmrPr-r-r.rh$szSendResponse.successcCrO)zTA ``FirebaseError`` if an error occurs while sending the message to the FCM service.)rmrPr-r-r.rp)rQzSendResponse.exceptionN) rRrSrTrUrNrVrqrhrpr-r-r-r.rs  rc@seZdZdZdZdZdZddiZe Z e e e e edZdd Zed d Zd$d dZd$ddZddZddZddZddZddZddZeddZedd Zed!d"Zd#S)%r+zKService class that implements Firebase Cloud Messaging (FCM) functionality.z8https://fcm.googleapis.com/v1/projects/{0}/messages:sendz https://fcm.googleapis.com/batchzhttps://iid.googleapis.comaccess_token_authtrue)APNS_AUTH_ERRORQUOTA_EXCEEDEDSENDER_ID_MISMATCHTHIRD_PARTY_AUTH_ERROR UNREGISTEREDcCsp|j}|s tdtj||_ddtjd|_|j dt j }t j |j|d|_t|j|_dS)NzProject ID is required to access Cloud Messaging service. Either set the projectId option, or use service account credentials. Alternatively, set the GOOGLE_CLOUD_PROJECT environment variable.2zfire-admin-python/{0})zX-GOOG-API-FORMAT-VERSIONzX-FIREBASE-CLIENT httpTimeout) credentialtimeout) project_idrAr+FCM_URLr\_fcm_urlfirebase_admin __version__ _fcm_headersoptionsrorDEFAULT_TIMEOUT_SECONDSJsonHttpClientr{get_credential_clientrauthorized_http _transport)rKr,r}r|r-r-r.rN@s  z_MessagingService.__init__cCst|ts td|j|S)Nz7Message must be an instance of messaging.Message class.)r@rrA JSON_ENCODERdefault)clsr1r-r-r.encode_messageQs  z _MessagingService.encode_messageFc CsV|||}z|jjd|j|j|d}W|dStjjy*}z||d}~ww)z2Sends the given message to FCM via the FCM v1 API.post)urlheadersjsonNrl) _message_datarbodyrrrequests exceptionsRequestException_handle_fcm_error)rKr1r2r5rbrYr-r-r.r#Ws   z_MessagingService.sendc st|ts tdt|dkrtdgfdd}tj|tjd}|D]}t  ||}tj j j jd|jd}||q&z |WtSty`}z|d }~ww) z2Sends the given messages to FCM via the batch API.z7messages must be a list of messaging.Message instances.iz1messages must not contain more than 500 elements.cs*d}|r |}t||}|dSr()_handle_batch_errorrra)_responserYrp send_responserkrKr-r.batch_callbackos   z2_MessagingService.send_all..batch_callback)callback batch_uriPOST)rpostprocurimethodrrN)r@listrArjrBatchHttpRequestr+ FCM_BATCH_URLrdumpsr HttpRequestr _postprocrraddexecute Exceptionrr) rKr4r2rbatchr1rreqrYr-rr.r$fs6     z_MessagingService.send_allc Cst|tr|g}t|tr|stddd|D}|r tdt|tr'|s+td|ds5d|}||d}d tj|}z|jj d ||tj d }Wt|St j j yd}z||d }~ww) z;Invokes the IID service for topic management functionality.z7Tokens must be a string or a non-empty list of strings.cSsg|] }t|tr |s|qSr-)r@str)r<tr-r-r.r?szC_MessagingService.make_topic_management_request..z!Tokens must be non-empty strings.z!Topic must be a non-empty string.z/topics/z /topics/{0})toregistration_tokensz{0}/{1}r)rrrN)r@rrrA startswithr\r+IID_URLrr IID_HEADERSrrr_handle_iid_errorr) rKrBrG operation invalid_strr5rrbrYr-r-r.rEs6     z/_MessagingService.make_topic_management_requestcCsdt|i}|r d|d<|S)Nr1T validate_only)r+r)rKr1r2r5r-r-r.rsz_MessagingService._message_datacCst|S)z'Handle response from batch API request.)rloadsdecode)rKrrr-r-r.rsz_MessagingService._postproccCt|tjS)z)Handles errors received from the FCM API.)r#handle_platform_error_from_requestsr+_build_fcm_error_requestsrKrYr-r-r.rz#_MessagingService._handle_fcm_errorcCs|jdur t|i}z|j}t|tr|}Wn ty#Ynw|d}d}|r3d|}n d|jj |jj }t||S)z1Handles errors received from the Instance ID API.NrYz(Error while calling the IID service: {0}z4Unexpected HTTP response with status: {0}; body: {1}) rrhandle_requests_errorrr@r[rAror\ status_codecontentr)rKrYr5 parsed_bodycodemsgr-r-r.rs&        z#_MessagingService._handle_iid_errorcCr)zMHandles errors received from the googleapiclient while making batch requests.)r*handle_platform_error_from_googleapiclientr+ _build_fcm_error_googleapiclientrr-r-r.rrz%_MessagingService._handle_batch_errorcCs"||}|r||||jdSdSzfParses an error response from the FCM API and creates a FCM-specific exception if appropriate.)cause http_responseN)_build_fcm_errorr)rrYr1 error_dictexc_typer-r-r.rs z+_MessagingService._build_fcm_error_requestscCs ||}|r||||dSdSr)r)rrYr1rrrr-r-r.rs z2_MessagingService._build_fcm_error_googleapiclientcCsF|sdSd}|dgD]}|ddkr|d}nq tj|S)Ndetailsz@typez3type.googleapis.com/google.firebase.fcm.v1.FcmError errorCode)ror+FCM_ERROR_TYPES)rrfcm_codedetailr-r-r.rs  z"_MessagingService._build_fcm_errorN)F)rRrSrTrUr~rrrrMessageEncoderrrrrrrrN classmethodrr#r$rErrrrrrrrr-r-r-r.r+/s<   %  r+)FNr().rUrgoogleapiclientrrrrrrrrr*__all__r r r r r rrrrrrrrrrr r!r"rrrrr/r#r$r%r&r'rrrrr+r-r-r-r.sZ      $