From 027168e671a0bc9556e2e6a8e4ff381543bb7506 Mon Sep 17 00:00:00 2001
From: Yiwen-Chan <yiwen_chan@foxmail.com>
Date: Tue, 16 Feb 2021 18:09:18 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=B8=20=E5=A2=9E=E5=8A=A0=E4=BA=86?=
 =?UTF-8?q?=E6=B6=A9=E5=9B=BE=E8=A7=A6=E5=8F=91=E7=9A=84=E9=A2=91=E7=8E=87?=
 =?UTF-8?q?=E9=99=90=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 go.mod                 |  3 ++-
 go.sum                 |  2 ++
 setutime/setu_geter.go | 16 ++++++++++++++++
 3 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/go.mod b/go.mod
index 690281a3..ce81c3ab 100644
--- a/go.mod
+++ b/go.mod
@@ -5,7 +5,8 @@ go 1.15
 require (
 	github.com/antchfx/htmlquery v1.2.3
 	github.com/mattn/go-sqlite3 v1.14.6
+	github.com/sirupsen/logrus v1.7.0
 	github.com/tidwall/gjson v1.6.8
-	github.com/wdvxdr1123/ZeroBot v0.0.0-20210212165900-0c6c4d79b40e
+	github.com/wdvxdr1123/ZeroBot v0.0.0-20210214154537-b88c3977a0c1
 	gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b
 )
diff --git a/go.sum b/go.sum
index 9c502a2f..6c0defde 100644
--- a/go.sum
+++ b/go.sum
@@ -45,6 +45,8 @@ github.com/tidwall/pretty v1.0.2 h1:Z7S3cePv9Jwm1KwS0513MRaoUe3S01WPbLNV40pwWZU=
 github.com/tidwall/pretty v1.0.2/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk=
 github.com/wdvxdr1123/ZeroBot v0.0.0-20210212165900-0c6c4d79b40e h1:BaUnQOFcCU4l+uf+mifbPdzrBtYfGeDM9RgIFhMBRLs=
 github.com/wdvxdr1123/ZeroBot v0.0.0-20210212165900-0c6c4d79b40e/go.mod h1:MC6+vOkILANgHlEvZpDMKLNeOWSde5eRgiGFtNGix/4=
+github.com/wdvxdr1123/ZeroBot v0.0.0-20210214154537-b88c3977a0c1 h1:HxjJ5hOLS1Qmhnd+X0ZFhEspaVqgNs5FR2jR/UBnI10=
+github.com/wdvxdr1123/ZeroBot v0.0.0-20210214154537-b88c3977a0c1/go.mod h1:MC6+vOkILANgHlEvZpDMKLNeOWSde5eRgiGFtNGix/4=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
 golang.org/x/net v0.0.0-20200421231249-e086a090c8fd h1:QPwSajcTUrFriMF1nJ3XzgoqakqQEsnZf9LdXdi2nkI=
diff --git a/setutime/setu_geter.go b/setutime/setu_geter.go
index 77e65bf5..2302f087 100644
--- a/setutime/setu_geter.go
+++ b/setutime/setu_geter.go
@@ -3,12 +3,16 @@ package setutime
 import (
 	"errors"
 	"fmt"
+	"time"
 
 	utils "bot/setutime/utils"
 
 	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/extension/rate"
 )
 
+var limit = rate.NewManager(time.Minute*1, 5)
+
 type setuGet struct{} // setuGet 来份色图
 
 func (_ setuGet) GetPluginInfo() zero.PluginInfo { // 返回插件信息
@@ -66,6 +70,10 @@ type scenery struct {
 func (_ setuGet) Start() { // 插件主体
 	zero.OnFullMatchGroup([]string{"来份涩图", "setu", "来份色图"}).SetBlock(true).SetPriority(20).
 		Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response {
+			if limit.Load(event.UserID).Acquire() == false {
+				zero.Send(event, "请稍后重试0x0...")
+				return zero.FinishResponse
+			}
 			var (
 				type_  = "setu"
 				illust = &setu{}
@@ -110,6 +118,10 @@ func (_ setuGet) Start() { // 插件主体
 		})
 	zero.OnFullMatchGroup([]string{"二次元", "ecy", "来份二次元"}).SetBlock(true).SetPriority(20).
 		Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response {
+			if limit.Load(event.UserID).Acquire() == false {
+				zero.Send(event, "请稍后重试0x0...")
+				return zero.FinishResponse
+			}
 			var (
 				type_  = "ecy"
 				illust = &ecy{}
@@ -154,6 +166,10 @@ func (_ setuGet) Start() { // 插件主体
 		})
 	zero.OnFullMatchGroup([]string{"风景", "来份风景"}).SetBlock(true).SetPriority(20).
 		Handle(func(matcher *zero.Matcher, event zero.Event, state zero.State) zero.Response {
+			if limit.Load(event.UserID).Acquire() == false {
+				zero.Send(event, "请稍后重试0x0...")
+				return zero.FinishResponse
+			}
 			var (
 				type_  = "scenery"
 				illust = &scenery{}