--presets print names only if stdout is not a terminal

mazznoer · Nov 29, 2024 · 7c4460d · 7c4460d
1 parent 7e68d44
commit 7c4460d
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -4,7 +4,7 @@ use clap::{CommandFactory, Parser, ValueEnum};
 use lolcrab::{Gradient, Lolcrab, Opt};
 use std::{
     fs::File,
-    io::{self, BufReader, Write},
+    io::{self, BufReader, IsTerminal, Write},
     path::PathBuf,
 };
 
@@ -51,10 +51,14 @@ fn main() -> Result<(), io::Error> {
         for g in Gradient::value_variants() {
             let name = format!("{g:?}").to_lowercase();
             let name = if name == "rdylgn" { "rd-yl-gn" } else { &name };
-            writeln!(stdout, "\n{name}\n")?;
-            lol.gradient = g.to_gradient();
-            lol.randomize_position();
-            lol.colorize_str(SAMPLE_TEXT, &mut stdout)?;
+            if stdout.is_terminal() {
+                writeln!(stdout, "\n{name}\n")?;
+                lol.gradient = g.to_gradient();
+                lol.randomize_position();
+                lol.colorize_str(SAMPLE_TEXT, &mut stdout)?;
+            } else {
+                writeln!(stdout, "{name}")?;
+            }
         }
         return Ok(());
     }