feat: enable to disable log color

reearth · Feb 22, 2024 · 402668d · 402668d
1 parent 13a660b
commit 402668d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/log/gcp.go b/log/gcp.go
@@ -11,6 +11,7 @@ var GCPEnv = []string{
 	"CLOUD_RUN_JOB",
 	"K_SERVICE",
 	"GOOGLE_CLOUD_PROJECT",
+	"GCP",
 }
 
 type severity string

diff --git a/log/logger.go b/log/logger.go
@@ -65,7 +65,11 @@ func encoder() zapcore.Encoder {
 	if isGCP() {
 		return zapcore.NewJSONEncoder(gceEncoderConfig)
 	} else {
-		return zapcore.NewConsoleEncoder(consoleEncoderConfig)
+		conf := consoleEncoderConfig
+		if isColorDisabled() {
+			conf.EncodeLevel = zapcore.CapitalLevelEncoder
+		}
+		return zapcore.NewConsoleEncoder(conf)
 	}
 }
 
@@ -270,3 +274,7 @@ func (l *Logger) msg(msg string) string {
 	}
 	return p + msg + s
 }
+
+func isColorDisabled() bool {
+	return os.Getenv("NO_COLOR") != ""
+}