更新若干功能
This commit is contained in:
李琦
226
update.go
Normal file
226
update.go
Normal file
@@ -0,0 +1,226 @@
|
||||
package main
|
||||
|
||||
// update.go 客户端软件自动更新：每 30 分钟检查服务端最新版，下载 NSIS 后拉起安装。
|
||||
|
||||
import (
|
||||
"crypto/sha256"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"view/platform"
|
||||
)
|
||||
|
||||
type appLatestInfo struct {
|
||||
Version string `json:"version"`
|
||||
Channel string `json:"channel"`
|
||||
SHA256 string `json:"sha256"`
|
||||
SizeBytes int64 `json:"sizeBytes"`
|
||||
Changelog string `json:"changelog"`
|
||||
CreatedAt string `json:"createdAt"`
|
||||
}
|
||||
|
||||
// CheckAppUpdate 手动/定时检查更新；force=true 忽略“已忽略版本”。
|
||||
func (a *App) CheckAppUpdate(force bool) (map[string]any, error) {
|
||||
if e := a.ready(); e != nil {
|
||||
return nil, e
|
||||
}
|
||||
info, e := a.fetchLatestRelease()
|
||||
if e != nil {
|
||||
if e.Error() == "NO_RELEASE" {
|
||||
return map[string]any{"upToDate": true, "current": appVersion}, nil
|
||||
}
|
||||
return nil, e
|
||||
}
|
||||
if !versionNewer(info.Version, appVersion) {
|
||||
return map[string]any{"upToDate": true, "current": appVersion, "latest": info.Version}, nil
|
||||
}
|
||||
if !force {
|
||||
if skipped := strings.TrimSpace(a.store.Meta("app_update_skipped")); skipped == info.Version {
|
||||
return map[string]any{"upToDate": false, "skipped": true, "current": appVersion, "latest": info.Version}, nil
|
||||
}
|
||||
}
|
||||
payload := map[string]any{
|
||||
"upToDate": false,
|
||||
"current": appVersion,
|
||||
"latest": info.Version,
|
||||
"changelog": info.Changelog,
|
||||
"sizeBytes": info.SizeBytes,
|
||||
"sha256": info.SHA256,
|
||||
"channel": info.Channel,
|
||||
}
|
||||
a.emit("app:update-available", payload)
|
||||
return payload, nil
|
||||
}
|
||||
|
||||
// SkipAppUpdate 本次跳过该版本提示。
|
||||
func (a *App) SkipAppUpdate(version string) error {
|
||||
if e := a.ready(); e != nil {
|
||||
return e
|
||||
}
|
||||
return a.store.SetMeta("app_update_skipped", strings.TrimSpace(version))
|
||||
}
|
||||
|
||||
// DownloadAndInstallUpdate 下载最新安装包、校验哈希后启动并退出应用。
|
||||
func (a *App) DownloadAndInstallUpdate() error {
|
||||
if e := a.ready(); e != nil {
|
||||
return e
|
||||
}
|
||||
info, e := a.fetchLatestRelease()
|
||||
if e != nil {
|
||||
return e
|
||||
}
|
||||
if !versionNewer(info.Version, appVersion) {
|
||||
return errors.New("ALREADY_LATEST")
|
||||
}
|
||||
path, e := a.downloadRelease(info)
|
||||
if e != nil {
|
||||
return e
|
||||
}
|
||||
cmd := exec.Command(path, "/S")
|
||||
platform.ConfigureHidden(cmd)
|
||||
if e := cmd.Start(); e != nil {
|
||||
// 静默失败则普通启动
|
||||
cmd2 := exec.Command(path)
|
||||
if e2 := cmd2.Start(); e2 != nil {
|
||||
return errors.New("INSTALLER_START_FAILED")
|
||||
}
|
||||
}
|
||||
a.store.Log("info", "系统", "开始安装更新", info.Version)
|
||||
go func() {
|
||||
time.Sleep(800 * time.Millisecond)
|
||||
a.QuitApp()
|
||||
}()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *App) runAppUpdateLoop() {
|
||||
time.Sleep(60 * time.Second)
|
||||
a.checkAppUpdateQuiet()
|
||||
t := time.NewTicker(30 * time.Minute)
|
||||
defer t.Stop()
|
||||
for {
|
||||
select {
|
||||
case <-a.ctx.Done():
|
||||
return
|
||||
case <-t.C:
|
||||
a.checkAppUpdateQuiet()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (a *App) checkAppUpdateQuiet() {
|
||||
if a.store == nil || a.bootstrap.State != BootstrapReady {
|
||||
return
|
||||
}
|
||||
if a.syncUserID() <= 0 || a.store.Meta("sync_access_token") == "" {
|
||||
return
|
||||
}
|
||||
_, _ = a.CheckAppUpdate(false)
|
||||
}
|
||||
|
||||
func (a *App) fetchLatestRelease() (*appLatestInfo, error) {
|
||||
if a.apiBaseURL() == "" {
|
||||
return nil, errors.New("SYNC_NOT_CONFIGURED")
|
||||
}
|
||||
var info appLatestInfo
|
||||
if e := a.apiDecode(http.MethodGet, "/api/v1/app/latest?channel=stable", nil, &info, false); e != nil {
|
||||
return nil, e
|
||||
}
|
||||
if strings.TrimSpace(info.Version) == "" {
|
||||
return nil, errors.New("NO_RELEASE")
|
||||
}
|
||||
return &info, nil
|
||||
}
|
||||
|
||||
func (a *App) downloadRelease(info *appLatestInfo) (string, error) {
|
||||
base := a.apiBaseURL()
|
||||
if base == "" {
|
||||
return "", errors.New("SYNC_NOT_CONFIGURED")
|
||||
}
|
||||
url := base + "/api/v1/app/download/" + pathEscape(info.Version) + "?channel=stable"
|
||||
req, e := http.NewRequest(http.MethodGet, url, nil)
|
||||
if e != nil {
|
||||
return "", errors.New("SYNC_OFFLINE")
|
||||
}
|
||||
if tok := strings.TrimSpace(a.store.Meta("sync_access_token")); tok != "" {
|
||||
req.Header.Set("Authorization", "Bearer "+tok)
|
||||
}
|
||||
client := &http.Client{Timeout: 30 * time.Minute}
|
||||
resp, e := client.Do(req)
|
||||
if e != nil {
|
||||
return "", errors.New("SYNC_OFFLINE")
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode >= 400 {
|
||||
raw, _ := io.ReadAll(io.LimitReader(resp.Body, 4096))
|
||||
var er struct {
|
||||
Error string `json:"error"`
|
||||
}
|
||||
if json.Unmarshal(raw, &er) == nil && er.Error != "" {
|
||||
return "", errors.New(er.Error)
|
||||
}
|
||||
return "", errors.New("SYNC_HTTP_" + strconv.Itoa(resp.StatusCode))
|
||||
}
|
||||
dir := filepath.Join(os.TempDir(), "code-count-updates")
|
||||
_ = os.MkdirAll(dir, 0755)
|
||||
outPath := filepath.Join(dir, info.Version+"-installer.exe")
|
||||
f, e := os.Create(outPath)
|
||||
if e != nil {
|
||||
return "", errors.New("SAVE_FAILED")
|
||||
}
|
||||
h := sha256.New()
|
||||
n, e := io.Copy(io.MultiWriter(f, h), resp.Body)
|
||||
_ = f.Close()
|
||||
if e != nil {
|
||||
_ = os.Remove(outPath)
|
||||
return "", errors.New("DOWNLOAD_FAILED")
|
||||
}
|
||||
sum := hex.EncodeToString(h.Sum(nil))
|
||||
if info.SHA256 != "" && !strings.EqualFold(sum, info.SHA256) {
|
||||
_ = os.Remove(outPath)
|
||||
return "", errors.New("SHA256_MISMATCH")
|
||||
}
|
||||
if info.SizeBytes > 0 && n != info.SizeBytes {
|
||||
a.store.Log("warning", "系统", "更新包大小与元数据不一致", fmt.Sprintf("%d vs %d", n, info.SizeBytes))
|
||||
}
|
||||
return outPath, nil
|
||||
}
|
||||
|
||||
// versionNewer 判断 remote 是否比 local 新（简单 x.y.z 比较）。
|
||||
func versionNewer(remote, local string) bool {
|
||||
rp := parseSemver(remote)
|
||||
lp := parseSemver(local)
|
||||
for i := 0; i < 3; i++ {
|
||||
if rp[i] > lp[i] {
|
||||
return true
|
||||
}
|
||||
if rp[i] < lp[i] {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func parseSemver(v string) [3]int {
|
||||
v = strings.TrimPrefix(strings.TrimSpace(v), "v")
|
||||
if i := strings.IndexAny(v, "-+"); i >= 0 {
|
||||
v = v[:i]
|
||||
}
|
||||
parts := strings.Split(v, ".")
|
||||
var out [3]int
|
||||
for i := 0; i < 3 && i < len(parts); i++ {
|
||||
n, _ := strconv.Atoi(parts[i])
|
||||
out[i] = n
|
||||
}
|
||||
return out
|
||||
}
|
||||
Reference in New Issue
Block a user