forked from ebhomengo/niki
41 lines
845 B
Go
41 lines
845 B
Go
|
package opts
|
||
|
|
|
||
|
|
// QuotedString is a string that may have extra quotes around the value. The
|
||
|
|
// quotes are stripped from the value.
|
||
|
|
type QuotedString struct {
|
||
|
|
value *string
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set sets a new value
|
||
|
|
func (s *QuotedString) Set(val string) error {
|
||
|
|
*s.value = trimQuotes(val)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Type returns the type of the value
|
||
|
|
func (s *QuotedString) Type() string {
|
||
|
|
return "string"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *QuotedString) String() string {
|
||
|
|
return *s.value
|
||
|
|
}
|
||
|
|
|
||
|
|
func trimQuotes(value string) string {
|
||
|
|
if len(value) < 2 {
|
||
|
|
return value
|
||
|
|
}
|
||
|
|
lastIndex := len(value) - 1
|
||
|
|
for _, char := range []byte{'\'', '"'} {
|
||
|
|
if value[0] == char && value[lastIndex] == char {
|
||
|
|
return value[1:lastIndex]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return value
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewQuotedString returns a new quoted string option
|
||
|
|
func NewQuotedString(value *string) *QuotedString {
|
||
|
|
return &QuotedString{value: value}
|
||
|
|
}
|