edit: check edited file and allow continued edit

1 files changed, 9 insertions, 4 deletions

diff --git a/src/utils/fileutil.rs b/src/utils/fileutil.rs
index 25051cb..1db5fb8 100644
--- a/src/utils/fileutil.rs
+++ b/src/utils/fileutil.rs
@@ -46,11 +46,16 @@ pub fn read_lines_from_file(filepath: &Path) -> io::Result<impl Iterator<Item =
   }
 }
 
-pub fn read_single_char(mut source: impl Read) -> Result<char, io::Error> {
-  let mut buffer = [0; 1];
+pub fn read_single_char(mut source: impl BufRead) -> Result<char, String> {
+  let mut buf = String::new();
+  if let Err(error) = source.read_line(&mut buf) {
+    return Err(format!("{}", error));
+  }
 
-  source.read(&mut buffer[..])?;
-  Ok(char::from(buffer[0]))
+  match buf.chars().next() {
+    Some(c) => Ok(c),
+    None => Err("failed to read from stdin".to_string()),
+  }
 }
 
 pub fn read_lines_from_stdin() -> Result<Vec<String>, io::Error> {