135 lines
3.3 KiB
Go
135 lines
3.3 KiB
Go
package bilibili
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/CuteReimu/bilibili/v2"
|
|
"github.com/ssdomei232/goodBaby/handler/db"
|
|
"github.com/ssdomei232/goodBaby/model"
|
|
)
|
|
|
|
// 检查cookie是否过期
|
|
func isCookiesExpired(cookiesString string) bool {
|
|
// 从cookie字符串中提取过期时间
|
|
lines := strings.SplitSeq(cookiesString, "\n")
|
|
for line := range lines {
|
|
if strings.Contains(line, "Expires=") {
|
|
// 提取过期时间字符串
|
|
expireStr := ""
|
|
parts := strings.SplitSeq(line, ";")
|
|
for part := range parts {
|
|
part = strings.TrimSpace(part)
|
|
if after, ok := strings.CutPrefix(part, "Expires="); ok {
|
|
expireStr = after
|
|
break
|
|
}
|
|
}
|
|
|
|
if expireStr != "" {
|
|
// 解析过期时间
|
|
expireTime, err := http.ParseTime(expireStr)
|
|
if err != nil {
|
|
log.Printf("解析过期时间失败: %v", err)
|
|
return true // 如果解析失败，cookie已过期
|
|
}
|
|
if time.Now().After(expireTime) {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
// 检查cookie是否即将过期（提前1天）
|
|
func isCookiesExpiringSoon(cookiesString string) bool {
|
|
lines := strings.SplitSeq(cookiesString, "\n")
|
|
for line := range lines {
|
|
if strings.Contains(line, "Expires=") {
|
|
// 提取过期时间字符串
|
|
expireStr := ""
|
|
parts := strings.SplitSeq(line, ";")
|
|
for part := range parts {
|
|
part = strings.TrimSpace(part)
|
|
if after, ok := strings.CutPrefix(part, "Expires="); ok {
|
|
expireStr = after
|
|
break
|
|
}
|
|
}
|
|
|
|
if expireStr != "" {
|
|
// 解析过期时间
|
|
expireTime, err := http.ParseTime(expireStr)
|
|
if err != nil {
|
|
log.Printf("解析过期时间失败: %v", err)
|
|
return true // 如果解析失败，认为需要重新登录
|
|
}
|
|
|
|
// 检查是否即将过期（提前1天）
|
|
if time.Now().Add(24 * time.Hour).After(expireTime) {
|
|
return true // 即将过期
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
// 触发登录请求
|
|
func triggerLoginRequest(biliClient *bilibili.Client) {
|
|
|
|
// 生成新的二维码
|
|
qrCode, err := biliClient.GetQRCode()
|
|
if err != nil {
|
|
log.Printf("获取二维码失败: %v", err)
|
|
return
|
|
}
|
|
|
|
// 在命令行打印二维码
|
|
log.Println("需要重新登录哔哩哔哩")
|
|
qrCode.Print()
|
|
|
|
// 通过Dingtalk发送登录请求
|
|
// title := "B站登录请求 - Cookie即将过期"
|
|
// msg := "您的B站cookie即将过期,请尽快重启摇篮系统进行登陆,请尽快完成登录操作。"
|
|
|
|
// 发送Dingtalk通知
|
|
// if config.DingtalkBot.AccessToken != "" && config.DingtalkBot.Secret != "" {
|
|
// SendDingTalkMsg(title, msg, config.DingtalkBot.AccessToken, config.DingtalkBot.Secret)
|
|
// }
|
|
|
|
}
|
|
|
|
// 获取 Bilibili Client
|
|
func GetBiliClient(rule *model.Rule) (*bilibili.Client, error) {
|
|
client := bilibili.New()
|
|
gormDB, err := db.GetGormDB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 1. 通过 account_id 获取到对应的 account 配置
|
|
var biliAccount model.Account
|
|
result := gormDB.Where("id = ?", rule.AccountID).First(&biliAccount)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
|
|
// 2. 通过 account 配置中的 config 字段获取到 CookiesString
|
|
var biliAccountConfig BiliAccount
|
|
err = json.Unmarshal([]byte(biliAccount.Config), &biliAccountConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
client.SetCookiesString(biliAccountConfig.CookiesString)
|
|
|
|
return client, nil
|
|
}
|