miaosic/providers/bilivideo/bilivideo.go

package bilivideo

import (
	"errors"
	"github.com/aynakeya/deepcolor"
	"github.com/aynakeya/deepcolor/dphttp"
	"github.com/spf13/cast"
	"github.com/tidwall/gjson"
	"miaosic"
	"miaosic/providers"
	"regexp"
)

var _ = (miaosic.MediaProvider)(&BilibiliVideo{})

type BilibiliVideo struct {
	providers.DeepcolorProvider
	requester dphttp.IRequester
	BVRegex   *regexp.Regexp
	IdRegex   *regexp.Regexp
	PageRegex *regexp.Regexp
	header    map[string]string
}

func NewBilibiliViedo(requester dphttp.IRequester) *BilibiliVideo {
	headers := map[string]string{
		"User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64; rv:51.0) Gecko/20100101 Firefox/51.0",
		"Referer":    "https://www.bilibili.com/",
		"Origin":     "https://www.bilibili.com",
		"Cookie":     "buvid3=9A8B3564-BDA9-407F-B45F-D5C40786CA49167618infoc;",
	}
	pvdr := &BilibiliVideo{
		BVRegex:   regexp.MustCompile("^BV[0-9A-Za-z]+"),
		IdRegex:   regexp.MustCompile("^BV[0-9A-Za-z]+(\\?p=[0-9]+)?"),
		PageRegex: regexp.MustCompile("p=[0-9]+"),
		header:    headers,
	}
	pvdr.InfoFunc = pvdr.buildInfoApi()
	//pvdr.FileFunc = buildFileApi(requester, headers)
	//pvdr.LyricFunc = buildLyricApi(requester, headers)
	//pvdr.PlaylistFunc = buildPlaylistApi(requester, headers)
	pvdr.SearchFunc = pvdr.buildSearchApi()
	return pvdr
}

func (b *BilibiliVideo) getPage(bv string) int {
	if page := b.PageRegex.FindString(bv); page != "" {
		return cast.ToInt(page[2:])
	}
	return 1
}

func (b *BilibiliVideo) getBv(bv string) string {
	return b.BVRegex.FindString(bv)
}

func (b *BilibiliVideo) GetName() string {
	return "bilibili-video"
}

func (b *BilibiliVideo) MatchMedia(keyword string) *miaosic.Media {
	if id := b.IdRegex.FindString(keyword); id != "" {
		return &miaosic.Media{
			Meta: miaosic.MediaMeta{
				Provider:   b.GetName(),
				Identifier: id,
			},
		}
	}
	return nil
}

func (b *BilibiliVideo) MatchPlaylist(keyword string) *miaosic.Playlist {
	return nil
}

func (b *BilibiliVideo) UpdateMediaLyric(media *miaosic.Media) error {
	return nil
}

func (b *BilibiliVideo) buildInfoApi() dphttp.ApiFunc[*miaosic.Media, *miaosic.Media] {
	return deepcolor.CreateApiFunc(
		b.requester,
		func(params *miaosic.Media) (*dphttp.Request, error) {
			return deepcolor.NewGetRequestWithSingleQuery(
				"https://api.bilibili.com/x/web-interface/view/detail?&aid=&jsonp=jsonp",
				"bvid", b.header,
			)(params.Meta.Identifier)
		},
		deepcolor.ParserGJson,
		func(result *gjson.Result, media *miaosic.Media) error {
			if result.Get("data.View.title").String() == "" {
				return errors.New("failed to find required data")
			}
			media.Title = result.Get("data.View.title").String()
			media.Artist = result.Get("data.View.owner.name").String()
			media.Cover.Url = result.Get("data.View.pic").String()
			return nil
		})
}

//func buildFileApi(requester dphttp.IRequester, headers map[string]string) dphttp.ApiFunc[*miaosic.Media, *miaosic.Media] {
//	return deepcolor.CreateApiFunc(
//		requester,
//		func(params *miaosic.Media) (*dphttp.Request, error) {
//			return deepcolor.NewGetRequestWithSingleQuery(
//				"https://api.bilibili.com/x/player/playurl?&avid=&cid=&qn=80&type=&otype=json",
//				"bvid", headers,
//			)(params.Meta.Identifier)
//		}
//}

func (b *BilibiliVideo) buildSearchApi() dphttp.ApiFuncResult[string, []*miaosic.Media] {
	return deepcolor.CreateApiResultFunc(
		b.requester,
		deepcolor.NewGetRequestWithSingleQuery(
			"https://api.bilibili.com/x/web-interface/search/type?search_type=video&page=1",
			"keyword", b.header),
		deepcolor.ParserGJson,
		func(resp *gjson.Result, result *[]*miaosic.Media) error {
			if resp.Get("code").String() != "0" {
				return errors.New("failed to find required data")
			}
			r := regexp.MustCompile("</?em[^>]*>")
			resp.Get("data.result").ForEach(func(key, value gjson.Result) bool {
				*result = append(*result, &miaosic.Media{
					Title:  r.ReplaceAllString(value.Get("title").String(), ""),
					Cover:  miaosic.Picture{Url: "https:" + value.Get("pic").String()},
					Artist: value.Get("author").String(),
					Meta: miaosic.MediaMeta{
						Provider:   b.GetName(),
						Identifier: value.Get("bvid").String(),
					},
				})
				return true
			})
			return nil
		})
}

//pvdr.cidApi = dphttp.CreateResultAPI(
//requester,
//&dphttp.ApiInfo[string, *gjson.Result, []string]{
//Request: deepcolor.NewGetRequestWithSingleQuery(
//"https://api.bilibili.com/x/web-interface/view/detail?&aid=&jsonp=jsonp",
//"bvid", pvdr.header,
//),
//Parser: deepcolor.ParserGJson,
//Selector: func (result *gjson.Result) ([]string, error) {
//rcids := result.Get("data.View.pages.#.cid").Array()
//cids := make([]string, 0)
//if len(cids) == 0 {
//cid := result.Get("data.View.cid").String()
//if cid == "" {
//return nil, providers.ErrorExternalApi
//}
//cids = append(cids, cid)
//} else {
//for _, r := range rcids {
//cids = append(cids, r.String())
//}
//}
//return cids, nil
//},
//},
//)

//var fileApi = dphttp.CreateReceiverAPI(
//	requester,
//	&dphttp.ApiInfo[[]string, *gjson.Result, *model.Media]{
//		Request: deepcolor.NewGetRequestWithQuery(
//			"https://api.bilibili.com/x/player/playurl?type=&otype=json&fourk=1&qn=32&avid=",
//			[]string{"bvid", "cid"}, pvdr.header),
//		Parser: deepcolor.ParserGJson,
//		Receiver: func(result *gjson.Result, container *model.Media) error {
//			uri := result.Get("data.durl.0.url").String()
//			if uri == "" {
//				return providers.ErrorExternalApi
//			}
//			container.Url = uri
//			header := make(map[string]string)
//			_ = copier.Copy(&header, &pvdr.header)
//			header["Referer"] = fmt.Sprintf("https://www.bilibili.com/video/%s", pvdr.getBv(container.Meta.(model.Meta).Id))
//			container.Header = header
//			return nil
//		},
//	})
//return pvdr
//}

//
//var BilibiliVideoAPI *BilibiliVideo
//
//func init() {
//	BilibiliVideoAPI = _newBilibiliVideo()
//	Providers[BilibiliVideoAPI.GetName()] = BilibiliVideoAPI
//}
//

//
//func (b *BilibiliVideo) GetPlaylist(playlist *model.Meta) ([]*model.Media, error) {
//	return nil, providers.ErrorExternalApi
//}
//
//func (b *BilibiliVideo) FormatPlaylistUrl(uri string) string {
//	return ""
//}
//
//func (b *BilibiliVideo) Search(keyword string) ([]*model.Media, error) {
//	return b.searchApi(keyword)
//}
//
//func (b *BilibiliVideo) UpdateMedia(media *model.Media) error {
//	err := b.infoApi(b.getBv(media.Meta.(model.Meta).Id), media)
//	if err != nil {
//		return providers.ErrorExternalApi
//	}
//	return nil
//}
//
//func (b *BilibiliVideo) UpdateMediaUrl(media *model.Media) error {
//	page := b.getPage(media.Meta.(model.Meta).Id) - 1
//	cids, err := b.cidApi(b.getBv(media.Meta.(model.Meta).Id))
//	if err != nil || page > len(cids) {
//		return providers.ErrorExternalApi
//	}
//	return b.fileApi([]string{b.getBv(media.Meta.(model.Meta).Id), cids[page]}, media)
//}