85 lines
2.0 KiB
Rust
85 lines
2.0 KiB
Rust
use crate::{
|
|
error::*,
|
|
models::{
|
|
GrandCompany,
|
|
character::GrandCompanyInfo,
|
|
},
|
|
};
|
|
|
|
use scraper::{
|
|
Html,
|
|
ElementRef,
|
|
node::Element,
|
|
};
|
|
|
|
macro_rules! selectors {
|
|
($($name:ident => $selector:expr);+$(;)?) => {
|
|
use lazy_static::lazy_static;
|
|
|
|
lazy_static! {
|
|
$(
|
|
static ref $name: scraper::Selector = scraper::Selector::parse($selector).unwrap();
|
|
)+
|
|
}
|
|
}
|
|
}
|
|
|
|
pub mod character;
|
|
pub mod free_company;
|
|
pub mod linkshell;
|
|
pub mod search;
|
|
|
|
pub use self::{
|
|
character::parse as parse_character,
|
|
free_company::parse as parse_free_company,
|
|
linkshell::parse as parse_linkshell,
|
|
search::*,
|
|
};
|
|
|
|
pub(crate) fn plain_parse(html: &Html, select: &scraper::Selector) -> Result<String> {
|
|
let string = html
|
|
.select(select)
|
|
.next()
|
|
.ok_or_else(|| Error::missing_element(select))?
|
|
.text()
|
|
.collect();
|
|
Ok(string)
|
|
}
|
|
|
|
pub(crate) fn plain_parse_elem<'a>(html: ElementRef<'a>, select: &scraper::Selector) -> Result<String> {
|
|
let string = html
|
|
.select(select)
|
|
.next()
|
|
.ok_or_else(|| Error::missing_element(select))?
|
|
.text()
|
|
.collect();
|
|
Ok(string)
|
|
}
|
|
|
|
pub(crate) fn parse_id(a: &Element) -> Result<u64> {
|
|
let href = a.attr("href").ok_or_else(|| Error::invalid_content("href on link", None))?;
|
|
let last = href
|
|
.split('/')
|
|
.filter(|x| !x.is_empty())
|
|
.last()
|
|
.ok_or_else(|| Error::invalid_content("href separated by `/`", Some(&href)))?;
|
|
last.parse().map_err(Error::InvalidNumber)
|
|
}
|
|
|
|
pub(crate) fn parse_grand_company(text: &str) -> Result<GrandCompanyInfo> {
|
|
let mut x = text.split(" / ");
|
|
let gc_str = x
|
|
.next()
|
|
.ok_or_else(|| Error::invalid_content("gc/rank separated by `/`", Some(&text)))?;
|
|
let name = GrandCompany::parse(gc_str)
|
|
.ok_or_else(|| Error::invalid_content("valid grand company", Some(&text)))?;
|
|
let rank = x
|
|
.next()
|
|
.ok_or_else(|| Error::invalid_content("gc/rank separated by `/`", Some(&text)))?
|
|
.to_string();
|
|
Ok(GrandCompanyInfo {
|
|
name,
|
|
rank,
|
|
})
|
|
}
|