9ae8bd79c5
* remove unused method * whitespace, redundant comments * use test helpers * move DumpConsoleConfig() from pkg/csconfig to cscli * package doc header * var -> const * rename ./tests -> ./testdata * shorter tests with more error checks * lint/formatting * use helpers; fix tests that didn't actually test * lint; rename expectedResult -> expected
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
package csconfig
|
|
|
|
import (
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/crowdsecurity/crowdsec/pkg/fflag"
|
|
)
|
|
|
|
// LoadFeatureFlagsEnv parses the environment variables to enable feature flags.
|
|
func LoadFeatureFlagsEnv(logger *log.Logger) error {
|
|
if err := fflag.Crowdsec.SetFromEnv(logger); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// LoadFeatureFlags parses feature.yaml to enable feature flags.
|
|
// The file is in the same directory as config.yaml, which is provided
|
|
// as the fist parameter. This can be different than ConfigPaths.ConfigDir
|
|
func LoadFeatureFlagsFile(configPath string, logger *log.Logger) error {
|
|
dir := filepath.Dir(configPath)
|
|
featurePath := filepath.Join(dir, "feature.yaml")
|
|
|
|
if err := fflag.Crowdsec.SetFromYamlFile(featurePath, logger); err != nil {
|
|
return fmt.Errorf("file %s: %s", featurePath, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ListFeatureFlags returns a list of the enabled feature flags.
|
|
func ListFeatureFlags() string {
|
|
enabledFeatures := fflag.Crowdsec.GetEnabledFeatures()
|
|
|
|
msg := "<none>"
|
|
if len(enabledFeatures) > 0 {
|
|
msg = strings.Join(enabledFeatures, ", ")
|
|
}
|
|
|
|
return msg
|
|
}
|