134 lines
2.5 KiB
Go
134 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"net/http"
|
|
"os"
|
|
"pmail/config"
|
|
"pmail/dto/parsemail"
|
|
"pmail/hooks/framework"
|
|
"pmail/utils/context"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type WebPushHook struct {
|
|
url string
|
|
token string
|
|
}
|
|
|
|
// EmailData 用于存储解析后的邮件数据
|
|
type EmailData struct {
|
|
From string `json:"from"`
|
|
To []string `json:"to"`
|
|
Subject string `json:"subject"`
|
|
Body string `json:"body"`
|
|
Token string `json:"token"`
|
|
}
|
|
|
|
func (w *WebPushHook) SendBefore(ctx *context.Context, email *parsemail.Email) {
|
|
|
|
}
|
|
|
|
func (w *WebPushHook) SendAfter(ctx *context.Context, email *parsemail.Email, err map[string]error) {
|
|
|
|
}
|
|
|
|
func (w *WebPushHook) ReceiveParseBefore(ctx *context.Context, email *[]byte) {
|
|
|
|
}
|
|
|
|
func (w *WebPushHook) ReceiveParseAfter(ctx *context.Context, email *parsemail.Email) {
|
|
if w.url == "" {
|
|
return
|
|
}
|
|
|
|
content := string(email.Text)
|
|
|
|
if content == "" {
|
|
content = email.Subject
|
|
}
|
|
|
|
webhookURL := w.url // 替换为您的 Webhook URL
|
|
|
|
to := make([]string, len(email.To))
|
|
for i, user := range email.To {
|
|
to[i] = user.EmailAddress
|
|
}
|
|
|
|
data := EmailData{
|
|
From: email.From.EmailAddress,
|
|
To: to,
|
|
Subject: email.Subject,
|
|
Body: content,
|
|
Token: w.token,
|
|
}
|
|
|
|
jsonData, err := json.Marshal(data)
|
|
|
|
if err != nil {
|
|
log.WithContext(ctx).Errorf("web push error %+v", err)
|
|
}
|
|
|
|
resp, err := http.Post(webhookURL, "application/json", bytes.NewBuffer(jsonData))
|
|
if err != nil {
|
|
log.WithContext(ctx).Errorf("web push error %+v", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
}
|
|
|
|
type Config struct {
|
|
WebPushUrl string `json:"webPushUrl"`
|
|
WebPushToken string `json:"webPushToken"`
|
|
}
|
|
|
|
func NewWebPushHook() *WebPushHook {
|
|
var cfgData []byte
|
|
var err error
|
|
|
|
cfgData, err = os.ReadFile("../config/config.json")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
var mainConfig *config.Config
|
|
err = json.Unmarshal(cfgData, &mainConfig)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
var pluginConfig *Config
|
|
if _, err := os.Stat("./web_push_config.json"); err == nil {
|
|
cfgData, err = os.ReadFile("./web_push_config.json")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
err = json.Unmarshal(cfgData, &pluginConfig)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
}
|
|
|
|
token := ""
|
|
pushURL := ""
|
|
if pluginConfig != nil {
|
|
pushURL = pluginConfig.WebPushUrl
|
|
token = pluginConfig.WebPushToken
|
|
} else {
|
|
pushURL = mainConfig.WebPushUrl
|
|
token = mainConfig.WebPushToken
|
|
}
|
|
|
|
ret := &WebPushHook{
|
|
url: pushURL,
|
|
token: token,
|
|
}
|
|
return ret
|
|
|
|
}
|
|
|
|
func main() {
|
|
framework.CreatePlugin(NewWebPushHook()).Run()
|
|
}
|