handle EOF correctly

Lodestone-Team · Jun 19, 2024 · b981e37 · b981e37
1 parent 84d9752
commit b981e37
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/core/src/command_console/mod.rs b/core/src/command_console/mod.rs
@@ -38,15 +38,19 @@ pub fn init(app_state: AppState) {
             loop {
                 let app_state = app_state.clone();
                 let mut input = String::new();
-                std::io::stdin().read_line(&mut input).unwrap();
+                let buf_size = std::io::stdin().read_line(&mut input).unwrap();
+                if buf_size == 0 {
+                    println!("EOF");
+                    break;
+                }
                 let input_tokens = input.split_whitespace().collect::<Vec<&str>>();
                 match input_tokens.first() {
                     Some(&"di") => {
                         // send input tokens without the first element
                         handle_docker_command(&input_tokens[1..], app_state).await;
                     }
                     _ => {
-                        println!("Unknown command");
+                        println!("Unknown command: {}", input);
                         continue;
                     }
                 }