1.4.5.2

EdgeAPI/internal/senders/mediasenders/media_telegram.go

@@ -0,0 +1,75 @@
+//go:build plus
+
+package mediasenders
+
+import (
+	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
+	"github.com/iwind/TeaGo/types"
+	"net/http"
+	"net/url"
+	"time"
+)
+
+// TelegramMedia Telegram媒介
+type TelegramMedia struct {
+	Token    string `yaml:"token" json:"token"`
+	ProxyURL string `yaml:"proxyURL" json:"proxyURL"`
+}
+
+// NewTelegramMedia 获取新对象
+func NewTelegramMedia() *TelegramMedia {
+	return &TelegramMedia{}
+}
+
+// Send 发送消息
+func (this *TelegramMedia) Send(user string, subject string, body string, productName string, datetime string) (respBytes []byte, err error) {
+	var proxyFunc func(req *http.Request) (*url.URL, error)
+	if len(this.ProxyURL) > 0 {
+		proxyFunc = func(req *http.Request) (*url.URL, error) {
+			return url.Parse(this.ProxyURL)
+		}
+	}
+
+	var httpClient = &http.Client{
+		Timeout: 10 * time.Second,
+		Transport: &http.Transport{
+			MaxIdleConns:          4096,
+			MaxIdleConnsPerHost:   32,
+			MaxConnsPerHost:       32,
+			IdleConnTimeout:       2 * time.Minute,
+			ExpectContinueTimeout: 1 * time.Second,
+			TLSHandshakeTimeout:   0,
+			Proxy:                 proxyFunc,
+		},
+	}
+
+	defer httpClient.CloseIdleConnections()
+
+	bot, err := tgbotapi.NewBotAPIWithClient(this.Token, httpClient)
+	if err != nil {
+		return nil, err
+	}
+
+	bot.Client = httpClient
+
+	var text string
+	if len(productName) > 0 {
+		text += productName
+		if len(datetime) > 0 {
+			text += " AT " + datetime
+		}
+		text += "\n----\n"
+	} else if len(datetime) > 0 {
+		text += "AT " + datetime
+		text += "\n----\n"
+	}
+	text += subject + "\n" + body
+	var msg = tgbotapi.NewMessage(types.Int64(user), text)
+	_, err = bot.Send(msg)
+	return nil, err
+}
+
+// RequireUser 是否需要用户标识
+func (this *TelegramMedia) RequireUser() bool {
+	return true
+}