From b6d3e4ba5a4e714405cca6f66d9b0433abf74654 Mon Sep 17 00:00:00 2001
From: Ryan Geary <rtgnj42@gmail.com>
Date: Sat, 14 Mar 2020 23:02:31 -0400
Subject: Add error messages for stdout writes

---
 src/choice.rs | 10 ++++++++--
 src/main.rs   |  7 +++++--
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/choice.rs b/src/choice.rs
index b7bf879..80e0a8c 100644
--- a/src/choice.rs
+++ b/src/choice.rs
@@ -62,8 +62,14 @@ impl Choice {
     }
 
     fn write_bytes<WriterType: Write>(handle: &mut BufWriter<WriterType>, b: &[u8]) {
-        handle.write(b).unwrap();
-        handle.write(b" ").unwrap();
+        match handle.write(b) {
+            Ok(_) => (),
+            Err(e) => eprintln!("Failed to write to output: {}", e),
+        }
+        match handle.write(b" ") {
+            Ok(_) => (),
+            Err(e) => eprintln!("Failed to write to output: {}", e),
+        }
     }
 
     #[cfg_attr(feature = "flame_it", flame)]
diff --git a/src/main.rs b/src/main.rs
index 51f1b1f..551a572 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -28,9 +28,12 @@ fn main() {
                 for choice in &config.opt.choice {
                     choice.print_choice(&l, &config, &mut handle);
                 }
-                handle.write(b"\n").unwrap();
+                match handle.write(b"\n") {
+                    Ok(_) => (),
+                    Err(e) => eprintln!("Failed to write to output: {}", e)
+                }
             }
-            Err(e) => println!("ERROR: {}", e),
+            Err(e) => println!("Failed to read line: {}", e),
         }
     }
 }
-- 
cgit v1.2.3