summaryrefslogtreecommitdiffstats
path: root/src/print.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/print.rs')
-rw-r--r--src/print.rs12
1 files changed, 10 insertions, 2 deletions
diff --git a/src/print.rs b/src/print.rs
index baf3ebe01..55e05ed82 100644
--- a/src/print.rs
+++ b/src/print.rs
@@ -1,15 +1,23 @@
+use std::io::{self, Write};
+use clap::ArgMatches;
+
use crate::modules;
use crate::modules::Segment;
-use clap::ArgMatches;
pub fn prompt(args: ArgMatches) {
let default_prompt = vec!["directory", "line_break", "character"];
+ let stdout = io::stdout();
+ let mut handle = stdout.lock();
+
+ // Write a new line before the prompt
+ write!(handle, "{}", "\n").unwrap();
+
default_prompt
.into_iter()
.map(|module| modules::handle(module, &args))
.map(stringify_segment)
- .for_each(|segment_string| print!("{}", segment_string));
+ .for_each(|segment_string| write!(handle, "{}", segment_string).unwrap());
}
/// Create a string with the formatted contents of a segment