feat: enable update checksum validation

Author: kldzj

commands/update.go

@@ -15,8 +15,11 @@ func cmdUpdate() *cobra.Command {
 				cobra.CheckErr(util.ErrVerNotSet)
 			}
 
+			updater, err := version.NewUpdater()
+			cobra.CheckErr(err)
+
 			ver := version.Get()
-			latest, err := version.GetLatestRelease()
+			latest, err := version.GetLatestRelease(updater)
 			cobra.CheckErr(err)
 
 			if version.IsLatest(ver, latest) {
@@ -30,7 +33,7 @@ func cmdUpdate() *cobra.Command {
 				return
 			}
 
-			err = version.Update(ver, latest)
+			err = version.Update(ver, latest, updater)
 			cobra.CheckErr(err)
 		},
 	}

main.go

@@ -23,8 +23,13 @@ pzmod --file server.ini set name "My Server"`,
 				return
 			}
 
+			updater, err := version.NewUpdater()
+			if err != nil {
+				return
+			}
+
 			ver := version.Get()
-			latest, err := version.GetLatestRelease()
+			latest, err := version.GetLatestRelease(updater)
 			if err != nil {
 				return
 			}

version/update.go

@@ -11,8 +11,16 @@ import (
 
 const repoSlug = "kldzj/pzmod"
 
-func GetLatestRelease() (*selfupdate.Release, error) {
-	latest, found, err := selfupdate.DetectLatest(context.Background(), selfupdate.ParseSlug(repoSlug))
+func NewUpdater() (*selfupdate.Updater, error) {
+	return selfupdate.NewUpdater(selfupdate.Config{
+		Validator: &selfupdate.ChecksumValidator{
+			UniqueFilename: "checksums.txt",
+		},
+	})
+}
+
+func GetLatestRelease(updater *selfupdate.Updater) (*selfupdate.Release, error) {
+	latest, found, err := updater.DetectLatest(context.Background(), selfupdate.ParseSlug(repoSlug))
 	if err != nil {
 		return nil, fmt.Errorf("error occurred while fetching release: %w", err)
 	}
@@ -28,7 +36,7 @@ func IsLatest(current string, latest *selfupdate.Release) bool {
 	return latest != nil && latest.LessOrEqual(current)
 }
 
-func Update(current string, latest *selfupdate.Release) error {
+func Update(current string, latest *selfupdate.Release, updater *selfupdate.Updater) error {
 	if IsLatest(current, latest) {
 		fmt.Printf("Current version %s is the latest", current)
 		return nil
@@ -39,7 +47,7 @@ func Update(current string, latest *selfupdate.Release) error {
 		return errors.New("could not locate executable path")
 	}
 
-	if err := selfupdate.UpdateTo(context.Background(), latest.AssetURL, latest.AssetName, exe); err != nil {
+	if err := updater.UpdateTo(context.Background(), latest, exe); err != nil {
 		return fmt.Errorf("error occurred while updating binary: %w", err)
 	}