107 lines
3.0 KiB
Rust
107 lines
3.0 KiB
Rust
use crate::{optional_data_line, AnimationFrames, Image};
|
|
use crate::image::animation_frames_from_str;
|
|
use std::fmt;
|
|
|
|
#[derive(Clone, Debug, Eq, PartialEq)]
|
|
pub struct Item {
|
|
pub id: String,
|
|
pub animation_frames: Vec<Image>,
|
|
pub name: Option<String>,
|
|
pub dialogue_id: Option<String>,
|
|
pub colour_id: Option<u64>,
|
|
pub blip: Option<String>,
|
|
}
|
|
|
|
impl Item {
|
|
fn name_line(&self) -> String {
|
|
optional_data_line("NAME", self.name.as_ref())
|
|
}
|
|
|
|
fn dialogue_line(&self) -> String {
|
|
optional_data_line("DLG", self.dialogue_id.as_ref())
|
|
}
|
|
|
|
fn colour_line(&self) -> String {
|
|
optional_data_line("COL", self.colour_id.as_ref())
|
|
}
|
|
|
|
fn blip_line(&self) -> String {
|
|
optional_data_line("BLIP", self.blip.as_ref())
|
|
}
|
|
|
|
pub fn from_str(str: &str) -> Result<(Item, Vec<crate::Error>), crate::Error> {
|
|
let mut lines: Vec<&str> = str.lines().collect();
|
|
|
|
if lines.is_empty() || !lines[0].starts_with("ITM ") {
|
|
return Err(crate::Error::Item);
|
|
}
|
|
|
|
let id = lines[0].replace("ITM ", "");
|
|
let mut name = None;
|
|
let mut dialogue_id = None;
|
|
let mut colour_id: Option<u64> = None;
|
|
let mut blip = None;
|
|
|
|
let mut warnings = Vec::new();
|
|
|
|
loop {
|
|
let last_line = lines.pop().unwrap();
|
|
|
|
if last_line.starts_with("BLIP") {
|
|
blip = Some(last_line.replace("BLIP ", "").to_string());
|
|
} else if last_line.starts_with("NAME") {
|
|
name = Some(last_line.replace("NAME ", "").to_string());
|
|
} else if last_line.starts_with("DLG") {
|
|
dialogue_id = Some(last_line.replace("DLG ", "").to_string());
|
|
} else if last_line.starts_with("COL") {
|
|
colour_id = Some(last_line.replace("COL ", "").parse().unwrap());
|
|
} else {
|
|
lines.push(last_line);
|
|
break;
|
|
}
|
|
}
|
|
|
|
let (animation_frames, mut animation_warnings) = animation_frames_from_str(
|
|
&lines[1..].join("\n")
|
|
).unwrap();
|
|
|
|
warnings.append(&mut animation_warnings);
|
|
|
|
Ok((Item { id, name, animation_frames, dialogue_id, colour_id, blip }, warnings))
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for Item {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
write!(
|
|
f,
|
|
"ITM {}\n{}{}{}{}{}",
|
|
self.id,
|
|
self.animation_frames.to_string(),
|
|
self.name_line(),
|
|
self.dialogue_line(),
|
|
self.colour_line(),
|
|
self.blip_line(),
|
|
)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use crate::{Item, mock};
|
|
|
|
#[test]
|
|
fn item_from_string() {
|
|
let (output, _err) = Item::from_str(include_str!("test-resources/item")).unwrap();
|
|
let expected = mock::item();
|
|
assert_eq!(output, expected);
|
|
}
|
|
|
|
#[test]
|
|
fn item_to_string() {
|
|
let output = mock::item().to_string();
|
|
let expected = include_str!("test-resources/item").to_string();
|
|
assert_eq!(output, expected);
|
|
}
|
|
}
|