"""National Grid MA utility implementation.

This module handles the authentication and API interactions for National Grid.
It uses OAuth 2.0 with PKCE for secure authentication.
"""

import logging
from typing import Optional

import aiohttp

from .base import UtilityBase
from .oidchelper import async_auth_oidc

_LOGGER = logging.getLogger(__name__)


class NationalGridMA(UtilityBase):
    """National Grid MA utility implementation.

    This class handles the authentication and API interactions for National Grid.
    It uses OAuth 2.0 with PKCE for secure authentication.
    """

    @staticmethod
    def name() -> str:
        """Return the name of the utility."""
        return "National Grid MA"

    @staticmethod
    def subdomain() -> str:
        """Return the opower.com subdomain for this utility."""
        return "ngma"

    @staticmethod
    def timezone() -> str:
        """Return the timezone."""
        return "America/New_York"

    BASE_URL = "https://login.nationalgrid.com"
    TENANT_ID = "0e1366c5-731c-42b3-90d3-508039d9e70f"
    POLICY = "B2C_1A_UWP_NationalGrid_convert_merge_signin"
    CLIENT_ID = "36488660-e86a-4a0d-8316-3df49af8d06d"
    REDIRECT_URI = "https://myaccount.nationalgrid.com/auth-landing"
    APPLICATION_URI = f"{BASE_URL}/"
    SCOPE_AUTH = "openid profile offline_access"
    SCOPE_ACCESS = f"{CLIENT_ID} openid profile offline_access"
    SELF_ASSERTED_ENDPOINT = "SelfAsserted"
    POLICY_CONFIRM_ENDPOINT = "api/CombinedSigninAndSignup/confirmed"

    @staticmethod
    async def async_login(
        session: aiohttp.ClientSession,
        username: str,
        password: str,
        optional_mfa_secret: Optional[str],
    ) -> Optional[str]:
        """Perform the login process and return an access token."""
        _LOGGER.debug("Starting login process for National Grid")
        return await async_auth_oidc(
            username,
            password,
            NationalGridMA.BASE_URL,
            NationalGridMA.TENANT_ID,
            NationalGridMA.POLICY,
            NationalGridMA.CLIENT_ID,
            NationalGridMA.REDIRECT_URI,
            NationalGridMA.SCOPE_AUTH,
            NationalGridMA.SCOPE_ACCESS,
            NationalGridMA.SELF_ASSERTED_ENDPOINT,
            NationalGridMA.POLICY_CONFIRM_ENDPOINT,
        )