mirror of
https://github.com/kubescape/kubescape.git
synced 2026-02-14 18:09:55 +00:00
90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/armosec/kubescape/core/cautils/getter"
|
|
"github.com/armosec/kubescape/core/cautils/logger"
|
|
"github.com/armosec/kubescape/core/cautils/logger/helpers"
|
|
|
|
"github.com/mattn/go-isatty"
|
|
)
|
|
|
|
const envFlagUsage = "Send report results to specific URL. Format:<ReportReceiver>,<Backend>,<Frontend>.\n\t\tExample:report.armo.cloud,api.armo.cloud,portal.armo.cloud"
|
|
|
|
func initLogger() {
|
|
logger.DisableColor(rootInfo.DisableColor)
|
|
|
|
if rootInfo.LoggerName == "" {
|
|
if l := os.Getenv("KS_LOGGER_NAME"); l != "" {
|
|
rootInfo.LoggerName = l
|
|
} else {
|
|
if isatty.IsTerminal(os.Stdout.Fd()) {
|
|
rootInfo.LoggerName = "pretty"
|
|
} else {
|
|
rootInfo.LoggerName = "zap"
|
|
}
|
|
}
|
|
}
|
|
|
|
logger.InitLogger(rootInfo.LoggerName)
|
|
|
|
}
|
|
func initLoggerLevel() {
|
|
if rootInfo.Logger == helpers.InfoLevel.String() {
|
|
} else if l := os.Getenv("KS_LOGGER"); l != "" {
|
|
rootInfo.Logger = l
|
|
}
|
|
|
|
if err := logger.L().SetLevel(rootInfo.Logger); err != nil {
|
|
logger.L().Fatal(fmt.Sprintf("supported levels: %s", strings.Join(helpers.SupportedLevels(), "/")), helpers.Error(err))
|
|
}
|
|
}
|
|
|
|
func initCacheDir() {
|
|
if rootInfo.CacheDir == getter.DefaultLocalStore {
|
|
getter.DefaultLocalStore = rootInfo.CacheDir
|
|
} else if cacheDir := os.Getenv("KS_CACHE_DIR"); cacheDir != "" {
|
|
getter.DefaultLocalStore = cacheDir
|
|
} else {
|
|
return // using default cache dir location
|
|
}
|
|
|
|
logger.L().Debug("cache dir updated", helpers.String("path", getter.DefaultLocalStore))
|
|
}
|
|
func initEnvironment() {
|
|
if rootInfo.ArmoBEURLs == "" {
|
|
rootInfo.ArmoBEURLs = rootInfo.ArmoBEURLsDep
|
|
}
|
|
urlSlices := strings.Split(rootInfo.ArmoBEURLs, ",")
|
|
if len(urlSlices) != 1 && len(urlSlices) < 3 {
|
|
logger.L().Fatal("expected at least 3 URLs (report, api, frontend, auth)")
|
|
}
|
|
switch len(urlSlices) {
|
|
case 1:
|
|
switch urlSlices[0] {
|
|
case "dev", "development":
|
|
getter.SetARMOAPIConnector(getter.NewARMOAPIDev())
|
|
case "stage", "staging":
|
|
getter.SetARMOAPIConnector(getter.NewARMOAPIStaging())
|
|
case "":
|
|
getter.SetARMOAPIConnector(getter.NewARMOAPIProd())
|
|
default:
|
|
logger.L().Fatal("--environment flag usage: " + envFlagUsage)
|
|
}
|
|
case 2:
|
|
logger.L().Fatal("--environment flag usage: " + envFlagUsage)
|
|
case 3, 4:
|
|
var armoAUTHURL string
|
|
armoERURL := urlSlices[0] // mandatory
|
|
armoBEURL := urlSlices[1] // mandatory
|
|
armoFEURL := urlSlices[2] // mandatory
|
|
if len(urlSlices) <= 4 {
|
|
armoAUTHURL = urlSlices[3]
|
|
}
|
|
getter.SetARMOAPIConnector(getter.NewARMOAPICustomized(armoERURL, armoBEURL, armoFEURL, armoAUTHURL))
|
|
}
|
|
}
|