From cc3db831ae857909849b9150bc02ea0c8517114b Mon Sep 17 00:00:00 2001
From: tison <wander4096@gmail.com>
Date: Wed, 14 Aug 2024 17:20:27 +0800
Subject: [PATCH] feat: support turn off color of TextLayout in code (#48)

Signed-off-by: tison <wander4096@gmail.com>
---
 examples/simple_stdio.rs |  6 +++
 src/layout/text.rs       | 80 +++++++++++++++++++++++++++++++++++-----
 2 files changed, 76 insertions(+), 10 deletions(-)

diff --git a/examples/simple_stdio.rs b/examples/simple_stdio.rs
index 1589851..e29136b 100644
--- a/examples/simple_stdio.rs
+++ b/examples/simple_stdio.rs
@@ -26,6 +26,12 @@ fn main() {
                 .layout(TextLayout::default())
                 .append(append::Stdout),
         )
+        .dispatch(
+            Dispatch::new()
+                .filter(LevelFilter::Trace)
+                .layout(TextLayout::default().no_color())
+                .append(append::Stderr),
+        )
         .apply()
         .unwrap();
 
diff --git a/src/layout/text.rs b/src/layout/text.rs
index e06d683..9b5f1da 100644
--- a/src/layout/text.rs
+++ b/src/layout/text.rs
@@ -37,7 +37,7 @@ use crate::layout::Layout;
 /// ```
 ///
 /// By default, log levels are colored. You can turn on the `no-color` feature flag to disable this
-/// feature.
+/// feature. Instead, you can also set the `no_color` field to `true` to disable coloring.
 ///
 /// You can also customize the color of each log level by setting the `colors` field with a
 /// [`LevelColor`] instance.
@@ -47,9 +47,66 @@ use crate::layout::Layout;
 #[derive(Default, Debug, Clone)]
 pub struct TextLayout {
     pub colors: LevelColor,
+    pub no_color: bool,
     pub tz: Option<TimeZone>,
 }
 
+impl TextLayout {
+    /// Turn off coloring of log levels.
+    pub fn no_color(mut self) -> Self {
+        self.no_color = true;
+        self
+    }
+
+    /// Set the timezone of the timestamp; default to the system timezone.
+    pub fn timezone(mut self, tz: TimeZone) -> Self {
+        self.tz = Some(tz);
+        self
+    }
+
+    /// Customize the color of each log level; no effect if `no_color` is set to `true`,
+    /// or the `no-color` feature flag is enabled.
+    pub fn colors(mut self, colors: LevelColor) -> Self {
+        self.colors = colors;
+        self
+    }
+
+    /// Customize the color of the error log level; no effect if `no_color` is set to `true`,
+    /// or the `no-color` feature flag is enabled. Default to red.
+    pub fn error_color(mut self, color: Color) -> Self {
+        self.colors.error = color;
+        self
+    }
+
+    /// Customize the color of the warn log level; no effect if `no_color` is set to `true`,
+    /// or the `no-color` feature flag is enabled. Default to yellow.
+    pub fn warn_color(mut self, color: Color) -> Self {
+        self.colors.warn = color;
+        self
+    }
+
+    /// Customize the color of the info log level; no effect if `no_color` is set to `true`,
+    /// or the `no-color` feature flag is enabled. Default to green.
+    pub fn info_color(mut self, color: Color) -> Self {
+        self.colors.info = color;
+        self
+    }
+
+    /// Customize the color of the debug log level; no effect if `no_color` is set to `true`,
+    /// or the `no-color` feature flag is enabled. Default to blue.
+    pub fn debug_color(mut self, color: Color) -> Self {
+        self.colors.debug = color;
+        self
+    }
+
+    /// Customize the color of the trace log level; no effect if `no_color` is set to `true`,
+    /// or the `no-color` feature flag is enabled. Default to magenta.
+    pub fn trace_color(mut self, color: Color) -> Self {
+        self.colors.trace = color;
+        self
+    }
+}
+
 /// Customize the color of each log level.
 #[derive(Debug, Clone)]
 pub struct LevelColor {
@@ -77,20 +134,23 @@ impl TextLayout {
     where
         F: Fn(Arguments) -> anyhow::Result<()>,
     {
-        let color = match record.level() {
-            Level::Error => self.colors.error,
-            Level::Warn => self.colors.warn,
-            Level::Info => self.colors.info,
-            Level::Debug => self.colors.debug,
-            Level::Trace => self.colors.trace,
-        };
-
         let time = match self.tz.clone() {
             Some(tz) => Zoned::now().with_time_zone(tz),
             None => Zoned::now(),
         }
         .strftime("%Y-%m-%dT%H:%M:%S.%6f%:z");
-        let level = ColoredString::from(record.level().to_string()).color(color);
+        let level = if self.no_color {
+            ColoredString::from(record.level().to_string())
+        } else {
+            let color = match record.level() {
+                Level::Error => self.colors.error,
+                Level::Warn => self.colors.warn,
+                Level::Info => self.colors.info,
+                Level::Debug => self.colors.debug,
+                Level::Trace => self.colors.trace,
+            };
+            ColoredString::from(record.level().to_string()).color(color)
+        };
         let module = record.module_path().unwrap_or_default();
         let file = record.file().unwrap_or_default();
         let line = record.line().unwrap_or_default();