mirror of https://github.com/hashicorp/consul
38 lines
652 B
Go
38 lines
652 B
Go
|
package flags
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Ensure implements
|
||
|
|
var _ flag.Value = (*FlagMapValue)(nil)
|
||
|
|
|
||
|
|
// FlagMapValue is a flag implementation used to provide key=value semantics
|
||
|
|
// multiple times.
|
||
|
|
type FlagMapValue map[string]string
|
||
|
|
|
||
|
|
func (h *FlagMapValue) String() string {
|
||
|
|
return fmt.Sprintf("%v", *h)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *FlagMapValue) Set(value string) error {
|
||
|
|
idx := strings.Index(value, "=")
|
||
|
|
if idx == -1 {
|
||
|
|
return fmt.Errorf("Missing \"=\" value in argument: %s", value)
|
||
|
|
}
|
||
|
|
|
||
|
|
key, value := value[0:idx], value[idx+1:]
|
||
|
|
|
||
|
|
if *h == nil {
|
||
|
|
*h = make(map[string]string)
|
||
|
|
}
|
||
|
|
|
||
|
|
headers := *h
|
||
|
|
headers[key] = value
|
||
|
|
*h = headers
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|