feat: 增加续命保活

Author: yqchilde

engine/go.mod

@@ -1,11 +1,15 @@
 package engine
 
 import (
+	"encoding/json"
 	"fmt"
+	"os"
 	"sync/atomic"
+	"time"
 
 	"github.com/eatmoreapple/openwechat"
 	"github.com/yqchilde/pkgs/log"
+	"github.com/yqchilde/pkgs/timer"
 
 	"github.com/yqchilde/wxbot/engine/robot"
 )
@@ -103,9 +107,79 @@ func InitRobot() {
 		robot.Groups = groups
 	}
 
+	robot.Bot = bot
+	go keepalive()
 	bot.Block()
 }
 
+func keepalive() {
+	task := timer.NewTimerTask()
+	_, err := task.AddTaskByFunc("keepalive", "0 0/30 * * * *", func() {
+		if robot.Bot.Alive() {
+			if checkWhetherNeedToLogin() {
+				reloadStorage := &robot.JsonLocalStorage{FileName: "storage.json"}
+				if err := robot.Bot.HotLogin(reloadStorage, false); err != nil {
+					log.Errorf("热登录续命失败, err: %v", err)
+					return
+				}
+				log.Println("热登录续命成功")
+				if err := robot.Bot.DumpHotReloadStorage(); err != nil {
+					log.Errorf("热登录数据持久化失败, err: %v", err)
+					return
+				}
+				log.Println("热登录数据持久化成功")
+			}
+
+			helper, err := robot.Self.FileHelper()
+			if err != nil {
+				log.Errorf("获取文件助手失败, err: %v", err)
+				return
+			}
+			if _, err := helper.SendText(openwechat.ZombieText); err != nil {
+				log.Errorf("Robot保活失败, err: %v", err)
+				return
+			}
+			log.Println("Robot保活成功")
+		}
+	})
+	if err != nil {
+		log.Errorf("NewScheduled add task error: %v", err)
+	}
+}
+
+func checkWhetherNeedToLogin() bool {
+	storage, err := os.ReadFile("storage.json")
+	if err != nil {
+		log.Errorf("获取热登录配置失败, err: %v", err)
+		return false
+	}
+
+	var hotLoginData openwechat.HotReloadStorageItem
+	err = json.Unmarshal(storage, &hotLoginData)
+	if err != nil {
+		log.Errorf("unmarshal hot login storage err: %v", err)
+		return false
+	}
+
+	for _, cookies := range hotLoginData.Cookies {
+		if len(cookies) <= 0 {
+			continue
+		}
+
+		for _, cookie := range cookies {
+			if cookie.Name == "wxsid" {
+				gmtLocal, _ := time.LoadLocation("GMT")
+				expiresGMTTime, _ := time.ParseInLocation("Mon, 02-Jan-2006 15:04:05 GMT", cookie.RawExpires, gmtLocal)
+				expiresLocalTime := expiresGMTTime.In(time.Local)
+				overHours := expiresLocalTime.Sub(time.Now().Local()).Hours()
+				log.Printf("距离登录失效还剩%v小时", overHours)
+				return overHours < 3
+			}
+		}
+	}
+	return false
+}
+
 func getGroupNicknameByGroupUsername(username string) string {
 	groups := robot.Groups.SearchByUserName(1, username)
 	return groups[0].NickName

engine/robot.go

@@ -3,6 +3,7 @@ package robot
 import "github.com/eatmoreapple/openwechat"
 
 var (
+	Bot     *openwechat.Bot
 	Self    *openwechat.Self
 	Friends openwechat.Friends
 	Groups  openwechat.Groups

engine/robot/robot.go

@@ -2,19 +2,17 @@ module github.com/yqchilde/wxbot
 
 go 1.19
 
-replace github.com/yqchilde/wxbot/engine v0.0.1 => ./engine
-
 require (
+	github.com/eatmoreapple/openwechat v1.1.14
+	github.com/gabriel-vasile/mimetype v1.4.1
+	github.com/pkg/errors v0.9.1
 	github.com/tidwall/gjson v1.14.2
 	github.com/yqchilde/pkgs v0.0.2
-	github.com/yqchilde/wxbot/engine v0.0.1
+	gopkg.in/yaml.v3 v3.0.1
 )
 
 require (
 	github.com/arthurkiller/rollingwriter v1.1.3 // indirect
-	github.com/eatmoreapple/openwechat v1.1.14 // indirect
-	github.com/gabriel-vasile/mimetype v1.4.1 // indirect
-	github.com/pkg/errors v0.9.1 // indirect
 	github.com/robfig/cron v1.1.0 // indirect
 	github.com/robfig/cron/v3 v3.0.1 // indirect
 	github.com/tidwall/match v1.1.1 // indirect
@@ -23,5 +21,4 @@ require (
 	go.uber.org/multierr v1.6.0 // indirect
 	go.uber.org/zap v1.22.0 // indirect
 	golang.org/x/net v0.0.0-20220624214902-1bab6f366d9e // indirect
-	gopkg.in/yaml.v3 v3.0.1 // indirect
 )