#coding: utf-8
# +-------------------------------------------------------------------
# | yakpanel
# +-------------------------------------------------------------------
# | Copyright (c) 2015-2020 yakpanel(http://www.yakpanel.com) All rights reserved.
# +-------------------------------------------------------------------
# | Author: 沐落 <cjx@yakpanel.com>
# | Author: lx
# | 消息通道邮箱模块
# +-------------------------------------------------------------------

import smtplib
import traceback
from email.mime.text import MIMEText
from email.utils import formataddr
from typing import Tuple, Union, Optional
import public

from mod.base.msg.util import write_push_log, write_mail_push_log, get_test_msg


class MailMsg:

    def __init__(self, mail_data):
        self.id = mail_data["id"]
        self.config = mail_data["data"]

    @classmethod
    def check_args(cls, args: dict) -> Tuple[bool, Union[dict, str]]:
        if "send" not in args or "receive" not in args or len(args["receive"]) < 1:
            return False, "Incomplete information, there must be a sender and at least one receiver"

        if "title" not in args:
            return False, "There is no necessary remark information"

        title = args["title"]
        if len(title) > 15:
            return False, 'Note names cannot be longer than 15 characters'

        send_data = args["send"]
        send = {}
        for i in ("qq_mail", "qq_stmp_pwd", "hosts", "port"):
            if i not in send_data:
                return False, "The sender configuration information is incomplete"
            send[i] = send_data[i].strip()

        receive_data = args["receive"]
        if isinstance(receive_data, str):
            receive_list = [i.strip() for i in receive_data.split("\n") if i.strip()]
        else:
            receive_list = [i.strip() for i in receive_data if i.strip()]

        data = {
            "send": send,
            "title": title,
            "receive": receive_list,
        }

        test_obj = cls({"data": data, "id": None})
        test_msg = {
            "msg_list": ['>configuration state: Success<br>']
        }

        test_task = get_test_msg("Message channel configuration reminders")

        res = test_obj.send_msg(
            test_task.to_mail_msg(test_msg, test_task.the_push_public_data()),
            "Message channel configuration reminders"
        )
        if res is None or res.find("Failed to send mail to some recipients") != -1:
            return True, data

        return False, res

    def send_msg(self, msg: str, title: str):
        """
        邮箱发送
        @msg 消息正文
        @title 消息标题
        """
        if not self.config:
            return public.lang('Mailbox information is not configured correctly')

        if 'port' not in self.config['send']: 
            self.config['send']['port'] = 465

        receive_list = self.config['receive']

        error_list, success_list = [], []
        error_msg_dict = {}
        for email in receive_list:
            if not email.strip():
                continue
            server = None
            try:
                data = MIMEText(msg, 'html', 'utf-8')
                data['From'] = formataddr((self.config['send']['qq_mail'], self.config['send']['qq_mail']))
                data['To'] = formataddr((self.config['send']['qq_mail'], email.strip()))
                data['Subject'] = title

                port = int(self.config['send']['port'])
                host = str(self.config['send']['hosts'])
                user = self.config['send']['qq_mail']
                pwd = self.config['send']['qq_stmp_pwd']

                if port == 465:
                    # SSL direct connection
                    server = smtplib.SMTP_SSL(host, port, timeout=10)
                else:
                    # Standard connection, possibly with STARTTLS
                    server = smtplib.SMTP(host, port, timeout=10)
                    try:
                        # Attempt to upgrade to a secure connection
                        server.starttls()
                    except smtplib.SMTPNotSupportedError:
                        # The server does not support STARTTLS, proceed with an insecure connection
                        pass

                server.login(user, pwd)
                server.sendmail(user, [email.strip(), ], data.as_string())
                success_list.append(email)
            except:
                err_msg = traceback.format_exc()
                # public.print_log(f"邮件发送失败: {err_msg}")
                error_list.append(email)
                error_msg_dict[email] = err_msg
            finally:
                if server:
                    server.quit()

        if not error_list and not success_list:
            return public.lang('The receiving mailbox is not configured')
        if not error_list:
            write_push_log("mail", True, title, success_list)
            return None
        if not success_list:
            write_push_log("mail", False, title, error_list)
            first_error_msg = list(error_msg_dict.values())[0]
            # 修复 IndexError
            return public.lang('Failed to send message, Recipient of failed to send:{}, Error: {}', error_list, first_error_msg)

        write_mail_push_log(title, error_list, success_list)

        return public.lang('Failed to send mail to some recipients, including:{}',error_list)

    def test_send_msg(self) -> Optional[str]:
        test_msg = {
            "msg_list": ['>configuration state: <font color=#20a53a> Success </font>\n\n']
        }
        test_task = get_test_msg("Message channel configuration reminders")
        res = self.send_msg(
            test_task.to_mail_msg(test_msg, test_task.the_push_public_data()),
            "Message channel configuration reminders"
        )
        if res is None:
            return None
        return res