goodBaby/internal/bilibili.go

package internal

import (
	"fmt"
	"log"
	"net/http"
	"os"
	"strings"
	"time"

	"github.com/CuteReimu/bilibili/v2"
	"github.com/ssdomei232/goodBaby/configs"
)

// 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本消息由自动程序发送(摇篮系统)", 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登录")
	}
}