feat: 发送bilibili动态

2025-10-06 20:02:47 +08:00
parent d4ffc72c8b
commit 5e7af981ad
6 changed files with 196 additions and 0 deletions
@@ -0,0 +1,146 @@
+package internal
+
+import (
+	"log"
+	"net/http"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/CuteReimu/bilibili/v2"
+	"github.com/ssdomei232/goodBaby/configs"
+)
+
+func isCookiesExpired(cookiesString string) bool {
+	// 解析cookie中的过期时间
+	lines := strings.SplitSeq(cookiesString, "\n")
+	for line := range lines {
+		if strings.Contains(line, "Expires=") {
+			// 提取过期时间字符串
+			expireStr := ""
+			parts := strings.Split(line, ";")
+			for _, part := range parts {
+				part = strings.TrimSpace(part)
+				if strings.HasPrefix(part, "Expires=") {
+					expireStr = strings.TrimPrefix(part, "Expires=")
+					break
+				}
+			}
+
+			if expireStr != "" {
+				// 解析过期时间
+				expireTime, err := http.ParseTime(expireStr)
+				if err != nil {
+					log.Printf("解析过期时间失败: %v", err)
+					return true // 如果解析失败，认为已过期
+				}
+
+				// 比较过期时间和当前时间
+				if time.Now().After(expireTime) {
+					return true // 已过期
+				}
+			}
+		}
+	}
+
+	return false // 未过期
+}
+
+// 二维码登录
+func LoginWithQRCode(biliClient *bilibili.Client) {
+	qrCode, err := biliClient.GetQRCode()
+	if err != nil {
+		log.Printf("获取二维码失败: %v", err)
+		return
+	}
+
+	log.Println("请使用哔哩哔哩APP扫码登录")
+	qrCode.Print()
+
+	result, err := biliClient.LoginWithQRCode(bilibili.LoginWithQRCodeParam{
+		QrcodeKey: qrCode.QrcodeKey,
+	})
+	if err != nil || result.Code != 0 {
+		log.Printf("登录失败: %v", err)
+		return
+	}
+
+	log.Println("登录成功")
+
+	// 保存新获取的cookie
+	saveCookies(biliClient)
+}
+
+// 保存cookie到文件
+func saveCookies(client *bilibili.Client) {
+	cookiesString := client.GetCookiesString()
+	err := os.WriteFile("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("cookies.txt"); os.IsNotExist(err) {
+		return false
+	}
+
+	// 读取cookie
+	cookiesBytes, err := os.ReadFile("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
+}
+
+func SendBili(biliClient *bilibili.Client) {
+	config, err := configs.GetConfig()
+	if err != nil {
+		log.Printf("获取配置文件失败: %v", err)
+		return
+	}
+
+	var dynamicParams bilibili.CreateDynamicParam
+	dynamicParams = bilibili.CreateDynamicParam{
+		DynamicId: 0,
+		Type:      4,
+		Rid:       0,
+		Content:   config.BiliMsg,
+	}
+
+	// 重试机制
+	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("动态发送最终失败，已达到最大重试次数")
+}