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

laniakea v0.6

Browse Source
This commit is contained in:
ScuroNeko
2026-02-18 14:06:58 +03:00
parent dd83373689
commit 57fb633893
12 changed files with 101 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
plugins/admin.go
View File

@@ -14,16 +14,16 @@ import (
func RegisterAdmin(b *laniakea.Bot) {
p := laniakea.NewPlugin("Admin")
p.Command(uploadPhoto, "uploadPhoto")
p.Command(emojiId, "emojiId")
p.Command(getProxy, "proxy")
p.Command(execSql, "sql")
p.AddCommand(laniakea.NewCommand(uploadPhoto, "uploadPhoto").SkipCommandAutoGen())
p.AddCommand(laniakea.NewCommand(emojiId, "emojiId").SkipCommandAutoGen())
p.AddCommand(laniakea.NewCommand(getProxy, "proxy").SkipCommandAutoGen())
p.AddCommand(laniakea.NewCommand(execSql, "sql").SkipCommandAutoGen())
p.AddMiddleware(AdminMiddleware())
b.AddPlugins(p.Build())
b.AddPlugins(p)
}
func AdminMiddleware() *laniakea.PluginMiddleware {
m := laniakea.NewPluginMiddleware(func(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) bool {
func AdminMiddleware() laniakea.Middleware {
m := laniakea.NewMiddleware("AdminMiddleware", func(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) bool {
rep := psql.NewUserRepository(db)
u, err := rep.GetById(ctx.FromID)
if err != nil {
@@ -31,7 +31,7 @@ func AdminMiddleware() *laniakea.PluginMiddleware {
}
return u.Group.IsAdmin
})
return m
return *m
}
func execSql(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {