优化截图提示

Author: Azusa-Yuan

plugin/dnf/dnf.go

@@ -18,8 +18,9 @@ func init() {
 订阅colg资讯 指令 订阅colg资讯`,
 	})
 
-	engine.OnPrefixGroup([]string{"比例", "金币", "游戏币"}).SetBlock(true).Handle(
+	engine.OnPrefixGroup([]string{"比例", "金币", "游戏币"}, IfSever).SetBlock(true).Handle(
 		func(ctx *zero.Ctx) {
+			ctx.SendChain(message.Text(zero.BotConfig.GetNickName(ctx.Event.SelfID)[0] + "在查了在查了。。。"))
 			arg := ctx.State["args"].(string)
 			data, url, err := service.Screenshot(arg, "youxibi")
 			if err != nil {
@@ -32,8 +33,9 @@ func init() {
 			ctx.SendChain(message.At(ctx.Event.UserID), message.ImageBytes(data), message.Text(url))
 		})
 
-	engine.OnPrefixGroup([]string{"矛盾"}).SetBlock(true).Handle(
+	engine.OnPrefixGroup([]string{"矛盾"}, IfSever).SetBlock(true).Handle(
 		func(ctx *zero.Ctx) {
+			ctx.SendChain(message.Text(zero.BotConfig.GetNickName(ctx.Event.SelfID)[0] + "在查了在查了。。。"))
 			arg := ctx.State["args"].(string)
 			data, url, err := service.Screenshot(arg, "maodun")
 			if err != nil {
@@ -129,3 +131,8 @@ func init() {
 		}
 	}()
 }
+
+func IfSever(ctx *zero.Ctx) bool {
+	_, ok := service.ReportRegions[ctx.State["args"].(string)]
+	return ok
+}

plugin/dnf/service/yxdr.go

@@ -52,7 +52,6 @@ func Screenshot(server string, productType string) ([]byte, string, error) {
 
 	// 导航到指定的URL
 	var buf []byte
-	// url := "https://www.baidu.com"
 	url := fmt.Sprintf("https://www.yxdr.com/bijiaqi/dnf/%s/kua%s", productType, ReportRegions[server])
 	err := chromedp.Run(ScCtx,
 		chromedp.Navigate(url),