# +----------------------------------------------------------------------
# | WaitAdmin(fastapi)快速开发后台管理系统
# +----------------------------------------------------------------------
# | 欢迎阅读学习程序代码,建议反馈是我们前进的动力
# | 程序完全开源可支持商用,允许去除界面版权信息
# | gitee:   https://gitee.com/wafts/waitadmin-python
# | github:  https://github.com/topwait/waitadmin-python
# | 官方网站: https://www.waitadmin.cn
# | WaitAdmin团队版权所有并拥有最终解释权
# +----------------------------------------------------------------------
# | Author: WaitAdmin Team <2474369941@qq.com>
# +----------------------------------------------------------------------
import aiosmtplib
from aiosmtplib import SMTP
from typing import Union, List
from email.message import EmailMessage
from common.utils.config import ConfigUtil


class MailDriver:

    def __init__(self, host: str = "", port: int = 25, username: str = "", password: str = "", ssl: bool = False):
        self.port: int = port
        self.host: str = host
        self.username: str = username
        self.password: str = password
        self.verify_type: str = "ssl" if ssl else "default"
        self.messages = EmailMessage()

    async def init_config(self):
        """
        初始化邮件配置
        """
        if not self.host:
            config = await ConfigUtil.get("email") or {}
            self.port = int(config.get("smtp_port", 25))
            self.host = str(config.get("smtp_host", ""))
            self.username = config.get("smtp_user", "")
            self.password = config.get("smtp_pass", "")
            self.verify_type = config.get("verify_type", "default")

    async def smtp(self):
        """
        获取邮件服务连接

        Warning:
            连接使用完后记得把连接关闭
            关闭方式: smtp_client.quit()

        Author:
            zero
        """
        await self.init_config()
        smtp_client = SMTP(
            hostname=self.host,
            port=self.port,
            username=self.username,
            password=self.password,
            use_tls=True if self.verify_type == "ssl" else False
        )
        await smtp_client.connect()
        return smtp_client

    async def send(self):
        """
        发送邮件消息

        Raises:
            RuntimeError

        Author:
            zero
        """
        await self.init_config()

        if not self.messages["From"]:
            self.messages["From"] = self.username

        result = await aiosmtplib.send(
            self.messages,
            hostname=self.host,
            port=25,
            username=self.username,
            password=self.password,
            use_tls=True if self.verify_type == "ssl" else False
        )

        if not result[1].startswith("Mail OK"):
            raise Exception(result[1])

    def subject(self, title: str):
        """
        设置邮件标题

        Args:
            title (str): 邮件的标题。

        Author:
            zero
        """
        self.messages["subject"] = title
        return self

    def body(self, body: str):
        """
        设置邮件内容

        Args:
            body (str): 邮件文本内容。

        Author:
            zero
        """
        self.messages.set_content(body)
        return self

    def add_alternative(self, body: str, subtype: str = "html"):
        """
        设置HTML内容

        Args:
            body (str): HTML文本内容。
            subtype (str): 文本类型。

        Author:
            zero
        """
        self.messages.add_alternative(body, subtype=subtype)
        return self

    def add_attachment(self, path: str, filename: str, maintype: str, subtype: str):
        """
        设置邮件附件

        Args:
            path (str): 文件的本地路径: 如: /www/aa.png。
            filename (str): 文件的名称: 如: wa.png。
            maintype (str): 文件的类型, 如: image。
            subtype (str): 文件子类型, 如: png。

        Author:
            zero
        """
        with open(path, "rb") as f:
            data = f.read()
        self.messages.add_attachment(data, maintype=maintype, subtype=subtype, filename=filename)
        return self

    def fromm(self, address: str):
        """
        设置发件人 (发出的邮箱号)

        Args:
            address (Union[str, List[str]]): 发送者的邮箱。

        Example:
            address = "wa@qq.com"

        Author:
            zero
        """
        self.messages["From"] = address
        return self

    def to(self, address: Union[str, List[str]]):
        """
        设置收件人 (发送给谁?)

        Args:
            address (Union[str, List[str]]): 接收者的邮箱。

        Example:
            address = "wa@qq.com"
            address = ["wa.qq.com", "wa.163.com"]

        Author:
            zero
        """
        self.messages["To"] = address
        return self

    def cc(self, address: Union[str, List[str]]):
        """
        设置抄送人 (抄送给谁?)

        Args:
            address (Union[str, List[str]]): 接收者的邮箱。

        Example:
            address = "wa@qq.com"
            address = ["wa.qq.com", "wa.163.com"]

        Author:
            zero
        """
        self.messages["Cc"] = address
        return self