integrate clustertool sourcecode

2024-10-16 14:06:31 +02:00
parent bc2a642e7a
commit 6365c6205f
265 changed files with 24073 additions and 45 deletions
@@ -0,0 +1,81 @@
+package main
+
+import (
+	"os"
+	"time"
+
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+
+	"github.com/truecharts/private/clustertool/cmd"
+	"github.com/truecharts/private/clustertool/embed"
+	"github.com/truecharts/private/clustertool/pkg/helper"
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+	ctrllog "sigs.k8s.io/controller-runtime/pkg/log"
+	ctrllogzap "sigs.k8s.io/controller-runtime/pkg/log/zap"
+)
+
+func main() {
+	// Configure zerolog
+	zerolog.DurationFieldUnit = time.Second
+
+	var zerologLevel zerolog.Level
+	var zapLevel zapcore.Level
+
+	// Switch-case for setting the global log level
+	switch os.Getenv("DEBUG") {
+	case "trace":
+		zerologLevel = zerolog.TraceLevel
+		zapLevel = zapcore.DebugLevel // zap does not have a Trace level, use Debug as equivalent
+	case "debug":
+		zerologLevel = zerolog.DebugLevel
+		zapLevel = zapcore.DebugLevel
+	case "warn":
+		zerologLevel = zerolog.WarnLevel
+		zapLevel = zapcore.WarnLevel
+	case "error":
+		zerologLevel = zerolog.ErrorLevel
+		zapLevel = zapcore.ErrorLevel
+	case "fatal":
+		zerologLevel = zerolog.FatalLevel
+		zapLevel = zapcore.FatalLevel
+	case "panic":
+		zerologLevel = zerolog.PanicLevel
+		zapLevel = zapcore.PanicLevel
+	default:
+		zerologLevel = zerolog.InfoLevel
+		zapLevel = zapcore.InfoLevel
+	}
+
+	// Set zerolog level
+	zerolog.SetGlobalLevel(zerologLevel)
+	log.Logger = log.Output(zerolog.ConsoleWriter{
+		Out:        os.Stdout,
+		TimeFormat: time.RFC3339,
+		NoColor:    true,
+	})
+
+	// Configure zap logger
+	zapConfig := zap.NewProductionConfig()
+	zapConfig.Level = zap.NewAtomicLevelAt(zapLevel)
+	zapLogger, err := zapConfig.Build()
+	if err != nil {
+		panic(err)
+	}
+	defer zapLogger.Sync()
+
+	// Set controller-runtime logger to use zap
+	ctrlLogger := ctrllogzap.New(ctrllogzap.UseDevMode(true), ctrllogzap.Level(zapLevel))
+	ctrllog.SetLogger(ctrlLogger)
+
+	embed.AllToCache()
+
+	helper.CheckSystemTime()
+	helper.CheckReqDomains()
+
+	err = cmd.Execute()
+	if err != nil {
+		log.Fatal().Err(err).Msg("Failed to execute command")
+	}
+}