mirror of
https://github.com/binwiederhier/ntfy.git
synced 2024-09-01 15:00:25 +02:00
155 lines
3.7 KiB
Go
155 lines
3.7 KiB
Go
package server
|
|
|
|
import (
|
|
_ "embed" // required by go:embed
|
|
"encoding/json"
|
|
"fmt"
|
|
"mime"
|
|
"net"
|
|
"net/smtp"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"heckel.io/ntfy/v2/log"
|
|
"heckel.io/ntfy/v2/util"
|
|
)
|
|
|
|
type mailer interface {
|
|
Send(v *visitor, m *message, to string) error
|
|
Counts() (total int64, success int64, failure int64)
|
|
}
|
|
|
|
type smtpSender struct {
|
|
config *Config
|
|
success int64
|
|
failure int64
|
|
mu sync.Mutex
|
|
}
|
|
|
|
func (s *smtpSender) Send(v *visitor, m *message, to string) error {
|
|
return s.withCount(v, m, func() error {
|
|
host, _, err := net.SplitHostPort(s.config.SMTPSenderAddr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
message, err := formatMail(s.config.BaseURL, v.ip.String(), s.config.SMTPSenderFrom, to, m)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var auth smtp.Auth
|
|
if s.config.SMTPSenderUser != "" {
|
|
auth = smtp.PlainAuth("", s.config.SMTPSenderUser, s.config.SMTPSenderPass, host)
|
|
}
|
|
ev := logvm(v, m).
|
|
Tag(tagEmail).
|
|
Fields(log.Context{
|
|
"email_via": s.config.SMTPSenderAddr,
|
|
"email_user": s.config.SMTPSenderUser,
|
|
"email_to": to,
|
|
})
|
|
if ev.IsTrace() {
|
|
ev.Field("email_body", message).Trace("Sending email")
|
|
} else if ev.IsDebug() {
|
|
ev.Debug("Sending email")
|
|
}
|
|
return smtp.SendMail(s.config.SMTPSenderAddr, auth, s.config.SMTPSenderFrom, []string{to}, []byte(message))
|
|
})
|
|
}
|
|
|
|
func (s *smtpSender) Counts() (total int64, success int64, failure int64) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
return s.success + s.failure, s.success, s.failure
|
|
}
|
|
|
|
func (s *smtpSender) withCount(v *visitor, m *message, fn func() error) error {
|
|
err := fn()
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
if err != nil {
|
|
logvm(v, m).Err(err).Debug("Sending mail failed")
|
|
s.failure++
|
|
} else {
|
|
s.success++
|
|
}
|
|
return err
|
|
}
|
|
|
|
func formatMail(baseURL, senderIP, from, to string, m *message) (string, error) {
|
|
topicURL := baseURL + "/" + m.Topic
|
|
subject := m.Title
|
|
if subject == "" {
|
|
subject = m.Message
|
|
}
|
|
subject = strings.ReplaceAll(strings.ReplaceAll(subject, "\r", ""), "\n", " ")
|
|
message := m.Message
|
|
trailer := ""
|
|
if len(m.Tags) > 0 {
|
|
emojis, tags, err := toEmojis(m.Tags)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(emojis) > 0 {
|
|
subject = strings.Join(emojis, " ") + " " + subject
|
|
}
|
|
if len(tags) > 0 {
|
|
trailer = "Tags: " + strings.Join(tags, ", ")
|
|
}
|
|
}
|
|
if m.Priority != 0 && m.Priority != 3 {
|
|
priority, err := util.PriorityString(m.Priority)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if trailer != "" {
|
|
trailer += "\n"
|
|
}
|
|
trailer += fmt.Sprintf("Priority: %s", priority)
|
|
}
|
|
if trailer != "" {
|
|
message += "\n\n" + trailer
|
|
}
|
|
subject = mime.BEncoding.Encode("utf-8", subject)
|
|
body := `From: "{shortTopicURL}" <{from}>
|
|
To: {to}
|
|
Subject: {subject}
|
|
Content-Type: text/plain; charset="utf-8"
|
|
|
|
{message}
|
|
|
|
--
|
|
This message was sent by {ip} at {time} via {topicURL}`
|
|
body = strings.ReplaceAll(body, "{from}", from)
|
|
body = strings.ReplaceAll(body, "{to}", to)
|
|
body = strings.ReplaceAll(body, "{subject}", subject)
|
|
body = strings.ReplaceAll(body, "{message}", message)
|
|
body = strings.ReplaceAll(body, "{topicURL}", topicURL)
|
|
body = strings.ReplaceAll(body, "{shortTopicURL}", util.ShortTopicURL(topicURL))
|
|
body = strings.ReplaceAll(body, "{time}", time.Unix(m.Time, 0).UTC().Format(time.RFC1123))
|
|
body = strings.ReplaceAll(body, "{ip}", senderIP)
|
|
return body, nil
|
|
}
|
|
|
|
var (
|
|
//go:embed "mailer_emoji_map.json"
|
|
emojisJSON string
|
|
)
|
|
|
|
func toEmojis(tags []string) (emojisOut []string, tagsOut []string, err error) {
|
|
var emojiMap map[string]string
|
|
if err = json.Unmarshal([]byte(emojisJSON), &emojiMap); err != nil {
|
|
return nil, nil, err
|
|
}
|
|
tagsOut = make([]string, 0)
|
|
emojisOut = make([]string, 0)
|
|
for _, t := range tags {
|
|
if emoji, ok := emojiMap[t]; ok {
|
|
emojisOut = append(emojisOut, emoji)
|
|
} else {
|
|
tagsOut = append(tagsOut, t)
|
|
}
|
|
}
|
|
return
|
|
}
|