mirror of
https://github.com/hashicorp/packer.git
synced 2026-02-24 18:30:41 -05:00
b7df3ca36f
Co-authored-by: hashicorp-copywrite[bot] <110428419+hashicorp-copywrite[bot]@users.noreply.github.com>
31 lines
610 B
Go
31 lines
610 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package enumflag
|
|
|
|
import "fmt"
|
|
|
|
type enumFlag struct {
|
|
target *string
|
|
options []string
|
|
}
|
|
|
|
// New returns a flag.Value implementation for parsing flags with a one-of-a-set value
|
|
func New(target *string, options ...string) *enumFlag {
|
|
return &enumFlag{target: target, options: options}
|
|
}
|
|
|
|
func (f *enumFlag) String() string {
|
|
return *f.target
|
|
}
|
|
|
|
func (f *enumFlag) Set(value string) error {
|
|
for _, v := range f.options {
|
|
if v == value {
|
|
*f.target = value
|
|
return nil
|
|
}
|
|
}
|
|
|
|
return fmt.Errorf("expected one of %q", f.options)
|
|
}
|