oauth.py

Go to the documentation of this file.

"""Provide oauth implementations for the Tesla Fleet integration."""
 
import base64
import hashlib
import secrets
from typing import Any
 
from homeassistant.components.application_credentials import (
    AuthImplementation,
    AuthorizationServer,
    ClientCredential,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_entry_oauth2_flow
 
from .const import AUTHORIZE_URL, CLIENT_ID, DOMAIN, SCOPES, TOKEN_URL
 
 
class TeslaSystemImplementation(config_entry_oauth2_flow.LocalOAuth2Implementation):
    """Tesla Fleet API open source Oauth2 implementation."""
 
    code_verifier: str
    code_challenge: str
 
    def __init__(self, hass: HomeAssistant) -> None:
        """Initialize open source Oauth2 implementation."""
 
        # Setup PKCE
        self.code_verifiercode_verifier = secrets.token_urlsafe(32)
        hashed_verifier = hashlib.sha256(self.code_verifiercode_verifier.encode()).digest()
        self.code_challengecode_challenge = (
            base64.urlsafe_b64encode(hashed_verifier).decode().replace("=", "")
        )
        super().__init__(
            hass,
            DOMAIN,
            CLIENT_ID,
            "",
            AUTHORIZE_URL,
            TOKEN_URL,
        )
 
    @property
    def name(self) -> str:
        """Name of the implementation."""
        return "Built-in open source client ID"
 
    @property
    def extra_authorize_data(self) -> dict[str, Any]:
        """Extra data that needs to be appended to the authorize url."""
        return {
            "prompt": "login",
            "scope": " ".join(SCOPES),
            "code_challenge": self.code_challengecode_challenge,  # PKCE
        }
 
    async def async_resolve_external_data(self, external_data: Any) -> dict:
        """Resolve the authorization code to tokens."""
        return await self._token_request(
            {
                "grant_type": "authorization_code",
                "code": external_data["code"],
                "redirect_uri": external_data["state"]["redirect_uri"],
                "code_verifier": self.code_verifiercode_verifier,  # PKCE
            }
        )
 
 
class TeslaUserImplementation(AuthImplementation):
    """Tesla Fleet API user Oauth2 implementation."""
 
    def __init__(
        self, hass: HomeAssistant, auth_domain: str, credential: ClientCredential
    ) -> None:
        """Initialize user Oauth2 implementation."""
 
        super().__init__(
            hass,
            auth_domain,
            credential,
            AuthorizationServer(AUTHORIZE_URL, TOKEN_URL),
        )
 
    @property
    def extra_authorize_data(self) -> dict[str, Any]:
        """Extra data that needs to be appended to the authorize url."""
        return {"prompt": "login", "scope": " ".join(SCOPES)}