🎨⚡️✨ 优化代码

plugin_lolicon/lolicon.go

@@ -0,0 +1,62 @@
+/*
+基于 https://api.lolicon.app 随机图片
+*/
+package plugin_lolicon
+
+import (
+	"io/ioutil"
+	"net/http"
+	"time"
+
+	"github.com/tidwall/gjson"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+)
+
+const (
+	API = "https://api.lolicon.app/setu/v2"
+	CAP = 10
+)
+
+var (
+	QUEUE = make(chan string, CAP)
+)
+
+func init() {
+	zero.OnFullMatch("来份萝莉").
+		Handle(func(ctx *zero.Ctx) {
+			go func() {
+				min := func(a, b int) int {
+					if a > b {
+						return a
+					} else {
+						return b
+					}
+				}
+				for i := 0; i < min(10-cap(QUEUE), 2); i++ {
+					resp, err := http.Get(API)
+					if err != nil {
+						ctx.SendChain(message.Text("ERROR: ", err))
+					}
+					defer resp.Body.Close()
+					if resp.StatusCode != http.StatusOK {
+						ctx.SendChain(message.Text("ERROR: code ", resp.StatusCode))
+					}
+					data, _ := ioutil.ReadAll(resp.Body)
+					json := gjson.ParseBytes(data)
+					if e := json.Get("error").Str; e != "" {
+						ctx.SendChain(message.Text("ERROR: ", e))
+					}
+					url := json.Get("data.0.urls.original").Str
+					ctx.SendGroupMessage(0, message.Image(url))
+					QUEUE <- url
+				}
+			}()
+			select {
+			case <-time.After(time.Second * 10):
+				ctx.SendChain(message.Text("ERROR: 等待填充，请稍后再试......"))
+			case url := <-QUEUE:
+				ctx.SendChain(message.Image(url))
+			}
+		})
+}