add log-level flag to server command

Signed-off-by: Steve Kriss <steve@heptio.com>

docs/cli-reference/ark_server.md

@@ -14,7 +14,8 @@ ark server [flags]
 ### Options
 
 ```
-  -h, --help   help for server
+  -h, --help        help for server
+      --log-level   the level at which to log. Valid values are debug, info, warning, error, fatal, panic. (default info)
 ```
 
 ### Options inherited from parent commands

pkg/cmd/server/server.go

@@ -22,6 +22,8 @@ import (
 	"io/ioutil"
 	"os"
 	"reflect"
+	"sort"
+	"strings"
 	"sync"
 	"time"
 
@@ -48,6 +50,7 @@ import (
 	"github.com/heptio/ark/pkg/cloudprovider/azure"
 	"github.com/heptio/ark/pkg/cloudprovider/gcp"
 	"github.com/heptio/ark/pkg/cmd"
+	"github.com/heptio/ark/pkg/cmd/util/flag"
 	"github.com/heptio/ark/pkg/controller"
 	arkdiscovery "github.com/heptio/ark/pkg/discovery"
 	"github.com/heptio/ark/pkg/generated/clientset"
@@ -60,15 +63,30 @@ import (
 )
 
 func NewCommand() *cobra.Command {
-	var kubeconfig string
+	var (
+		kubeconfig      string
+		sortedLogLevels = getSortedLogLevels()
+		logLevelFlag    = flag.NewEnum(logrus.InfoLevel.String(), sortedLogLevels...)
+	)
 
 	var command = &cobra.Command{
 		Use:   "server",
 		Short: "Run the ark server",
 		Long:  "Run the ark server",
 		Run: func(c *cobra.Command, args []string) {
-			logger := logrus.New()
+			logLevel := logrus.InfoLevel
-			logger.Hooks.Add(&logging.ErrorLocationHook{})
+
+			if parsed, err := logrus.ParseLevel(logLevelFlag.String()); err == nil {
+				logLevel = parsed
+			} else {
+				// This should theoretically never happen assuming the enum flag
+				// is constructed correctly because the enum flag will not allow
+				//  an invalid value to be set.
+				logrus.Errorf("log-level flag has invalid value %s", strings.ToUpper(logLevelFlag.String()))
+			}
+			logrus.Infof("setting log-level to %s", strings.ToUpper(logLevel.String()))
+
+			logger := newLogger(logLevel, &logging.ErrorLocationHook{})
 
 			s, err := newServer(kubeconfig, fmt.Sprintf("%s-%s", c.Parent().Name(), c.Name()), logger)
 
@@ -78,11 +96,43 @@ func NewCommand() *cobra.Command {
 		},
 	}
 
+	command.Flags().Var(logLevelFlag, "log-level", fmt.Sprintf("the level at which to log. Valid values are %s.", strings.Join(sortedLogLevels, ", ")))
 	command.Flags().StringVar(&kubeconfig, "kubeconfig", "", "Path to the kubeconfig file to use to talk to the Kubernetes apiserver. If unset, try the environment variable KUBECONFIG, as well as in-cluster configuration")
 
 	return command
 }
 
+func newLogger(level logrus.Level, hooks ...logrus.Hook) *logrus.Logger {
+	logger := logrus.New()
+	logger.Level = level
+
+	for _, hook := range hooks {
+		logger.Hooks.Add(hook)
+	}
+
+	return logger
+}
+
+// getSortedLogLevels returns a string slice containing all of the valid logrus
+// log levels (based on logrus.AllLevels), sorted in ascending order of severity.
+func getSortedLogLevels() []string {
+	var (
+		sortedLogLevels  = make([]logrus.Level, len(logrus.AllLevels))
+		logLevelsStrings []string
+	)
+
+	copy(sortedLogLevels, logrus.AllLevels)
+
+	// logrus.Panic has the lowest value, so the compare function uses ">"
+	sort.Slice(sortedLogLevels, func(i, j int) bool { return sortedLogLevels[i] > sortedLogLevels[j] })
+
+	for _, level := range sortedLogLevels {
+		logLevelsStrings = append(logLevelsStrings, level.String())
+	}
+
+	return logLevelsStrings
+}
+
 type server struct {
 	kubeClient            kubernetes.Interface
 	arkClient             clientset.Interface

pkg/cmd/util/flag/enum.go

@@ -31,12 +31,12 @@ type Enum struct {
 }
 
 // NewEnum returns a new enum flag with the specified list
-// of allowed values. The first value specified is used
+// of allowed values, and the specified default value if
-// as the default.
+// none is set.
-func NewEnum(allowedValues ...string) Enum {
+func NewEnum(defaultValue string, allowedValues ...string) *Enum {
-	return Enum{
+	return &Enum{
 		allowedValues: sets.NewString(allowedValues...),
-		value:         allowedValues[0],
+		value:         defaultValue,
 	}
 }
 
@@ -61,5 +61,8 @@ func (e *Enum) Set(s string) error {
 // Type returns a string representation of the
 // Enum type.
 func (e *Enum) Type() string {
-	return "enum"
+	// we don't want the help text to display anything regarding
+	// the type because the usage text for the flag should capture
+	// the possible options.
+	return ""
 }