Implement OTA functionality for version checking

Author: ixugo

.gitignore

@@ -42,4 +42,5 @@ cover/
 *.jpg
 __pycache__/
 *.pt
-*.onnx
+*.onnx
+*.remember/

analysis/logger.py

@@ -59,7 +59,7 @@ def setup_logging(level_str: str = "INFO", retention_days: int = 3):
     )
 
     # 设置后缀格式，例如 app.log.2023-12-31
-    file_handler.suffix = "%Y-%m-%d"
+    file_handler.suffix = "%Y-%m-%d.log"
     file_handler.setFormatter(formatter)
     root_logger.addHandler(file_handler)
 

configs/config.toml

@@ -2,18 +2,26 @@
   Debug = false
   # rtmp 推流秘钥
   RTMPSecret = '123'
-
+  # 登录用户名
   Username = 'admin'
+  # 登录密码
   Password = 'admin'
 
+  # ai 分析服务
+  [Server.AI]
+    # 是否禁用 ai 分析服务
+    Disabled = false
+    # 保留天数
+    RetainDays = 0
+
   # 对外提供的服务，建议由 nginx 代理
   [Server.HTTP]
     # http 端口
     Port = 15123
     # 请求超时时间
     Timeout = '1m0s'
     # jwt 秘钥，空串时，每次启动程序将随机赋值
-    JwtSecret = ''
+    JwtSecret = '6caOiETMs8SPWNHgEKA1Jhmn9wxpjAj9'
 
     [Server.HTTP.PProf]
       # 是否启用 pprof, 建议设置为 true
@@ -22,7 +30,7 @@
       AccessIps = ['::1', '127.0.0.1']
 
 [Data]
-  # 数据库支持 sqlite/postgres/mysql ，使用 sqlite 时 dsn 应当填写文件存储路径
+  # 数据库支持 sqlite/postgres/mysql, 使用 sqlite 时 dsn 应当填写文件存储路径
   # postgres://postgres:123456@127.0.0.1:5432/gb28181?sslmode=disable
   # mysql://root:123456@127.0.0.1:5432/gb28181?sslmode=disable
   [Data.Database]
@@ -61,9 +69,11 @@
   HTTPPort = 8080
   # 媒体服务器密钥
   Secret = 'jvRqCAzEg7AszBi4gm1cfhwXpmnVmJMG'
+  # 媒体服务器类型 zlm/lalmax
+  Type = 'zlm'
   # 用于流媒体 webhook 回调
-  WebHookIP = '192.168.10.10'
+  WebHookIP = '192.168.1.3'
   # 媒体服务器 RTP 端口范围
   RTPPortRange = '20000-20100'
   # 媒体服务器 SDP IP
-  SDPIP = '192.168.10.10'
+  SDPIP = '192.168.1.3'

docker-compose.yml

@@ -3,7 +3,7 @@ services:
     # 如果拉不到 docker hub 镜像，也可以尝试
     # registry.cn-shanghai.aliyuncs.com/ixugo/homenvr:latest
     image: gospace/gowvp:latest
-    restart: unless-stopped
+    restart: always
     # linux 解开下行注释，并将 ports 全部注释
     # network_mode: host
     ports:

internal/web/api/api.go

@@ -19,6 +19,7 @@ import (
 	"github.com/gin-contrib/gzip"
 	"github.com/gin-gonic/gin"
 	"github.com/gowvp/owl/internal/core/sms"
+	"github.com/gowvp/owl/pkg/ota"
 	"github.com/gowvp/owl/plugin/stat"
 	"github.com/gowvp/owl/plugin/stat/statapi"
 	"github.com/ixugo/goddd/domain/version/versionapi"
@@ -94,6 +95,8 @@ func setupRouter(r *gin.Engine, uc *Usecase) {
 	auth := web.AuthMiddleware(uc.Conf.Server.HTTP.JwtSecret)
 	r.GET("/health", web.WrapH(uc.getHealth))
 	r.GET("/app/metrics/api", web.WrapH(uc.getMetricsAPI))
+	r.GET("/app/version/check", web.WrapH(uc.checkVersion))
+	r.POST("/app/upgrade", auth, uc.upgradeApp)
 
 	versionapi.Register(r, uc.Version, auth)
 	statapi.Register(r)
@@ -202,6 +205,109 @@ func sortExpvarMap(data *expvar.Map, top int) []KV {
 	return kvs[:idx]
 }
 
+const repoName = "gowvp/owl"
+
+type checkVersionOutput struct {
+	HasNewVersion  bool   `json:"has_new_version"`
+	CurrentVersion string `json:"current_version"`
+	NewVersion     string `json:"new_version"`
+	Description    string `json:"description"`
+}
+
+// checkVersion 检查是否有新版本
+// 通过 GitHub API 获取最新 release 信息，与当前版本比较
+func (uc *Usecase) checkVersion(_ *gin.Context, _ *struct{}) (checkVersionOutput, error) {
+	currentVersion := uc.Conf.BuildVersion
+	newVersion, body, err := ota.GetLastVersion(repoName)
+	if err != nil {
+		return checkVersionOutput{}, err
+	}
+
+	hasNew := compareVersion(currentVersion, newVersion) < 0
+
+	return checkVersionOutput{
+		HasNewVersion:  hasNew,
+		CurrentVersion: currentVersion,
+		NewVersion:     newVersion,
+		Description:    body,
+	}, nil
+}
+
+// compareVersion 比较两个版本号
+// 返回值: -1 表示 v1 < v2, 0 表示相等, 1 表示 v1 > v2
+func compareVersion(v1, v2 string) int {
+	v1 = strings.TrimPrefix(v1, "v")
+	v2 = strings.TrimPrefix(v2, "v")
+
+	parts1 := strings.Split(v1, ".")
+	parts2 := strings.Split(v2, ".")
+
+	maxLen := len(parts1)
+	if len(parts2) > maxLen {
+		maxLen = len(parts2)
+	}
+
+	for i := 0; i < maxLen; i++ {
+		var n1, n2 int
+		if i < len(parts1) {
+			fmt.Sscanf(parts1[i], "%d", &n1)
+		}
+		if i < len(parts2) {
+			fmt.Sscanf(parts2[i], "%d", &n2)
+		}
+		if n1 < n2 {
+			return -1
+		}
+		if n1 > n2 {
+			return 1
+		}
+	}
+	return 0
+}
+
+// upgradeApp 执行应用升级
+// 通过 SSE 返回下载进度，下载完成后由回调决定如何升级
+func (uc *Usecase) upgradeApp(c *gin.Context) {
+	c.Header("Content-Type", "text/event-stream")
+	c.Header("Cache-Control", "no-cache")
+	c.Header("Connection", "keep-alive")
+	c.Header("Access-Control-Allow-Origin", "*")
+
+	flusher, ok := c.Writer.(http.Flusher)
+	if !ok {
+		c.JSON(http.StatusInternalServerError, gin.H{"msg": "不支持 SSE"})
+		return
+	}
+
+	sendEvent := func(event, data string) {
+		fmt.Fprintf(c.Writer, "event: %s\ndata: %s\n\n", event, data)
+		flusher.Flush()
+	}
+
+	sendEvent("start", `{"msg":"开始下载升级包"}`)
+
+	filename := "linux_amd64"
+	if runtime.GOARCH == "arm64" {
+		filename = "linux_arm64"
+	}
+
+	o := ota.NewOTA(repoName, filename)
+	o.SetProgressCallback(func(current, total int64) {
+		percent := 0
+		if total > 0 {
+			percent = int(current * 100 / total)
+		}
+		sendEvent("progress", fmt.Sprintf(`{"current":%d,"total":%d,"percent":%d}`, current, total, percent))
+	})
+
+	if err := o.Download().Error(); err != nil {
+		sendEvent("error", fmt.Sprintf(`{"msg":"%s"}`, err.Error()))
+		return
+	}
+
+	sendEvent("complete", `{"msg":"下载完成，请手动重启服务"}`)
+}
+
 func (uc *Usecase) proxySMS(c *gin.Context) {
 	defer func() {
 		_ = recover()