mattermost-community-enterprise/vendor/github.com/splitio/go-split-commons/v7/engine/grammar/startswith.go

package grammar

import (
	"fmt"
	"strings"
)

// StartsWithMatcher matches strings which start with one of the prefixes in the feature flag
type StartsWithMatcher struct {
	Matcher
	prefixes []string
}

// Match returns true if the key provided starts with one of the prefixes in the feature flag.
func (m *StartsWithMatcher) Match(key string, attributes map[string]interface{}, bucketingKey *string) bool {
	matchingKey, err := m.matchingKey(key, attributes)
	if err != nil {
		m.logger.Warning(fmt.Sprintf("StartsWithMatcher: %s", err.Error()))
		return false
	}

	asString, ok := matchingKey.(string)
	if !ok {
		m.logger.Error("StartsWithMatcher: Failed to type-assert key to string")
		return false
	}

	for _, prefix := range m.prefixes {
		if strings.HasPrefix(asString, prefix) {
			return true
		}
	}

	return false
}

// NewStartsWithMatcher returns a new instance of StartsWithMatcher
func NewStartsWithMatcher(negate bool, prefixes []string, attributeName *string) *StartsWithMatcher {
	return &StartsWithMatcher{
		Matcher: Matcher{
			negate:        negate,
			attributeName: attributeName,
		},
		prefixes: prefixes,
	}
}