x

core/init.go

@@ -1,10 +1,15 @@
 package core
 
 import (
+	"fmt"
 	"os"
+	"regexp"
+	"runtime"
+	"syscall"
 	"time"
 
 	"github.com/beego/beego/v2/adapter/httplib"
+	"github.com/cdle/sillyplus/core/storage"
 	"github.com/cdle/sillyplus/utils"
 )
 
@@ -18,8 +23,105 @@ func Init() {
 	}
 	// utils.ReadYaml(utils.ExecPath+"/conf/", &Config, "https://raw.githubusercontent.com/cdle/sillyplus/main/conf/demo_config.yaml")
 	initToHandleMessage()
-
+	sillyGirl.Set("compiled_at", compiled_at)
+	console.Log("编译版本：%s", compiled_at)
 	sillyGirl.Set("started_at", time.Now().Format("2006-01-02 15:04:05"))
+	storage.Watch(sillyGirl, "compiled_at", func(old, new, key string) *storage.Final {
+		if old != new {
+			console.Debug("正在从 cdle/binary 获取版本号...")
+			data, err := httplib.Get("https://raw.githubusercontent.com/cdle/binary/main/compile_time.go").Bytes()
+			if err != nil {
+				console.Error("获取版本号错误：%s", err)
+				return &storage.Final{
+					Error: fmt.Errorf("貌似网络不太行啊：%s", err),
+				}
+			}
+			latest_version := regexp.MustCompile(`\d{13}`).FindString(string(data))
+			if latest_version <= compiled_at {
+				console.Debug("当前版本 %s 已是最新，无需升级", compiled_at)
+				return &storage.Final{
+					Message: fmt.Sprintf("当前版本 %s 已是最新，无需升级", compiled_at),
+				}
+			}
+			console.Debug("正在从 cdle/binary 获取最新版本 %s 编译文件...", latest_version)
+			qurl := "https://raw.githubusercontent.com/cdle/binary/master/sillyGirl_linux_" + runtime.GOARCH + "_" + latest_version
+			if runtime.GOARCH == "windows" {
+				qurl += ".exe"
+			}
+			req := httplib.Get(qurl)
+			req.SetTimeout(time.Minute*5, time.Minute*5)
+			data, err = req.Bytes()
+			if err != nil {
+				console.Error("获取最新编译文件错误：%s", err)
+				return &storage.Final{
+					Error: fmt.Errorf("升级时貌似网络不太行啊：%v", err),
+				}
+			}
+			if len(data) < 2646140 {
+				console.Error("获取最新编译文件错误：%v", len(data))
+				return &storage.Final{
+					Error: fmt.Errorf("升级时貌似网络不太行啊！%v", len(data)),
+				}
+			}
+			console.Debug("正在创建编译文件...")
+			filename := utils.ExecPath + "/" + utils.ProcessName
+			ready := filename + ".ready"
+			if f, err := os.OpenFile(ready, syscall.O_CREAT, 0777); err != nil {
+				console.Error("创建编译文件错误：%v", err)
+				return &storage.Final{
+					Error: fmt.Errorf("创建编译文件错误：%v", err),
+				}
+			} else {
+				_, err := f.Write(data)
+				f.Close()
+				if err != nil {
+					des := err.Error()
+					if err = os.WriteFile(ready, data, 0777); err != nil {
+						console.Error("写入编译文件错误：%s || %s", des, err)
+						return &storage.Final{
+							Error: fmt.Errorf("写入编译文件错误：%s || %s", des, err),
+						}
+					}
+				}
+			}
+			console.Debug("正在删除旧程序错误...")
+			if err = os.RemoveAll(filename); err != nil {
+				console.Error("删除旧程序错误：%v", err)
+				return &storage.Final{
+					Error: fmt.Errorf("删除旧程序错误：%v", err),
+				}
+			}
+			console.Debug("正在移动新程序错误...")
+			if err = os.Rename(ready, filename); err != nil {
+				console.Error("移动新程序错误：%v", err)
+				return &storage.Final{
+					Error: fmt.Errorf("移动新程序错误：%v", err),
+				}
+			}
+			go func() {
+				console.Debug("正在重启...")
+				time.Sleep(time.Second)
+				utils.Daemon()
+			}()
+			return &storage.Final{
+				Message: "升级成功，即将重启！",
+			}
+		}
+		return nil
+	})
+	storage.Watch(sillyGirl, "started_at", func(old, new, key string) *storage.Final {
+		if old != new {
+			go func() {
+				time.Sleep(time.Second)
+				utils.Daemon()
+			}()
+			return &storage.Final{
+				Message: "马上重启！",
+			}
+		}
+		return nil
+	})
+
 	api_key := sillyGirl.GetString("api_key")
 	if api_key == "" {
 		api_key := time.Now().UnixNano()
@@ -40,7 +142,4 @@ func Init() {
 	initWebPluginList()
 	go initPluginList()
 	initPluginPublish()
-	if compiled_at != "" {
-		console.Log("编译时间戳，", compiled_at)
-	}
 }