kyverno/cmd/cli/kubectl-kyverno/main.go

package main

import (
	"flag"
	"fmt"
	"os"

	"github.com/kyverno/kyverno/cmd/cli/kubectl-kyverno/commands"
	"github.com/kyverno/kyverno/cmd/cli/kubectl-kyverno/commands/apply"
	"github.com/kyverno/kyverno/cmd/cli/kubectl-kyverno/experimental"
	"github.com/kyverno/kyverno/cmd/cli/kubectl-kyverno/log"
	"github.com/spf13/cobra"
)

func main() {
	cmd, err := setup()
	if err != nil {
		fmt.Println("Error:", err)
		os.Exit(1)
	}
	if err := cmd.Execute(); err != nil {
		switch e := err.(type) {
		case apply.WarnExitCodeError:
			os.Exit(e.ExitCode)
		default:
			os.Exit(1)
		}
	}
}

func setup() (*cobra.Command, error) {
	cmd := commands.RootCommand(experimental.IsEnabled())
	if err := configureLogs(cmd); err != nil {
		return nil, fmt.Errorf("Failed to setup logging (%w)", err)
	}
	return cmd, nil
}

func configureLogs(cli *cobra.Command) error {
	if err := log.Configure(); err != nil {
		return err
	}
	cli.PersistentFlags().AddGoFlagSet(flag.CommandLine)
	return nil
}