mirror of
https://github.com/kyverno/policy-reporter.git
synced 2024-12-14 11:57:32 +00:00
77 lines
1.6 KiB
Go
77 lines
1.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"log"
|
|
|
|
"github.com/kyverno/policy-reporter/pkg/config"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// NewCLI creates a new instance of the root CLI
|
|
func NewCLI() *cobra.Command {
|
|
rootCmd := &cobra.Command{
|
|
Use: "policyreporter",
|
|
Short: "Generates PolicyReport Metrics and Send Results to different targets",
|
|
Long: `Generates Prometheus Metrics from PolicyReports, ClusterPolicyReports and PolicyReportResults.
|
|
Sends notifications to different targets like Grafana's Loki.`,
|
|
}
|
|
|
|
rootCmd.AddCommand(newRunCMD())
|
|
|
|
return rootCmd
|
|
}
|
|
|
|
func loadConfig(cmd *cobra.Command) (*config.Config, error) {
|
|
v := viper.New()
|
|
|
|
cfgFile := ""
|
|
|
|
configFlag := cmd.Flags().Lookup("config")
|
|
if configFlag != nil {
|
|
cfgFile = configFlag.Value.String()
|
|
}
|
|
|
|
if cfgFile != "" {
|
|
v.SetConfigFile(cfgFile)
|
|
} else {
|
|
v.AddConfigPath(".")
|
|
v.SetConfigName("config")
|
|
}
|
|
|
|
v.AutomaticEnv()
|
|
|
|
if err := v.ReadInConfig(); err != nil {
|
|
log.Println("[INFO] No configuration file found")
|
|
}
|
|
|
|
if flag := cmd.Flags().Lookup("kubeconfig"); flag != nil {
|
|
v.BindPFlag("kubeconfig", flag)
|
|
}
|
|
|
|
if flag := cmd.Flags().Lookup("port"); flag != nil {
|
|
v.BindPFlag("api.port", flag)
|
|
}
|
|
|
|
if flag := cmd.Flags().Lookup("rest-enabled"); flag != nil {
|
|
v.BindPFlag("rest.enabled", flag)
|
|
}
|
|
|
|
if flag := cmd.Flags().Lookup("metrics-enabled"); flag != nil {
|
|
v.BindPFlag("metrics.enabled", flag)
|
|
}
|
|
|
|
if flag := cmd.Flags().Lookup("dbfile"); flag != nil {
|
|
v.BindPFlag("dbfile", flag)
|
|
}
|
|
|
|
c := &config.Config{}
|
|
|
|
err := v.Unmarshal(c)
|
|
|
|
if c.DBFile == "" {
|
|
c.DBFile = "sqlite-database.db"
|
|
}
|
|
|
|
return c, err
|
|
}
|