synctv/internal/bootstrap/update.go

package bootstrap

import (
	"context"
	"time"

	log "github.com/sirupsen/logrus"
	sysnotify "github.com/synctv-org/synctv/internal/sysNotify"
	"github.com/synctv-org/synctv/internal/version"
)

func InitCheckUpdate(ctx context.Context) error {
	v, err := version.NewVersionInfo()
	if err != nil {
		log.Fatalf("get version info error: %v", err)
	}

	go func() {
		execFile, err := version.ExecutableFile()
		if err != nil {
			execFile = "synctv"
		}

		var (
			need   bool
			latest string
			url    string
		)
		need, latest, url, err = check(ctx, v)
		if err != nil {
			log.Errorf("check update error: %v", err)
		} else {
			if need {
				log.Infof("new version (%s) available: %s", latest, url)
				log.Infof("run '%s self-update' to auto update", execFile)
			}
		}

		sysnotify.RegisterSysNotifyTask(0, sysnotify.NewSysNotifyTask(
			"check-update",
			sysnotify.NotifyTypeEXIT,
			func() error {
				if need {
					log.Infof("new version (%s) available: %s", latest, url)
					log.Infof("run '%s self-update' to auto update", execFile)
				}
				return nil
			},
		))

		t := time.NewTicker(time.Hour * 6)
		defer t.Stop()
		for range t.C {
			need, latest, url, err = check(ctx, v)
			if err != nil {
				log.Errorf("check update error: %v", err)
			}
		}
	}()

	return nil
}

func check(ctx context.Context, v *version.VersionInfo) (need bool, latest string, url string, err error) {
	l, err := v.CheckLatest(ctx)
	if err != nil {
		return false, "", "", err
	}
	latest = l
	b, err := v.NeedUpdate(ctx)
	if err != nil {
		return false, "", "", err
	}
	need = b
	if b {
		u, err := v.LatestBinaryURL(ctx)
		if err != nil {
			return false, "", "", err
		}
		url = u
	}
	return need, latest, url, nil
}