feat: support auto check upgrade (#53)

internal/cli/update.go

@@ -0,0 +1,56 @@
+package cli
+
+import (
+	"fmt"
+	"os"
+	"strings"
+
+	"github.com/li4n0/revsuit/internal/update"
+	"github.com/li4n0/revsuit/pkg/server"
+	"github.com/pkg/errors"
+	"github.com/rhysd/go-github-selfupdate/selfupdate"
+	log "unknwon.dev/clog/v2"
+)
+
+func checkUpdateFromCli() {
+	log.Info("Checking for updates...")
+	if upgradeable, release, err := update.CheckUpgrade(server.VERSION); err != nil {
+		log.Warn(err.Error())
+	} else if upgradeable {
+		err := updateFromCli(release)
+		if err != nil {
+			log.Warn(err.Error())
+		}
+	} else {
+		log.Info("The current version is the latest")
+	}
+}
+
+func updateFromCli(release *selfupdate.Release) error {
+	releaseNotes := release.ReleaseNotes
+	releaseNotes = strings.Replace(releaseNotes, "\n\n", "\n", -1)
+	releaseNotes = strings.Replace(releaseNotes, "\r\n\r\n", "\n", -1)
+	releaseNotes = strings.Replace(releaseNotes, "**", "", -1)
+
+	log.Info("New version detected: %s\n%s", release.URL, releaseNotes)
+	log.Warn("Do you want to update to v%s?[Y/n]:", release.Version)
+
+	var upgrade string
+	_, _ = fmt.Scanln(&upgrade)
+	if upgrade != "n" {
+		exe, err := os.Executable()
+		if err != nil {
+			log.Error("error when locate executable path, err: %v", err)
+		}
+
+		assetURL := "https://upgrade.revsuit.pro" + strings.TrimPrefix(release.AssetURL, "https://github.com/Li4n0/revsuit/releases") + "?from=" + server.VERSION
+
+		log.Info("Downloading from %s", assetURL)
+		if err := selfupdate.UpdateTo(assetURL, exe); err != nil {
+			return errors.Wrap(err, "error when download asset")
+		}
+		log.Info("Successfully updated to v%s", release.Version)
+		return nil
+	}
+	return nil
+}