fumiama/ReiBot
1
0
Fork 0
mirror of https://github.com/fumiama/ReiBot.git synced 2026-06-05 00:50:25 +08:00
Code Activity

init

Browse Source
This commit is contained in:
源文雨
2022-05-31 21:09:38 +08:00
parent caaf7035d9
commit e37701840d
8 changed files with 297 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
README.md
View File

@@ -1,2 +1,63 @@
# ReiBot
Lightweight Telegram bot framework
## Instructions
This framework is a simple wrapper for [go-telegram-bot-api](https://github.com/go-telegram-bot-api/telegram-bot-api), aiming to make the event processing easier.
## Example
See under `example` folder or below.
![example](https://user-images.githubusercontent.com/41315874/171180885-c888a031-7797-4b4b-a232-9ff23f031b32.png)
```go
package main
import (
"log"
"strings"
rei "github.com/fumiama/ReiBot"
tgba "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)
func main() {
rei.Run(rei.Bot{
Token: "",
Buffer: 256,
UpdateConfig: tgba.UpdateConfig{
Offset: 0,
Limit: 0,
Timeout: 60,
},
Debug: true,
Handler: rei.Handler{
OnMessage: func(updateid int, bot *rei.TelegramClient, msg *tgba.Message) {
if len(msg.Text) <= len("测试") {
return
}
if !strings.HasPrefix(msg.Text, "测试") {
return
}
_, err := bot.Send(tgba.NewMessage(msg.Chat.ID, msg.Text[len("测试"):]))
if err != nil {
log.Println("[ERRO]", err)
}
},
OnEditedMessage: func(updateid int, bot *rei.TelegramClient, msg *tgba.Message) {
if len(msg.Text) <= len("测试") {
return
}
if !strings.HasPrefix(msg.Text, "测试") {
return
}
_, err := bot.Send(tgba.NewMessage(msg.Chat.ID, "已编辑："+msg.Text[len("测试"):]))
if err != nil {
log.Println("[ERRO]", err)
}
},
},
})
}
```

52
bot.go Normal file
View File

@@ -0,0 +1,52 @@
// Package rei ReiBot created in 2022.5.31
package rei
import (
tgba "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)
// Bot bot 的配置
type Bot struct {
// Token bot 的 token
// see https://core.telegram.org/bots#3-how-do-i-create-a-bot
Token string `json:"token"`
// Buffer 控制消息队列的长度
Buffer int `json:"buffer"`
// UpdateConfig 配置消息获取
tgba.UpdateConfig
// Debug 控制调试信息的输出与否
Debug bool `json:"debug"`
// Handler 注册对各种事件的处理
Handler Handler
}
// Start clients without blocking
func Start(bots ...Bot) {
for _, c := range bots {
tc := NewTelegramClient(&c)
tc.Connect()
go tc.Listen()
}
}
// Run clients and block self in listening last one
func Run(bots ...Bot) {
var tc TelegramClient
switch len(bots) {
case 0:
return
case 1:
c := bots[0]
tc = NewTelegramClient(&c)
default:
for _, c := range bots[:len(bots)-1] {
tc := NewTelegramClient(&c)
tc.Connect()
go tc.Listen()
}
c := bots[len(bots)-1]
tc = NewTelegramClient(&c)
}
tc.Connect()
tc.Listen()
}

52
client.go Normal file
View File

@@ -0,0 +1,52 @@
package rei
import (
"log"
"time"
tgba "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)
// TelegramClient ...
type TelegramClient struct {
tgba.BotAPI
b Bot
}
// NewTelegramClient ...
func NewTelegramClient(c *Bot) (tc TelegramClient) {
tc.b = *c
return
}
// Connect ...
func (tc *TelegramClient) Connect() {
log.Println("[INFO] 开始尝试连接到Telegram服务器, token:", tc.b.Token)
for {
ba, err := tgba.NewBotAPI(tc.b.Token)
if err != nil {
log.Println("[WARN] 连接到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
}
log.Println("[INFO] 连接到Telegram服务器成功, token:", tc.b.Token)
}
// Listen 开始监听事件
func (tc *TelegramClient) Listen() {
log.Println("[INFO] 开始监听", tc.Self.UserName, "的事件")
for {
updates := tc.GetUpdatesChan(tc.b.UpdateConfig)
for update := range updates {
tc.processEvent(update)
}
log.Println("[WARN] Telegram服务器连接断开...")
time.Sleep(time.Millisecond * time.Duration(3))
tc.Connect()
}
}

38
event.go Normal file
View File

@@ -0,0 +1,38 @@
package rei
import (
"log"
"reflect"
"unsafe"
tgba "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)
// Event ...
type Event struct {
// Type is the non-null field name in Update
Type string
// UpdateID is the update's unique identifier.
UpdateID int
// Value is the non-null field value in Update
Value reflect.Value
}
func (tc *TelegramClient) processEvent(update tgba.Update) {
v := reflect.ValueOf(&update).Elem()
t := reflect.ValueOf(&update).Elem().Type()
h := reflect.ValueOf(tc.b.Handler)
for i := 1; i < v.NumField(); i++ {
f := v.Field(i)
if f.IsZero() {
continue
}
m := h.FieldByName("On" + t.Field(i).Name)
if m.IsZero() {
continue
}
log.Println("[INFO] processEvent call", "On"+t.Field(i).Name)
handler := m.Interface()
go (*(*GeneralHandleType)(unsafe.Add(unsafe.Pointer(&handler), unsafe.Sizeof(uintptr(0)))))(update.UpdateID, tc, f.UnsafePointer())
}
}

48
example/main.go Normal file
View File

@@ -0,0 +1,48 @@
package main
import (
"log"
"strings"
rei "github.com/fumiama/ReiBot"
tgba "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)
func main() {
rei.Run(rei.Bot{
Token: "",
Buffer: 256,
UpdateConfig: tgba.UpdateConfig{
Offset: 0,
Limit: 0,
Timeout: 60,
},
Debug: true,
Handler: rei.Handler{
OnMessage: func(updateid int, bot *rei.TelegramClient, msg *tgba.Message) {
if len(msg.Text) <= len("测试") {
return
}
if !strings.HasPrefix(msg.Text, "测试") {
return
}
_, err := bot.Send(tgba.NewMessage(msg.Chat.ID, msg.Text[len("测试"):]))
if err != nil {
log.Println("[ERRO]", err)
}
},
OnEditedMessage: func(updateid int, bot *rei.TelegramClient, msg *tgba.Message) {
if len(msg.Text) <= len("测试") {
return
}
if !strings.HasPrefix(msg.Text, "测试") {
return
}
_, err := bot.Send(tgba.NewMessage(msg.Chat.ID, "已编辑："+msg.Text[len("测试"):]))
if err != nil {
log.Println("[ERRO]", err)
}
},
},
})
}

5
go.mod Normal file
View File

@@ -0,0 +1,5 @@
module github.com/fumiama/ReiBot
go 1.18
require github.com/go-telegram-bot-api/telegram-bot-api/v5 v5.5.1

2
go.sum Normal file
View File

@@ -0,0 +1,2 @@
github.com/go-telegram-bot-api/telegram-bot-api/v5 v5.5.1 h1:wG8n/XJQ07TmjbITcGiUaOtXxdrINDz1b0J1w0SzqDc=
github.com/go-telegram-bot-api/telegram-bot-api/v5 v5.5.1/go.mod h1:A2S0CWkNylc2phvKXWBBdD3K0iGnDBGbzRpISP2zBl8=

39
handler.go Normal file
View File

@@ -0,0 +1,39 @@
package rei
import (
"unsafe"
tgba "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)
type GeneralHandleType func(int, *TelegramClient, unsafe.Pointer)
type Handler struct {
OnMessage func(updateid int, bot *TelegramClient, msg *tgba.Message)
OnEditedMessage func(updateid int, bot *TelegramClient, msg *tgba.Message)
OnChannelPost func(updateid int, bot *TelegramClient, msg *tgba.Message)
OnEditedChannelPost func(updateid int, bot *TelegramClient, msg *tgba.Message)
OnInlineQuery func(updateid int, bot *TelegramClient, q *tgba.InlineQuery)
OnChosenInlineResult func(updateid int, bot *TelegramClient, r *tgba.ChosenInlineResult)
OnCallbackQuery func(updateid int, bot *TelegramClient, q *tgba.CallbackQuery)
OnShippingQuery func(updateid int, bot *TelegramClient, q *tgba.ShippingQuery)
OnPreCheckoutQuery func(updateid int, bot *TelegramClient, q *tgba.PreCheckoutQuery)
OnPoll func(updateid int, bot *TelegramClient, p *tgba.Poll)
OnPollAnswer func(updateid int, bot *TelegramClient, pa *tgba.PollAnswer)
OnMyChatMember func(updateid int, bot *TelegramClient, m *tgba.ChatMemberUpdated)
OnChatMember func(updateid int, bot *TelegramClient, m *tgba.ChatMemberUpdated)
OnChatJoinRequest func(updateid int, bot *TelegramClient, r *tgba.ChatJoinRequest)
}