mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 22:00:11 +08:00
565 lines
19 KiB
Go
565 lines
19 KiB
Go
// Package bilibilipush b站推送
|
||
package bilibilipush
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"strconv"
|
||
"time"
|
||
|
||
log "github.com/sirupsen/logrus"
|
||
"github.com/tidwall/gjson"
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
|
||
"github.com/FloatTech/zbputils/binary"
|
||
"github.com/FloatTech/zbputils/control"
|
||
"github.com/FloatTech/zbputils/control/order"
|
||
"github.com/FloatTech/zbputils/ctxext"
|
||
"github.com/FloatTech/zbputils/img/text"
|
||
"github.com/FloatTech/zbputils/web"
|
||
)
|
||
|
||
const (
|
||
ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.107 Safari/537.36"
|
||
referer = "https://www.bilibili.com/"
|
||
infoURL = "https://api.bilibili.com/x/space/acc/info?mid=%d"
|
||
userDynamicURL = "https://api.vc.bilibili.com/dynamic_svr/v1/dynamic_svr/space_history?host_uid=%d&offset_dynamic_id=0&need_top=0"
|
||
liveListURL = "https://api.live.bilibili.com/room/v1/Room/get_status_info_by_uids"
|
||
tURL = "https://t.bilibili.com/"
|
||
liveURL = "https://live.bilibili.com/"
|
||
serviceName = "bilibilipush"
|
||
)
|
||
|
||
// bdb bilibili推送数据库
|
||
var bdb *bilibilipushdb
|
||
|
||
var (
|
||
lastTime = map[int64]int64{}
|
||
typeMsg = map[int64]string{
|
||
1: "转发了一条动态",
|
||
2: "有图营业",
|
||
4: "无图营业",
|
||
8: "发布了新投稿",
|
||
16: "发布了短视频",
|
||
64: "发布了新专栏",
|
||
256: "发布了新音频",
|
||
}
|
||
liveStatus = map[int64]int{}
|
||
uidErrorMsg = map[int]string{
|
||
0: "输入的uid有效",
|
||
-400: "uid不存在，注意uid不是房间号",
|
||
-402: "uid不存在，注意uid不是房间号",
|
||
-412: "操作过于频繁IP暂时被风控，请半小时后再尝试",
|
||
}
|
||
upMap = map[int64]string{}
|
||
)
|
||
|
||
func init() {
|
||
go bilibiliPushDaily()
|
||
en := control.Register(serviceName, order.AcquirePrio(), &control.Options{
|
||
DisableOnDefault: false,
|
||
Help: "bilibilipush\n" +
|
||
"- 添加订阅[uid]\n" +
|
||
"- 取消订阅[uid]\n" +
|
||
"- 取消动态订阅[uid]\n" +
|
||
"- 取消直播订阅[uid]\n" +
|
||
"- 推送列表",
|
||
PrivateDataFolder: serviceName,
|
||
})
|
||
|
||
// 加载数据库
|
||
go func() {
|
||
dbpath := en.DataFolder()
|
||
dbfile := dbpath + "push.db"
|
||
defer order.DoneOnExit()()
|
||
bdb = initialize(dbfile)
|
||
log.Println("[bilibilipush]加载bilibilipush数据库")
|
||
}()
|
||
|
||
en.OnRegex(`^添加订阅\s?(\d+)$`, ctxext.UserOrGrpAdmin).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
||
buid, _ := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64)
|
||
var name string
|
||
var ok bool
|
||
if name, ok = upMap[buid]; !ok {
|
||
var status int
|
||
status, name = checkBuid(buid)
|
||
if status != 0 {
|
||
msg, ok := uidErrorMsg[status]
|
||
if !ok {
|
||
msg = "未知错误，请私聊反馈给" + zero.BotConfig.NickName[0]
|
||
}
|
||
ctx.SendChain(message.Text(msg))
|
||
return
|
||
}
|
||
}
|
||
gid := ctx.Event.GroupID
|
||
if gid == 0 {
|
||
gid = -ctx.Event.UserID
|
||
}
|
||
if err := subscribe(buid, gid); err != nil {
|
||
log.Errorln("[bilibilipush]:", err)
|
||
} else {
|
||
ctx.SendChain(message.Text("已添加" + name + "的订阅"))
|
||
}
|
||
})
|
||
en.OnRegex(`^取消订阅\s?(\d+)$`, ctxext.UserOrGrpAdmin).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
||
buid, _ := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64)
|
||
var name string
|
||
var ok bool
|
||
if name, ok = upMap[buid]; !ok {
|
||
var status int
|
||
status, name = checkBuid(buid)
|
||
if status != 0 {
|
||
msg, ok := uidErrorMsg[status]
|
||
if !ok {
|
||
msg = "未知错误，请私聊反馈给" + zero.BotConfig.NickName[0]
|
||
}
|
||
ctx.SendChain(message.Text(msg))
|
||
return
|
||
}
|
||
}
|
||
gid := ctx.Event.GroupID
|
||
if gid == 0 {
|
||
gid = -ctx.Event.UserID
|
||
}
|
||
if err := unsubscribe(buid, gid); err != nil {
|
||
log.Errorln("[bilibilipush]:", err)
|
||
} else {
|
||
ctx.SendChain(message.Text("已取消" + name + "的订阅"))
|
||
}
|
||
})
|
||
en.OnRegex(`^取消动态订阅\s?(\d+)$`, ctxext.UserOrGrpAdmin).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
||
buid, _ := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64)
|
||
var name string
|
||
var ok bool
|
||
if name, ok = upMap[buid]; !ok {
|
||
var status int
|
||
status, name = checkBuid(buid)
|
||
if status != 0 {
|
||
msg, ok := uidErrorMsg[status]
|
||
if !ok {
|
||
msg = "未知错误，请私聊反馈给" + zero.BotConfig.NickName[0]
|
||
}
|
||
ctx.SendChain(message.Text(msg))
|
||
return
|
||
}
|
||
}
|
||
gid := ctx.Event.GroupID
|
||
if gid == 0 {
|
||
gid = -ctx.Event.UserID
|
||
}
|
||
if err := unsubscribeDynamic(buid, gid); err != nil {
|
||
log.Errorln("[bilibilipush]:", err)
|
||
} else {
|
||
ctx.SendChain(message.Text("已取消" + name + "的动态订阅"))
|
||
}
|
||
})
|
||
en.OnRegex(`^取消直播订阅\s?(\d+)$`, ctxext.UserOrGrpAdmin).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
||
buid, _ := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64)
|
||
var name string
|
||
var ok bool
|
||
if name, ok = upMap[buid]; !ok {
|
||
var status int
|
||
status, name = checkBuid(buid)
|
||
if status != 0 {
|
||
msg, ok := uidErrorMsg[status]
|
||
if !ok {
|
||
msg = "未知错误，请私聊反馈给" + zero.BotConfig.NickName[0]
|
||
}
|
||
ctx.SendChain(message.Text(msg))
|
||
return
|
||
}
|
||
}
|
||
gid := ctx.Event.GroupID
|
||
if gid == 0 {
|
||
gid = -ctx.Event.UserID
|
||
}
|
||
if err := unsubscribeLive(buid, gid); err != nil {
|
||
log.Errorln("[bilibilipush]:", err)
|
||
} else {
|
||
ctx.SendChain(message.Text("已取消" + name + "的直播订阅"))
|
||
}
|
||
})
|
||
en.OnFullMatch("推送列表", ctxext.UserOrGrpAdmin).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
||
gid := ctx.Event.GroupID
|
||
if gid == 0 {
|
||
gid = -ctx.Event.UserID
|
||
}
|
||
bpl := bdb.getAllPushByGroup(gid)
|
||
fmt.Println(bpl)
|
||
msg := "--------推送列表--------"
|
||
for _, v := range bpl {
|
||
if _, ok := upMap[v.BilibiliUID]; !ok {
|
||
bdb.updateAllUp()
|
||
fmt.Println(upMap)
|
||
}
|
||
msg += fmt.Sprintf("\nuid:%-12d 动态：", v.BilibiliUID)
|
||
if v.DynamicDisable == 0 {
|
||
msg += "●"
|
||
} else {
|
||
msg += "○"
|
||
}
|
||
msg += " 直播："
|
||
if v.LiveDisable == 0 {
|
||
msg += "●"
|
||
} else {
|
||
msg += "○"
|
||
}
|
||
msg += " up主：" + upMap[v.BilibiliUID]
|
||
}
|
||
data, err := text.RenderToBase64(msg, text.FontFile, 600, 20)
|
||
if err != nil {
|
||
log.Errorln("[bilibilipush]:", err)
|
||
}
|
||
if id := ctx.SendChain(message.Image("base64://" + binary.BytesToString(data))); id.ID() == 0 {
|
||
ctx.SendChain(message.Text("ERROR: 可能被风控了"))
|
||
}
|
||
})
|
||
}
|
||
|
||
func bilibiliPushDaily() {
|
||
t := time.NewTicker(time.Second * 10)
|
||
defer t.Stop()
|
||
for range t.C {
|
||
log.Println("-----bilibilipush拉取推送信息-----")
|
||
sendDynamic()
|
||
sendLive()
|
||
}
|
||
}
|
||
|
||
func checkBuid(buid int64) (status int, name string) {
|
||
data, err := web.GetDataWith(web.NewDefaultClient(), fmt.Sprintf(infoURL, buid), "GET", referer, ua)
|
||
if err != nil {
|
||
log.Errorln("[bilibilipush]:", err)
|
||
}
|
||
status = int(gjson.Get(binary.BytesToString(data), "code").Int())
|
||
name = gjson.Get(binary.BytesToString(data), "data.name").String()
|
||
if status == 0 {
|
||
bdb.insertBilibiliUp(buid, name)
|
||
upMap[buid] = name
|
||
}
|
||
return
|
||
}
|
||
|
||
// subscribe 订阅
|
||
func subscribe(buid, groupid int64) (err error) {
|
||
bpMap := map[string]interface{}{
|
||
"bilibili_uid": buid,
|
||
"group_id": groupid,
|
||
"live_disable": 0,
|
||
"dynamic_disable": 0,
|
||
}
|
||
err = bdb.insertOrUpdateLiveAndDynamic(bpMap)
|
||
return
|
||
}
|
||
|
||
// unsubscribe 取消订阅
|
||
func unsubscribe(buid, groupid int64) (err error) {
|
||
bpMap := map[string]interface{}{
|
||
"bilibili_uid": buid,
|
||
"group_id": groupid,
|
||
"live_disable": 1,
|
||
"dynamic_disable": 1,
|
||
}
|
||
err = bdb.insertOrUpdateLiveAndDynamic(bpMap)
|
||
return
|
||
}
|
||
|
||
func unsubscribeDynamic(buid, groupid int64) (err error) {
|
||
bpMap := map[string]interface{}{
|
||
"bilibili_uid": buid,
|
||
"group_id": groupid,
|
||
"dynamic_disable": 1,
|
||
}
|
||
err = bdb.insertOrUpdateLiveAndDynamic(bpMap)
|
||
return
|
||
}
|
||
|
||
func unsubscribeLive(buid, groupid int64) (err error) {
|
||
bpMap := map[string]interface{}{
|
||
"bilibili_uid": buid,
|
||
"group_id": groupid,
|
||
"live_disable": 1,
|
||
}
|
||
err = bdb.insertOrUpdateLiveAndDynamic(bpMap)
|
||
return
|
||
}
|
||
|
||
func getUserDynamicCard(buid int64) (cardList []gjson.Result) {
|
||
data, err := web.GetDataWith(web.NewDefaultClient(), fmt.Sprintf(userDynamicURL, buid), "GET", referer, ua)
|
||
if err != nil {
|
||
log.Errorln("[bilibilipush]:", err)
|
||
}
|
||
cardList = gjson.Get(binary.BytesToString(data), "data.cards").Array()
|
||
return
|
||
}
|
||
|
||
func getLiveList(uids ...int64) string {
|
||
m := make(map[string]interface{})
|
||
m["uids"] = uids
|
||
b, _ := json.Marshal(m)
|
||
client := &http.Client{}
|
||
// 提交请求
|
||
request, err := http.NewRequest("POST", liveListURL, bytes.NewBuffer(b))
|
||
if err != nil {
|
||
log.Errorln("[bilibilipush]:", err)
|
||
}
|
||
request.Header.Add("Referer", referer)
|
||
request.Header.Add("User-Agent", ua)
|
||
response, err := client.Do(request)
|
||
if err != nil {
|
||
log.Errorln("[bilibilipush]:", err)
|
||
}
|
||
defer response.Body.Close()
|
||
data, err := io.ReadAll(response.Body)
|
||
if err != nil {
|
||
log.Errorln("[bilibilipush]:", err)
|
||
}
|
||
return binary.BytesToString(data)
|
||
}
|
||
|
||
func sendDynamic() {
|
||
uids := bdb.getAllBuidByDynamic()
|
||
for _, buid := range uids {
|
||
cardList := getUserDynamicCard(buid)
|
||
if len(cardList) == 0 {
|
||
return
|
||
}
|
||
t, ok := lastTime[buid]
|
||
if !ok {
|
||
lastTime[buid] = cardList[0].Get("desc.timestamp").Int()
|
||
return
|
||
}
|
||
for i := len(cardList) - 1; i >= 0; i-- {
|
||
ct := cardList[i].Get("desc.timestamp").Int()
|
||
if ct > t && ct > time.Now().Unix()-600 {
|
||
lastTime[buid] = ct
|
||
m, ok := control.Lookup(serviceName)
|
||
if ok {
|
||
groupList := bdb.getAllGroupByBuidAndDynamic(buid)
|
||
var msg []message.MessageSegment
|
||
cType := cardList[i].Get("desc.type").Int()
|
||
cardStr := cardList[i].Get("card").String()
|
||
switch cType {
|
||
case 0:
|
||
cName := cardList[i].Get("desc.user_profile.info.uname").String()
|
||
cTime := time.Unix(cardList[i].Get("desc.timestamp").Int(), 0).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cName+"在"+cTime+typeMsg[cType]+"\n"))
|
||
case 1:
|
||
cName := gjson.Get(cardStr, "user.uname").String()
|
||
msg = append(msg, message.Text(cName+typeMsg[cType]+"\n"))
|
||
cContent := gjson.Get(cardStr, "item.content").String()
|
||
msg = append(msg, message.Text(cContent+"\n"))
|
||
msg = append(msg, message.Text("转发的内容：\n"))
|
||
cOrigType := gjson.Get(cardStr, "item.orig_type").Int()
|
||
cOrigin := gjson.Get(cardStr, "origin").String()
|
||
switch cOrigType {
|
||
case 1:
|
||
cName := gjson.Get(cOrigin, "user.uname").String()
|
||
msg = append(msg, message.Text(cName+typeMsg[cOrigType]+"\n"))
|
||
case 2:
|
||
cName := gjson.Get(cOrigin, "user.name").String()
|
||
cUploadTime := time.Unix(gjson.Get(cOrigin, "item.upload_time").Int(), 0).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cName+"在"+cUploadTime+typeMsg[cOrigType]+"\n"))
|
||
cDescription := gjson.Get(cOrigin, "item.description")
|
||
msg = append(msg, message.Text(cDescription))
|
||
if gjson.Get(cOrigin, "item.pictures.#").Int() != 0 {
|
||
gjson.Get(cOrigin, "item.pictures").ForEach(func(_, v gjson.Result) bool {
|
||
msg = append(msg, message.Image(v.Get("img_src").String()))
|
||
return true
|
||
})
|
||
}
|
||
case 4:
|
||
cName := gjson.Get(cOrigin, "user.uname").String()
|
||
cTimestamp := time.Unix(gjson.Get(cOrigin, "item.timestamp").Int(), 0).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cName+"在"+cTimestamp+typeMsg[cOrigType]+"\n"))
|
||
cContent := gjson.Get(cOrigin, "item.content").String()
|
||
msg = append(msg, message.Text(cContent+"\n"))
|
||
case 8:
|
||
cName := gjson.Get(cOrigin, "owner.name").String()
|
||
cTime := time.Unix(gjson.Get(cOrigin, "pubdate").Int(), 0).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cName+"在"+cTime+typeMsg[cOrigType]+"\n"))
|
||
cTitle := gjson.Get(cOrigin, "title").String()
|
||
msg = append(msg, message.Text(cTitle))
|
||
cPic := gjson.Get(cOrigin, "pic").String()
|
||
msg = append(msg, message.Image(cPic))
|
||
cDesc := gjson.Get(cOrigin, "desc").String()
|
||
msg = append(msg, message.Text(cDesc+"\n"))
|
||
cShareSubtitle := gjson.Get(cOrigin, "share_subtitle").String()
|
||
msg = append(msg, message.Text(cShareSubtitle+"\n"))
|
||
cShortLink := gjson.Get(cOrigin, "short_link").String()
|
||
msg = append(msg, message.Text("视频链接："+cShortLink+"\n"))
|
||
case 16:
|
||
cName := gjson.Get(cOrigin, "user.name").String()
|
||
cUploadTime := gjson.Get(cOrigin, "item.upload_time").String()
|
||
msg = append(msg, message.Text(cName+"在"+cUploadTime+typeMsg[cOrigType]+"\n"))
|
||
cDescription := gjson.Get(cOrigin, "item.description")
|
||
msg = append(msg, message.Text(cDescription))
|
||
cCover := gjson.Get(cOrigin, "item.cover.default").String()
|
||
msg = append(msg, message.Image(cCover))
|
||
case 64:
|
||
cName := gjson.Get(cOrigin, "author.name").String()
|
||
cPublishTime := time.Unix(gjson.Get(cOrigin, "publish_time").Int(), 0).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cName+"在"+cPublishTime+typeMsg[cOrigType]+"\n"))
|
||
cTitle := gjson.Get(cOrigin, "title").String()
|
||
msg = append(msg, message.Text(cTitle+"\n"))
|
||
cSummary := gjson.Get(cOrigin, "summary").String()
|
||
msg = append(msg, message.Text(cSummary))
|
||
cBannerURL := gjson.Get(cOrigin, "banner_url").String()
|
||
msg = append(msg, message.Image(cBannerURL))
|
||
case 256:
|
||
cUpper := gjson.Get(cOrigin, "upper").String()
|
||
cTime := time.UnixMilli(gjson.Get(cOrigin, "ctime").Int()).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cUpper+"在"+cTime+typeMsg[cOrigType]+"\n"))
|
||
cTitle := gjson.Get(cOrigin, "title").String()
|
||
msg = append(msg, message.Text(cTitle))
|
||
cCover := gjson.Get(cOrigin, "cover").String()
|
||
msg = append(msg, message.Image(cCover))
|
||
default:
|
||
msg = append(msg, message.Text("未知动态类型"+strconv.FormatInt(cOrigType, 10)+"\n"))
|
||
}
|
||
case 2:
|
||
cName := gjson.Get(cardStr, "user.name").String()
|
||
cUploadTime := time.Unix(gjson.Get(cardStr, "item.upload_time").Int(), 0).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cName+"在"+cUploadTime+typeMsg[cType]+"\n"))
|
||
cDescription := gjson.Get(cardStr, "item.description")
|
||
msg = append(msg, message.Text(cDescription))
|
||
if gjson.Get(cardStr, "item.pictures.#").Int() != 0 {
|
||
gjson.Get(cardStr, "item.pictures").ForEach(func(_, v gjson.Result) bool {
|
||
msg = append(msg, message.Image(v.Get("img_src").String()))
|
||
return true
|
||
})
|
||
}
|
||
case 4:
|
||
cName := gjson.Get(cardStr, "user.uname").String()
|
||
cTimestamp := time.Unix(gjson.Get(cardStr, "item.timestamp").Int(), 0).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cName+"在"+cTimestamp+typeMsg[cType]+"\n"))
|
||
cContent := gjson.Get(cardStr, "item.content").String()
|
||
msg = append(msg, message.Text(cContent+"\n"))
|
||
case 8:
|
||
cName := gjson.Get(cardStr, "owner.name").String()
|
||
cTime := time.Unix(gjson.Get(cardStr, "ctime").Int(), 0).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cName+"在"+cTime+typeMsg[cType]+"\n"))
|
||
cTitle := gjson.Get(cardStr, "title").String()
|
||
msg = append(msg, message.Text(cTitle))
|
||
cPic := gjson.Get(cardStr, "pic").String()
|
||
msg = append(msg, message.Image(cPic))
|
||
cDesc := gjson.Get(cardStr, "desc").String()
|
||
msg = append(msg, message.Text(cDesc+"\n"))
|
||
cShareSubtitle := gjson.Get(cardStr, "share_subtitle").String()
|
||
msg = append(msg, message.Text(cShareSubtitle+"\n"))
|
||
cShortLink := gjson.Get(cardStr, "short_link").String()
|
||
msg = append(msg, message.Text("视频链接："+cShortLink+"\n"))
|
||
case 16:
|
||
cName := gjson.Get(cardStr, "user.name").String()
|
||
cUploadTime := gjson.Get(cardStr, "item.upload_time").String()
|
||
msg = append(msg, message.Text(cName+"在"+cUploadTime+typeMsg[cType]+"\n"))
|
||
cDescription := gjson.Get(cardStr, "item.description")
|
||
msg = append(msg, message.Text(cDescription))
|
||
cCover := gjson.Get(cardStr, "item.cover.default").String()
|
||
msg = append(msg, message.Image(cCover))
|
||
case 64:
|
||
cName := gjson.Get(cardStr, "author.name").String()
|
||
cPublishTime := time.Unix(gjson.Get(cardStr, "publish_time").Int(), 0).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cName+"在"+cPublishTime+typeMsg[cType]+"\n"))
|
||
cTitle := gjson.Get(cardStr, "title").String()
|
||
msg = append(msg, message.Text(cTitle+"\n"))
|
||
cSummary := gjson.Get(cardStr, "summary").String()
|
||
msg = append(msg, message.Text(cSummary))
|
||
cBannerURL := gjson.Get(cardStr, "banner_url").String()
|
||
msg = append(msg, message.Image(cBannerURL))
|
||
case 256:
|
||
cUpper := gjson.Get(cardStr, "upper").String()
|
||
cTime := time.UnixMilli(gjson.Get(cardStr, "ctime").Int()).Format("2006-01-02 15:04:05")
|
||
msg = append(msg, message.Text(cUpper+"在"+cTime+typeMsg[cType]+"\n"))
|
||
cTitle := gjson.Get(cardStr, "title").String()
|
||
msg = append(msg, message.Text(cTitle))
|
||
cCover := gjson.Get(cardStr, "cover").String()
|
||
msg = append(msg, message.Image(cCover))
|
||
default:
|
||
msg = append(msg, message.Text("未知动态类型"+strconv.FormatInt(cType, 10)+"\n"))
|
||
}
|
||
cID := cardList[i].Get("desc.dynamic_id").String()
|
||
msg = append(msg, message.Text("动态链接：", tURL+cID))
|
||
|
||
zero.RangeBot(func(id int64, ctx *zero.Ctx) bool {
|
||
for _, gid := range groupList {
|
||
if m.IsEnabledIn(gid) {
|
||
switch {
|
||
case gid > 0:
|
||
ctx.SendGroupMessage(gid, msg)
|
||
case gid < 0:
|
||
ctx.SendPrivateMessage(-gid, msg)
|
||
default:
|
||
log.Errorln("[bilibilipush]:gid为0")
|
||
}
|
||
}
|
||
}
|
||
return true
|
||
})
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
func sendLive() {
|
||
uids := bdb.getAllBuidByLive()
|
||
gjson.Get(getLiveList(uids...), "data").ForEach(func(key, value gjson.Result) bool {
|
||
newStatus := int(value.Get("live_status").Int())
|
||
if newStatus == 2 {
|
||
newStatus = 0
|
||
}
|
||
if _, ok := liveStatus[key.Int()]; !ok {
|
||
liveStatus[key.Int()] = newStatus
|
||
return true
|
||
}
|
||
oldStatus := liveStatus[key.Int()]
|
||
if newStatus != oldStatus && newStatus == 1 {
|
||
liveStatus[key.Int()] = newStatus
|
||
m, ok := control.Lookup(serviceName)
|
||
if ok {
|
||
groupList := bdb.getAllGroupByBuidAndLive(key.Int())
|
||
roomID := value.Get("short_id").Int()
|
||
if roomID == 0 {
|
||
roomID = value.Get("room_id").Int()
|
||
}
|
||
lURL := liveURL + strconv.FormatInt(roomID, 10)
|
||
lName := value.Get("uname").String()
|
||
lTitle := value.Get("title").String()
|
||
lCover := value.Get("cover_from_user").String()
|
||
if lCover == "" {
|
||
lCover = value.Get("keyframe").String()
|
||
}
|
||
var msg []message.MessageSegment
|
||
msg = append(msg, message.Text(lName+" 正在直播：\n"))
|
||
msg = append(msg, message.Text(lTitle))
|
||
msg = append(msg, message.Image(lCover))
|
||
msg = append(msg, message.Text("直播链接：", lURL))
|
||
zero.RangeBot(func(id int64, ctx *zero.Ctx) bool {
|
||
for _, gid := range groupList {
|
||
if m.IsEnabledIn(gid) {
|
||
switch {
|
||
case gid > 0:
|
||
ctx.SendGroupMessage(gid, msg)
|
||
case gid < 0:
|
||
ctx.SendPrivateMessage(-gid, msg)
|
||
default:
|
||
log.Errorln("[bilibilipush]:gid为0")
|
||
}
|
||
}
|
||
}
|
||
return true
|
||
})
|
||
}
|
||
} else if newStatus != oldStatus {
|
||
liveStatus[key.Int()] = newStatus
|
||
}
|
||
return true
|
||
})
|
||
}
|