nuclei/v2/internal/severity/misc.go

package severity

import (
	"errors"
	"fmt"
	"strings"

	"github.com/projectdiscovery/goflags"
)

type Severities []Severity

func (severities Severities) String() string {
	return strings.Join(severities.ToStringArray(), ", ")
}

func (severities *Severities) Set(value string) error {
	if inputSeverities, err := goflags.ToStringSlice(value); err != nil {
		return err
	} else {
		for _, inputSeverity := range inputSeverities {
			if err := setSeverity(severities, inputSeverity); err != nil {
				return err
			}
		}
		return nil
	}
}

func setSeverity(severities *Severities, value string) error {
	computedSeverity, err := toSeverity(value)
	if err != nil {
		return errors.New(fmt.Sprintf("'%s' is not a valid severity!", value))
	}
	// TODO change the Severities type to map[Severity]interface{}, where the values are struct{}{}, to "simulates" a "set" data structure
	*severities = append(*severities, computedSeverity)
	return nil
}

func (severities *Severities) ToStringArray() []string {
	var result []string
	for _, severity := range *severities {
		result = append(result, severity.String())
	}
	return result
}