# example.py

from abc import ABC, abstractmethod
from dataclasses import dataclass


# TODO: use Pydantic for the Email object.
@dataclass
class Email:
    recipients: list[str]
    sender: str
    sender_name: str
    subject: str
    body: str
    cc: list[str] = None
    bcc: list[str] = None


class EmailHandler(ABC):  # interface
    @abstractmethod
    async def send(self, email: Email) -> None:
        pass


# ----------------------------------------------
# SendGrid implementation of EmailHandler
# ----------------------------------------------
import httpx


class SendGridClient(EmailHandler):
    def __init__(self, api_key: str, http_client: httpx.AsyncClient):
        if not api_key:
            raise ValueError("API key is required")
        self.http_client = http_client
        self.api_key = api_key

    async def send(self, email: Email) -> None:
        response = await self.http_client.post(
            "https://api.sendgrid.com/v3/mail/send",
            headers={
                "Authorization": f"Bearer {self.api_key}",
                "Content-Type": "application/json",
            },
            json=self.get_body(email),
        )
        # Note: in case of error, inspect response.text
        response.raise_for_status()  # Raise an error for bad responses

    def get_body(self, email: Email) -> dict:
        return {
            "personalizations": [
                {
                    "to": [{"email": recipient} for recipient in email.recipients],
                    "subject": email.subject,
                    "cc": [{"email": cc} for cc in email.cc] if email.cc else None,
                    "bcc": [{"email": bcc} for bcc in email.bcc] if email.bcc else None,
                }
            ],
            "from": {"email": email.sender, "name": email.sender_name},
            "content": [{"type": "text/html", "value": email.body}],
        }

# ----------------------------------------------
# Test
# ----------------------------------------------
import os

async def example():

    async with httpx.AsyncClient() as http_client:
        sendgrid = SendGridClient(
            api_key=os.environ.get("SENDGRID_API_KEY"), http_client=http_client
        )

        await sendgrid.send(
            Email(
                recipients=["hello-world@gmail.com"],
                sender="example@somedomain.com",
                sender_name="Example",
                subject="Test email",
                body="<h1>Hello</h1>",
            )
        )


if __name__ == "__main__":
    import asyncio

    asyncio.run(example())