This commit is contained in:
mei
2025-10-07 17:43:50 +08:00
parent 950e2fada5
commit 98b193030c
3 changed files with 24 additions and 25 deletions
+1
View File
@@ -45,6 +45,7 @@ type Basic struct {
CauseStop string `json:"cause_stop"` CauseStop string `json:"cause_stop"`
} }
// Get config from json file
func GetConfig() (config Config, err error) { func GetConfig() (config Config, err error) {
content, err := os.ReadFile("config.json") content, err := os.ReadFile("config.json")
if err != nil { if err != nil {
+14 -13
View File
@@ -12,6 +12,7 @@ import (
"github.com/ssdomei232/goodBaby/configs" "github.com/ssdomei232/goodBaby/configs"
) )
// check tmp dir exist
func EnsureTmpDirectory() error { func EnsureTmpDirectory() error {
_, err := os.Stat("tmp") _, err := os.Stat("tmp")
if os.IsNotExist(err) { if os.IsNotExist(err) {
@@ -29,41 +30,41 @@ func EnsureTmpDirectory() error {
} }
func isCookiesExpired(cookiesString string) bool { func isCookiesExpired(cookiesString string) bool {
// 解析cookie中的过期时间 // parase expires date from cookie string
lines := strings.SplitSeq(cookiesString, "\n") lines := strings.SplitSeq(cookiesString, "\n")
for line := range lines { for line := range lines {
if strings.Contains(line, "Expires=") { if strings.Contains(line, "Expires=") {
// 提取过期时间字符串 // 提取过期时间字符串
expireStr := "" expireStr := ""
parts := strings.Split(line, ";") parts := strings.SplitSeq(line, ";")
for _, part := range parts { for part := range parts {
part = strings.TrimSpace(part) part = strings.TrimSpace(part)
if strings.HasPrefix(part, "Expires=") { if after, ok := strings.CutPrefix(part, "Expires="); ok {
expireStr = strings.TrimPrefix(part, "Expires=") expireStr = after
break break
} }
} }
if expireStr != "" { if expireStr != "" {
// 解析过期时间 // phrase expirres date
expireTime, err := http.ParseTime(expireStr) expireTime, err := http.ParseTime(expireStr)
if err != nil { if err != nil {
log.Printf("解析过期时间失败: %v", err) log.Printf("解析过期时间失败: %v", err)
return true // 如果解析失败,认为已过期 return true // if phrase failed, cookie expired
} }
// 比较过期时间和当前时间 // compare expires date and now
if time.Now().After(expireTime) { if time.Now().After(expireTime) {
return true // 已过期 return true // expired
} }
} }
} }
} }
return false // 未过期 return false // not expired
} }
// 二维码登录 // Qrcode login
func LoginWithQRCode(biliClient *bilibili.Client) { func LoginWithQRCode(biliClient *bilibili.Client) {
qrCode, err := biliClient.GetQRCode() qrCode, err := biliClient.GetQRCode()
if err != nil { if err != nil {
@@ -86,11 +87,11 @@ func LoginWithQRCode(biliClient *bilibili.Client) {
log.Println("登录成功") log.Println("登录成功")
// 保存新获取的cookie // save new cookie
saveCookies(biliClient) saveCookies(biliClient)
} }
// 保存cookie到文件 // save cookie to file
func saveCookies(client *bilibili.Client) { func saveCookies(client *bilibili.Client) {
cookiesString := client.GetCookiesString() cookiesString := client.GetCookiesString()
err := os.WriteFile("tmp/cookies.txt", []byte(cookiesString), 0644) err := os.WriteFile("tmp/cookies.txt", []byte(cookiesString), 0644)
+9 -12
View File
@@ -27,6 +27,7 @@ var biliClient *bilibili.Client
var cookieCheckTimer *time.Ticker var cookieCheckTimer *time.Ticker
func init() { func init() {
// Get COnfig
config, err := configs.GetConfig() config, err := configs.GetConfig()
if err != nil { if err != nil {
log.Printf("获取配置文件失败: %v", err) log.Printf("获取配置文件失败: %v", err)
@@ -38,15 +39,15 @@ func init() {
duration = time.Duration(config.DisconnectDuration) * time.Hour duration = time.Duration(config.DisconnectDuration) * time.Hour
} }
// 检查并创建tmp目录 // Check and create tmp dir
if err := internal.EnsureTmpDirectory(); err != nil { if err := internal.EnsureTmpDirectory(); err != nil {
log.Printf("创建tmp目录失败: %v", err) log.Printf("创建tmp目录失败: %v", err)
} }
// 初始化bilibili客户端 // Init Bilibili Clinet
biliClient = bilibili.New() biliClient = bilibili.New()
initBilibili() initBilibili()
startCookieChecker() // 启动定期检查cookie有效性 startCookieChecker() // Check Bilibili cookie
internal.InitTimerManager(duration) internal.InitTimerManager(duration)
timer = time.NewTimer(duration) timer = time.NewTimer(duration)
@@ -69,12 +70,8 @@ func trigger(config configs.Config) {
func main() { func main() {
r := gin.Default() r := gin.Default()
// 设置嵌入的模板
templFS, _ := fs.Sub(templateFiles, "templates") templFS, _ := fs.Sub(templateFiles, "templates")
r.SetHTMLTemplate(loadTemplates(templFS)) r.SetHTMLTemplate(loadTemplates(templFS))
// 提供嵌入的静态文件
staticFS, _ := fs.Sub(staticFiles, "static") staticFS, _ := fs.Sub(staticFiles, "static")
r.StaticFS("/static", http.FS(staticFS)) r.StaticFS("/static", http.FS(staticFS))
@@ -89,18 +86,18 @@ func indexPage(c *gin.Context) {
} }
func initBilibili() { func initBilibili() {
// 尝试加载已存储的cookie // try cached cookie
if !internal.LoadCookies(biliClient) { if !internal.LoadCookies(biliClient) {
// 如果没有有效cookie,则进行二维码登录 // if cookie check failed, request qrcode login
internal.LoginWithQRCode(biliClient) internal.LoginWithQRCode(biliClient)
} else { } else {
log.Println("使用已存储的有效cookie登录") log.Println("使用已存储的有效cookie登录")
} }
} }
// 启动cookie定期检查 // Enable periodic cookie checks
func startCookieChecker() { func startCookieChecker() {
// 每小时检查一次cookie有效性 // check cookie per hour
cookieCheckTimer = time.NewTicker(1 * time.Hour) cookieCheckTimer = time.NewTicker(1 * time.Hour)
go func() { go func() {
for { for {
@@ -133,7 +130,7 @@ func handleSignal(c *gin.Context) {
log.Println("触发信号") log.Println("触发信号")
} }
// loadTemplates 从嵌入的文件系统加载模板 // loadTemplates loads templates from the embedded file system
func loadTemplates(filesystem fs.FS) *template.Template { func loadTemplates(filesystem fs.FS) *template.Template {
templ := template.Must(template.New("").ParseFS(filesystem, "*.html")) templ := template.Must(template.New("").ParseFS(filesystem, "*.html"))
return templ return templ