From c7c02a6718ae37c16a6f5f3325c5acb770056b6f Mon Sep 17 00:00:00 2001
From: Mark Bailey <mark.bailey@moyerautogroup.com>
Date: Sat, 14 Sep 2024 15:35:54 -0400
Subject: [PATCH] refactor: make mask/unmask more intuitive

---
 cmd/root.go | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/cmd/root.go b/cmd/root.go
index b036f0f..4c9ae4e 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -43,24 +43,27 @@ var rootCmd = &cobra.Command{
 func downgrade() error {
 	fmt.Println("Downgrading ", PkgName, "to ", Version)
 
+	if Mask && Unmask {
+		return errors.New("please choose either --mask or --unmask, not both")
+	}
+
 	cmd := exec.Command("flatpak", "update", "--commit="+Version, PkgName)
 
 	cmd.Stdin = os.Stdin
 	cmd.Stdout = os.Stdout
 	cmd.Stderr = os.Stderr
 
+	if Unmask {
+		unmask()
+	}
+
 	err := cmd.Run()
 	if err != nil {
 		return err
 	}
 
-	if Mask && Unmask {
-		return errors.New("please choose either --mask or --unmask, not both")
-	}
 	if Mask {
 		mask()
-	} else if Unmask {
-		unmask()
 	}
 
 	return nil