package mail

import (
	"gitee.com/progress_dream/utools/idhelper"
	"gitee.com/progress_dream/utools/logger"
	"gitee.com/progress_dream/utools/timehelper"
	"gopkg.in/gomail.v2"
)

type Store struct {
	sendCh chan MailItem
}

func NewStore() *Store {
	return &Store{
		sendCh: make(chan MailItem, 100),
	}
}

func (s *Store) Start() {
	for mailItem := range s.sendCh {
		s.sendMailItem(mailItem)
	}
}
func (s *Store) sendMailItem(mailItem MailItem) {
	mailMess := gomail.NewMessage()
	mailMess.SetHeader("From", mailItem.From)
	mailMess.SetHeader("To", mailItem.To...)
	mailMess.SetHeader("Subject", mailItem.Subject)
	mailMess.SetBody("text/html", mailItem.Body)
	dialer := gomail.NewDialer(mailItem.sender.host, mailItem.sender.port,
		mailItem.sender.username, mailItem.sender.password)
	err := dialer.DialAndSend(mailMess)
	if err != nil {
		logger.Error("send mail error", err, mailItem, dialer)
	}
}

func (s *Store) SendMail(from string, to []string, subject, body string, sender MailSender) uint64 {
	mailItem := NewMailItem(from, to, subject, body, sender)
	s.addMailItem(mailItem)
	return mailItem.Id
}
func (s *Store) addMailItem(mailItem MailItem) {
	s.sendCh <- mailItem
}

type MailSender struct {
	host     string
	port     int
	username string
	password string
}

func NewMailSender(host string, port int, username, password string) MailSender {
	return MailSender{
		host:     host,
		port:     port,
		username: username,
		password: password,
	}
}

type MailItem struct {
	From    string   `json:"from,omitempty"`
	To      []string `json:"to,omitempty"`
	Subject string   `json:"subject,omitempty"`
	Body    string   `json:"body,omitempty"`
	Id      uint64   `json:"id,omitempty"`
	t       int64
	sender  MailSender
}

func NewMailItem(from string, to []string, subject, body string, sender MailSender) MailItem {
	return MailItem{
		Id:      idhelper.NextId(),
		From:    from,
		To:      to,
		Subject: subject,
		Body:    body,
		t:       timehelper.ServerNowSeconds(),
		sender:  sender,
	}
}
func (m MailItem) GetMailTime() int64 {
	return m.t
}