refactor(cmd/fw): wtf is that error handling

cmd/banforge/command/fw.go

@@ -16,13 +16,14 @@ var (
 	port     int
 	protocol string
 )
+
 var UnbanCmd = &cobra.Command{
 	Use:   "unban",
 	Short: "Unban IP",
 	Run: func(cmd *cobra.Command, args []string) {
+		err := func() error {
 			if len(args) == 0 {
-			fmt.Println("IP can't be empty")
-			os.Exit(1)
+				return fmt.Errorf("IP can't be empty")
 			}
 			if ttl_fw == "" {
 				ttl_fw = "1y"
@@ -30,39 +31,38 @@ var UnbanCmd = &cobra.Command{
 			ip := args[0]
 			db, err := storage.NewBanWriter()
 			if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
+				return err
 			}
 			cfg, err := config.LoadConfig()
 			if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
+				return err
 			}
 			fw := cfg.Firewall.Name
 			b := blocker.GetBlocker(fw, cfg.Firewall.Config)
 			if ip == "" {
-			fmt.Println("IP can't be empty")
-			os.Exit(1)
+				return fmt.Errorf("IP can't be empty")
 			}
 			if net.ParseIP(ip) == nil {
-			fmt.Println("Invalid IP")
-			os.Exit(1)
+				return fmt.Errorf("Invalid IP")
 			}
 			if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
+				return err
 			}
 			err = b.Unban(ip)
 			if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
+				return err
 			}
 			err = db.RemoveBan(ip)
+			if err != nil {
+				return err
+			}
+			fmt.Println("IP unblocked successfully!")
+			return nil
+		}()
 		if err != nil {
 			fmt.Println(err)
 			os.Exit(1)
 		}
-		fmt.Println("IP unblocked successfully!")
 	},
 }
 
@@ -70,9 +70,9 @@ var BanCmd = &cobra.Command{
 	Use:   "ban",
 	Short: "Ban IP",
 	Run: func(cmd *cobra.Command, args []string) {
+		err := func() error {
 			if len(args) == 0 {
-			fmt.Println("IP can't be empty")
-			os.Exit(1)
+				return fmt.Errorf("IP can't be empty")
 			}
 			if ttl_fw == "" {
 				ttl_fw = "1y"
@@ -80,39 +80,38 @@ var BanCmd = &cobra.Command{
 			ip := args[0]
 			db, err := storage.NewBanWriter()
 			if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
+				return err
 			}
 			cfg, err := config.LoadConfig()
 			if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
+				return err
 			}
 			fw := cfg.Firewall.Name
 			b := blocker.GetBlocker(fw, cfg.Firewall.Config)
 			if ip == "" {
-			fmt.Println("IP can't be empty")
-			os.Exit(1)
+				return fmt.Errorf("IP can't be empty")
 			}
 			if net.ParseIP(ip) == nil {
-			fmt.Println("Invalid IP")
-			os.Exit(1)
+				return fmt.Errorf("Invalid IP")
 			}
 			if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
+				return err
 			}
 			err = b.Ban(ip)
 			if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
+				return err
 			}
 			err = db.AddBan(ip, ttl_fw, "manual ban")
+			if err != nil {
+				return err
+			}
+			fmt.Println("IP blocked successfully!")
+			return err
+		}()
 		if err != nil {
 			fmt.Println(err)
 			os.Exit(1)
 		}
-		fmt.Println("IP blocked successfully!")
 	},
 }