feat(sender/onebot): 新增onebot协议支持
This commit is contained in:
mei
@@ -0,0 +1,283 @@
|
||||
package sender
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/CuteReimu/bilibili/v2"
|
||||
"github.com/ssdomei232/goodBaby/configs"
|
||||
"github.com/ssdomei232/goodBaby/internal/base"
|
||||
)
|
||||
|
||||
// check tmp dir exist
|
||||
func EnsureTmpDirectory() error {
|
||||
_, err := os.Stat("tmp")
|
||||
if os.IsNotExist(err) {
|
||||
err = os.Mkdir("tmp", 0755)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
log.Println("已创建tmp目录")
|
||||
} else if err != nil {
|
||||
return err
|
||||
} else {
|
||||
log.Println("tmp目录已存在")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func isCookiesExpired(cookiesString string) bool {
|
||||
// parase expires date from cookie string
|
||||
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 != "" {
|
||||
// phrase expirres date
|
||||
expireTime, err := http.ParseTime(expireStr)
|
||||
if err != nil {
|
||||
log.Printf("解析过期时间失败: %v", err)
|
||||
return true // if phrase failed, cookie expired
|
||||
}
|
||||
|
||||
// compare expires date and now
|
||||
if time.Now().After(expireTime) {
|
||||
return true // expired
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return false // not expired
|
||||
}
|
||||
|
||||
// Qrcode login
|
||||
func LoginWithQRCode(biliClient *bilibili.Client) {
|
||||
qrCode, err := biliClient.GetQRCode()
|
||||
if err != nil {
|
||||
log.Printf("获取二维码失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
log.Println("请使用哔哩哔哩APP扫码登录或进入文件目录寻找qrcode.png查看二维码")
|
||||
buf, _ := qrCode.Encode()
|
||||
os.WriteFile("tmp/qrcode.png", buf, 0644)
|
||||
qrCode.Print()
|
||||
|
||||
result, err := biliClient.LoginWithQRCode(bilibili.LoginWithQRCodeParam{
|
||||
QrcodeKey: qrCode.QrcodeKey,
|
||||
})
|
||||
if err != nil || result.Code != 0 {
|
||||
log.Printf("登录失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
log.Println("登录成功")
|
||||
|
||||
// save new cookie
|
||||
saveCookies(biliClient)
|
||||
}
|
||||
|
||||
// save cookie to file
|
||||
func saveCookies(client *bilibili.Client) {
|
||||
cookiesString := client.GetCookiesString()
|
||||
err := os.WriteFile("tmp/cookies.txt", []byte(cookiesString), 0644)
|
||||
if err != nil {
|
||||
log.Printf("保存cookie失败: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// 从文件加载cookie并验证有效性
|
||||
func LoadCookies(client *bilibili.Client) bool {
|
||||
// 检查cookie文件是否存在
|
||||
if _, err := os.Stat("tmp/cookies.txt"); os.IsNotExist(err) {
|
||||
return false
|
||||
}
|
||||
|
||||
// 读取cookie
|
||||
cookiesBytes, err := os.ReadFile("tmp/cookies.txt")
|
||||
if err != nil {
|
||||
log.Printf("读取cookie文件失败: %v", err)
|
||||
return false
|
||||
}
|
||||
|
||||
cookiesString := string(cookiesBytes)
|
||||
client.SetCookiesString(cookiesString)
|
||||
|
||||
// 验证cookie是否过期
|
||||
if isCookiesExpired(cookiesString) {
|
||||
log.Println("cookie已过期")
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
// 在Bilibili动态发送死亡通告
|
||||
func SendBili(biliClient *bilibili.Client) {
|
||||
config, err := configs.GetConfig()
|
||||
if err != nil {
|
||||
log.Printf("获取配置文件失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
stopMsg := fmt.Sprintf("%s\n\n%s\n\n本消息由自动程序发送(摇篮系统)", base.GetBasicInfo(), config.BiliMsg)
|
||||
|
||||
var dynamicParams bilibili.CreateDynamicParam
|
||||
dynamicParams = bilibili.CreateDynamicParam{
|
||||
DynamicId: 0,
|
||||
Type: 4,
|
||||
Rid: 0,
|
||||
Content: stopMsg,
|
||||
}
|
||||
|
||||
// 重试机制
|
||||
retryCount := 10 // 重试次数
|
||||
delay := time.Second * 2 // 初始重试延迟
|
||||
|
||||
for i := 0; i <= retryCount; i++ {
|
||||
_, err := biliClient.CreateDynamic(dynamicParams)
|
||||
if err == nil {
|
||||
log.Println("动态发送成功")
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf("动态发送失败 (尝试 %d/%d): %v", i+1, retryCount+1, err)
|
||||
|
||||
// 如果不是最后一次尝试，则等待后重试
|
||||
if i < retryCount {
|
||||
log.Printf("等待 %v 后进行第 %d 次重试", delay, i+2)
|
||||
time.Sleep(delay)
|
||||
delay *= 2 // 指数退避
|
||||
}
|
||||
}
|
||||
|
||||
log.Println("动态发送最终失败，已达到最大重试次数")
|
||||
}
|
||||
|
||||
// 检查cookie有效性
|
||||
func CheckCookieValidity(biliClient *bilibili.Client) {
|
||||
// 读取存储的cookie文件
|
||||
if _, err := os.Stat("tmp/cookies.txt"); os.IsNotExist(err) {
|
||||
log.Println("cookie文件不存在,需要重新登录")
|
||||
triggerLoginRequest(biliClient)
|
||||
return
|
||||
}
|
||||
|
||||
cookiesBytes, err := os.ReadFile("tmp/cookies.txt")
|
||||
if err != nil {
|
||||
log.Printf("读取cookie文件失败: %v", err)
|
||||
triggerLoginRequest(biliClient)
|
||||
return
|
||||
}
|
||||
|
||||
cookiesString := string(cookiesBytes)
|
||||
|
||||
// 检查cookie是否即将过期（提前1天提醒）
|
||||
if isCookiesExpiringSoon(cookiesString) {
|
||||
log.Println("cookie即将过期,需要重新登录")
|
||||
triggerLoginRequest(biliClient)
|
||||
}
|
||||
}
|
||||
|
||||
// 检查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) {
|
||||
// 获取配置
|
||||
config, err := configs.GetConfig()
|
||||
if err != nil {
|
||||
log.Printf("获取配置文件失败: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
// 生成新的二维码
|
||||
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)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// Enable periodic cookie checks
|
||||
func StartCookieChecker(cookieCheckTimer *time.Ticker, biliClient *bilibili.Client) {
|
||||
// check cookie per hour
|
||||
cookieCheckTimer = time.NewTicker(1 * time.Hour)
|
||||
go func() {
|
||||
for {
|
||||
<-cookieCheckTimer.C
|
||||
CheckCookieValidity(biliClient)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
func InitBilibili(biliClient *bilibili.Client) {
|
||||
// try cached cookie
|
||||
if !LoadCookies(biliClient) {
|
||||
// if cookie check failed, request qrcode login
|
||||
LoginWithQRCode(biliClient)
|
||||
} else {
|
||||
log.Println("使用已存储的有效cookie登录")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user