Add debug log flag

alenkacz · Dec 30, 2020 · 3caa9a2 · 3caa9a2
1 parent 17a6947
commit 3caa9a2
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/pkg/cmd/verify/deployments.go b/pkg/cmd/verify/deployments.go
@@ -3,21 +3,23 @@ package verify
 import (
 	"context"
 	"fmt"
+	"os"
+	"time"
+
 	"github.com/alenkacz/cert-manager-verifier/pkg/verify"
+	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 	"k8s.io/cli-runtime/pkg/genericclioptions"
 	"k8s.io/client-go/dynamic"
 	"k8s.io/client-go/kubernetes"
-	"os"
-	"github.com/sirupsen/logrus"
-	"time"
 )
 
 const defaultTimeout = 2 * time.Minute
 
 type Options struct {
 	ConfigFlags *genericclioptions.ConfigFlags
 	Streams     *genericclioptions.IOStreams
+	DebugLogs   bool
 }
 
 func NewOptions() *Options {
@@ -46,6 +48,8 @@ func NewCmd() *cobra.Command {
 		},
 	}
 
+	rootCmd.Flags().BoolVar(&options.DebugLogs, "debug", false, "If true, will print out debug logs (default false)")
+
 	options.ConfigFlags.AddFlags(rootCmd.Flags())
 	rootCmd.SetOut(options.Streams.Out)
 	rootCmd.SilenceUsage = true
@@ -59,6 +63,9 @@ func NewCmd() *cobra.Command {
 func (o *Options) Execute() error {
 	logrus.SetOutput(o.Streams.Out)
 	logrus.SetFormatter(SimpleFormatter{})
+	if o.DebugLogs {
+		logrus.SetLevel(logrus.DebugLevel)
+	}
 
 	ctx, cancel := context.WithTimeout(context.Background(), defaultTimeout)
 	defer cancel()
@@ -127,8 +134,8 @@ func formatDeploymentResult(result []verify.DeploymentResult) string {
 	return formattedResult
 }
 
-type SimpleFormatter struct {}
+type SimpleFormatter struct{}
 
 func (SimpleFormatter) Format(entry *logrus.Entry) ([]byte, error) {
 	return []byte(entry.Message), nil
-}
+}