use crate::get_name;
use crate::utils::cli;
use crate::utils::enums::{LanguageId, Type};
use clap::error::ErrorKind;
use itertools::izip;
use rustemon::Follow;
use rustemon::client::RustemonClient;
use rustemon::pokemon::*;
pub async fn print_matchups(
client: &RustemonClient,
primary: Type,
secondary: Option<Type>,
list: bool,
fast: bool,
lang: LanguageId,
) -> Result<Vec<String>, clap::Error> {
let primary = match type_::get_by_name(&primary.to_string(), &client).await {
Ok(x) => x,
Err(_) => {
return Err(cli::error(
ErrorKind::InvalidValue,
format!("API error: could not retrieve type {primary}"),
));
},
};
let secondary = match secondary {
Some(t) => match type_::get_by_name(&t.to_string(), &client).await {
Ok(x) => Some(x),
Err(_) => {
return Err(cli::error(
ErrorKind::InvalidValue,
format!("API error: could not retrieve type {t}"),
));
},
},
None => None,
};
let mut no_damage_from = Vec::new();
let mut half_damage_from = Vec::new();
let mut double_damage_from = Vec::new();
let mut quarter_damage_from = Vec::new();
let mut quad_damage_from = Vec::new();
for other_type in primary.damage_relations.no_damage_from.iter() {
no_damage_from.push(if !fast {
get_name!(follow other_type, client, lang.to_string())
} else {
other_type.name.clone()
});
}
for other_type in primary.damage_relations.half_damage_from.iter() {
half_damage_from.push(if !fast {
get_name!(follow other_type, client, lang.to_string())
} else {
other_type.name.clone()
});
}
for other_type in primary.damage_relations.double_damage_from.iter() {
double_damage_from.push(if !fast {
get_name!(follow other_type, client, lang.to_string())
} else {
other_type.name.clone()
});
}
if let Some(ref second) = secondary {
for other_type in second.damage_relations.no_damage_from.iter() {
let name = if !fast {
get_name!(follow other_type, client, lang.to_string())
} else {
other_type.name.clone()
};
if let Some(idx) = half_damage_from.iter().position(|x| *x == name) {
half_damage_from.remove(idx);
no_damage_from.push(name.clone());
} else if let Some(idx) = double_damage_from.iter().position(|x| *x == name) {
double_damage_from.remove(idx);
no_damage_from.push(name.clone());
} else if let None = no_damage_from.iter().position(|x| *x == name) {
no_damage_from.push(name.clone());
}
}
for other_type in second.damage_relations.half_damage_from.iter() {
let name = if !fast {
get_name!(follow other_type, client, lang.to_string())
} else {
other_type.name.clone()
};
if let Some(idx) = half_damage_from.iter().position(|x| *x == name) {
quarter_damage_from.push(name.clone());
half_damage_from.remove(idx);
} else if let Some(idx) = double_damage_from.iter().position(|x| *x == name) {
double_damage_from.remove(idx);
} else if let None = no_damage_from.iter().position(|x| *x == name) {
half_damage_from.push(name.clone());
}
}
for other_type in second.damage_relations.double_damage_from.iter() {
let name = if !fast {
get_name!(follow other_type, client, lang.to_string())
} else {
other_type.name.clone()
};
if let Some(idx) = half_damage_from.iter().position(|x| *x == name) {
half_damage_from.remove(idx);
} else if let Some(idx) = double_damage_from.iter().position(|x| *x == name) {
quad_damage_from.push(name.clone());
double_damage_from.remove(idx);
} else if let None = no_damage_from.iter().position(|x| *x == name) {
double_damage_from.push(name.clone());
}
}
}
let maxlen = itertools::max(vec![
no_damage_from.len(),
half_damage_from.len(),
double_damage_from.len(),
])
.unwrap();
while no_damage_from.len() < maxlen {
no_damage_from.push(String::new());
}
while half_damage_from.len() < maxlen {
half_damage_from.push(String::new());
}
while double_damage_from.len() < maxlen {
double_damage_from.push(String::new());
}
while quarter_damage_from.len() < maxlen {
quarter_damage_from.push(String::new());
}
while quad_damage_from.len() < maxlen {
quad_damage_from.push(String::new());
}
let mut result = Vec::new();
match secondary {
None => {
if !list {
result.push(format!("{:^12} {:^12} {:^12}", "*0", "*0.5", "*2"));
result.push(format!("{:-<12} {:-<12} {:-<12}", "", "", ""));
for (no_dmg, half_dmg, double_dmg) in
izip!(&no_damage_from, &half_damage_from, &double_damage_from)
{
result.push(format!(
"{:<12} {:<12} {:<12}",
no_dmg, half_dmg, double_dmg
));
}
} else {
result.push(format!(
"{}:",
if !fast {
get_name!(primary, client, lang.to_string())
} else {
primary.name.clone()
},
));
let mut add_separator = false;
if no_damage_from.iter().filter(|x| !x.is_empty()).count() != 0 {
add_separator = true;
result.push(String::from(" - 0x:"));
for no_dmg in no_damage_from.iter() {
if no_dmg.is_empty() {
break;
}
result.push(format!(" * {}", no_dmg));
}
}
if half_damage_from.iter().filter(|x| !x.is_empty()).count() != 0 {
if add_separator {
result.push(String::new());
}
add_separator = true;
result.push(String::from(" - 0.5x:"));
for half_dmg in half_damage_from.iter() {
if half_dmg.is_empty() {
break;
}
result.push(format!(" * {}", half_dmg));
}
}
if double_damage_from.iter().filter(|x| !x.is_empty()).count() != 0 {
if add_separator {
result.push(String::new());
}
result.push(String::from(" - 2x:"));
for double_dmg in double_damage_from.iter() {
if double_dmg.is_empty() {
break;
}
result.push(format!(" * {}", double_dmg));
}
}
}
},
Some(second) => {
if !list {
result.push(format!(
"{:^12} {:^12} {:^12} {:^12} {:^12}",
"*0", "*0.25", "*0.5", "*2", "*4"
));
result.push(format!(
"{:-<12} {:-<12} {:-<12} {:-<12} {:-<12}",
"", "", "", "", ""
));
for (no_dmg, quarter_dmg, half_dmg, double_dmg, quad_dmg) in izip!(
&no_damage_from, &quarter_damage_from, &half_damage_from, &double_damage_from,
&quad_damage_from
) {
result.push(format!(
"{:<12} {:<12} {:<12} {:<12} {:<12}",
no_dmg, quarter_dmg, half_dmg, double_dmg, quad_dmg
));
}
} else {
result.push(format!(
"{}:",
vec![
if !fast {
get_name!(primary, client, lang.to_string())
} else {
primary.name.clone()
},
if !fast {
get_name!(second, client, lang.to_string())
} else {
second.name.clone()
},
]
.join("/"),
));
let mut add_separator = false;
if no_damage_from.iter().filter(|x| !x.is_empty()).count() != 0 {
add_separator = true;
result.push(String::from(" - 0x:"));
for no_dmg in no_damage_from.iter() {
if no_dmg.is_empty() {
break;
}
result.push(format!(" * {}", no_dmg));
}
}
if quarter_damage_from.iter().filter(|x| !x.is_empty()).count() != 0 {
if add_separator {
result.push(String::new());
}
add_separator = true;
result.push(String::from(" - 0.25x:"));
for quarter_dmg in quarter_damage_from.iter() {
if quarter_dmg.is_empty() {
break;
}
result.push(format!(" * {}", quarter_dmg));
}
}
if half_damage_from.iter().filter(|x| !x.is_empty()).count() != 0 {
if add_separator {
result.push(String::new());
}
add_separator = true;
result.push(String::from(" - 0.5x:"));
for half_dmg in half_damage_from.iter() {
if half_dmg.is_empty() {
break;
}
result.push(format!(" * {}", half_dmg));
}
}
if double_damage_from.iter().filter(|x| !x.is_empty()).count() != 0 {
if add_separator {
result.push(String::new());
}
add_separator = true;
result.push(String::from(" - 2x:"));
for double_dmg in double_damage_from.iter() {
if double_dmg.is_empty() {
break;
}
result.push(format!(" * {}", double_dmg));
}
}
if quad_damage_from.iter().filter(|x| !x.is_empty()).count() != 0 {
if add_separator {
result.push(String::new());
}
result.push(String::from(" - 4x:"));
for quad_dmg in quad_damage_from.iter() {
if quad_dmg.is_empty() {
break;
}
result.push(format!(" * {}", quad_dmg));
}
}
}
},
}
Ok(result)
}
#[cfg(test)]
mod tests {
use super::*;
#[tokio::test]
async fn test_matchups() {
let client = RustemonClient::default();
let success = vec![
" *0 *0.5 *2 ",
"------------ ------------ ------------",
"Dragon Fighting Poison ",
" Bug Steel ",
" Dark ",
];
let primary = Type::Fairy;
let secondary = None;
let fast = false;
let lang = LanguageId::En;
let list = false;
match print_matchups(&client, primary, secondary, list, fast, lang).await {
Ok(res) => assert_eq!(res, success),
Err(err) => panic!("{}", err.render()),
}
}
#[tokio::test]
async fn test_matchups_dual() {
let client = RustemonClient::default();
let success = vec![
" *0 *0.25 *0.5 *2 *4 ",
"------------ ------------ ------------ ------------ ------------",
"Electric Flying Ground ",
" Steel Water ",
" Poison Grass ",
" Rock Ice ",
];
let primary = Type::Electric;
let secondary = Some(Type::Ground);
let fast = false;
let lang = LanguageId::En;
let list = false;
match print_matchups(&client, primary, secondary, list, fast, lang).await {
Ok(res) => assert_eq!(res, success),
Err(err) => panic!("{}", err.render()),
}
}
#[tokio::test]
async fn test_matchups_list() {
let client = RustemonClient::default();
let success = vec![
"Hada/Acero:", " - 0x:", " * Dragón", " * Veneno", "", " - 0.25x:", " * Bicho", "",
" - 0.5x:", " * Siniestro", " * Normal", " * Volador", " * Roca", " * Planta",
" * Psíquico", " * Hielo", " * Hada", "", " - 2x:", " * Tierra", " * Fuego",
];
let primary = Type::Fairy;
let secondary = Some(Type::Steel);
let fast = false;
let lang = LanguageId::Es;
let list = true;
match print_matchups(&client, primary, secondary, list, fast, lang).await {
Ok(res) => assert_eq!(res, success),
Err(err) => panic!("{}", err.render()),
}
}
}