✨ 增加集成度

2026-02-12 02:00:24 +00:00 · 2021-08-06 23:24:02 +08:00
parent 9ea41412ad
commit a861a1d169
5 changed files with 126 additions and 67 deletions
--- a/plugin_acgimage/classify.go
+++ b/plugin_acgimage/classify.go
@@ -0,0 +1,118 @@
+package acgimage
+
+import (
+	"net/url"
+	"strconv"
+	"strings"
+
+	"github.com/FloatTech/AnimeAPI/classify"
+	"github.com/FloatTech/AnimeAPI/picture"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+)
+
+var (
+	lolipxy = "http://sayuri.fumiama.top:62002/dice?class=0&loli=true&r18=true"
+	// r18有一定保护，一般不会发出图片
+	randapi   = "&loli=true&r18=true"
+	msgof     = make(map[int64]int64)
+	datapath  = "data/acgimage/"
+	cachefile = datapath + "cache"
+	cacheuri  = "file:///" + cachefile
+)
+
+func init() { // 插件主体
+	zero.OnRegex(`^设置随机图片网址(.*)$`, zero.SuperUserPermission).SetBlock(true).SetPriority(20).
+		Handle(func(ctx *zero.Ctx) {
+			url := ctx.State["regex_matched"].([]string)[1]
+			if !strings.HasPrefix(url, "http") {
+				ctx.Send("URL非法!")
+			} else {
+				randapi = url
+			}
+		})
+	// 有保护的随机图片
+	zero.OnFullMatch("随机图片").SetBlock(true).SetPriority(24).
+		Handle(func(ctx *zero.Ctx) {
+			if ctx.Event.GroupID > 0 {
+				if classify.CanVisit(5) {
+					go func() {
+						class, lastvisit, dhash, comment := classify.Classify(randapi, false)
+						replyClass(ctx, dhash, class, false, lastvisit, comment)
+					}()
+				} else {
+					ctx.Send("你太快啦!")
+				}
+			}
+		})
+	// 直接随机图片，无r18保护，后果自负。如果出r18图可尽快通过发送"太涩了"撤回
+	zero.OnFullMatch("直接随机", zero.AdminPermission).SetBlock(true).SetPriority(24).
+		Handle(func(ctx *zero.Ctx) {
+			if ctx.Event.GroupID > 0 {
+				if classify.CanVisit(5) {
+					ctx.Send("请稍后再试哦")
+				} else if randapi != "" {
+					var url string
+					if randapi[0] == '&' {
+						url = lolipxy
+					} else {
+						url = randapi
+					}
+					setLastMsg(ctx.Event.GroupID, ctx.Send(message.Image(url).Add("cache", "0")))
+				}
+			}
+		})
+	// 撤回最后的直接随机图片
+	zero.OnFullMatch("太涩了").SetBlock(true).SetPriority(24).
+		Handle(func(ctx *zero.Ctx) {
+			go cancel(ctx)
+		})
+	// 上传一张图进行评价
+	zero.OnKeywordGroup([]string{"评价图片"}, picture.CmdMatch(), picture.MustGiven()).SetBlock(true).SetPriority(24).
+		Handle(func(ctx *zero.Ctx) {
+			if ctx.Event.GroupID > 0 {
+				ctx.Send("少女祈祷中...")
+				for _, url := range ctx.State["image_url"].([]string) {
+					go func(target string) {
+						class, lastvisit, dhash, comment := classify.Classify(target, true)
+						replyClass(ctx, dhash, class, true, lastvisit, comment)
+					}(url)
+				}
+			}
+		})
+}
+
+func setLastMsg(id int64, msg int64) {
+	msgof[id] = msg
+}
+
+func cancel(ctx *zero.Ctx) {
+	msg, ok := msgof[ctx.Event.GroupID]
+	if ok {
+		ctx.DeleteMessage(msg)
+		delete(msgof, ctx.Event.GroupID)
+	}
+}
+
+func replyClass(ctx *zero.Ctx, dhash string, class int, noimg bool, lv int64, comment string) {
+	img := message.Image(cacheuri + strconv.FormatInt(lv, 10))
+	if class > 5 {
+		if dhash != "" && !noimg {
+			b14, err3 := url.QueryUnescape(dhash)
+			if err3 == nil {
+				ctx.Send(comment + "\n给你点提示哦：" + b14)
+				ctx.Event.GroupID = 0
+				ctx.Send(img)
+			}
+		} else {
+			ctx.Send(comment)
+		}
+	} else {
+		comment := message.Text(comment)
+		if !noimg {
+			ctx.SendChain(img, comment)
+		} else {
+			ctx.SendChain(message.Reply(ctx.Event.MessageID), comment)
+		}
+	}
+}