From 9bfa21acdd75c8fdce72f67cef6a994820e31033 Mon Sep 17 00:00:00 2001
From: cdle <798731886@qq.com>
Date: Fri, 20 Aug 2021 11:41:58 +0800
Subject: [PATCH] x

---
 models/tbot.go | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/models/tbot.go b/models/tbot.go
index e1be994..b21a6ed 100644
--- a/models/tbot.go
+++ b/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() {