U
    	ڲg.+                  	   @   s   d Z ddlmZ ddlZddlZddlmZ ddlmZ edddZ	dd	d
ddddddZ
dZdZdZdZdZdd ZG dd deZG dd deZdd Zedd!dd ZdS )"a  Channel notifications support.

Classes and functions to support channel subscriptions and notifications
on those channels.

Notes:
  - This code is based on experimental APIs and is subject to change.
  - Notification does not do deduplication of notification ids, that's up to
    the receiver.
  - Storing the Channel between calls is up to the caller.


Example setting up a channel:

  # Create a new channel that gets notifications via webhook.
  channel = new_webhook_channel("https://example.com/my_web_hook")

  # Store the channel, keyed by 'channel.id'. Store it before calling the
  # watch method because notifications may start arriving before the watch
  # method returns.
  ...

  resp = service.objects().watchAll(
    bucket="some_bucket_id", body=channel.body()).execute()
  channel.update(resp)

  # Store the channel, keyed by 'channel.id'. Store it after being updated
  # since the resource_id value will now be correct, and that's needed to
  # stop a subscription.
  ...


An example Webhook implementation using webapp2. Note that webapp2 puts
headers in a case insensitive dictionary, as headers aren't guaranteed to
always be upper case.

  id = self.request.headers[X_GOOG_CHANNEL_ID]

  # Retrieve the channel by id.
  channel = ...

  # Parse notification from the headers, including validating the id.
  n = notification_from_headers(channel, self.request.headers)

  # Do app specific stuff with the notification here.
  if n.resource_state == 'sync':
    # Code to handle sync state.
  elif n.resource_state == 'exists':
    # Code to handle the exists state.
  elif n.resource_state == 'not_exists':
    # Code to handle the not exists state.


Example of unsubscribing.

  service.channels().stop(channel.body()).execute()
    )absolute_importN)_helpers)errorsi     addressid
expirationparamsresource_idresource_uritypetoken)r   r   r   r	   
resourceIdresourceUrir   r   zX-GOOG-CHANNEL-IDzX-GOOG-MESSAGE-NUMBERzX-GOOG-RESOURCE-STATEzX-GOOG-RESOURCE-URIzX-GOOG-RESOURCE-IDc                 C   s&   i }|   D ]\}}||| < q|S )N)itemsupper)headersnew_headerskv r   ;/tmp/pip-unpacked-wheel-rfrvjkbl/googleapiclient/channel.py_upper_header_keysg   s    r   c                   @   s"   e Zd ZdZeddd ZdS )Notificationa  A Notification from a Channel.

    Notifications are not usually constructed directly, but are returned
    from functions like notification_from_headers().

    Attributes:
      message_number: int, The unique id number of this notification.
      state: str, The state of the resource being monitored.
      uri: str, The address of the resource being monitored.
      resource_id: str, The unique identifier of the version of the resource at
        this event.
       c                 C   s   || _ || _|| _|| _dS )a  Notification constructor.

        Args:
          message_number: int, The unique id number of this notification.
          state: str, The state of the resource being monitored. Can be one
            of "exists", "not_exists", or "sync".
          resource_uri: str, The address of the resource being monitored.
          resource_id: str, The identifier of the watched resource.
        N)message_numberstater   r
   )selfr   r   r   r
   r   r   r   __init__|   s    zNotification.__init__N)__name__
__module____qualname____doc__util
positionalr   r   r   r   r   r   n   s   r   c                   @   s4   e Zd ZdZeddddZdd Zd	d
 ZdS )Channela  A Channel for notifications.

    Usually not constructed directly, instead it is returned from helper
    functions like new_webhook_channel().

    Attributes:
      type: str, The type of delivery mechanism used by this channel. For
        example, 'web_hook'.
      id: str, A UUID for the channel.
      token: str, An arbitrary string associated with the channel that
        is delivered to the target address with each event delivered
        over this channel.
      address: str, The address of the receiving entity where events are
        delivered. Specific to the channel type.
      expiration: int, The time, in milliseconds from the epoch, when this
        channel will expire.
      params: dict, A dictionary of string to string, with additional parameters
        controlling delivery channel behavior.
      resource_id: str, An opaque id that identifies the resource that is
        being watched. Stable across different API versions.
      resource_uri: str, The canonicalized ID of the watched resource.
    r   N c	           	      C   s4   || _ || _|| _|| _|| _|| _|| _|| _dS )a  Create a new Channel.

        In user code, this Channel constructor will not typically be called
        manually since there are functions for creating channels for each specific
        type with a more customized set of arguments to pass.

        Args:
          type: str, The type of delivery mechanism used by this channel. For
            example, 'web_hook'.
          id: str, A UUID for the channel.
          token: str, An arbitrary string associated with the channel that
            is delivered to the target address with each event delivered
            over this channel.
          address: str,  The address of the receiving entity where events are
            delivered. Specific to the channel type.
          expiration: int, The time, in milliseconds from the epoch, when this
            channel will expire.
          params: dict, A dictionary of string to string, with additional parameters
            controlling delivery channel behavior.
          resource_id: str, An opaque id that identifies the resource that is
            being watched. Stable across different API versions.
          resource_uri: str, The canonicalized ID of the watched resource.
        N)r   r   r   r   r   r	   r
   r   )	r   r   r   r   r   r   r	   r
   r   r   r   r   r      s    #zChannel.__init__c                 C   sZ   | j | j| j| jd}| jr&| j|d< | jr6| j|d< | jrF| j|d< | jrV| j|d< |S )zBuild a body from the Channel.

        Constructs a dictionary that's appropriate for passing into watch()
        methods as the value of body argument.

        Returns:
          A dictionary representation of the channel.
        )r   r   r   r   r	   r   r   r   )r   r   r   r   r	   r
   r   r   )r   resultr   r   r   body   s    




zChannel.bodyc                 C   s4   t  D ]&\}}||}|dk	rt| || qdS )a  Update a channel with information from the response of watch().

        When a request is sent to watch() a resource, the response returned
        from the watch() request is a dictionary with updated channel information,
        such as the resource_id, which is needed when stopping a subscription.

        Args:
          resp: dict, The response from a watch() method.
        N)CHANNEL_PARAMSr   getsetattr)r   respZ	json_name
param_namevaluer   r   r   update   s    

zChannel.update)NNr&   r&   )	r   r    r!   r"   r#   r$   r   r(   r/   r   r   r   r   r%      s       +r%   c                 C   sf   t |}|t }| j|kr0td| j|f n2t|t }|t }|t }|t	 }t
||||S dS )a  Parse a notification from the webhook request headers, validate
      the notification, and return a Notification object.

    Args:
      channel: Channel, The channel that the notification is associated with.
      headers: dict, A dictionary like object that contains the request headers
        from the webhook HTTP request.

    Returns:
      A Notification object.

    Raises:
      errors.InvalidNotificationError if the notification is invalid.
      ValueError if the X-GOOG-MESSAGE-NUMBER can't be converted to an int.
    zChannel id mismatch: %s != %sN)r   X_GOOG_CHANNEL_IDr   r   ZInvalidNotificationErrorintX_GOOG_MESSAGE_NUMBERX_GOOG_RESOURCE_STATEX_GOOG_RESOURCE_URIX_GOOG_RESOURCE_IDr   )Zchannelr   Z
channel_idr   r   r   r
   r   r   r   notification_from_headers   s    
r6      c                 C   sZ   d}|r>|t  }|jd |j|jd d  d  }|dk r>d}tdtt || ||dS )a  Create a new webhook Channel.

    Args:
      url: str, URL to post notifications to.
      token: str, An arbitrary string associated with the channel that
        is delivered to the target address with each notification delivered
        over this channel.
      expiration: datetime.datetime, A time in the future when the channel
        should expire. Can also be None if the subscription should use the
        default expiration. Note that different services may have different
        limits on how long a subscription lasts. Check the response from the
        watch() method to see the value the service has set for an expiration
        time.
      params: dict, Extra parameters to pass on channel creation. Currently
        not used for webhook channels.
    r   i     i  Zweb_hook)r   r	   )EPOCHmicrosecondssecondsdaysr%   struuidZuuid4)urlr   r   r	   Zexpiration_msdeltar   r   r   new_webhook_channel  s     
rA   )NNN)r"   
__future__r   datetimer>   Zgoogleapiclientr   r#   r   r9   r)   r0   r2   r3   r4   r5   r   objectr   r%   r6   r$   rA   r   r   r   r   <module>   s4   9n