string_slice.go

package goflags

import (
	sliceutil "github.com/projectdiscovery/utils/slice"
)

var (
	optionMap           map[*StringSlice]Options
	optionDefaultValues map[*StringSlice][]string
)

func init() {
	optionMap = make(map[*StringSlice]Options)
	optionDefaultValues = make(map[*StringSlice][]string)
}

// StringSlice is a slice of strings
type StringSlice []string

// Set appends a value to the string slice.
func (stringSlice *StringSlice) Set(value string) error {
	option, ok := optionMap[stringSlice]
	if !ok {
		option = StringSliceOptions
	}
	values, err := ToStringSlice(value, option)
	if err != nil {
		return err
	}
	// if new values are provided, we remove default ones
	if defaultValue, ok := optionDefaultValues[stringSlice]; ok {
		if sliceutil.Equal(*stringSlice, defaultValue) {
			*stringSlice = []string{}
		}
	}

	*stringSlice = append(*stringSlice, values...)
	return nil
}

func (stringSlice StringSlice) String() string {
	return ToString(stringSlice)
}