diff --git a/src/main.rs b/src/main.rs index 9e7b6b1..bcf870b 100644 --- a/src/main.rs +++ b/src/main.rs @@ -65,7 +65,12 @@ fn main() { stdin().read_line(&mut input_text).unwrap(); let action: Vec<_> = input_text.trim().split_ascii_whitespace().collect(); if input_text.starts_with("input ") { - interpreter.input.extend(&input_text.as_bytes()[6..]); + interpreter.input.extend( + &input_text + .strip_suffix('\n') + .unwrap_or(&input_text) + .as_bytes()[6..], + ); continue; } match action.as_slice() {