93 lines
4.4 KiB
Rust
93 lines
4.4 KiB
Rust
use crate::{
|
|
data::Card,
|
|
parser::{Field, Operator, RawCardFilter, Value, OPERATOR_CHARS},
|
|
};
|
|
|
|
/// A struct derived from `Card` that has all fields lowercased for easier search
|
|
#[derive(Debug, PartialEq, Eq, Clone)]
|
|
pub struct SearchCard {
|
|
pub id: usize,
|
|
card_type: String,
|
|
name: String,
|
|
text: String,
|
|
atk: Option<i32>,
|
|
def: Option<i32>,
|
|
attribute: Option<String>,
|
|
r#type: String,
|
|
// also includes rank
|
|
level: Option<i32>,
|
|
link_rating: Option<i32>,
|
|
link_arrows: Option<Vec<String>>,
|
|
}
|
|
|
|
impl From<&Card> for SearchCard {
|
|
fn from(card: &Card) -> Self {
|
|
Self {
|
|
id: card.id,
|
|
card_type: card.card_type.to_lowercase(),
|
|
name: card.name.to_lowercase(),
|
|
text: card.text.to_lowercase(),
|
|
atk: card.atk,
|
|
def: card.def,
|
|
attribute: card.attribute.as_ref().map(|s| s.to_lowercase()),
|
|
r#type: card.r#type.to_lowercase(),
|
|
level: card.level,
|
|
link_rating: card.link_rating,
|
|
link_arrows: card.link_arrows.as_ref().map(|arrows| arrows.iter().map(|a| a.to_lowercase()).collect()),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub type CardFilter = Box<dyn Fn(&SearchCard) -> bool>;
|
|
|
|
pub fn fallback_filter(query: &str) -> Result<RawCardFilter, String> {
|
|
if query.contains(OPERATOR_CHARS) {
|
|
return Err(format!("Invalid query: {query}"));
|
|
}
|
|
let q = query.to_lowercase();
|
|
Ok(RawCardFilter(Field::Name, Operator::Equal, Value::String(q)))
|
|
}
|
|
|
|
pub fn build_filter(query: RawCardFilter) -> Result<CardFilter, String> {
|
|
Ok(match query {
|
|
RawCardFilter(Field::Atk, op, Value::Numerical(n)) => Box::new(move |card| op.filter_number(card.atk, n)),
|
|
RawCardFilter(Field::Def, op, Value::Numerical(n)) => Box::new(move |card| op.filter_number(card.def, n)),
|
|
// ? ATK/DEF is modeled as None in the source json. At least for some monsters.
|
|
// Let’s at least find those.
|
|
RawCardFilter(Field::Atk, _, Value::String(s)) if s == "?" => {
|
|
Box::new(move |card| card.atk.is_none() && card.card_type.contains("monster"))
|
|
}
|
|
RawCardFilter(Field::Def, _, Value::String(s)) if s == "?" => {
|
|
Box::new(move |card| card.def.is_none() && card.link_rating.is_none() && card.card_type.contains("monster"))
|
|
}
|
|
RawCardFilter(Field::Level, op, Value::Numerical(n)) => Box::new(move |card| op.filter_number(card.level, n)),
|
|
RawCardFilter(Field::LinkRating, op, Value::Numerical(n)) => Box::new(move |card| op.filter_number(card.link_rating, n)),
|
|
RawCardFilter(Field::Type, Operator::Equal, Value::String(s)) => Box::new(move |card| card.r#type == s),
|
|
RawCardFilter(Field::Type, Operator::NotEqual, Value::String(s)) => Box::new(move |card| card.r#type != s),
|
|
RawCardFilter(Field::Attribute, Operator::Equal, Value::String(s)) => Box::new(move |card| card.attribute.contains(&s)),
|
|
RawCardFilter(Field::Attribute, Operator::NotEqual, Value::String(s)) => Box::new(move |card| !card.attribute.contains(&s)),
|
|
RawCardFilter(Field::Class, Operator::Equal, Value::String(s)) => Box::new(move |card| card.card_type.contains(&s)),
|
|
RawCardFilter(Field::Class, Operator::NotEqual, Value::String(s)) => Box::new(move |card| !card.card_type.contains(&s)),
|
|
RawCardFilter(Field::Text, Operator::Equal, Value::String(s)) => Box::new(move |card| card.text.contains(&s)),
|
|
RawCardFilter(Field::Text, Operator::NotEqual, Value::String(s)) => Box::new(move |card| !card.text.contains(&s)),
|
|
RawCardFilter(Field::Name, Operator::Equal, Value::String(s)) => Box::new(move |card| card.name.contains(&s)),
|
|
RawCardFilter(Field::Name, Operator::NotEqual, Value::String(s)) => Box::new(move |card| !card.name.contains(&s)),
|
|
q => Err(format!("unknown query: {q:?}"))?,
|
|
})
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use crate::{data::tests::RAW_MONSTER, parser::parse_filters};
|
|
|
|
#[test]
|
|
fn level_filter_test() {
|
|
let lacooda = SearchCard::from(&serde_json::from_str::<Card>(RAW_MONSTER).unwrap());
|
|
let filter_level_3 = parse_filters("l=3").unwrap();
|
|
assert!(filter_level_3[0].1(&lacooda));
|
|
let filter_level_5 = parse_filters("l=5").unwrap();
|
|
assert!(!filter_level_5[0].1(&lacooda));
|
|
}
|
|
}
|