feat(bilibili): move methods back from zbp

Author: fumiama

bilibili/api.go

@@ -11,6 +11,7 @@ import (
 	"github.com/FloatTech/floatbox/binary"
 	"github.com/FloatTech/floatbox/web"
 	"github.com/tidwall/gjson"
+	"github.com/wdvxdr1123/ZeroBot/message"
 )
 
 // ErrAPINeedCookie ...
@@ -20,15 +21,15 @@ var ErrAPINeedCookie = errors.New("api need cookie")
 var Ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.107 Safari/537.36"
 
 // SearchUser 查找b站用户
-func SearchUser(cookiecfg *CookieConfig, keyword string) (r []SearchResult, err error) {
+func (cfg *CookieConfig) SearchUser(keyword string) (r []SearchResult, err error) {
 	client := &http.Client{}
 	req, err := http.NewRequest("GET", fmt.Sprintf(SearchUserURL, keyword), nil)
 	if err != nil {
 		return
 	}
-	if cookiecfg != nil {
+	if cfg != nil {
 		cookie := ""
-		cookie, err = cookiecfg.Load()
+		cookie, err = cfg.Load()
 		if err != nil {
 			return
 		}
@@ -76,12 +77,12 @@ func LoadDynamicDetail(str string) (card DynamicCard, err error) {
 }
 
 // GetDynamicDetail 用动态id查动态信息
-func GetDynamicDetail(cookiecfg *CookieConfig, dynamicIDStr string) (card DynamicCard, err error) {
+func (cfg *CookieConfig) GetDynamicDetail(dynamicIDStr string) (card DynamicCard, err error) {
 	var data []byte
 	data, err = web.RequestDataWithHeaders(web.NewDefaultClient(), fmt.Sprintf(DynamicDetailURL, dynamicIDStr), "GET", func(req *http.Request) error {
-		if cookiecfg != nil {
+		if cfg != nil {
 			cookie := ""
-			cookie, err = cookiecfg.Load()
+			cookie, err = cfg.Load()
 			if err != nil {
 				return err
 			}
@@ -107,15 +108,15 @@ func GetMemberCard(uid any) (result MemberCard, err error) {
 }
 
 // GetMedalWall 用b站uid获得牌子
-func GetMedalWall(cookiecfg *CookieConfig, uid string) (result []Medal, err error) {
+func (cfg *CookieConfig) GetMedalWall(uid string) (result []Medal, err error) {
 	client := &http.Client{}
 	req, err := http.NewRequest("GET", fmt.Sprintf(MedalWallURL, uid), nil)
 	if err != nil {
 		return
 	}
-	if cookiecfg != nil {
+	if cfg != nil {
 		cookie := ""
-		cookie, err = cookiecfg.Load()
+		cookie, err = cfg.Load()
 		if err != nil {
 			return
 		}
@@ -199,12 +200,12 @@ func GetVideoInfo(id string) (card Card, err error) {
 	if err != nil {
 		return
 	}
-	err = json.Unmarshal(binary.StringToBytes(gjson.ParseBytes(data).Get("data").Raw), &card)
+	err = json.Unmarshal(binary.StringToBytes(gjson.ParseBytes(data).Raw), &card)
 	return
 }
 
 // GetVideoSummary 用av或bv查看AI视频总结
-func GetVideoSummary(cookiecfg *CookieConfig, id string) (videoSummary VideoSummary, err error) {
+func (cfg *CookieConfig) GetVideoSummary(id string) (videoSummary VideoSummary, err error) {
 	var (
 		data []byte
 		card Card
@@ -223,9 +224,9 @@ func GetVideoSummary(cookiecfg *CookieConfig, id string) (videoSummary VideoSumm
 		return
 	}
 	data, err = web.RequestDataWithHeaders(web.NewDefaultClient(), SignURL(fmt.Sprintf(VideoSummaryURL, card.BvID, card.CID, card.Owner.Mid)), "GET", func(req *http.Request) error {
-		if cookiecfg != nil {
+		if cfg != nil {
 			cookie := ""
-			cookie, err = cookiecfg.Load()
+			cookie, err = cfg.Load()
 			if err != nil {
 				return err
 			}
@@ -240,3 +241,12 @@ func GetVideoSummary(cookiecfg *CookieConfig, id string) (videoSummary VideoSumm
 	err = json.Unmarshal(data, &videoSummary)
 	return
 }
+
+// GetDetailMessage 用动态id查动态信息
+func (cfg *CookieConfig) GetDetailMessage(dynamicIDStr string) (msg []message.Segment, err error) {
+	dyc, err := cfg.GetDynamicDetail(dynamicIDStr)
+	if err != nil {
+		return
+	}
+	return dyc.ToMessage()
+}

bilibili/api_test.go

@@ -14,9 +14,56 @@ func TestGetAllGuard(t *testing.T) {
 
 func TestGetDynamicDetail(t *testing.T) {
 	cfg := NewCookieConfig("config.json")
-	detail, err := GetDynamicDetail(cfg, "851252197280710664")
+	detail, err := cfg.GetDynamicDetail("851252197280710664")
 	if err != nil {
 		t.Fatal(err)
 	}
 	t.Logf("%+v\n", detail)
 }
+
+func TestArticleInfo(t *testing.T) {
+	card, err := GetArticleInfo("17279244")
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(card.ToArticleMessage("17279244"))
+}
+
+func TestMemberCard(t *testing.T) {
+	card, err := GetMemberCard(2)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Logf("%+v\n", card)
+}
+
+func TestVideoInfo(t *testing.T) {
+	card, err := GetVideoInfo("10007")
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(card.ToVideoMessage())
+	card, err = GetVideoInfo("BV1xx411c7mD")
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(card.ToVideoMessage())
+	card, err = GetVideoInfo("bv1xx411c7mD")
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(card.ToVideoMessage())
+	card, err = GetVideoInfo("BV1mF411j7iU")
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(card.ToVideoMessage())
+}
+
+func TestLiveRoomInfo(t *testing.T) {
+	card, err := GetLiveRoomInfo("83171", "b_ut=7;buvid3=0;i-wanna-go-back=-1;innersign=0;")
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(card.ToMessage())
+}

bilibili/ctx.go

@@ -0,0 +1,62 @@
+package bilibili
+
+import (
+	"regexp"
+	"strconv"
+	"time"
+
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+)
+
+var re = regexp.MustCompile(`^\d+$`)
+
+func RequireUser(cfg *CookieConfig) func(ctx *zero.Ctx) bool {
+	return func(ctx *zero.Ctx) bool {
+		keyword := ctx.State["regex_matched"].([]string)[1]
+		if !re.MatchString(keyword) {
+			searchRes, err := cfg.SearchUser(keyword)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR: ", err))
+				return false
+			}
+			ctx.State["uid"] = strconv.FormatInt(searchRes[0].Mid, 10)
+			return true
+		}
+		next := zero.NewFutureEvent("message", 999, false, ctx.CheckSession())
+		recv, cancel := next.Repeat()
+		defer cancel()
+		ctx.SendChain(message.Text("输入为纯数字, 请选择查询uid还是用户名, 输入对应序号：\n0. 查询uid\n1. 查询用户名"))
+		for {
+			select {
+			case <-time.After(time.Second * 10):
+				ctx.SendChain(message.Text("时间太久啦！", zero.BotConfig.NickName[0], "帮你选择查询uid"))
+				ctx.State["uid"] = keyword
+				return true
+			case c := <-recv:
+				msg := c.Event.Message.ExtractPlainText()
+				num, err := strconv.Atoi(msg)
+				if err != nil {
+					ctx.SendChain(message.Text("请输入数字!"))
+					continue
+				}
+				if num < 0 || num > 1 {
+					ctx.SendChain(message.Text("序号非法!"))
+					continue
+				}
+				if num == 0 {
+					ctx.State["uid"] = keyword
+					return true
+				} else if num == 1 {
+					searchRes, err := cfg.SearchUser(keyword)
+					if err != nil {
+						ctx.SendChain(message.Text("ERROR: ", err))
+						return false
+					}
+					ctx.State["uid"] = strconv.FormatInt(searchRes[0].Mid, 10)
+					return true
+				}
+			}
+		}
+	}
+}