feat: support auto check upgrade (#53)

internal/cli/cli.go

@@ -42,6 +42,15 @@ func Start() {
 				Usage: "load configuration from FILE (default: config.yaml)",
 			},
 		},
+		Commands: []*cli.Command{
+			{
+				Name: "upgrade",
+				Action: func(c *cli.Context) error {
+					checkUpdateFromCli()
+					return nil
+				},
+			},
+		},
 		Action: func(c *cli.Context) error {
 			var configFile = "config.yaml"
 			if c.Path("config") != "" {
@@ -81,6 +90,11 @@ func Start() {
 			if c.String("log") != "" {
 				conf.LogLevel = c.String("log")
 			}
+
+			if conf.CheckUpgrade {
+				checkUpdateFromCli()
+			}
+
 			server.New(conf).Run()
 			return nil
 		},

internal/cli/config.tpl.yaml

@@ -6,6 +6,7 @@ external_ip:                        # You need to make sure that your target can
 admin_path_prefix: "/revsuit"
 database: revsuit.db
 log_level: info
+check_upgrade: true
 
 ip_location_database:
   database: "qqwry"                 # qqwry or geoip.

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
+}

internal/update/update.go

@@ -0,0 +1,25 @@
+package update
+
+import (
+	"strings"
+
+	"github.com/blang/semver"
+	"github.com/pkg/errors"
+	"github.com/rhysd/go-github-selfupdate/selfupdate"
+)
+
+func CheckUpgrade(currentVersion string) (bool, *selfupdate.Release, error) {
+	latest, found, err := selfupdate.DetectLatest("Li4n0/revsuit")
+	if err != nil {
+		return false, nil, errors.Wrap(err, "error when detecting version")
+	}
+
+	v, err := semver.Parse(strings.TrimPrefix(currentVersion, "v"))
+	if err != nil {
+		return false, nil, errors.Wrap(err, "error when parse version")
+	}
+	if !found || latest.Version.LTE(v) {
+		return false, nil, nil
+	}
+	return true, latest, nil
+}