Do not crash on empty terminal

quentinlesceller · May 3, 2019 · b76c18d · b76c18d
1 parent e673fd9
commit b76c18d
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/controller/src/display.rs b/controller/src/display.rs
@@ -35,6 +35,10 @@ pub fn outputs(
 		account, cur_height
 	);
 	println!();
+	if term::stdout().is_none() {
+		println!("Could not open terminal");
+		return Ok(());
+	}
 	let mut t = term::stdout().unwrap();
 	t.fg(term::color::MAGENTA).unwrap();
 	writeln!(t, "{}", title).unwrap();
@@ -132,6 +136,10 @@ pub fn txs(
 		account, cur_height
 	);
 	println!();
+	if term::stdout().is_none() {
+		println!("Could not open terminal");
+		return Ok(());
+	}
 	let mut t = term::stdout().unwrap();
 	t.fg(term::color::MAGENTA).unwrap();
 	writeln!(t, "{}", title).unwrap();
@@ -409,6 +417,10 @@ pub fn accounts(acct_mappings: Vec<AcctPathMapping>) {
 pub fn tx_messages(tx: &TxLogEntry, dark_background_color_scheme: bool) -> Result<(), Error> {
 	let title = format!("Transaction Messages - Transaction '{}'", tx.id,);
 	println!();
+	if term::stdout().is_none() {
+		println!("Could not open terminal");
+		return Ok(());
+	}
 	let mut t = term::stdout().unwrap();
 	t.fg(term::color::MAGENTA).unwrap();
 	writeln!(t, "{}", title).unwrap();