Add ascii flag

o2sh · Oct 20, 2020 · 2e48d98 · 2e48d98
1 parent b7d51f3
commit 2e48d98
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/onefetch/cli.rs b/src/onefetch/cli.rs
@@ -10,6 +10,7 @@ use {
 
 pub struct Cli {
     pub path: String,
+    pub ascii_input: Option<String>,
     pub ascii_language: Language,
     pub ascii_colors: Vec<String>,
     pub disabled_fields: info_fields::InfoFieldOn,
@@ -73,6 +74,13 @@ impl Cli {
                         .collect::<Vec<&str>>()
                 ),
         )
+        .arg(
+            Arg::with_name("ascii-input")
+                .long("ascii")
+                .value_name("STRING")
+                .takes_value(true)
+                .help("Takes a STRING as input to replace the ASCII logo")
+        )
         .arg(
             Arg::with_name("ascii-colors")
                 .short("c")
@@ -182,6 +190,9 @@ impl Cli {
         };
 
         let path = String::from(matches.value_of("input").unwrap());
+
+        let ascii_input = matches.value_of("ascii-input").map(String::from);
+
         let ascii_language = if let Some(ascii_language) = matches.value_of("ascii-language") {
             Language::from_str(&ascii_language.to_lowercase()).unwrap()
         } else {
@@ -210,6 +221,7 @@ impl Cli {
 
         Ok(Cli {
             path,
+            ascii_input,
             ascii_language,
             ascii_colors,
             disabled_fields,

diff --git a/src/onefetch/info.rs b/src/onefetch/info.rs
@@ -272,6 +272,11 @@ impl std::fmt::Display for Info {
         } else {
             let mut logo_lines =
                 AsciiArt::new(self.get_ascii(), self.colors(), !self.config.no_bold);
+
+            if let Some(custom_ascii) = &self.config.ascii_input {
+                logo_lines = AsciiArt::new(custom_ascii, Vec::new(), !self.config.no_bold);
+            }
+
             loop {
                 match (logo_lines.next(), info_lines.next()) {
                     (Some(logo_line), Some(info_line)) => {