REPOSITORY / ScuroNeko/Laniakea

Compare commits

DIFF REPOSITORY

Compare commits

...
Author SHA1 Message Date
ScuroNeko 2de46a27c8 v1.0.0 beta 5 2026-03-01 23:40:27 +03:00
ScuroNeko ae7426c36a 1.0.0 beta 4 2026-03-01 23:08:22 +03:00
ScuroNeko 61562e8a3b 1.0.0 beta 3 2026-03-01 23:01:06 +03:00
ScuroNeko a84e24ff25 small fix 2026-02-27 13:53:00 +03:00
ScuroNeko c0a26024f4 v1.0.0 beta 2 2026-02-26 15:15:35 +03:00
ScuroNeko 786da652e6 v1.0.0 beta 1 2026-02-26 15:12:36 +03:00
14 changed files with 267 additions and 94 deletions
+68 -35
View File
@@ -1,16 +1,18 @@
package laniakea package laniakea
import ( import (
"context"
"fmt" "fmt"
"os" "os"
"sort" "sort"
"strconv" "strconv"
"strings" "strings"
"time" "sync"
"git.nix13.pw/scuroneko/extypes" "git.nix13.pw/scuroneko/extypes"
"git.nix13.pw/scuroneko/laniakea/tgapi" "git.nix13.pw/scuroneko/laniakea/tgapi"
"git.nix13.pw/scuroneko/slog" "git.nix13.pw/scuroneko/slog"
"github.com/alitto/pond/v2"
"golang.org/x/time/rate" "golang.org/x/time/rate"
) )
@@ -82,18 +84,20 @@ type Bot[T DbContext] struct {
prefixes []string prefixes []string
runners []Runner[T] runners []Runner[T]
api *tgapi.API api *tgapi.API
uploader *tgapi.Uploader uploader *tgapi.Uploader
dbContext *T dbContext *T
l10n *L10n l10n *L10n
draftProvider *DraftProvider
updateOffset int updateOffsetMu sync.Mutex
updateTypes []tgapi.UpdateType updateOffset int
updateQueue *extypes.Queue[*tgapi.Update] updateTypes []tgapi.UpdateType
updateQueue chan *tgapi.Update
} }
func NewBot[T any](opts *BotOpts) *Bot[T] { func NewBot[T any](opts *BotOpts) *Bot[T] {
updateQueue := extypes.CreateQueue[*tgapi.Update](512) updateQueue := make(chan *tgapi.Update, 512)
var limiter *rate.Limiter var limiter *rate.Limiter
if opts.RateLimit > 0 { if opts.RateLimit > 0 {
@@ -119,6 +123,7 @@ func NewBot[T any](opts *BotOpts) *Bot[T] {
runners: make([]Runner[T], 0), runners: make([]Runner[T], 0),
extraLoggers: make([]*slog.Logger, 0), extraLoggers: make([]*slog.Logger, 0),
l10n: &L10n{}, l10n: &L10n{},
draftProvider: NewRandomDraftProvider(api),
} }
bot.extraLoggers = bot.extraLoggers.Push(api.GetLogger()).Push(uploader.GetLogger()) bot.extraLoggers = bot.extraLoggers.Push(api.GetLogger()).Push(uploader.GetLogger())
@@ -185,13 +190,24 @@ func (bot *Bot[T]) initLoggers(opts *BotOpts) {
} }
} }
func (bot *Bot[T]) GetUpdateOffset() int { return bot.updateOffset } func (bot *Bot[T]) GetUpdateOffset() int {
func (bot *Bot[T]) SetUpdateOffset(offset int) { bot.updateOffset = offset } bot.updateOffsetMu.Lock()
func (bot *Bot[T]) GetUpdateTypes() []tgapi.UpdateType { return bot.updateTypes } defer bot.updateOffsetMu.Unlock()
func (bot *Bot[T]) GetQueue() *extypes.Queue[*tgapi.Update] { return bot.updateQueue } return bot.updateOffset
func (bot *Bot[T]) GetLogger() *slog.Logger { return bot.logger } }
func (bot *Bot[T]) GetDBContext() *T { return bot.dbContext } func (bot *Bot[T]) SetUpdateOffset(offset int) {
func (bot *Bot[T]) L10n(lang, key string) string { return bot.l10n.Translate(lang, key) } bot.updateOffsetMu.Lock()
defer bot.updateOffsetMu.Unlock()
bot.updateOffset = offset
}
func (bot *Bot[T]) GetUpdateTypes() []tgapi.UpdateType { return bot.updateTypes }
func (bot *Bot[T]) GetLogger() *slog.Logger { return bot.logger }
func (bot *Bot[T]) GetDBContext() *T { return bot.dbContext }
func (bot *Bot[T]) L10n(lang, key string) string { return bot.l10n.Translate(lang, key) }
func (bot *Bot[T]) SetDraftProvider(p *DraftProvider) *Bot[T] {
bot.draftProvider = p
return bot
}
type DbLogger[T DbContext] func(db *T) slog.LoggerWriter type DbLogger[T DbContext] func(db *T) slog.LoggerWriter
@@ -235,7 +251,7 @@ func (bot *Bot[T]) Debug(debug bool) *Bot[T] {
func (bot *Bot[T]) AddPlugins(plugin ...*Plugin[T]) *Bot[T] { func (bot *Bot[T]) AddPlugins(plugin ...*Plugin[T]) *Bot[T] {
for _, p := range plugin { for _, p := range plugin {
bot.plugins = append(bot.plugins, *p) bot.plugins = append(bot.plugins, *p)
bot.logger.Debugln(fmt.Sprintf("plugins with name \"%s\" registered", p.Name)) bot.logger.Debugln(fmt.Sprintf("plugins with name \"%s\" registered", p.name))
} }
return bot return bot
} }
@@ -266,7 +282,15 @@ func (bot *Bot[T]) AddL10n(l *L10n) *Bot[T] {
return bot return bot
} }
func (bot *Bot[T]) Run() { func (bot *Bot[T]) enqueueUpdate(u *tgapi.Update) error {
select {
case bot.updateQueue <- u:
return nil
default:
return extypes.QueueFullErr
}
}
func (bot *Bot[T]) RunWithContext(ctx context.Context) {
if len(bot.prefixes) == 0 { if len(bot.prefixes) == 0 {
bot.logger.Fatalln("no prefixes defined") bot.logger.Fatalln("no prefixes defined")
return return
@@ -282,26 +306,35 @@ func (bot *Bot[T]) Run() {
bot.logger.Infoln("Bot running. Press CTRL+C to exit.") bot.logger.Infoln("Bot running. Press CTRL+C to exit.")
go func() { go func() {
for { for {
_, err := bot.Updates() select {
if err != nil { case <-ctx.Done():
bot.logger.Errorln(err) return
default:
updates, err := bot.Updates()
if err != nil {
bot.logger.Errorln(err)
continue
}
for _, u := range updates {
select {
case bot.updateQueue <- new(u):
case <-ctx.Done():
return
}
}
} }
} }
}() }()
for { pool := pond.NewPool(16)
queue := bot.updateQueue for update := range bot.updateQueue {
if queue.IsEmpty() { update := update
time.Sleep(time.Millisecond * 25) pool.Submit(func() {
continue bot.handle(update)
} })
u := queue.Dequeue()
if u == nil {
bot.logger.Errorln("update is nil")
continue
}
bot.handle(u)
} }
} }
func (bot *Bot[T]) Run() {
bot.RunWithContext(context.Background())
}
+5 -1
View File
@@ -27,7 +27,7 @@ func generateBotCommand[T any](cmd Command[T]) tgapi.BotCommand {
func generateBotCommandForPlugin[T any](pl Plugin[T]) []tgapi.BotCommand { func generateBotCommandForPlugin[T any](pl Plugin[T]) []tgapi.BotCommand {
commands := make([]tgapi.BotCommand, 0) commands := make([]tgapi.BotCommand, 0)
for _, cmd := range pl.Commands { for _, cmd := range pl.commands {
if cmd.skipAutoCmd { if cmd.skipAutoCmd {
continue continue
} }
@@ -46,6 +46,10 @@ func (bot *Bot[T]) AutoGenerateCommands() error {
commands := make([]tgapi.BotCommand, 0) commands := make([]tgapi.BotCommand, 0)
for _, pl := range bot.plugins { for _, pl := range bot.plugins {
if pl.skipAutoCmd {
continue
}
commands = append(commands, generateBotCommandForPlugin(pl)...) commands = append(commands, generateBotCommandForPlugin(pl)...)
} }
if len(commands) > 100 { if len(commands) > 100 {
+101
View File
@@ -0,0 +1,101 @@
package laniakea
import (
"math"
"math/rand/v2"
"sync/atomic"
"git.nix13.pw/scuroneko/laniakea/tgapi"
)
type draftIdGenerator interface {
Next() uint64
}
type RandomDraftIdGenerator struct {
draftIdGenerator
}
func (g *RandomDraftIdGenerator) Next() uint64 {
return rand.Uint64N(math.MaxUint64)
}
type LinearDraftIdGenerator struct {
draftIdGenerator
lastId uint64
}
func (g *LinearDraftIdGenerator) Next() uint64 {
return atomic.AddUint64(&g.lastId, 1)
}
type DraftProvider struct {
api *tgapi.API
chatID int
messageThreadID int
parseMode tgapi.ParseMode
entities []tgapi.MessageEntity
drafts map[uint64]*Draft
generator draftIdGenerator
}
type Draft struct {
api *tgapi.API
chatID int
messageThreadID int
parseMode tgapi.ParseMode
entities []tgapi.MessageEntity
ID uint64
Message string
}
func NewRandomDraftProvider(api *tgapi.API) *DraftProvider {
return &DraftProvider{
api: api, generator: &RandomDraftIdGenerator{},
drafts: make(map[uint64]*Draft),
}
}
func NewLinearDraftProvider(api *tgapi.API, startValue uint64) *DraftProvider {
return &DraftProvider{
api: api,
generator: &LinearDraftIdGenerator{lastId: startValue},
drafts: make(map[uint64]*Draft),
}
}
func (d *DraftProvider) NewDraft() *Draft {
id := d.generator.Next()
draft := &Draft{d.api, d.chatID, d.messageThreadID, d.parseMode, d.entities, id, ""}
d.drafts[id] = draft
return draft
}
func (d *Draft) Push(newText string) error {
d.Message += newText
params := tgapi.SendMessageDraftP{
ChatID: d.chatID,
DraftID: d.ID,
Text: d.Message,
ParseMode: d.parseMode,
Entities: d.entities,
}
if d.messageThreadID > 0 {
params.MessageThreadID = d.messageThreadID
}
_, err := d.api.SendMessageDraft(params)
return err
}
func (d *Draft) Flush() error {
params := tgapi.SendMessageP{
ChatID: d.chatID,
ParseMode: d.parseMode,
Entities: d.entities,
}
if d.messageThreadID > 0 {
params.MessageThreadID = d.messageThreadID
}
_, err := d.api.SendMessage(params)
return err
}
+2 -2
View File
@@ -3,13 +3,13 @@ module git.nix13.pw/scuroneko/laniakea
go 1.26 go 1.26
require ( require (
git.nix13.pw/scuroneko/extypes v1.2.0 git.nix13.pw/scuroneko/extypes v1.2.1
git.nix13.pw/scuroneko/slog v1.0.2 git.nix13.pw/scuroneko/slog v1.0.2
github.com/alitto/pond/v2 v2.6.2
golang.org/x/time v0.14.0 golang.org/x/time v0.14.0
) )
require ( require (
github.com/alitto/pond/v2 v2.6.2 // indirect
github.com/fatih/color v1.18.0 // indirect github.com/fatih/color v1.18.0 // indirect
github.com/mattn/go-colorable v0.1.14 // indirect github.com/mattn/go-colorable v0.1.14 // indirect
github.com/mattn/go-isatty v0.0.20 // indirect github.com/mattn/go-isatty v0.0.20 // indirect
+2 -2
View File
@@ -1,5 +1,5 @@
git.nix13.pw/scuroneko/extypes v1.2.0 h1:2n2hD6KsMAted+6MGhAyeWyli2Qzc9G2y+pQNB7C1dM= git.nix13.pw/scuroneko/extypes v1.2.1 h1:IYrOjnWKL2EAuJYtYNa+luB1vBe6paE8VY/YD+5/RpQ=
git.nix13.pw/scuroneko/extypes v1.2.0/go.mod h1:uZVs8Yo3RrYAG9dMad6qR6lsYY67t+459D9c65QAYAw= git.nix13.pw/scuroneko/extypes v1.2.1/go.mod h1:uZVs8Yo3RrYAG9dMad6qR6lsYY67t+459D9c65QAYAw=
git.nix13.pw/scuroneko/slog v1.0.2 h1:vZyUROygxC2d5FJHUQM/30xFEHY1JT/aweDZXA4rm2g= git.nix13.pw/scuroneko/slog v1.0.2 h1:vZyUROygxC2d5FJHUQM/30xFEHY1JT/aweDZXA4rm2g=
git.nix13.pw/scuroneko/slog v1.0.2/go.mod h1:3Qm2wzkR5KjwOponMfG7TcGSDjmYaFqRAmLvSPTuWJI= git.nix13.pw/scuroneko/slog v1.0.2/go.mod h1:3Qm2wzkR5KjwOponMfG7TcGSDjmYaFqRAmLvSPTuWJI=
github.com/alitto/pond/v2 v2.6.2 h1:Sphe40g0ILeM1pA2c2K+Th0DGU+pt0A/Kprr+WB24Pw= github.com/alitto/pond/v2 v2.6.2 h1:Sphe40g0ILeM1pA2c2K+Th0DGU+pt0A/Kprr+WB24Pw=
+9 -3
View File
@@ -9,7 +9,13 @@ import (
) )
func (bot *Bot[T]) handle(u *tgapi.Update) { func (bot *Bot[T]) handle(u *tgapi.Update) {
ctx := &MsgContext{Update: *u, Api: bot.api, botLogger: bot.logger, errorTemplate: bot.errorTemplate, l10n: bot.l10n} ctx := &MsgContext{
Update: *u, Api: bot.api,
botLogger: bot.logger,
errorTemplate: bot.errorTemplate,
l10n: bot.l10n,
draftProvider: bot.draftProvider,
}
for _, middleware := range bot.middlewares { for _, middleware := range bot.middlewares {
middleware.Execute(ctx, bot.dbContext) middleware.Execute(ctx, bot.dbContext)
} }
@@ -46,7 +52,7 @@ func (bot *Bot[T]) handleMessage(update *tgapi.Update, ctx *MsgContext) {
text = strings.TrimSpace(text[len(prefix):]) text = strings.TrimSpace(text[len(prefix):])
for _, plugin := range bot.plugins { for _, plugin := range bot.plugins {
for cmd := range plugin.Commands { for cmd := range plugin.commands {
if !strings.HasPrefix(text, cmd) { if !strings.HasPrefix(text, cmd) {
continue continue
} }
@@ -96,7 +102,7 @@ func (bot *Bot[T]) handleCallback(update *tgapi.Update, ctx *MsgContext) {
ctx.Args = data.Args ctx.Args = data.Args
for _, plugin := range bot.plugins { for _, plugin := range bot.plugins {
_, ok := plugin.Payloads[data.Command] _, ok := plugin.payloads[data.Command]
if !ok { if !ok {
continue continue
} }
+5 -8
View File
@@ -19,14 +19,8 @@ func (bot *Bot[T]) Updates() ([]tgapi.Update, error) {
return nil, err return nil, err
} }
for _, u := range updates { if bot.RequestLogger != nil {
bot.SetUpdateOffset(u.UpdateID + 1) for _, u := range updates {
err = bot.GetQueue().Enqueue(&u)
if err != nil {
return nil, err
}
if bot.RequestLogger != nil {
j, err := json.Marshal(u) j, err := json.Marshal(u)
if err != nil { if err != nil {
bot.GetLogger().Error(err) bot.GetLogger().Error(err)
@@ -34,5 +28,8 @@ func (bot *Bot[T]) Updates() ([]tgapi.Update, error) {
bot.RequestLogger.Debugf("UPDATE %s\n", j) bot.RequestLogger.Debugf("UPDATE %s\n", j)
} }
} }
if len(updates) > 0 {
bot.SetUpdateOffset(updates[len(updates)-1].UpdateID + 1)
}
return updates, err return updates, err
} }
+28 -23
View File
@@ -24,6 +24,7 @@ type MsgContext struct {
errorTemplate string errorTemplate string
botLogger *slog.Logger botLogger *slog.Logger
l10n *L10n l10n *L10n
draftProvider *DraftProvider
} }
type AnswerMessage struct { type AnswerMessage struct {
@@ -77,6 +78,7 @@ func (ctx *MsgContext) editPhotoText(messageId int, text string, kb *InlineKeybo
if kb != nil { if kb != nil {
params.ReplyMarkup = kb.Get() params.ReplyMarkup = kb.Get()
} }
msg, _, err := ctx.Api.EditMessageCaption(params) msg, _, err := ctx.Api.EditMessageCaption(params)
if err != nil { if err != nil {
ctx.botLogger.Errorln(err) ctx.botLogger.Errorln(err)
@@ -105,6 +107,9 @@ func (ctx *MsgContext) answer(text string, keyboard *InlineKeyboard) *AnswerMess
if keyboard != nil { if keyboard != nil {
params.ReplyMarkup = keyboard.Get() params.ReplyMarkup = keyboard.Get()
} }
if ctx.Msg.MessageThreadID > 0 {
params.MessageThreadID = ctx.Msg.MessageThreadID
}
msg, err := ctx.Api.SendMessage(params) msg, err := ctx.Api.SendMessage(params)
if err != nil { if err != nil {
@@ -135,6 +140,10 @@ func (ctx *MsgContext) answerPhoto(photoId, text string, kb *InlineKeyboard) *An
if kb != nil { if kb != nil {
params.ReplyMarkup = kb.Get() params.ReplyMarkup = kb.Get()
} }
if ctx.Msg.MessageThreadID > 0 {
params.MessageThreadID = ctx.Msg.MessageThreadID
}
msg, err := ctx.Api.SendPhoto(params) msg, err := ctx.Api.SendPhoto(params)
if err != nil { if err != nil {
ctx.botLogger.Errorln(err) ctx.botLogger.Errorln(err)
@@ -162,12 +171,8 @@ func (ctx *MsgContext) delete(messageId int) {
ctx.botLogger.Errorln(err) ctx.botLogger.Errorln(err)
} }
} }
func (m *AnswerMessage) Delete() { func (m *AnswerMessage) Delete() { m.ctx.delete(m.MessageID) }
m.ctx.delete(m.MessageID) func (ctx *MsgContext) CallbackDelete() { ctx.delete(ctx.CallbackMsgId) }
}
func (ctx *MsgContext) CallbackDelete() {
ctx.delete(ctx.CallbackMsgId)
}
func (ctx *MsgContext) answerCallbackQuery(url, text string, showAlert bool) { func (ctx *MsgContext) answerCallbackQuery(url, text string, showAlert bool) {
if len(ctx.CallbackQueryId) == 0 { if len(ctx.CallbackQueryId) == 0 {
@@ -181,23 +186,19 @@ func (ctx *MsgContext) answerCallbackQuery(url, text string, showAlert bool) {
ctx.botLogger.Errorln(err) ctx.botLogger.Errorln(err)
} }
} }
func (ctx *MsgContext) AnswerCbQuery() { func (ctx *MsgContext) AnswerCbQuery() { ctx.answerCallbackQuery("", "", false) }
ctx.answerCallbackQuery("", "", false) func (ctx *MsgContext) AnswerCbQueryText(text string) { ctx.answerCallbackQuery("", text, false) }
} func (ctx *MsgContext) AnswerCbQueryAlert(text string) { ctx.answerCallbackQuery("", text, true) }
func (ctx *MsgContext) AnswerCbQueryText(text string) { func (ctx *MsgContext) AnswerCbQueryUrl(u string) { ctx.answerCallbackQuery(u, "", false) }
ctx.answerCallbackQuery("", text, false)
}
func (ctx *MsgContext) AnswerCbQueryAlert(text string) {
ctx.answerCallbackQuery("", text, true)
}
func (ctx *MsgContext) AnswerCbQueryUrl(u string) {
ctx.answerCallbackQuery(u, "", false)
}
func (ctx *MsgContext) SendAction(action tgapi.ChatActionType) { func (ctx *MsgContext) SendAction(action tgapi.ChatActionType) {
_, err := ctx.Api.SendChatAction(tgapi.SendChatActionP{ params := tgapi.SendChatActionP{
ChatID: ctx.Msg.Chat.ID, Action: action, ChatID: ctx.Msg.Chat.ID, Action: action,
}) }
if ctx.Msg.MessageThreadID > 0 {
params.MessageThreadID = ctx.Msg.MessageThreadID
}
_, err := ctx.Api.SendChatAction(params)
if err != nil { if err != nil {
ctx.botLogger.Errorln(err) ctx.botLogger.Errorln(err)
} }
@@ -213,10 +214,14 @@ func (ctx *MsgContext) error(err error) {
} }
ctx.botLogger.Errorln(err) ctx.botLogger.Errorln(err)
} }
func (ctx *MsgContext) Error(err error) { func (ctx *MsgContext) Error(err error) { ctx.error(err) }
ctx.error(err)
}
func (ctx *MsgContext) NewDraft() *Draft {
draft := ctx.draftProvider.NewDraft()
draft.chatID = ctx.Msg.Chat.ID
draft.messageThreadID = ctx.Msg.MessageThreadID
return draft
}
func (ctx *MsgContext) Translate(key string) string { func (ctx *MsgContext) Translate(key string) string {
if ctx.From == nil { if ctx.From == nil {
return key return key
+16 -11
View File
@@ -93,37 +93,42 @@ func (c *Command[T]) validateArgs(args []string) error {
} }
type Plugin[T DbContext] struct { type Plugin[T DbContext] struct {
Name string name string
Commands map[string]Command[T] commands map[string]Command[T]
Payloads map[string]Command[T] payloads map[string]Command[T]
Middlewares extypes.Slice[Middleware[T]] middlewares extypes.Slice[Middleware[T]]
skipAutoCmd bool
} }
func NewPlugin[T DbContext](name string) *Plugin[T] { func NewPlugin[T DbContext](name string) *Plugin[T] {
return &Plugin[T]{ return &Plugin[T]{
name, map[string]Command[T]{}, name, map[string]Command[T]{},
map[string]Command[T]{}, extypes.Slice[Middleware[T]]{}, map[string]Command[T]{}, extypes.Slice[Middleware[T]]{}, false,
} }
} }
func (p *Plugin[T]) AddCommand(command *Command[T]) *Plugin[T] { func (p *Plugin[T]) AddCommand(command *Command[T]) *Plugin[T] {
p.Commands[command.command] = *command p.commands[command.command] = *command
return p return p
} }
func (p *Plugin[T]) NewCommand(exec CommandExecutor[T], command string, args ...CommandArg) *Command[T] { func (p *Plugin[T]) NewCommand(exec CommandExecutor[T], command string, args ...CommandArg) *Command[T] {
return NewCommand(exec, command, args...) return NewCommand(exec, command, args...)
} }
func (p *Plugin[T]) AddPayload(command *Command[T]) *Plugin[T] { func (p *Plugin[T]) AddPayload(command *Command[T]) *Plugin[T] {
p.Payloads[command.command] = *command p.payloads[command.command] = *command
return p return p
} }
func (p *Plugin[T]) AddMiddleware(middleware Middleware[T]) *Plugin[T] { func (p *Plugin[T]) AddMiddleware(middleware Middleware[T]) *Plugin[T] {
p.Middlewares = p.Middlewares.Push(middleware) p.middlewares = p.middlewares.Push(middleware)
return p
}
func (p *Plugin[T]) SkipCommandAutoGen() *Plugin[T] {
p.skipAutoCmd = true
return p return p
} }
func (p *Plugin[T]) executeCmd(cmd string, ctx *MsgContext, dbContext *T) { func (p *Plugin[T]) executeCmd(cmd string, ctx *MsgContext, dbContext *T) {
command := p.Commands[cmd] command := p.commands[cmd]
if err := command.validateArgs(ctx.Args); err != nil { if err := command.validateArgs(ctx.Args); err != nil {
ctx.error(err) ctx.error(err)
return return
@@ -131,7 +136,7 @@ func (p *Plugin[T]) executeCmd(cmd string, ctx *MsgContext, dbContext *T) {
command.exec(ctx, dbContext) command.exec(ctx, dbContext)
} }
func (p *Plugin[T]) executePayload(payload string, ctx *MsgContext, dbContext *T) { func (p *Plugin[T]) executePayload(payload string, ctx *MsgContext, dbContext *T) {
pl := p.Payloads[payload] pl := p.payloads[payload]
if err := pl.validateArgs(ctx.Args); err != nil { if err := pl.validateArgs(ctx.Args); err != nil {
ctx.error(err) ctx.error(err)
return return
@@ -139,7 +144,7 @@ func (p *Plugin[T]) executePayload(payload string, ctx *MsgContext, dbContext *T
pl.exec(ctx, dbContext) pl.exec(ctx, dbContext)
} }
func (p *Plugin[T]) executeMiddlewares(ctx *MsgContext, db *T) bool { func (p *Plugin[T]) executeMiddlewares(ctx *MsgContext, db *T) bool {
for _, m := range p.Middlewares { for _, m := range p.middlewares {
if !m.Execute(ctx, db) { if !m.Execute(ctx, db) {
return false return false
} }
+12
View File
@@ -56,6 +56,7 @@ type PromoteChatMember struct {
CanPinMessages bool `json:"can_pin_messages,omitempty"` CanPinMessages bool `json:"can_pin_messages,omitempty"`
CanManageTopics bool `json:"can_manage_topics,omitempty"` CanManageTopics bool `json:"can_manage_topics,omitempty"`
CanManageDirectMessages bool `json:"can_manage_direct_messages,omitempty"` CanManageDirectMessages bool `json:"can_manage_direct_messages,omitempty"`
CanManageTags bool `json:"can_manage_tags,omitempty"`
} }
func (api *API) PromoteChatMember(params PromoteChatMember) (bool, error) { func (api *API) PromoteChatMember(params PromoteChatMember) (bool, error) {
@@ -74,6 +75,17 @@ func (api *API) SetChatAdministratorCustomTitle(params SetChatAdministratorCusto
return req.Do(api) return req.Do(api)
} }
type SetChatMemberTagP struct {
ChatID int `json:"chat_id"`
UserID int `json:"user_id"`
Tag string `json:"tag,omitempty"`
}
func (api *API) SetChatMemberTag(params SetChatMemberTagP) (bool, error) {
req := NewRequest[bool]("setChatMemberTag", params)
return req.Do(api)
}
type BanChatSenderChatP struct { type BanChatSenderChatP struct {
ChatID int `json:"chat_id"` ChatID int `json:"chat_id"`
SenderChatID int `json:"sender_chat_id"` SenderChatID int `json:"sender_chat_id"`
+5
View File
@@ -99,6 +99,7 @@ type ChatPermissions struct {
CanSendPolls bool `json:"can_send_polls"` CanSendPolls bool `json:"can_send_polls"`
CanSendOtherMessages bool `json:"can_send_other_messages"` CanSendOtherMessages bool `json:"can_send_other_messages"`
CanAddWebPagePreview bool `json:"can_add_web_page_preview"` CanAddWebPagePreview bool `json:"can_add_web_page_preview"`
CatEditTag bool `json:"cat_edit_tag"`
CanChangeInfo bool `json:"can_change_info"` CanChangeInfo bool `json:"can_change_info"`
CanInviteUsers bool `json:"can_invite_users"` CanInviteUsers bool `json:"can_invite_users"`
CanPinMessages bool `json:"can_pin_messages"` CanPinMessages bool `json:"can_pin_messages"`
@@ -137,6 +138,7 @@ const (
type ChatMember struct { type ChatMember struct {
Status ChatMemberStatusType `json:"status"` Status ChatMemberStatusType `json:"status"`
User User `json:"user"` User User `json:"user"`
Tag string `json:"tag,omitempty"`
// Owner // Owner
IsAnonymous *bool `json:"is_anonymous"` IsAnonymous *bool `json:"is_anonymous"`
@@ -160,6 +162,7 @@ type ChatMember struct {
CanPinMessages *bool `json:"can_pin_messages,omitempty"` CanPinMessages *bool `json:"can_pin_messages,omitempty"`
CanManageTopics *bool `json:"can_manage_topics,omitempty"` CanManageTopics *bool `json:"can_manage_topics,omitempty"`
CanManageDirectMessages *bool `json:"can_manage_direct_messages,omitempty"` CanManageDirectMessages *bool `json:"can_manage_direct_messages,omitempty"`
CanManageTags *bool `json:"can_manage_tags,omitempty"`
// Member // Member
UntilDate *int `json:"until_date,omitempty"` UntilDate *int `json:"until_date,omitempty"`
@@ -175,6 +178,7 @@ type ChatMember struct {
CanSendPolls *bool `json:"can_send_polls,omitempty"` CanSendPolls *bool `json:"can_send_polls,omitempty"`
CanSendOtherMessages *bool `json:"can_send_other_messages,omitempty"` CanSendOtherMessages *bool `json:"can_send_other_messages,omitempty"`
CanAddWebPagePreview *bool `json:"can_add_web_page_preview,omitempty"` CanAddWebPagePreview *bool `json:"can_add_web_page_preview,omitempty"`
CanEditTag *bool `json:"can_edit_tag,omitempty"`
} }
type ChatBoostSource struct { type ChatBoostSource struct {
@@ -215,6 +219,7 @@ type ChatAdministratorRights struct {
CanPinMessages *bool `json:"can_pin_messages,omitempty"` CanPinMessages *bool `json:"can_pin_messages,omitempty"`
CanManageTopics *bool `json:"can_manage_topics,omitempty"` CanManageTopics *bool `json:"can_manage_topics,omitempty"`
CanManageDirectMessages *bool `json:"can_manage_direct_messages,omitempty"` CanManageDirectMessages *bool `json:"can_manage_direct_messages,omitempty"`
CanManageTags *bool `json:"can_manage_tags,omitempty"`
} }
type ChatBoostUpdated struct { type ChatBoostUpdated struct {
+1 -1
View File
@@ -268,7 +268,7 @@ func (api *API) SendDice(params SendDiceP) (Message, error) {
type SendMessageDraftP struct { type SendMessageDraftP struct {
ChatID int `json:"chat_id"` ChatID int `json:"chat_id"`
MessageThreadID int `json:"message_thread_id,omitempty"` MessageThreadID int `json:"message_thread_id,omitempty"`
DraftID int `json:"draft_id"` DraftID uint64 `json:"draft_id"`
Text string `json:"text"` Text string `json:"text"`
ParseMode ParseMode `json:"parse_mode,omitempty"` ParseMode ParseMode `json:"parse_mode,omitempty"`
Entities []MessageEntity `json:"entities,omitempty"` Entities []MessageEntity `json:"entities,omitempty"`
+9 -4
View File
@@ -12,10 +12,11 @@ type Message struct {
BusinessConnectionId string `json:"business_connection_id,omitempty"` BusinessConnectionId string `json:"business_connection_id,omitempty"`
From *User `json:"from,omitempty"` From *User `json:"from,omitempty"`
SenderChat *Chat `json:"sender_chat,omitempty"` SenderChat *Chat `json:"sender_chat,omitempty"`
SenderBoostCount int `json:"sender_boost_count,omitempty"` SenderBoostCount int `json:"sender_boost_count,omitempty"`
SenderBusinessBot *User `json:"sender_business_bot,omitempty"` SenderBusinessBot *User `json:"sender_business_bot,omitempty"`
Chat *Chat `json:"chat,omitempty"` SenderTag string `json:"sender_tag,omitempty"`
Chat *Chat `json:"chat,omitempty"`
IsTopicMessage bool `json:"is_topic_message,omitempty"` IsTopicMessage bool `json:"is_topic_message,omitempty"`
IsAutomaticForward bool `json:"is_automatic_forward,omitempty"` IsAutomaticForward bool `json:"is_automatic_forward,omitempty"`
@@ -74,6 +75,7 @@ const (
MessageEntityTextLink MessageEntityType = "text_link" MessageEntityTextLink MessageEntityType = "text_link"
MessageEntityTextMention MessageEntityType = "text_mention" MessageEntityTextMention MessageEntityType = "text_mention"
MessageEntityCustomEmoji MessageEntityType = "custom_emoji" MessageEntityCustomEmoji MessageEntityType = "custom_emoji"
MessageEntityDateTime MessageEntityType = "date_time"
) )
type MessageEntity struct { type MessageEntity struct {
@@ -85,6 +87,9 @@ type MessageEntity struct {
User *User `json:"user,omitempty"` User *User `json:"user,omitempty"`
Language string `json:"language,omitempty"` Language string `json:"language,omitempty"`
CustomEmojiID string `json:"custom_emoji_id,omitempty"` CustomEmojiID string `json:"custom_emoji_id,omitempty"`
UnixTime int `json:"unix_time,omitempty"`
DateTimeFormat string `json:"date_time_format,omitempty"`
} }
type ReplyParameters struct { type ReplyParameters struct {
+4 -4
View File
@@ -1,9 +1,9 @@
package utils package utils
const ( const (
VersionString = "0.8.0-beta.4" VersionString = "1.0.0-beta.5"
VersionMajor = 0 VersionMajor = 1
VersionMinor = 8 VersionMinor = 0
VersionPatch = 0 VersionPatch = 0
Beta = 4 Beta = 5
) )