init

2022-04-04 16:01:36 +08:00
commit 20c576b9bc
72 changed files with 4680 additions and 0 deletions
@@ -0,0 +1,58 @@
+package telegram
+
+import (
+	"github.com/assimon/luuu/config"
+	"github.com/assimon/luuu/util/log"
+	tb "gopkg.in/telebot.v3"
+	"gopkg.in/telebot.v3/middleware"
+	"time"
+)
+
+var bots *tb.Bot
+
+// BotStart 机器人启动
+func BotStart() {
+	var err error
+	botSetting := tb.Settings{
+		Token:  config.TgBotToken,
+		Poller: &tb.LongPoller{Timeout: 10 * time.Second},
+	}
+	if config.TgProxy != "" {
+		botSetting.URL = config.TgProxy
+	}
+	bots, err = tb.NewBot(botSetting)
+	if err != nil {
+		log.Sugar.Error(err.Error())
+		return
+	}
+	err = bots.SetCommands(Cmds)
+	if err != nil {
+		log.Sugar.Error(err.Error())
+		return
+	}
+	RegisterHandle()
+	bots.Start()
+}
+
+// RegisterHandle 注册处理器
+func RegisterHandle() {
+	adminOnly := bots.Group()
+	adminOnly.Use(middleware.Whitelist(config.TgManage))
+	adminOnly.Handle(START_CMD, WalletList)
+	adminOnly.Handle(tb.OnText, OnTextMessageHandle)
+}
+
+// SendToBot 主动发送消息机器人消息
+func SendToBot(msg string) {
+	go func() {
+		user := tb.User{
+			ID: config.TgManage,
+		}
+		_, err := bots.Send(&user, msg, &tb.SendOptions{
+			ParseMode: tb.ModeHTML,
+		})
+		if err != nil {
+			log.Sugar.Error(err)
+		}
+	}()
+}