From 5c96f877ed3b32d1576d3ff04e05bb8e3bf60f15 Mon Sep 17 00:00:00 2001
From: Aditya Pratap Singh <adityapratapsingh51@gmail.com>
Date: Mon, 5 Sep 2022 11:33:38 +0530
Subject: [PATCH] Option to force enable color output (closes #560) (#767)

* Option to force enable color output (closes #560)

(cherry picked from commit 4f951781ee8dd6bb451ac7d159787f47e4b07379)

* Update go.mod
---
 cmd/root.go              | 1 +
 cmd/rootutils.go         | 1 +
 core/cautils/rootinfo.go | 1 +
 3 files changed, 3 insertions(+)

diff --git a/cmd/root.go b/cmd/root.go
index a94c5f5e..c4d0da28 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -63,6 +63,7 @@ func getRootCmd(ks meta.IKubescape) *cobra.Command {
 	rootCmd.PersistentFlags().StringVarP(&rootInfo.Logger, "logger", "l", helpers.InfoLevel.String(), fmt.Sprintf("Logger level. Supported: %s [$KS_LOGGER]", strings.Join(helpers.SupportedLevels(), "/")))
 	rootCmd.PersistentFlags().StringVar(&rootInfo.CacheDir, "cache-dir", getter.DefaultLocalStore, "Cache directory [$KS_CACHE_DIR]")
 	rootCmd.PersistentFlags().BoolVarP(&rootInfo.DisableColor, "disable-color", "", false, "Disable Color output for logging")
+	rootCmd.PersistentFlags().BoolVarP(&rootInfo.EnableColor, "enable-color", "", false, "Force enable Color output for logging")
 
 	cobra.OnInitialize(initLogger, initLoggerLevel, initEnvironment, initCacheDir)
 
diff --git a/cmd/rootutils.go b/cmd/rootutils.go
index cf6e533f..3b0140df 100644
--- a/cmd/rootutils.go
+++ b/cmd/rootutils.go
@@ -16,6 +16,7 @@ const envFlagUsage = "Send report results to specific URL. Format:<ReportReceive
 
 func initLogger() {
 	logger.DisableColor(rootInfo.DisableColor)
+	logger.EnableColor(rootInfo.EnableColor)
 
 	if rootInfo.LoggerName == "" {
 		if l := os.Getenv("KS_LOGGER_NAME"); l != "" {
diff --git a/core/cautils/rootinfo.go b/core/cautils/rootinfo.go
index 5b2c8a69..333f474d 100644
--- a/core/cautils/rootinfo.go
+++ b/core/cautils/rootinfo.go
@@ -5,6 +5,7 @@ type RootInfo struct {
 	LoggerName   string // logger name ("pretty"/"zap"/"none")
 	CacheDir     string // cached dir
 	DisableColor bool   // Disable Color
+	EnableColor  bool   // Force enable Color
 
 	KSCloudBEURLs    string // Kubescape Cloud URL
 	KSCloudBEURLsDep string // Kubescape Cloud URL