from email.message import EmailMessage

from pydantic import EmailStr

from app.config import settings


def create_registration_confirmation_template(
        username: str,
        email_to: EmailStr,
        confirmation_code: str,
):
    email = EmailMessage()

    email["Subject"] = "Подтверждение регистрации"
    email["From"] = settings.SMTP_USER
    email["To"] = email_to

    email.set_content(
        f"""
             <h1>{username}, лови аптечку</h1>
             <h2>{confirmation_code}</h2>
        """,
        subtype="html"
    )
    return email


def create_password_change_confirmation_template(
        username: str,
        email_to: EmailStr,
):
    email = EmailMessage()

    email["Subject"] = "Смена пароля к аккаунту"
    email["From"] = settings.SMTP_USER
    email["To"] = email_to

    email.set_content(
        f"""
             <h1>{username}, ты менял пароль?</h1>
             <h2>Если нет, то пидора ответ</h2>
        """,
        subtype="html"
    )
    return email


def create_password_recover_template(
        username: str,
        email_to: EmailStr,
        confirmation_code: str,
):
    email = EmailMessage()

    email["Subject"] = "Восстановление пароля"
    email["From"] = settings.SMTP_USER
    email["To"] = email_to

    email.set_content(
        f"""
             <h1>{username}, ты тут хотел восстановить пароль?</h1>
             <h2>{confirmation_code}</h2>
        """,
        subtype="html"
    )
    return email