1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
use crate::optional_data_line; #[derive(Debug, Eq, PartialEq)] pub struct Dialogue { pub id: String, pub contents: String, pub name: Option<String>, } impl From<String> for Dialogue { #[inline] fn from(string: String) -> Dialogue { let mut lines: Vec<&str> = string.lines().collect(); let id = lines[0].replace("DLG ", "").to_string(); let name = if lines.last().unwrap().starts_with("NAME ") { Some(lines.pop().unwrap().replace("NAME ", "")) } else { None }; let contents = lines[1..].join("\n"); Dialogue { id, contents, name } } } impl ToString for Dialogue { #[inline] fn to_string(&self) -> String { format!( "DLG {}\n{}{}", self.id, self.contents, optional_data_line("NAME", self.name.as_ref()) ) } } #[cfg(test)] mod test { use crate::dialogue::Dialogue; #[test] fn test_dialogue_from_string() { let output = Dialogue::from( "DLG h\nhello\nNAME not a dialogue name\nNAME a dialogue name".to_string() ); let expected = Dialogue { id: "h".to_string(), contents: "hello\nNAME not a dialogue name".to_string(), name: Some("a dialogue name".to_string()) }; assert_eq!(output, expected); } #[test] fn test_dialogue_to_string() { let output = Dialogue { id: "y".to_string(), contents: "This is a bit of dialogue,\nblah blah\nblah blah".to_string(), name: Some("a dialogue name".to_string()) }.to_string(); let expected = "DLG y\nThis is a bit of dialogue,\nblah blah\nblah blah\nNAME a dialogue name".to_string(); assert_eq!(output, expected); } }