Merge pull request #14 from J3n5en/master

添加telegram消息推送功能

README.md

@@ -66,6 +66,10 @@ use [https://www.mail-tester.com/](https://www.mail-tester.com/) for checking.
 Open the `config/config.json` file in the run directory, edit a few configuration items at the beginning of `weChatPush`
 and restart the service.
 
+## 6、Telegram Message Push
+Create bot and get token from [BotFather](https://t.me/BotFather)
+Open the `config/config.json` file in the run directory, edit a few configuration items at the beginning of `tg`and restart the service.
+
 # For Developer
 
 ## Project Framework

README_CN.md

@@ -67,6 +67,9 @@ PMail是一个追求极简部署流程、极致资源占用的个人域名邮箱
 
 打开运行目录下的 `config/config.json`文件，编辑 `weChatPush` 开头的几个配置项，重启服务即可。
 
+## 6、Telegram推送
+从 [BotFather](https://t.me/BotFather) 创建并获取令牌机器人。 打开运行目录下的 config/config.json 文件，编辑 `tg` 开头的几个配置项，重启服务即可。
+
 # 参与开发
 
 ## 项目架构

server/config/config.dev.json

@@ -12,6 +12,8 @@
   "weChatPushSecret": "",
   "weChatPushTemplateId": "",
   "weChatPushUserId": "",
+  "tgChatId": "",
+  "tgBotToken": "",
   "isInit": true,
   "httpsEnabled": 2
 }

server/config/config.go

@@ -24,6 +24,8 @@ type Config struct {
 	WeChatPushSecret     string            `json:"weChatPushSecret"`
 	WeChatPushTemplateId string            `json:"weChatPushTemplateId"`
 	WeChatPushUserId     string            `json:"weChatPushUserId"`
+	TgBotToken           string            `json:"tgBotToken"`
+	TgChatId             string            `json:"tgChatId"`
 	IsInit               bool              `json:"isInit"`
 	HttpsEnabled         int               `json:"httpsEnabled"` //后台页面是否启用https，0默认（启用），1启用，2不启用
 	Tables               map[string]string `json:"-"`

server/config/config.json

@@ -12,6 +12,8 @@
   "weChatPushSecret": "",
   "weChatPushTemplateId": "",
   "weChatPushUserId": "",
+  "tgChatId": "",
+  "tgBotToken": "",
   "isInit": true,
   "httpsEnabled": 2
 }

server/config/config_mysql.json

@@ -12,6 +12,8 @@
   "weChatPushSecret": "",
   "weChatPushTemplateId": "",
   "weChatPushUserId": "",
+  "tgChatId": "",
+  "tgBotToken": "",
   "isInit": true,
   "httpsEnabled": 2
 }

server/hooks/base.go

@@ -3,6 +3,7 @@ package hooks
 import (
 	"pmail/dto"
 	"pmail/dto/parsemail"
+	"pmail/hooks/telegram_push"
 	"pmail/hooks/wechat_push"
 )
 
@@ -24,5 +25,6 @@ var HookList []EmailHook
 func Init() {
 	HookList = []EmailHook{
 		wechat_push.NewWechatPushHook(),
+		telegram_push.NewTelegramPushHook(),
 	}
 }

server/hooks/telegram_push/telegram_push.go

@@ -0,0 +1,96 @@
+package telegram_push
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"pmail/config"
+	"pmail/dto"
+	"pmail/dto/parsemail"
+	"strings"
+
+	log "github.com/sirupsen/logrus"
+)
+
+type TelegramPushHook struct {
+	chatId       string
+	botToken     string
+	httpsEnabled int
+	webDomain    string
+}
+
+func (w *TelegramPushHook) SendBefore(ctx *dto.Context, email *parsemail.Email) {
+
+}
+
+func (w *TelegramPushHook) SendAfter(ctx *dto.Context, email *parsemail.Email, err map[string]error) {
+
+}
+
+func (w *TelegramPushHook) ReceiveParseBefore(email []byte) {
+
+}
+
+func (w *TelegramPushHook) ReceiveParseAfter(email *parsemail.Email) {
+	if w.chatId == "" || w.botToken == "" {
+		return
+	}
+
+	w.sendUserMsg(nil, email)
+}
+
+type SendMessageRequest struct {
+	ChatID      string      `json:"chat_id"`
+	Text        string      `json:"text"`
+	ReplyMarkup ReplyMarkup `json:"reply_markup"`
+	ParseMode   string      `json:"parse_mode"`
+}
+
+type ReplyMarkup struct {
+	InlineKeyboard [][]InlineKeyboardButton `json:"inline_keyboard"`
+}
+
+type InlineKeyboardButton struct {
+	Text string `json:"text"`
+	URL  string `json:"url"`
+}
+
+func (w *TelegramPushHook) sendUserMsg(ctx *dto.Context, email *parsemail.Email) {
+	url := w.webDomain
+	if w.httpsEnabled > 1 {
+		url = "http://" + url
+	} else {
+		url = "https://" + url
+	}
+	sendMsgReq, _ := json.Marshal(SendMessageRequest{
+		ChatID:    w.chatId,
+		Text:      fmt.Sprintf("📧<b>%s</b>&#60;%s&#62;\n\n%s", email.Subject, email.From.EmailAddress, string(email.Text)),
+		ParseMode: "HTML",
+		ReplyMarkup: ReplyMarkup{
+			InlineKeyboard: [][]InlineKeyboardButton{
+				{
+					{
+						Text: "查收邮件",
+						URL:  url,
+					},
+				},
+			},
+		},
+	})
+
+	_, err := http.Post(fmt.Sprintf("https://api.telegram.org/bot%s/sendMessage", w.botToken), "application/json", strings.NewReader(string(sendMsgReq)))
+	if err != nil {
+		log.WithContext(ctx).Errorf("telegram push error %+v", err)
+	}
+
+}
+func NewTelegramPushHook() *TelegramPushHook {
+	ret := &TelegramPushHook{
+		botToken:     config.Instance.TgBotToken,
+		chatId:       config.Instance.TgChatId,
+		webDomain:    config.Instance.WebDomain,
+		httpsEnabled: config.Instance.HttpsEnabled,
+	}
+	return ret
+
+}

server/hooks/telegram_push/telegram_push_test.go

@@ -0,0 +1,21 @@
+package telegram_push
+
+import (
+	"pmail/config"
+	"pmail/dto/parsemail"
+	"testing"
+)
+
+func testInit() {
+
+	config.Init()
+
+}
+func TestWeChatPushHook_ReceiveParseAfter(t *testing.T) {
+	testInit()
+
+	w := NewTelegramPushHook()
+	w.ReceiveParseAfter(&parsemail.Email{Subject: "标题", Text: []byte("文本内容"), From: &parsemail.User{
+		EmailAddress: "hello@gmail.com",
+	}})
+}