ScuroNeko/YaeMikoBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

memory leak

Browse Source
This commit is contained in:
ScuroNeko
2026-02-05 15:14:47 +03:00
parent b894cbd9c3
commit 0fded455b8
22 changed files with 420 additions and 170 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
plugins/admin.go
View File

@@ -12,31 +12,26 @@ func RegisterAdmin(b *laniakea.Bot) {
p.Command(uploadPhoto, "uploadPhoto")
p.Command(test, "test")
p.Middleware(laniakea.NewPluginMiddleware(func(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) bool {
p.Middleware(AdminMiddleware())
b.AddPlugins(p.Build())
}
func AdminMiddleware() *laniakea.PluginMiddleware {
return laniakea.NewPluginMiddleware(func(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) bool {
rep := psql.NewUserRepository(db)
u, err := rep.GetById(ctx.FromID)
if err != nil {
return false
}
return u.Group.IsAdmin
}))
b.AddPlugins(p.Build())
})
}
func test(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
ctx.Answer("Ok")
}
func uploadPhoto(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
rep := psql.NewUserRepository(db)
user, err := rep.GetOrCreate(ctx.FromID, ctx.Msg.From.FirstName)
if err != nil {
ctx.Error(err)
return
}
if !user.Group.IsAdmin {
return
}
func uploadPhoto(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
ctx.SendAction(laniakea.ChatActionUploadPhoto)
photoId := ctx.Msg.Photo.Last().FileID
f, err := ctx.Bot.GetFile(&laniakea.GetFileP{FileId: photoId})
if err != nil {