diff options
Diffstat (limited to 'pkg/updates')
-rw-r--r-- | pkg/updates/updates.go | 74 |
1 files changed, 57 insertions, 17 deletions
diff --git a/pkg/updates/updates.go b/pkg/updates/updates.go index 7e745635c..ae2265c4e 100644 --- a/pkg/updates/updates.go +++ b/pkg/updates/updates.go @@ -9,6 +9,7 @@ import ( "os" "path/filepath" "runtime" + "strings" "time" "github.com/kardianos/osext" @@ -21,10 +22,9 @@ import ( // Update checks for updates and does updates type Updater struct { - LastChecked string - Log *logrus.Entry - Config config.AppConfigurer - OSCommand *commands.OSCommand + Log *logrus.Entry + Config config.AppConfigurer + OSCommand *commands.OSCommand } // Updater implements the check and update methods @@ -42,16 +42,14 @@ func NewUpdater(log *logrus.Logger, config config.AppConfigurer, osCommand *comm contextLogger := log.WithField("context", "updates") updater := &Updater{ - LastChecked: "today", - Log: contextLogger, - Config: config, - OSCommand: osCommand, + Log: contextLogger, + Config: config, + OSCommand: osCommand, } return updater, nil } func (u *Updater) getLatestVersionNumber() (string, error) { - time.Sleep(5) req, err := http.NewRequest("GET", projectUrl+"/releases/latest", nil) if err != nil { return "", err @@ -76,21 +74,41 @@ func (u *Updater) getLatestVersionNumber() (string, error) { return dat["tag_name"].(string), nil } +func (u *Updater) RecordLastUpdateCheck() error { + u.Config.GetAppState().LastUpdateCheck = time.Now().Unix() + return u.Config.SaveAppState() +} + +// expecting version to be of the form `v12.34.56` +func (u *Updater) majorVersionDiffers(oldVersion, newVersion string) bool { + return strings.Split(oldVersion, ".")[0] != strings.Split(newVersion, ".")[0] +} + func (u *Updater) checkForNewUpdate() (string, error) { u.Log.Info("Checking for an updated version") - // if u.Config.GetVersion() == "unversioned" { - // u.Log.Info("Current version is not built from an official release so we won't check for an update") - // return "", nil - // } + if u.Config.GetVersion() == "unversioned" { + u.Log.Info("Current version is not built from an official release so we won't check for an update") + return "", nil + } newVersion, err := u.getLatestVersionNumber() if err != nil { return "", err } u.Log.Info("Current version is " + u.Config.GetVersion()) u.Log.Info("New version is " + newVersion) - // if newVersion == u.Config.GetVersion() { - // return "", nil - // } + + if err := u.RecordLastUpdateCheck(); err != nil { + return "", err + } + + if newVersion == u.Config.GetVersion() { + return "", nil + } + + if u.majorVersionDiffers(u.Config.GetVersion(), newVersion) { + u.Log.Info("New version has non-backwards compatible changes.") + return "", nil + } rawUrl, err := u.getBinaryUrl(newVersion) if err != nil { @@ -102,11 +120,16 @@ func (u *Updater) checkForNewUpdate() (string, error) { return "", nil } u.Log.Info("Verified resource is available, ready to update") + return newVersion, nil } // CheckForNewUpdate checks if there is an available update -func (u *Updater) CheckForNewUpdate(onFinish func(string, error) error) { +func (u *Updater) CheckForNewUpdate(onFinish func(string, error) error, userRequested bool) { + if !userRequested && u.skipUpdateCheck() { + return + } + go func() { newVersion, err := u.checkForNewUpdate() if err = onFinish(newVersion, err); err != nil { @@ -115,6 +138,23 @@ func (u *Updater) CheckForNewUpdate(onFinish func(string, error) error) { }() } +func (u *Updater) skipUpdateCheck() bool { + if u.Config.GetBuildSource() != "buildBinary" { + return true + } + + userConfig := u.Config.GetUserConfig() + if userConfig.Get("update.method") == "never" { + return true + } + + currentTimestamp := time.Now().Unix() + lastUpdateCheck := u.Config.GetAppState().LastUpdateCheck + days := userConfig.GetInt64("update.days") + + return (currentTimestamp-lastUpdateCheck)/(60*60*24) < days +} + func (u *Updater) mappedOs(os string) string { osMap := map[string]string{ "darwin": "Darwin", |