only remove leading newlines

DJ-Laser · Dec 25, 2024 · 0b5b411 · 0b5b411
1 parent 2d068c6
commit 0b5b411
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/djlaser_catsay/src/cats.rs b/djlaser_catsay/src/cats.rs
@@ -10,7 +10,11 @@ pub struct Cat<'a> {
 impl<'a> Cat<'a> {
   pub const CATS: &'static [Cat<'static>; 6] = &CATS;
   pub fn get_art(&self) -> &str {
-    return &self.art[1..];
+    if matches!(self.art.chars().next(), Some('\n')) {
+      &self.art[1..]
+    } else {
+      &self.art
+    }
   }
 
   pub fn get_cat(name: &str) -> Option<&'static Cat<'static>> {

diff --git a/djlaser_catsay/src/cli.rs b/djlaser_catsay/src/cli.rs
@@ -49,10 +49,8 @@ where
     _ => CliError::CatFileNotFound(file.to_string(), error),
   })?;
 
-  // get_art removes a leading newline
-  let simple_text = " ".to_string() + &text;
   // TODO: Do parsing of complex catfiles else return simple_text
-  Ok(simple_text)
+  Ok(text)
 }
 
 fn catsay<I, O, E, F>(