"""

    AUTHOR:

        Khushal P Soonderji

    DATE:

        Monday, 25th Nov., 2024

    OBJECTIVE:

        To manage e-mails in a GMail account.

    REFERENCES:

        1. Quickstart: https://developers.google.com/gmail/api/quickstart/python
        2. Labels: https://developers.google.com/gmail/api/guides/labels
        3. Messages: https://developers.google.com/gmail/api/reference/rest/v1/users.messages

    DOWNLOADS:

        N/A

"""


#   *****************************************************************************************************************
#  *****                                                                                                          ****
# ***                                                   IMPORT                                                      ***
#  *****                                                                                                          ****
#   *****************************************************************************************************************


# To make sibling directories accessible for imports:
import sys
sys.path.append(".")
sys.path.append("..")

# System-level activities:
import io

# My utils:
from utils_v2.string import json
from utils_v2.string import regex
from utils_v2.date_time import date_time
from utils_v2.mail import mail_parser

# My Google utils:
from utils_v2.oauth.services.goog import GoogleOAuth
from utils_v2.goog.base import AsyncGoogleBase
from utils_v2.goog.models.data.auth_tokens import GoogleAuthTokens
from utils_v2.goog.models.data.api_call import GoogleApiResponse

# Related to Google:
from google.auth.transport.requests import Request
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build

# To make API calls:
import httpx

# For asynchronous activities:
import asyncio

# To work with date and time:
import datetime

# For working with datatypes:
from typing import Dict, Literal, List, Any

# For debugging:
from icecream import IceCreamDebugger

# For computational help:
import math

# For base64 encoding:
import base64


#   *****************************************************************************************************************
#  *****                                                                                                          ****
# ***                                           MACROS / ONE-TIME INIT                                              ***
#  *****                                                                                                          ****
#   *****************************************************************************************************************


# --- Nothing Yet


#   *****************************************************************************************************************
#  *****                                                                                                          ****
# ***                                                VARIABLES                                                      ***
#  *****                                                                                                          ****
#   *****************************************************************************************************************


# --- Nothing Yet


#   *****************************************************************************************************************
#  *****                                                                                                          ****
# ***                                                  FUNCTIONS                                                    ***
#  *****                                                                                                          ****
#   *****************************************************************************************************************


# --- Nothing Yet


#   *****************************************************************************************************************
#  *****                                                                                                          ****
# ***                                                   CLASSES                                                     ***
#  *****                                                                                                          ****
#   *****************************************************************************************************************


class AsyncGMailClient(AsyncGoogleBase):

    async def get_user_profile(
            self,
            tokens: GoogleAuthTokens,
            user_id: str = "me",
    ) -> GoogleApiResponse:

        # Ensure that the tokens are valid:
        await tokens.arefresh(
            http_client = self._http_client,
            client_id = self._client_id,
            client_secret = self._client_secret,
            force_refresh = False
        )

        # Make the API call:
        if not self._debug_only_errors: self._printer("Getting User Profile.")
        api_response = await self.get(
            url = f"https://gmail.googleapis.com/gmail/v1/users/{user_id}/profile",
            headers = {"Authorization": f"Bearer {tokens.accessToken}"}
        )

        # If the call was successful:
        if api_response.httpCode in [200]:
            api_response.data = await api_response.get_json()

        # Done here:
        return api_response

    # ┓   ┓   ┓
    # ┃ ┏┓┣┓┏┓┃┏
    # ┗┛┗┻┗┛┗ ┗┛

    async def list_labels(
            self,
            tokens: GoogleAuthTokens,
            user_id: str = "me"
    ) -> GoogleApiResponse:

        """
        To get the list of labels of this user.
        DOCUMENTATION:
            1. https://developers.google.com/gmail/api/reference/rest/v1/users.labels/list
            2. https://developers.google.com/gmail/api/reference/rest/v1/users.labels#Label
        :param tokens: The object that holds the access token to the service.
        :param user_id: The id of the user for which this service must be run. Typically, the e-mail id itself.
        :return: A structured response where the list of labels will be in the 'data' variable.
        """

        # Ensure that the tokens are valid:
        await tokens.arefresh(
            http_client = self._http_client,
            client_id = self._client_id,
            client_secret = self._client_secret,
            force_refresh = False
        )

        # Make the API call:
        if not self._debug_only_errors: self._printer("Listing All Labels.", user_id)
        api_response = await self.get(
            url = f"https://gmail.googleapis.com/gmail/v1/users/{user_id}/labels",
            headers = {"Authorization": f"Bearer {tokens.accessToken}"}
        )

        # If the call was successful:
        if api_response.httpCode in [200]:
            api_json = await api_response.get_json()
            api_response.data = {label.pop("name"): label for label in api_json.get("labels", [])}

        # Done here:
        return api_response


#   *****************************************************************************************************************
#  *****                                                                                                          ****
# ***                                               MAIN PROGRAM                                                    ***
#  *****                                                                                                          ****
#   *****************************************************************************************************************


if __name__ == "__main__":

    import dateparser
    from utils_v2.goog.models.data.auth_tokens import GoogleAuthTokens

    # Create an HTTP client:
    test_client = httpx.AsyncClient(
        limits = httpx.Limits(
            max_connections = 100,  # ............ Maximum number of connections allowed in the pool.
            max_keepalive_connections = 50,  # ... Maximum number of connections that can be kept alive.
        ),
        timeout = httpx.Timeout(
            connect = 2.5,  # ... Shorter connection timeout.
            read = 2.5,  # ...... Like what EasyEcom gives.
            write = 10.0,  # .... Time to wait for sending data.
            pool = 120.0  # ..... Time to wait for a free connection from the pool.
        )
    )

    secrets_file = r"../../../creds/google_tcaoff_test_oauth_20241125.json"
    secrets_dict = json.from_file(secrets_file)

    tokens = {
        "accessToken": "ya29.a0AeDClZAYoo85BXRId_n-hwo_amKshzi46c33GaJcsZZvGB7A7OGU2RFYcWBM_BleNBfAFUSJP2NHAvmd7Nsp_U5Kg68hXSy0iO99PNTm3pvKrJSzbkA-rXsVLsCnBIfPUMyNt2nOOVJmGwm17DNN0jAELkm1fPNTju7SZzmuaCgYKAZwSARMSFQHGX2Mis8TZui2rZT1gKySVds-N0w0175",
        "refreshToken": "1//0gnqzjMf9YT19CgYIARAAGBASNgF-L9Ir3rcY37nGrV45XyOUBRllEH7Txui7T1JbwevlmDoNw7PuMu149cCWQSwsScuKaZusUQ",
        "expiresIn": 3539,
        "expiresAt": dateparser.parse("2024-11-25 10:40:40.833699+00:00"),
        "scopes": [
            "https://www.googleapis.com/auth/gmail.labels",
            "https://www.googleapis.com/auth/gmail.modify"
        ]
    }

    async def main():

        my_gmail = AsyncGMailClient(
            service_name = "gmail",
            client_id = secrets_dict["web"]["client_id"],
            client_secret = secrets_dict["web"]["client_secret"],
            http_client = test_client,
            debug = True,
            debug_prefix = "GMail (M) | ",
            debug_only_errors = False
        )

        test_tokens = GoogleAuthTokens(**tokens)

        response = await my_gmail.get_user_profile(tokens = test_tokens)
        print("RESPONSE:", response)


    asyncio.run(main())