GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-07-12 16:02:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: spilt handlers file

Browse Source
This commit is contained in:
krau
2025-02-24 17:50:35 +08:00
parent b96340dd46
commit 9c42bee662
8 changed files with 375 additions and 313 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
bot/middlewares.go
View File

@@ -3,9 +3,13 @@ package bot
import (
"time"
"github.com/celestix/gotgproto/dispatcher"
"github.com/celestix/gotgproto/ext"
"github.com/duke-git/lancet/v2/slice"
"github.com/gotd/contrib/middleware/floodwait"
"github.com/gotd/contrib/middleware/ratelimit"
"github.com/gotd/td/telegram"
"github.com/krau/SaveAny-Bot/config"
"golang.org/x/time/rate"
)
@@ -17,3 +21,12 @@ func FloodWaitMiddleware() []telegram.Middleware {
ratelimiter,
}
}
func checkPermission(ctx *ext.Context, update *ext.Update) error {
userID := update.GetUserChat().GetID()
if !slice.Contain(config.Cfg.GetUsersID(), userID) {
ctx.Reply(update, ext.ReplyTextString(noPermissionText), nil)
return dispatcher.EndGroups
}
return dispatcher.ContinueGroups
}