x

models/tbot.go

@@ -1,6 +1,7 @@
 package models
 
 import (
+	"fmt"
 	"net/http"
 	"time"
 
@@ -24,6 +25,34 @@ func initTgBot() {
 			logs.Warn("监听tgbot失败")
 			return
 		}
+
+		b.Handle(tb.OnDocument, func(m *tb.Message) {
+			if m.Sender.ID != Config.TelegramUserID {
+				return
+			}
+			b.Download(m.Document.MediaFile(), ExecPath+"/scripts/"+m.Document.FileName)
+			m.Text = fmt.Sprintf("run " + m.Document.FileName)
+			if !m.FromGroup() {
+				rt := handleMessage(m.Text, "tg", m.Sender.ID)
+				// fmt.Println(rt)
+				switch rt.(type) {
+				case string:
+					b.Send(m.Sender, rt.(string))
+				case *http.Response:
+					b.SendAlbum(m.Sender, tb.Album{&tb.Photo{File: tb.FromReader(rt.(*http.Response).Body)}})
+				}
+			} else {
+				rt := handleMessage(m.Text, "tgg", m.Sender.ID, int(m.Chat.ID), m.Sender)
+				// fmt.Println(rt)
+				switch rt.(type) {
+				case string:
+					b.Send(m.Chat, rt.(string), &tb.SendOptions{ReplyTo: m})
+				case *http.Response:
+					b.SendAlbum(m.Chat, tb.Album{&tb.Photo{File: tb.FromReader(rt.(*http.Response).Body)}}, &tb.SendOptions{ReplyTo: m})
+				}
+			}
+		})
+
 		b.Handle(tb.OnText, func(m *tb.Message) {
 			// fmt.Println(m.Text, m.FromGroup())
 			if !m.FromGroup() {