terraform/internal/command/arguments/flags.go

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

97 lines
2 KiB
Go
Raw Permalink Normal View History

// Copyright IBM Corp. 2014, 2026
// SPDX-License-Identifier: BUSL-1.1
2021-02-18 17:23:34 -05:00
package arguments
import (
"flag"
"fmt"
)
// FlagStringSlice is a flag.Value implementation which allows collecting
2021-02-18 17:23:34 -05:00
// multiple instances of a single flag into a slice. This is used for flags
// such as -target=aws_instance.foo and -var x=y.
type FlagStringSlice []string
2021-02-18 17:23:34 -05:00
var _ flag.Value = (*FlagStringSlice)(nil)
2021-02-18 17:23:34 -05:00
func (v *FlagStringSlice) String() string {
2021-02-18 17:23:34 -05:00
return ""
}
func (v *FlagStringSlice) Set(raw string) error {
2021-02-18 17:23:34 -05:00
*v = append(*v, raw)
return nil
}
// FlagNameValueSlice is a flag.Value implementation that appends raw flag
2021-02-18 17:23:34 -05:00
// names and values to a slice. This is used to collect a sequence of flags
// with possibly different names, preserving the overall order.
type FlagNameValueSlice struct {
FlagName string
Items *[]FlagNameValue
2021-02-18 17:23:34 -05:00
}
var _ flag.Value = FlagNameValueSlice{}
2021-02-18 17:23:34 -05:00
func NewFlagNameValueSlice(flagName string) FlagNameValueSlice {
2021-02-18 17:23:34 -05:00
var items []FlagNameValue
return FlagNameValueSlice{
FlagName: flagName,
Items: &items,
2021-02-18 17:23:34 -05:00
}
}
func (f FlagNameValueSlice) Empty() bool {
if f.Items == nil {
2021-02-18 17:23:34 -05:00
return true
}
return len(*f.Items) == 0
2021-02-18 17:23:34 -05:00
}
func (f FlagNameValueSlice) AllItems() []FlagNameValue {
if f.Items == nil {
2021-02-18 17:23:34 -05:00
return nil
}
return *f.Items
2021-02-18 17:23:34 -05:00
}
func (f FlagNameValueSlice) Alias(flagName string) FlagNameValueSlice {
return FlagNameValueSlice{
FlagName: flagName,
Items: f.Items,
2021-02-18 17:23:34 -05:00
}
}
func (f FlagNameValueSlice) String() string {
2021-02-18 17:23:34 -05:00
return ""
}
func (f FlagNameValueSlice) Set(str string) error {
*f.Items = append(*f.Items, FlagNameValue{
Name: f.FlagName,
2021-02-18 17:23:34 -05:00
Value: str,
})
return nil
}
type FlagNameValue struct {
Name string
Value string
}
func (f FlagNameValue) String() string {
return fmt.Sprintf("%s=%q", f.Name, f.Value)
}
// FlagIsSet returns whether a flag is explicitly set in a set of flags
func FlagIsSet(flags *flag.FlagSet, name string) bool {
isSet := false
flags.Visit(func(f *flag.Flag) {
if f.Name == name {
isSet = true
}
})
return isSet
}