ReiBot/client.go

package rei

import (
	"reflect"
	"sync"
	"time"
	"unsafe"

	"github.com/RomiChan/syncx"
	tgba "github.com/go-telegram-bot-api/telegram-bot-api/v5"
	log "github.com/sirupsen/logrus"
)

// TelegramClient ...
type TelegramClient struct {
	tgba.BotAPI
	b Bot
}

var (
	logsetter = &sync.Once{}
	clients   = syncx.Map[int64, *TelegramClient]{}
)

func init() {
	logsetter.Do(func() {
		_ = tgba.SetLogger(log.StandardLogger())
	})
}

// NewTelegramClient ...
func NewTelegramClient(c *Bot) (tc TelegramClient) {
	tc.b = *c
	if tc.b.Handler != nil {
		h := reflect.ValueOf(tc.b.Handler).Elem()
		t := h.Type()
		tc.b.handlers = make(map[string]GeneralHandleType, 16)
		for i := 0; i < h.NumField(); i++ {
			f := h.Field(i)
			if f.IsZero() {
				continue
			}
			tp := t.Field(i).Name[2:]
			log.Println("注册处理函数", tp)
			handler := f.Interface()
			tc.b.handlers[tp] = *(*GeneralHandleType)(unsafe.Add(unsafe.Pointer(&handler), unsafe.Sizeof(uintptr(0))))
		}
	}
	return
}

// Connect ...
func (tc *TelegramClient) Connect() {
	log.Println("开始尝试连接到Telegram服务器, token:", tc.b.Token)
	for {
		ba, err := tgba.NewBotAPI(tc.b.Token)
		if err != nil {
			log.Warnln("连接到Telegram服务器时出现错误:", err, ", token:", tc.b.Token)
			time.Sleep(2 * time.Second) // 等待两秒后重新连接
			continue
		}
		tc.BotAPI = *ba
		tc.Debug = tc.b.Debug
		tc.Buffer = tc.b.Buffer
		break
	}
	clients.Store(tc.Self.ID, tc)
	log.Println("连接到Telegram服务器成功, token:", tc.b.Token)
}

// Listen 开始监听事件
func (tc *TelegramClient) Listen() {
	log.Println("开始监听", tc.Self.UserName, "的事件")
	for {
		updates := tc.GetUpdatesChan(tc.b.UpdateConfig)
		for update := range updates {
			tc.processEvent(update)
		}
		log.Warnln("Telegram服务器连接断开...")
		clients.Delete(tc.Self.ID)
		time.Sleep(time.Millisecond * time.Duration(3))
		tc.Connect()
	}
}