package internal import ( "fmt" "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 } stopMsg := fmt.Sprintf("%s\n\n%s\n\n本消息由自动程序发送(摇篮系统)", 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("cookies.txt"); os.IsNotExist(err) { log.Println("cookie文件不存在,需要重新登录") triggerLoginRequest(biliClient) return } cookiesBytes, err := os.ReadFile("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.Split(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 // 如果解析失败,认为需要重新登录 } // 检查是否即将过期(提前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() // 通过邮件发送登录请求 emailTitle := "B站登录请求 - Cookie即将过期" emailMsg := "您的B站cookie即将过期,请尽快重启摇篮系统进行登陆,请尽快完成登录操作。" // 发送邮件通知 if config.BiliWarnAddress != "" { err = sendMailMsg(config.BiliWarnAddress, emailMsg, emailTitle) if err != nil { log.Printf("发送登录邮件失败: %v", err) } else { log.Println("登录请求邮件已发送") } } }