Add verbose flag to enable debug logs (#27)

chanzuckerberg · Sep 25, 2018 · 0e697fa · 0e697fa
1 parent 14d6415
commit 0e697fa
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/cmd/root.go b/cmd/root.go
@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"github.com/pkg/errors"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
@@ -11,11 +12,22 @@ var (
 )
 
 func init() {
+	rootCmd.PersistentFlags().BoolP("verbose", "v", false, "Use this to enable verbose mode")
 }
 
 var rootCmd = &cobra.Command{
 	Use:   "blessclient",
 	Short: "",
+	PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
+		verbose, err := cmd.Flags().GetBool("verbose")
+		if err != nil {
+			return errors.Wrap(err, "Missing verbose flag")
+		}
+		if verbose {
+			log.SetLevel(log.DebugLevel)
+		}
+		return nil
+	},
 }
 
 // Execute executes the command