diff --git a/src/variable.rs b/src/variable.rs index b0a53e5..2473bf4 100644 --- a/src/variable.rs +++ b/src/variable.rs @@ -6,9 +6,14 @@ pub struct Variable { impl From for Variable { fn from(string: String) -> Variable { - let id_value: Vec<&str> = string.split('\n').collect(); + let id_value: Vec<&str> = string.lines().collect(); let id = id_value[0].replace("VAR ", "").to_string(); - let initial_value = id_value[1].to_string(); + + let initial_value = if id_value.len() == 1 { + "".to_string() + } else { + id_value[1..].join("") + }; Variable { id, initial_value } }