add plugin: Pixiv Rank

README.md

@@ -59,7 +59,8 @@
 
 ### 好康的
 - [x] 涩图 (pixiv)
-- [ ] Pixiv排行榜
+- [x] Pixiv排行榜 (pixiv_rank)
+- [ ] 查看Pixiv插图所有分P
 - [ ] coser
 
 </details>

cmd/main.go

@@ -29,6 +29,7 @@ import (
 	_ "github.com/RicheyJang/PaimengBot/plugins/inspection"
 	_ "github.com/RicheyJang/PaimengBot/plugins/music"
 	_ "github.com/RicheyJang/PaimengBot/plugins/pixiv"
+	_ "github.com/RicheyJang/PaimengBot/plugins/pixiv_rank"
 	_ "github.com/RicheyJang/PaimengBot/plugins/random"
 	_ "github.com/RicheyJang/PaimengBot/plugins/short_url"
 	_ "github.com/RicheyJang/PaimengBot/plugins/statistic"

plugins/pixiv/download.go

@@ -128,6 +128,12 @@ func (pic *PictureInfo) GenSinglePicMsg() (message.Message, error) {
 		return nil, err
 	}
 	// 文字
+	tip := pic.GetDescribe()
+	return message.Message{message.Text(pic.Title), picMsg, message.Text(tip)}, nil
+}
+
+// GetDescribe 获取图片说明
+func (pic *PictureInfo) GetDescribe() string {
 	var tags []string
 	for _, tag := range pic.Tags {
 		if isCNOrEn(tag) {
@@ -147,7 +153,7 @@ func (pic *PictureInfo) GenSinglePicMsg() (message.Message, error) {
 	if len(tags) > 0 {
 		tip += fmt.Sprintf("\n标签: %v", strings.Join(tags, ","))
 	}
-	return message.Message{message.Text(pic.Title), picMsg, message.Text(tip)}, nil
+	return tip
 }
 
 // GetURLByPID 通过PID获取图片下载URL
@@ -181,11 +187,7 @@ func (pic *PictureInfo) GetURLByPID() (err error) {
 		if err == nil && len(pic.URL) == 0 {
 			err = fmt.Errorf("unexpected error")
 		} else if len(pic.URL) > 0 {
-			p := proxy.GetConfigString("proxy")
-			if len(p) > 0 {
-				pic.URL = strings.ReplaceAll(pic.URL, "i.pximg.net", p)
-				pic.URL = strings.ReplaceAll(pic.URL, "i.pixiv.net", p)
-			}
+			pic.ReplaceURLToProxy()
 		}
 	}()
 	// 解析
@@ -197,6 +199,17 @@ func (pic *PictureInfo) GetURLByPID() (err error) {
 	return nil
 }
 
+// ReplaceURLToProxy 将图片URL替换为反代地址
+func (pic *PictureInfo) ReplaceURLToProxy() {
+	if len(pic.URL) > 0 {
+		p := proxy.GetConfigString("proxy")
+		if len(p) > 0 {
+			pic.URL = strings.ReplaceAll(pic.URL, "i.pximg.net", p)
+			pic.URL = strings.ReplaceAll(pic.URL, "i.pixiv.net", p)
+		}
+	}
+}
+
 func getTimeout() time.Duration {
 	s := proxy.GetConfigString("timeout")
 	if len(s) == 0 { // 未设置

plugins/pixiv/init.go

@@ -22,7 +22,7 @@ var info = manager.PluginInfo{
 另外，高级用法询问管理员哦~[dog]`,
 	SuperUsage: `特别用法：(在私聊中)
 	色图r [Tag]* [数量num]?：你懂得
-配置项：
+config-plugin配置项：
 pixiv.timeout： 下载图片超时时长，至少为1s；越长下载成功率越高、等待时间越长
 pixiv.proxy： Pixiv反代网站，默认为i.pixiv.re，令外可选i.pixiv.cat
 pixiv.scale：从各个图库取图的比例，导入Omega图库后，将pixiv.scale.omega设为非0值才可使其生效

plugins/pixiv_rank/hibi.go

@@ -0,0 +1,78 @@
+package pixiv_rank
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+
+	"github.com/RicheyJang/PaimengBot/plugins/pixiv"
+	"github.com/RicheyJang/PaimengBot/utils"
+	"github.com/RicheyJang/PaimengBot/utils/client"
+	"github.com/RicheyJang/PaimengBot/utils/consts"
+
+	"github.com/tidwall/gjson"
+)
+
+func getPixivRankByHIBI(rankType string, num int, date string) ([]pixiv.PictureInfo, error) {
+	// 整理API URL
+	api := proxy.GetAPIConfig(consts.APIOfHibiAPIKey)
+	if len(api) == 0 {
+		return nil, fmt.Errorf("API of HibiAPI is empty")
+	}
+	if !strings.HasPrefix(api, "http://") && !strings.HasPrefix(api, "https://") {
+		api = "https://" + api
+	}
+	if !strings.HasSuffix(api, "/") {
+		api += "/"
+	}
+	api = fmt.Sprintf("%sapi/pixiv/rank?mode=%v&size=%v", api, rankType, num)
+	if len(date) > 0 {
+		api += "&date=" + date
+	}
+	// 调用
+	c := client.NewHttpClient(nil)
+	rsp, err := c.GetGJson(api)
+	if err != nil {
+		return nil, err
+	}
+	rsp = rsp.Get("illusts")
+	if !rsp.Exists() {
+		return nil, fmt.Errorf("illusts is not found")
+	}
+	// 解析
+	var pics []pixiv.PictureInfo
+	illusts := rsp.Array()
+	for i, illust := range illusts {
+		if i >= num { // 限制数量
+			break
+		}
+		pics = append(pics, analysisHIBIIllust(illust))
+	}
+	return pics, nil
+}
+
+// 分析单张插图信息
+func analysisHIBIIllust(illust gjson.Result) pixiv.PictureInfo {
+	pic := pixiv.PictureInfo{
+		Title:  illust.Get("title").String(),
+		PID:    illust.Get("id").Int(),
+		P:      0,
+		Author: illust.Get("user.name").String(),
+		UID:    illust.Get("user.id").Int(),
+	}
+	for _, tag := range illust.Get("tags").Array() {
+		if tag.Get("name").Type != gjson.Null {
+			pic.Tags = append(pic.Tags, tag.Get("name").String())
+		}
+		if tag.Get("translated_name").Type != gjson.Null {
+			pic.Tags = append(pic.Tags, tag.Get("translated_name").String())
+		}
+	}
+	pic.Tags = utils.MergeStringSlices(pic.Tags) // 去重
+	if illust.Get("page_count").Int() == 1 {
+		pic.URL = illust.Get("meta_single_page.original_image_url").String()
+	} else if illust.Get("page_count").Int() > int64(pic.P) {
+		pic.URL = illust.Get("meta_pages." + strconv.Itoa(pic.P) + ".image_urls.original").String()
+	}
+	return pic
+}

plugins/pixiv_rank/init.go

@@ -0,0 +1,158 @@
+package pixiv_rank
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/RicheyJang/PaimengBot/manager"
+	"github.com/RicheyJang/PaimengBot/utils"
+	"github.com/RicheyJang/PaimengBot/utils/consts"
+
+	log "github.com/sirupsen/logrus"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+)
+
+var info = manager.PluginInfo{
+	Name: "pixiv排行榜",
+	Usage: `用法：
+	pixiv排行榜 [类型]? [数量]? [日期]?：获取指定日期指定数量指定类型的Pixiv排行榜图片
+可选类型（左侧数字）：
+	1. 日排行
+	2. 周排行
+	3. 月排行
+	4. 男性向日排行
+	5. 女性向日排行
+	6. 原创排行
+	7. 新人排行
+以下类型仅限私聊：
+	8. R18日排行
+	9. R18周排行
+	10. R18男性向日排行
+	11. R18女性向日排行
+示例：
+	pixiv排行榜：默认当日排行5张
+	pixiv排行榜 4：男性向当日排行5张
+	pixiv排行榜 4 20：男性向当日排行20张
+	pixiv排行榜 4 20 2022-01-28：男性向2022年1月28日排行10张`,
+	SuperUsage: `
+备注：config-plugin配置项沿用pixiv插件`,
+	Classify: "好康的",
+}
+var proxy *manager.PluginProxy
+
+func init() {
+	proxy = manager.RegisterPlugin(info)
+	if proxy == nil {
+		return
+	}
+	proxy.OnCommands([]string{"pixiv排行榜", "Pixiv排行榜"}).SetBlock(true).SetPriority(3).Handle(getRankPictures)
+
+	proxy.AddAPIConfig(consts.APIOfHibiAPIKey, "api.obfs.dev")
+}
+
+func getRankPictures(ctx *zero.Ctx) {
+	if proxy.LockUser(ctx.Event.UserID) {
+		ctx.Send("正在发送中呢，稍等一下嘛")
+		return
+	}
+	defer proxy.UnlockUser(ctx.Event.UserID)
+
+	rankType, num, date := analysisArgs(utils.GetArgs(ctx))
+	if !utils.IsMessagePrimary(ctx) && strings.Contains(rankType, "r18") {
+		ctx.Send("滚滚滚去私聊")
+		return
+	}
+	// 获取图片信息
+	pics, err := getPixivRankByHIBI(rankType, num, date)
+	if err != nil {
+		log.Errorf("getPixivRankByHIBI type=%v,num=%v,date=%v error: %v", rankType, num, date, err)
+		ctx.Send("失败了...")
+		return
+	}
+	if len(pics) == 0 {
+		log.Warnf("No Pics : type=%v,num=%v,date=%v", rankType, num, date)
+		ctx.Send("没图了...")
+		return
+	}
+	// 开始发送
+	if len(date) == 0 {
+		date = "今天"
+	}
+	if transType, ok := translateRankTypeMap[rankType]; ok {
+		rankType = transType
+	}
+	ctx.Send(fmt.Sprintf("开始发送%v的%v排行榜，共计%v张", date, rankType, len(pics)))
+	for _, pic := range pics {
+		pic.ReplaceURLToProxy() // 使用反代
+		// 生成消息
+		msg, err := pic.GenSinglePicMsg()
+		if err != nil { // 下载图片失败
+			log.Warnf("GenSinglePicMsg failed: pid=%v, err=%v", pic.PID, err)
+			msg = message.Message{message.Text(pic.Title + "\n图片下载失败\n" + pic.GetDescribe())}
+		}
+		ctx.Send(msg)
+	}
+}
+
+var rankTypeMap = map[int]string{
+	1:  "day",
+	2:  "week",
+	3:  "month",
+	4:  "day_male",
+	5:  "day_female",
+	6:  "week_original",
+	7:  "week_rookie",
+	8:  "day_r18",
+	9:  "week_r18",
+	10: "day_male_r18",
+	11: "day_female_r18",
+}
+
+var translateRankTypeMap = map[string]string{
+	"day":            "日",
+	"week":           "周",
+	"month":          "月",
+	"day_male":       "男性向日",
+	"day_female":     "女性向日",
+	"week_original":  "原创",
+	"week_rookie":    "新人",
+	"day_r18":        "日R-18",
+	"week_r18":       "周R-18",
+	"day_male_r18":   "男性向R-18",
+	"day_female_r18": "女性向R-18",
+}
+
+// 分析参数
+func analysisArgs(args string) (rankType string, num int, date string) {
+	rankType = rankTypeMap[1]
+	num = 5 // 默认5张
+	args = strings.TrimSpace(args)
+	if len(args) == 0 {
+		return
+	}
+	subs := strings.Split(args, " ")
+	if len(subs) >= 1 {
+		tpID, err := strconv.Atoi(subs[0])
+		if err != nil || tpID < 1 || tpID > len(rankTypeMap) {
+			tpID = 1
+		}
+		rankType = rankTypeMap[tpID]
+	}
+	if len(subs) >= 2 {
+		var err error
+		num, err = strconv.Atoi(subs[1])
+		if err != nil || num < 1 || num > 20 { // 最多20张
+			num = 5
+		}
+	}
+	if len(subs) >= 3 {
+		_, err := time.Parse("2006-01-02", subs[2])
+		if err == nil {
+			date = subs[2]
+		}
+	}
+	return
+}