139 lines
3.3 KiB
Go
139 lines
3.3 KiB
Go
package message
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"crypto/tls"
|
|
"encoding/base64"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/songquanpeng/one-api/common/config"
|
|
"net"
|
|
"net/smtp"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type loginAuth struct {
|
|
username, password string
|
|
}
|
|
|
|
func LoginAuth(username, password string) smtp.Auth {
|
|
return &loginAuth{username, password}
|
|
}
|
|
|
|
func (a *loginAuth) Start(_ *smtp.ServerInfo) (string, []byte, error) {
|
|
return "LOGIN", []byte(a.username), nil
|
|
}
|
|
|
|
func (a *loginAuth) Next(fromServer []byte, more bool) ([]byte, error) {
|
|
if more {
|
|
switch string(fromServer) {
|
|
case "Username:":
|
|
return []byte(a.username), nil
|
|
case "Password:":
|
|
return []byte(a.password), nil
|
|
default:
|
|
return nil, errors.New("unknown command from server during login auth")
|
|
}
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func shouldAuth() bool {
|
|
return config.SMTPAccount != "" || config.SMTPToken != ""
|
|
}
|
|
|
|
func SendEmail(subject string, receiver string, content string) error {
|
|
if receiver == "" {
|
|
return fmt.Errorf("receiver is empty")
|
|
}
|
|
if config.SMTPFrom == "" { // for compatibility
|
|
config.SMTPFrom = config.SMTPAccount
|
|
}
|
|
encodedSubject := fmt.Sprintf("=?UTF-8?B?%s?=", base64.StdEncoding.EncodeToString([]byte(subject)))
|
|
|
|
// Extract domain from SMTPFrom
|
|
parts := strings.Split(config.SMTPFrom, "@")
|
|
var domain string
|
|
if len(parts) > 1 {
|
|
domain = parts[1]
|
|
}
|
|
// Generate a unique Message-ID
|
|
buf := make([]byte, 16)
|
|
_, err := rand.Read(buf)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
messageId := fmt.Sprintf("<%x@%s>", buf, domain)
|
|
|
|
mail := []byte(fmt.Sprintf("To: %s\r\n"+
|
|
"From: %s<%s>\r\n"+
|
|
"Subject: %s\r\n"+
|
|
"Message-ID: %s\r\n"+ // add Message-ID header to avoid being treated as spam, RFC 5322
|
|
"Date: %s\r\n"+
|
|
"Content-Type: text/html; charset=UTF-8\r\n\r\n%s\r\n",
|
|
receiver, config.SystemName, config.SMTPFrom, encodedSubject, messageId, time.Now().Format(time.RFC1123Z), content))
|
|
|
|
var auth smtp.Auth
|
|
if config.SMTPAuthLoginEnabled {
|
|
auth = LoginAuth(config.SMTPAccount, config.SMTPToken)
|
|
} else {
|
|
auth = smtp.PlainAuth("", config.SMTPAccount, config.SMTPToken, config.SMTPServer)
|
|
}
|
|
|
|
addr := fmt.Sprintf("%s:%d", config.SMTPServer, config.SMTPPort)
|
|
to := strings.Split(receiver, ";")
|
|
|
|
if config.SMTPPort == 465 || !shouldAuth() {
|
|
// need advanced client
|
|
var conn net.Conn
|
|
var err error
|
|
if config.SMTPPort == 465 {
|
|
tlsConfig := &tls.Config{
|
|
InsecureSkipVerify: true,
|
|
ServerName: config.SMTPServer,
|
|
}
|
|
conn, err = tls.Dial("tcp", fmt.Sprintf("%s:%d", config.SMTPServer, config.SMTPPort), tlsConfig)
|
|
} else {
|
|
conn, err = net.Dial("tcp", fmt.Sprintf("%s:%d", config.SMTPServer, config.SMTPPort))
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
client, err := smtp.NewClient(conn, config.SMTPServer)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer client.Close()
|
|
if shouldAuth() {
|
|
if err = client.Auth(auth); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if err = client.Mail(config.SMTPFrom); err != nil {
|
|
return err
|
|
}
|
|
receiverEmails := strings.Split(receiver, ";")
|
|
for _, receiver := range receiverEmails {
|
|
if err = client.Rcpt(receiver); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
w, err := client.Data()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = w.Write(mail)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = w.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
err = smtp.SendMail(addr, auth, config.SMTPAccount, to, mail)
|
|
}
|
|
return err
|
|
}
|