forked from FloatTech/ZeroBot-Plugin
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
21 changed files
with
355 additions
and
48 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,4 +1,5 @@ | ||
| data/SetuTime/cache | ||
| data/control | ||
| data/SetuTime/search | ||
| data/manager | ||
| data/acgimage | ||
|
|
||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,14 @@ | ||
| package control | ||
|
|
||
| import ( | ||
| zero "github.com/wdvxdr1123/ZeroBot" | ||
| ) | ||
|
|
||
| var m Control | ||
|
|
||
| func Register(service string, o *Options) *zero.Engine { | ||
| engine := zero.New() | ||
| m = *New(service, o) | ||
| engine.UsePreHandler(m.Handler()) | ||
| return engine | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,177 @@ | ||
| // Package control 控制插件的启用与优先级等 | ||
| package control | ||
|
|
||
| import ( | ||
| "os" | ||
| "strconv" | ||
| "sync" | ||
|
|
||
| . "github.com/FloatTech/ZeroBot-Plugin/data" | ||
| "github.com/sirupsen/logrus" | ||
| zero "github.com/wdvxdr1123/ZeroBot" | ||
| "github.com/wdvxdr1123/ZeroBot/extension" | ||
| "github.com/wdvxdr1123/ZeroBot/message" | ||
| ) | ||
|
|
||
| var ( | ||
| db = &Sqlite{DBPath: "data/control/plugins.db"} | ||
| // managers 每个插件对应的管理 | ||
| managers = map[string]*Control{} | ||
| mu = sync.RWMutex{} | ||
| ) | ||
|
|
||
| // Control is to control the plugins. | ||
| type Control struct { | ||
| sync.RWMutex | ||
| service string | ||
| options Options | ||
| } | ||
|
|
||
| // New returns Manager with settings. | ||
| func New(service string, o *Options) *Control { | ||
| m := &Control{service: service, | ||
| options: func() Options { | ||
| if o == nil { | ||
| return Options{} | ||
| } | ||
| return *o | ||
| }(), | ||
| } | ||
| mu.Lock() | ||
| defer mu.Unlock() | ||
| managers[service] = m | ||
| err := db.Create(service, &grpcfg{}) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| return m | ||
| } | ||
|
|
||
| // Enable enables a group to pass the Manager. | ||
| func (m *Control) Enable(groupID int64) { | ||
| m.Lock() | ||
| err := db.Insert(m.service, &grpcfg{groupID, 0}) | ||
| if err != nil { | ||
| logrus.Errorf("[control] %v", err) | ||
| } | ||
| m.Unlock() | ||
| } | ||
|
|
||
| // Disable disables a group to pass the Manager. | ||
| func (m *Control) Disable(groupID int64) { | ||
| m.Lock() | ||
| err := db.Insert(m.service, &grpcfg{groupID, 1}) | ||
| if err != nil { | ||
| logrus.Errorf("[control] %v", err) | ||
| } | ||
| m.Unlock() | ||
| } | ||
|
|
||
| // Handler 返回 预处理器 | ||
| func (m *Control) Handler() zero.Rule { | ||
| return func(ctx *zero.Ctx) bool { | ||
| m.RLock() | ||
| ctx.State["manager"] = m | ||
| var c grpcfg | ||
| err := db.Find(m.service, &c, "WHERE gid = "+strconv.FormatInt(ctx.Event.GroupID, 10)) | ||
| if err == nil { | ||
| m.RUnlock() | ||
| return c.Disable == 0 | ||
| } else { | ||
| logrus.Errorf("[control] %v", err) | ||
| } | ||
| m.RUnlock() | ||
| if m.options.DisableOnDefault { | ||
| m.Disable(ctx.Event.GroupID) | ||
| } else { | ||
| m.Enable(ctx.Event.GroupID) | ||
| } | ||
| return !m.options.DisableOnDefault | ||
| } | ||
| } | ||
|
|
||
| // Lookup returns a Manager by the service name, if | ||
| // not exist, it will returns nil. | ||
| func Lookup(service string) (*Control, bool) { | ||
| mu.RLock() | ||
| defer mu.RUnlock() | ||
| m, ok := managers[service] | ||
| return m, ok | ||
| } | ||
|
|
||
| // ForEach iterates through managers. | ||
| func ForEach(iterator func(key string, manager *Control) bool) { | ||
| mu.RLock() | ||
| m := copyMap(managers) | ||
| mu.RUnlock() | ||
| for k, v := range m { | ||
| if !iterator(k, v) { | ||
| return | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func copyMap(m map[string]*Control) map[string]*Control { | ||
| ret := make(map[string]*Control, len(m)) | ||
| for k, v := range m { | ||
| ret[k] = v | ||
| } | ||
| return ret | ||
| } | ||
|
|
||
| func Init() { | ||
| err := os.MkdirAll("data/control", 0755) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| zero.OnCommandGroup([]string{"启用", "enable"}, zero.AdminPermission, zero.OnlyGroup). | ||
| Handle(func(ctx *zero.Ctx) { | ||
| model := extension.CommandModel{} | ||
| _ = ctx.Parse(&model) | ||
| service, ok := Lookup(model.Args) | ||
| if !ok { | ||
| ctx.Send("没有找到指定服务!") | ||
| } | ||
| service.Enable(ctx.Event.GroupID) | ||
| ctx.Send(message.Text("已启用服务: " + model.Args)) | ||
| }) | ||
|
|
||
| zero.OnCommandGroup([]string{"禁用", "disable"}, zero.AdminPermission, zero.OnlyGroup). | ||
| Handle(func(ctx *zero.Ctx) { | ||
| model := extension.CommandModel{} | ||
| _ = ctx.Parse(&model) | ||
| service, ok := Lookup(model.Args) | ||
| if !ok { | ||
| ctx.Send("没有找到指定服务!") | ||
| } | ||
| service.Disable(ctx.Event.GroupID) | ||
| ctx.Send(message.Text("已关闭服务: " + model.Args)) | ||
| }) | ||
|
|
||
| zero.OnCommandGroup([]string{"用法", "usage"}, zero.AdminPermission, zero.OnlyGroup). | ||
| Handle(func(ctx *zero.Ctx) { | ||
| model := extension.CommandModel{} | ||
| _ = ctx.Parse(&model) | ||
| service, ok := Lookup(model.Args) | ||
| if !ok { | ||
| ctx.Send("没有找到指定服务!") | ||
| } | ||
| if service.options.Help != "" { | ||
| ctx.Send(service.options.Help) | ||
| } else { | ||
| ctx.Send("该服务无帮助!") | ||
| } | ||
| }) | ||
|
|
||
| zero.OnCommandGroup([]string{"服务列表", "service_list"}, zero.AdminPermission, zero.OnlyGroup). | ||
| Handle(func(ctx *zero.Ctx) { | ||
| msg := `---服务列表---` | ||
| i := 0 | ||
| ForEach(func(key string, manager *Control) bool { | ||
| i++ | ||
| msg += "\n" + strconv.Itoa(i) + `: ` + key | ||
| return true | ||
| }) | ||
| ctx.Send(message.Text(msg)) | ||
| }) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,13 @@ | ||
| package control | ||
|
|
||
| // grpcfg holds the group config for the Manager. | ||
| type grpcfg struct { | ||
| GroupID int64 `db:"gid"` // GroupID 群号 | ||
| Disable int64 `db:"disable"` // Disable 默认启用该插件 | ||
| } | ||
|
|
||
| // Options holds the optional parameters for the Manager. | ||
| type Options struct { | ||
| DisableOnDefault bool | ||
| Help string // 帮助文本信息 | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.