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 }, },