bitsy-parser/src/item.rs

102 lines
2.5 KiB
Rust

use crate::{optional_data_line, AnimationFrames, Image};
use crate::image::animation_frames_from_string;
#[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>,
}
impl Item {
#[inline]
fn name_line(&self) -> String {
optional_data_line("NAME", self.name.as_ref())
}
#[inline]
fn dialogue_line(&self) -> String {
optional_data_line("DLG", self.dialogue_id.as_ref())
}
#[inline]
fn colour_line(&self) -> String {
optional_data_line("COL", self.colour_id.as_ref())
}
}
impl From<String> for Item {
#[inline]
fn from(string: String) -> Item {
let mut lines: Vec<&str> = string.lines().collect();
let id = lines[0].replace("ITM ", "");
let mut name = None;
let mut dialogue_id = None;
let mut colour_id: Option<u64> = None;
loop {
let last_line = lines.pop().unwrap();
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 = animation_frames_from_string(
lines[1..].join("\n")
);
Item {
id,
name,
animation_frames,
dialogue_id,
colour_id,
}
}
}
impl ToString for Item {
#[inline]
fn to_string(&self) -> String {
format!(
"ITM {}\n{}{}{}{}",
self.id,
self.animation_frames.to_string(),
self.name_line(),
self.dialogue_line(),
self.colour_line(),
)
}
}
#[cfg(test)]
mod test {
use crate::item::Item;
use crate::mock;
#[test]
fn test_item_from_string() {
let output = Item::from(include_str!("test-resources/item").to_string());
let expected = mock::item();
assert_eq!(output, expected);
}
#[test]
fn test_item_to_string() {
let output = mock::item().to_string();
let expected = include_str!("test-resources/item").to_string();
assert_eq!(output, expected);
}
}