Files
miaosic/providers/bilibili/bilibili.go
2024-04-09 05:33:38 -07:00

115 lines
3.3 KiB
Go

package bilibili
import (
"github.com/AynaLivePlayer/miaosic"
"github.com/AynaLivePlayer/miaosic/providers"
"github.com/aynakeya/deepcolor"
"github.com/aynakeya/deepcolor/dphttp"
"github.com/tidwall/gjson"
"regexp"
)
var _ = (miaosic.MediaProvider)(&Bilibili{})
type Bilibili struct {
providers.DeepcolorProvider
IdRegex0 *regexp.Regexp
IdRegex1 *regexp.Regexp
header map[string]string
}
func NewBilibili() *Bilibili {
bili := &Bilibili{
IdRegex0: regexp.MustCompile("^[0-9]+"),
IdRegex1: regexp.MustCompile("^au[0-9]+"),
header: map[string]string{
"user-agent": "BiliMusic/2.233.3",
},
}
bili.InfoApi = deepcolor.CreateApiResultFunc(
func(meta miaosic.MetaData) (*dphttp.Request, error) {
return deepcolor.NewGetRequestWithSingleQuery(
"https://www.bilibili.com/audio/music-service-c/web/song/info",
"sid", meta.Identifier,
bili.header)
},
deepcolor.ParserGJson,
func(resp *gjson.Result, media *miaosic.MediaInfo) error {
if resp.Get("data.title").String() == "" {
return miaosic.ErrorExternalApi
}
media.Title = resp.Get("data.title").String()
media.Cover.Url = resp.Get("data.cover").String()
media.Artist = resp.Get("data.author").String()
return nil
})
bili.FileApi = deepcolor.CreateApiResultFunc(
func(param providers.FileApiParam) (*dphttp.Request, error) {
// todo: handle quality
return deepcolor.NewGetRequestWithSingleQuery(
"https://api.bilibili.com/audio/music-service-c/url?device=phone&mid=8047632&mobi_app=iphone&platform=ios&privilege=2&quality=2",
"songid", param.Meta.Identifier,
bili.header)
},
deepcolor.ParserGJson,
func(resp *gjson.Result, media *[]miaosic.MediaUrl) error {
if resp.Get("data.cdns.0").String() == "" {
return miaosic.ErrorExternalApi
}
resp.Get("data.cdns").ForEach(func(key, value gjson.Result) bool {
*media = append(*media, miaosic.NewMediaUrl(value.String(), miaosic.QualityUnk))
return true
})
return nil
})
bili.SearchApi = deepcolor.CreateApiResultFunc(
func(param providers.MediaSearchParam) (*dphttp.Request, error) {
return deepcolor.NewGetRequestWithQuery(
"https://api.bilibili.com/audio/music-service-c/s?search_type=music",
map[string]any{
"keyword": param.Keyword,
"page": param.Page,
"pagesize": param.PageSize,
},
bili.header)
},
deepcolor.ParserGJson,
func(resp *gjson.Result, result *[]miaosic.MediaInfo) error {
// Assuming data contains a list of search results
for _, r := range resp.Get("data.result").Array() {
media := miaosic.MediaInfo{
Title: r.Get("title").String(),
Cover: miaosic.Picture{Url: r.Get("cover").String()},
Artist: r.Get("author").String(),
Meta: miaosic.MetaData{
Provider: bili.GetName(),
Identifier: r.Get("id").String(),
},
}
*result = append(*result, media)
}
return nil
})
return bili
}
func (b *Bilibili) GetName() string {
return "bilibili"
}
func (b *Bilibili) MatchMedia(keyword string) (miaosic.MetaData, bool) {
if id := b.IdRegex0.FindString(keyword); id != "" {
return miaosic.MetaData{
Provider: b.GetName(),
Identifier: id,
}, true
}
if id := b.IdRegex1.FindString(keyword); id != "" {
return miaosic.MetaData{
Provider: b.GetName(),
Identifier: id[2:],
}, true
}
return miaosic.MetaData{}, false
}