fix: danbooru

FloatTech · Nov 19, 2023 · 54c67e3 · 54c67e3
1 parent 61168ad
commit 54c67e3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/plugin/danbooru/tag.go b/plugin/danbooru/tag.go
@@ -3,6 +3,7 @@ package deepdanbooru
 import (
 	"bytes"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"image"
 	"net/url"
@@ -59,8 +60,12 @@ func tagurl(name, u string) (im image.Image, st *sorttags, err error) {
 	if err != nil {
 		return
 	}
+	if len(data) < 4 {
+		err = errors.New("data too short")
+		return
+	}
 	tags := make(map[string]float64)
-	err = json.Unmarshal(data, &tags)
+	err = json.Unmarshal(data[1:len(data)-1], &tags)
 	if err != nil {
 		return
 	}