From ba9eb9c0230b1e97f2a52bbedc40941ec8c658af Mon Sep 17 00:00:00 2001 From: fumiama Date: Sun, 31 Oct 2021 19:19:13 +0800 Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20acgimage=20=E5=A2=9E=E5=8A=A0?= =?UTF-8?q?=E9=99=90=E9=80=9F=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- plugin_acgimage/classify.go | 5 ++++- plugin_manager/manager.go | 2 ++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/plugin_acgimage/classify.go b/plugin_acgimage/classify.go index 0c060f92..8de4ba6e 100644 --- a/plugin_acgimage/classify.go +++ b/plugin_acgimage/classify.go @@ -5,10 +5,12 @@ import ( "net/url" "strconv" "strings" + "time" "github.com/FloatTech/AnimeAPI/classify" "github.com/FloatTech/AnimeAPI/picture" zero "github.com/wdvxdr1123/ZeroBot" + "github.com/wdvxdr1123/ZeroBot/extension/rate" "github.com/wdvxdr1123/ZeroBot/message" "github.com/FloatTech/ZeroBot-Plugin/control" @@ -27,6 +29,7 @@ var ( randapi = "&loli=true&r18=true" msgof = make(map[int64]int64) block = false + limit = rate.NewManager(time.Minute, 1) ) func init() { // 插件主体 @@ -54,7 +57,7 @@ func init() { // 插件主体 // 有保护的随机图片 engine.OnFullMatch("随机图片", zero.OnlyGroup).SetBlock(true).SetPriority(24). Handle(func(ctx *zero.Ctx) { - if classify.CanVisit(5) { + if classify.CanVisit(5) && limit.Load(ctx.Event.UserID).Acquire() { go func() { class, lastvisit, dhash, comment := classify.Classify(randapi, false) replyClass(ctx, dhash, class, false, lastvisit, comment) diff --git a/plugin_manager/manager.go b/plugin_manager/manager.go index a92f9a9d..a4894975 100644 --- a/plugin_manager/manager.go +++ b/plugin_manager/manager.go @@ -44,6 +44,8 @@ const ( "- 在MM月[每周|周几]的hh点mm分时(用http://url)提醒大家XXX\n" + "- 取消在MM月dd日的hh点mm分的提醒\n" + "- 取消在MM月[每周|周几]的hh点mm分的提醒\n" + + "- [x] 在\"cron\"时(用[url])提醒大家[xxx]\n" + + "- [x] 取消在\"cron\"的提醒\n" + "- 列出所有提醒\n" + "- 翻牌\n" + "- 设置欢迎语XXX\n" +