优化代码结构

plugin/danbooru/main.go

@@ -5,7 +5,6 @@ import (
 	"crypto/md5"
 	"encoding/hex"
 
-	"github.com/FloatTech/AnimeAPI/danbooru"
 	"github.com/FloatTech/floatbox/file"
 	"github.com/FloatTech/floatbox/img/writer"
 	ctrl "github.com/FloatTech/zbpctrl"
@@ -30,7 +29,7 @@ func init() { // 插件主体
 		Handle(func(ctx *zero.Ctx) {
 			ctx.SendChain(message.Text("少女祈祷中..."))
 			for _, url := range ctx.State["image_url"].([]string) {
-				t, err := danbooru.TagURL("", url)
+				t, err := tagurl("", url)
 				if err != nil {
 					ctx.SendChain(message.Text("ERROR: ", err))
 					return

plugin/danbooru/tag.go

@@ -0,0 +1,118 @@
+package deepdanbooru
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"image"
+	"net/url"
+	"sort"
+
+	"github.com/Coloured-glaze/gg"
+	"github.com/FloatTech/floatbox/file"
+	"github.com/FloatTech/floatbox/web"
+	imgutils "github.com/FloatTech/zbputils/img"
+	"github.com/FloatTech/zbputils/img/text" // jpg png gif
+	_ "golang.org/x/image/webp"
+)
+
+const api = "https://nsfwtag.azurewebsites.net/api/tag?limit=0.5&url="
+
+type sorttags struct {
+	tags map[string]float64
+	tseq []string
+}
+
+func newsorttags(tags map[string]float64) (s *sorttags) {
+	t := make([]string, 0, len(tags))
+	for k := range tags {
+		t = append(t, k)
+	}
+	return &sorttags{tags: tags, tseq: t}
+}
+
+func (s *sorttags) Len() int {
+	return len(s.tags)
+}
+
+func (s *sorttags) Less(i, j int) bool {
+	v1 := s.tseq[i]
+	v2 := s.tseq[j]
+	return s.tags[v1] >= s.tags[v2]
+}
+
+// Swap swaps the elements with indexes i and j.
+func (s *sorttags) Swap(i, j int) {
+	s.tseq[j], s.tseq[i] = s.tseq[i], s.tseq[j]
+}
+
+func tagurl(name, u string) (im image.Image, err error) {
+	ch := make(chan []byte, 1)
+	go func() {
+		var data []byte
+		data, err = web.GetData(u)
+		ch <- data
+	}()
+
+	data, err := web.GetData(api + url.QueryEscape(u))
+	if err != nil {
+		return
+	}
+	tags := make(map[string]float64)
+	err = json.Unmarshal(bytes.ReplaceAll(data, []byte("'"), []byte("\"")), &tags)
+	if err != nil {
+		return
+	}
+
+	longestlen := 0
+	for k := range tags {
+		if len(k) > longestlen {
+			longestlen = len(k)
+		}
+	}
+	longestlen++
+
+	st := newsorttags(tags)
+	sort.Sort(st)
+
+	_, err = file.GetLazyData(text.BoldFontFile, true)
+	if err != nil {
+		return
+	}
+	_, err = file.GetLazyData(text.ConsolasFontFile, true)
+	if err != nil {
+		return
+	}
+
+	data = <-ch
+	if err != nil {
+		return
+	}
+	img, _, err := image.Decode(bytes.NewReader(data))
+	if err != nil {
+		return
+	}
+
+	img = imgutils.Limit(img, 1280, 720)
+
+	canvas := gg.NewContext(img.Bounds().Size().X, img.Bounds().Size().Y+int(float64(img.Bounds().Size().X)*0.2)+len(tags)*img.Bounds().Size().X/25)
+	canvas.SetRGB(1, 1, 1)
+	canvas.Clear()
+	canvas.DrawImage(img, 0, 0)
+	if err = canvas.LoadFontFace(text.BoldFontFile, float64(img.Bounds().Size().X)*0.1); err != nil {
+		return
+	}
+	canvas.SetRGB(0, 0, 0)
+	canvas.DrawString(name, float64(img.Bounds().Size().X)*0.02, float64(img.Bounds().Size().Y)+float64(img.Bounds().Size().X)*0.1)
+	i := float64(img.Bounds().Size().Y) + float64(img.Bounds().Size().X)*0.2
+	if err = canvas.LoadFontFace(text.ConsolasFontFile, float64(img.Bounds().Size().X)*0.04); err != nil {
+		return
+	}
+	rate := float64(img.Bounds().Size().X) * 0.04
+	for _, k := range st.tseq {
+		canvas.DrawString(fmt.Sprintf("* %-*s -%.3f-", longestlen, k, tags[k]), float64(img.Bounds().Size().X)*0.04, i)
+		i += rate
+	}
+	im = canvas.Image()
+	return
+}