From f579b82146dfe0a33b9065b3803f85d399ef4da5 Mon Sep 17 00:00:00 2001
From: ZJY <764763903@qq.com>
Date: Tue, 28 Sep 2021 15:52:57 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E8=A7=A3=E5=86=B3=E7=AE=A1=E7=90=86?=
 =?UTF-8?q?=E5=91=98=E6=9F=A5=E8=AF=A2=E9=99=90=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/command.go | 28 +++++++++++++++++++++++-----
 1 file changed, 23 insertions(+), 5 deletions(-)

diff --git a/models/command.go b/models/command.go
index b559e67..2464588 100644
--- a/models/command.go
+++ b/models/command.go
@@ -299,22 +299,40 @@ var codeSignals = []CodeSignal{
 	{
 		Command: []string{"查询", "query"},
 		Handle: func(sender *Sender) interface{} {
-			if getLimit(sender.UserID, 1) {
+			if sender.IsAdmin {
 				sender.handleJdCookies(func(ck *JdCookie) {
 					sender.Reply(ck.Query())
 				})
 			} else {
-				sender.Reply(fmt.Sprintf("鉴于东哥对接口限流，为了不影响大家的任务正常运行，即日起每日限流%d次，已超过今日限制", Config.Lim))
+				if getLimit(sender.UserID, 1) {
+					sender.handleJdCookies(func(ck *JdCookie) {
+						sender.Reply(ck.Query())
+					})
+				} else {
+					sender.Reply(fmt.Sprintf("鉴于东哥对接口限流，为了不影响大家的任务正常运行，即日起每日限流%d次，已超过今日限制", Config.Lim))
+				}
 			}
+
 			return nil
 		},
 	},
 	{
 		Command: []string{"详细查询", "query"},
 		Handle: func(sender *Sender) interface{} {
-			sender.handleJdCookies(func(ck *JdCookie) {
-				sender.Reply(ck.Query1())
-			})
+			if sender.IsAdmin {
+				sender.handleJdCookies(func(ck *JdCookie) {
+					sender.Reply(ck.Query1())
+				})
+			} else {
+				if getLimit(sender.UserID, 1) {
+					sender.handleJdCookies(func(ck *JdCookie) {
+						sender.Reply(ck.Query1())
+					})
+				} else {
+					sender.Reply(fmt.Sprintf("鉴于东哥对接口限流，为了不影响大家的任务正常运行，即日起每日限流%d次，已超过今日限制", Config.Lim))
+				}
+			}
+
 			return nil
 		},
 	},