75 lines
1.9 KiB
Rust
75 lines
1.9 KiB
Rust
use crate::{
|
|
error::*,
|
|
models::search::Pagination,
|
|
};
|
|
|
|
use scraper::Html;
|
|
|
|
pub mod character;
|
|
pub mod free_company;
|
|
pub mod linkshell;
|
|
|
|
pub use self::{
|
|
character::parse as parse_character_search,
|
|
free_company::parse as parse_free_company_search,
|
|
linkshell::parse as parse_linkshell_search,
|
|
};
|
|
|
|
selectors!(
|
|
PAGINATION_TOTAL => ".parts__total";
|
|
PAGINATION_PAGES => ".btn__pager__current";
|
|
|
|
NO_RESULTS => "p.parts__zero";
|
|
);
|
|
|
|
pub(crate) fn parse_pagination(html: &Html) -> Result<Pagination> {
|
|
const LODESTONE_PER_PAGE: f32 = 50.0;
|
|
|
|
let total_str = crate::logic::plain_parse(&html, &*PAGINATION_TOTAL)?;
|
|
let total_results: u64 = total_str
|
|
.split(' ')
|
|
.next()
|
|
.unwrap() // will always have a first element
|
|
.parse()
|
|
.map_err(Error::InvalidNumber)?;
|
|
|
|
let (total_pages, current_page) = if parse_no_results(html) {
|
|
let total_pages = if total_results > 0 {
|
|
(total_results as f32 / LODESTONE_PER_PAGE).ceil() as u64
|
|
} else {
|
|
1
|
|
};
|
|
let current_page = if total_results > 0 { 0 } else { 1 };
|
|
(total_pages, current_page)
|
|
} else {
|
|
let pages_str = crate::logic::plain_parse(&html, &*PAGINATION_PAGES)?;
|
|
let mut pages_split = pages_str.split(' ');
|
|
|
|
let current_page: u64 = pages_split
|
|
.nth(1)
|
|
.ok_or_else(|| Error::invalid_content("4 items in pages string", None))?
|
|
.parse()
|
|
.map_err(Error::InvalidNumber)?;
|
|
|
|
let total_pages: u64 = pages_split
|
|
.nth(1)
|
|
.ok_or_else(|| Error::invalid_content("4 items in pages string", None))?
|
|
.parse()
|
|
.map_err(Error::InvalidNumber)?;
|
|
(total_pages, current_page)
|
|
};
|
|
|
|
Ok(Pagination {
|
|
current_page,
|
|
total_pages,
|
|
total_results,
|
|
})
|
|
}
|
|
|
|
pub(crate) fn parse_no_results(html: &Html) -> bool {
|
|
html.select(&*NO_RESULTS)
|
|
.next()
|
|
.map(|x| x.text().collect::<String>() == "Your search yielded no results.")
|
|
.unwrap_or(false)
|
|
}
|