85 lines
1.8 KiB
Go
85 lines
1.8 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"fmt"
|
|
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"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func StartTelegramBot() {
|
|
telegramKey := os.Getenv("telegram")
|
|
if len(telegramKey) == 0 {
|
|
getConfig := config.GetConfig()
|
|
if getConfig == nil {
|
|
return
|
|
}
|
|
botConfig := getConfig.ChatGpt
|
|
if botConfig.Telegram == nil {
|
|
return
|
|
}
|
|
telegramKey = *botConfig.Telegram
|
|
log.Info("读取本地本置文件中的telegram token:", telegramKey)
|
|
} else {
|
|
log.Info("找到环境变量: telegram token:", telegramKey)
|
|
}
|
|
bot, err := tgbotapi.NewBotAPI(telegramKey)
|
|
if err != nil {
|
|
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 := os.Getenv("tg_whitelist")
|
|
|
|
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)
|
|
return
|
|
}
|
|
}
|
|
|
|
responseMsg := telegram.Handle(text)
|
|
if responseMsg == nil {
|
|
continue
|
|
}
|
|
msg := tgbotapi.NewMessage(chatID, *responseMsg)
|
|
send, err := bot.Send(msg)
|
|
if err != nil {
|
|
log.Errorf("发送消息出错:%s", err.Error())
|
|
return
|
|
}
|
|
fmt.Println(send.Text)
|
|
}
|
|
}
|