🎨 桌面端自动下载更新安装包 https://github.com/siyuan-note/siyuan/issues/5837

kernel/model/liandi.go

@@ -144,11 +144,11 @@ func LoadUploadToken() (err error) {
 }
 
 var (
-	refreshUserTicker              = time.NewTicker(2 * time.Hour)
+	refreshCheckTicker             = time.NewTicker(2 * time.Hour)
 	subscriptionExpirationReminded bool
 )
 
-func AutoRefreshUser() {
+func AutoRefreshCheck() {
 	for {
 		if !subscriptionExpirationReminded {
 			subscriptionExpirationReminded = true
@@ -235,7 +235,12 @@ func AutoRefreshUser() {
 			}
 		}()
 
-		<-refreshUserTicker.C
+		go func() {
+			defer logging.Recover()
+			checkDownloadInstallPkg()
+		}()
+
+		<-refreshCheckTicker.C
 	}
 }
 

kernel/model/updater.go

@@ -17,16 +17,111 @@
 package model
 
 import (
+	"bufio"
+	"crypto/sha256"
 	"fmt"
-	"sync"
+	"io"
+	"os"
+	"path/filepath"
+	"runtime"
 
+	"github.com/88250/gulu"
+	"github.com/imroc/req/v3"
 	"github.com/siyuan-note/logging"
 	"github.com/siyuan-note/siyuan/kernel/util"
 )
 
-var (
-	checkUpdateLock = &sync.Mutex{}
-)
+func checkDownloadInstallPkg() {
+	if !Conf.System.DownloadInstallPkg {
+		return
+	}
+
+	result, err := util.GetRhyResult(false)
+	if nil != err {
+		return
+	}
+
+	installPkgSite := result["installPkg"].(string)
+	ver := result["ver"].(string)
+	if ver == util.Ver {
+		return
+	}
+
+	var suffix string
+	if gulu.OS.IsWindows() {
+		if "386" == runtime.GOARCH {
+			suffix = "win32.exe"
+		} else {
+			suffix = "win.exe"
+		}
+	} else if gulu.OS.IsDarwin() {
+		if "arm64" == runtime.GOARCH {
+			suffix = "mac-arm64.dmg"
+		} else {
+			suffix = "mac.dmg"
+		}
+	} else if gulu.OS.IsLinux() {
+		suffix = "linux.AppImage"
+	}
+	pkg := "siyuan-" + ver + "-" + suffix
+	installPkg := "siyuan/" + pkg
+	downloadPkgURL := installPkgSite + installPkg
+	localPkgPath := filepath.Join(util.TempDir, "install", pkg)
+	checksums := result["checksums"].(map[string]interface{})
+	checksum := checksums[pkg].(string)
+
+	downloadInstallPkg(downloadPkgURL, checksum, localPkgPath)
+}
+
+func downloadInstallPkg(pkgURL, checksum, savePath string) {
+	if gulu.File.IsExist(savePath) {
+		localChecksum, _ := sha256Hash(savePath)
+		if localChecksum == checksum {
+			return
+		}
+	}
+
+	client := req.C()
+	callback := func(info req.DownloadInfo) {
+		logging.LogInfof("downloading install package from [%s] %.2f%%", pkgURL, float64(info.DownloadedSize)/float64(info.Response.ContentLength)*100.0)
+	}
+	resp, err := client.R().SetOutputFile(savePath).SetDownloadCallback(callback).Get(pkgURL)
+	if nil != err {
+		logging.LogErrorf("download install package failed: %s", err)
+		return
+	}
+	if 200 != resp.StatusCode {
+		logging.LogErrorf("download install package [%s] failed [sc=%d]", pkgURL, resp.StatusCode)
+		return
+	}
+	logging.LogInfof("downloaded install package [%s] to [%s]", pkgURL, savePath)
+	localChecksum, _ := sha256Hash(savePath)
+	if checksum != localChecksum {
+		logging.LogErrorf("verify checksum failed, download install package [%s] checksum [%s] not equal to downloaded [%s] checksum [%s]", pkgURL, checksum, savePath, localChecksum)
+	}
+}
+
+func sha256Hash(filename string) (ret string, err error) {
+	file, err := os.Open(filename)
+	if nil != err {
+		return
+	}
+	defer file.Close()
+
+	hash := sha256.New()
+	reader := bufio.NewReader(file)
+	buf := make([]byte, 1024*1024*4)
+	for {
+		switch n, readErr := reader.Read(buf); readErr {
+		case nil:
+			hash.Write(buf[:n])
+		case io.EOF:
+			return fmt.Sprintf("%x", hash.Sum(nil)), nil
+		default:
+			return "", err
+		}
+	}
+}
 
 type Announcement struct {
 	Id    string `json:"id"`
@@ -58,9 +153,6 @@ func CheckUpdate(showMsg bool) {
 		return
 	}
 
-	checkUpdateLock.Lock()
-	defer checkUpdateLock.Unlock()
-
 	result, err := util.GetRhyResult(showMsg)
 	if nil != err {
 		return