Implement required flags

flag.go

@@ -34,6 +34,7 @@ type Flag interface {
 	// Apply Flag settings to the given flag set
 	Apply(*flag.FlagSet)
 	getName() string
+	IsRequired() bool
 }
 
 func flagSet(name string, flags []Flag) *flag.FlagSet {
@@ -61,10 +62,11 @@ type Generic interface {
 
 // GenericFlag is the flag type for types implementing Generic
 type GenericFlag struct {
-	Name   string
-	Value  Generic
-	Usage  string
-	EnvVar string
+	Name     string
+	Value    Generic
+	Usage    string
+	EnvVar   string
+	Required bool
 }
 
 func (f GenericFlag) String() string {
@@ -88,6 +90,10 @@ func (f GenericFlag) getName() string {
 	return f.Name
 }
 
+func (f GenericFlag) IsRequired() bool {
+	return f.Required
+}
+
 type StringSlice []string
 
 func (f *StringSlice) Set(value string) error {
@@ -104,10 +110,11 @@ func (f *StringSlice) Value() []string {
 }
 
 type StringSliceFlag struct {
-	Name   string
-	Value  *StringSlice
-	Usage  string
-	EnvVar string
+	Name     string
+	Value    *StringSlice
+	Usage    string
+	EnvVar   string
+	Required bool
 }
 
 func (f StringSliceFlag) String() string {
@@ -136,6 +143,10 @@ func (f StringSliceFlag) getName() string {
 	return f.Name
 }
 
+func (f StringSliceFlag) IsRequired() bool {
+	return f.Required
+}
+
 type IntSlice []int
 
 func (f *IntSlice) Set(value string) error {
@@ -158,10 +169,11 @@ func (f *IntSlice) Value() []int {
 }
 
 type IntSliceFlag struct {
-	Name   string
-	Value  *IntSlice
-	Usage  string
-	EnvVar string
+	Name     string
+	Value    *IntSlice
+	Usage    string
+	EnvVar   string
+	Required bool
 }
 
 func (f IntSliceFlag) String() string {
@@ -193,10 +205,15 @@ func (f IntSliceFlag) getName() string {
 	return f.Name
 }
 
+func (f IntSliceFlag) IsRequired() bool {
+	return f.Required
+}
+
 type BoolFlag struct {
-	Name   string
-	Usage  string
-	EnvVar string
+	Name     string
+	Usage    string
+	EnvVar   string
+	Required bool
 }
 
 func (f BoolFlag) String() string {
@@ -223,10 +240,15 @@ func (f BoolFlag) getName() string {
 	return f.Name
 }
 
+func (f BoolFlag) IsRequired() bool {
+	return f.Required
+}
+
 type BoolTFlag struct {
-	Name   string
-	Usage  string
-	EnvVar string
+	Name     string
+	Usage    string
+	EnvVar   string
+	Required bool
 }
 
 func (f BoolTFlag) String() string {
@@ -253,11 +275,16 @@ func (f BoolTFlag) getName() string {
 	return f.Name
 }
 
+func (f BoolTFlag) IsRequired() bool {
+	return f.Required
+}
+
 type StringFlag struct {
-	Name   string
-	Value  string
-	Usage  string
-	EnvVar string
+	Name     string
+	Value    string
+	Usage    string
+	EnvVar   string
+	Required bool
 }
 
 func (f StringFlag) String() string {
@@ -289,11 +316,16 @@ func (f StringFlag) getName() string {
 	return f.Name
 }
 
+func (f StringFlag) IsRequired() bool {
+	return f.Required
+}
+
 type IntFlag struct {
-	Name   string
-	Value  int
-	Usage  string
-	EnvVar string
+	Name     string
+	Value    int
+	Usage    string
+	EnvVar   string
+	Required bool
 }
 
 func (f IntFlag) String() string {
@@ -319,11 +351,16 @@ func (f IntFlag) getName() string {
 	return f.Name
 }
 
+func (f IntFlag) IsRequired() bool {
+	return f.Required
+}
+
 type DurationFlag struct {
-	Name   string
-	Value  time.Duration
-	Usage  string
-	EnvVar string
+	Name     string
+	Value    time.Duration
+	Usage    string
+	EnvVar   string
+	Required bool
 }
 
 func (f DurationFlag) String() string {
@@ -349,11 +386,16 @@ func (f DurationFlag) getName() string {
 	return f.Name
 }
 
+func (f DurationFlag) IsRequired() bool {
+	return f.Required
+}
+
 type Float64Flag struct {
-	Name   string
-	Value  float64
-	Usage  string
-	EnvVar string
+	Name     string
+	Value    float64
+	Usage    string
+	EnvVar   string
+	Required bool
 }
 
 func (f Float64Flag) String() string {
@@ -379,6 +421,10 @@ func (f Float64Flag) getName() string {
 	return f.Name
 }
 
+func (f Float64Flag) IsRequired() bool {
+	return f.Required
+}
+
 func prefixFor(name string) (prefix string) {
 	if len(name) == 1 {
 		prefix = "-"