Add error messages for stdout writes

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),
         }
     }
 }