mirror of
https://github.com/hashicorp/packer.git
synced 2026-02-03 20:39:29 -05:00
19055df3ec
* Updating the license from MPL to Business Source License Going forward, this project will be licensed under the Business Source License v1.1. Please see our blog post for more details at https://hashi.co/bsl-blog, FAQ at https://hashi.co/license-faq, and details of the license at www.hashicorp.com/bsl. * Update copyright file headers to BUSL-1.1 --------- Co-authored-by: hashicorp-copywrite[bot] <110428419+hashicorp-copywrite[bot]@users.noreply.github.com>
31 lines
611 B
Go
31 lines
611 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
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)
|
|
}
|