aboutsummaryrefslogtreecommitdiff
path: root/cmd/podman/validate/choice.go
blob: 8bb21c5916c04f0e9e83f4e021935e73b0ffd972 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
package validate

import (
	"fmt"
	"strings"
)

// Honors cobra.Value interface
type ChoiceValue struct {
	value   *string
	choices []string
}

// Value may be used in cobra FlagSet methods Var/VarP/VarPF() to select from a set of values
//
// Example:
// 	created := validate.ChoiceValue(&opts.Sort, "command", "created", "id", "image", "names", "runningfor", "size", "status")
//	flags.Var(created, "sort", "Sort output by: "+created.Choices())
func Value(p *string, choices ...string) *ChoiceValue {
	return &ChoiceValue{
		value:   p,
		choices: choices,
	}
}

func (c *ChoiceValue) String() string {
	return *c.value
}

func (c *ChoiceValue) Set(value string) error {
	for _, v := range c.choices {
		if v == value {
			*c.value = value
			return nil
		}
	}
	return fmt.Errorf("%q is not a valid value.  Choose from: %q", value, c.Choices())
}

func (c *ChoiceValue) Choices() string {
	return strings.Join(c.choices, ", ")
}

func (c *ChoiceValue) Type() string {
	return "choice"
}