103 lines
2.1 KiB
Go
103 lines
2.1 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/wechatgpt/wechatbot/config"
|
|
"github.com/wechatgpt/wechatbot/handler/telegram"
|
|
"github.com/wechatgpt/wechatbot/utils"
|
|
)
|
|
|
|
func StartTelegramBot() {
|
|
telegramKey := config.GetTelegram()
|
|
if telegramKey == nil {
|
|
log.Info("未找到tg token,不启动tg tot")
|
|
return
|
|
}
|
|
|
|
bot, err := tgbotapi.NewBotAPI(*telegramKey)
|
|
if err != nil {
|
|
log.Error("tg bot 启动失败：", err.Error())
|
|
return
|
|
}
|
|
|
|
bot.Debug = false
|
|
log.Info("Authorized on account: ", bot.Self.UserName)
|
|
u := tgbotapi.NewUpdate(0)
|
|
|
|
updates := bot.GetUpdatesChan(u)
|
|
time.Sleep(time.Millisecond * 500)
|
|
for len(updates) != 0 {
|
|
<-updates
|
|
}
|
|
|
|
for update := range updates {
|
|
if update.Message == nil {
|
|
continue
|
|
}
|
|
|
|
text := update.Message.Text
|
|
chatID := update.Message.Chat.ID
|
|
chatUserName := update.Message.Chat.UserName
|
|
|
|
tgUserNameStr := config.GetTelegramWhitelist()
|
|
if tgUserNameStr != nil {
|
|
tgUserNames := strings.Split(*tgUserNameStr, ",")
|
|
if len(tgUserNames) > 0 {
|
|
found := false
|
|
for _, name := range tgUserNames {
|
|
if name == chatUserName {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !found {
|
|
log.Error("用户设置了私人私用，白名单以外的人不生效: ", chatUserName)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
tgKeyWord := config.GetTelegramKeyword()
|
|
var reply *string
|
|
// 如果设置了关键字就以关键字为准，没设置就所有消息都监听
|
|
if tgKeyWord != nil {
|
|
content, key := utils.ContainsI(text, *tgKeyWord)
|
|
if len(key) == 0 {
|
|
continue
|
|
}
|
|
|
|
splitItems := strings.Split(content, key)
|
|
if len(splitItems) < 2 {
|
|
continue
|
|
}
|
|
|
|
requestText := strings.TrimSpace(splitItems[1])
|
|
log.Println("问题：", requestText)
|
|
reply = telegram.Handle(requestText)
|
|
} else {
|
|
reply = telegram.Handle(text)
|
|
}
|
|
|
|
if reply == nil {
|
|
continue
|
|
}
|
|
|
|
msg := tgbotapi.NewMessage(chatID, *reply)
|
|
send, err := bot.Send(msg)
|
|
if err != nil {
|
|
log.Errorf("发送消息出错:%s", err.Error())
|
|
continue
|
|
}
|
|
|
|
fmt.Println(send.Text)
|
|
}
|
|
|
|
select {}
|
|
}
|