86 lines
2.3 KiB
Rust
86 lines
2.3 KiB
Rust
use indexmap::IndexMap;
|
|
use serde_derive::Deserialize;
|
|
use serde_json::{json, Value};
|
|
|
|
use crate::data_source::{DataSource, Stat};
|
|
use crate::ChampInfo;
|
|
use crate::Champion as ChampionLoL;
|
|
|
|
pub struct PBDataSource;
|
|
|
|
#[derive(Deserialize, Debug)]
|
|
struct ChampionResponse {
|
|
champions: Vec<Champion>,
|
|
}
|
|
|
|
#[derive(Deserialize, Debug)]
|
|
struct Champion {
|
|
key: Option<String>,
|
|
}
|
|
|
|
impl DataSource for PBDataSource {
|
|
fn get_alias(&self) -> &str {
|
|
"PB"
|
|
}
|
|
|
|
fn get_timeout(&self) -> u64 {
|
|
300
|
|
}
|
|
|
|
fn get_champs_with_positions(
|
|
&self,
|
|
client: &ureq::Agent,
|
|
_champion: &ChampionLoL,
|
|
) -> IndexMap<u32, Vec<String>> {
|
|
let mut champs = IndexMap::new();
|
|
|
|
let page = client
|
|
.get("https://www.probuilds.net/ajax/championListNew")
|
|
.set("Accept", "application/json")
|
|
.call()
|
|
.unwrap()
|
|
.into_string()
|
|
.unwrap();
|
|
|
|
let data: ChampionResponse = serde_json::from_str(&page).unwrap();
|
|
|
|
for champ in data.champions {
|
|
if let Some(k) = champ.key {
|
|
champs.insert(k.parse::<u32>().unwrap(), vec!["ANY".to_owned()]);
|
|
}
|
|
}
|
|
|
|
champs
|
|
}
|
|
|
|
fn get_champ_data_with_win_pourcentage(
|
|
&self,
|
|
champ: &ChampInfo,
|
|
positions: &[String],
|
|
client: &ureq::Agent,
|
|
) -> Vec<(String, Vec<Value>, Stat)> {
|
|
let mut builds = vec![];
|
|
|
|
let page = client
|
|
.get(format!("https://www.probuilds.net/ajax/games?limit=1&sort=gameDate-desc&championId={}&olderThan=0&lane=®ion=", champ.key).as_str())
|
|
.set("Accept", "application/json")
|
|
.call().unwrap().into_string().unwrap();
|
|
|
|
let mut items = vec![];
|
|
let mut build = &page[page.find("'items'").unwrap()..];
|
|
while build.contains("items") {
|
|
let pos = build.find("data-id=").unwrap();
|
|
build = &build[pos + 10..];
|
|
let end = build.find('\\').unwrap();
|
|
items.push(&build[..end]);
|
|
build = &build[end + 20..];
|
|
}
|
|
|
|
builds.push((positions[0].to_owned(), vec![json!({
|
|
"items": items.iter().map(|x| json!({"id": x.to_string(), "count": 1})).collect::<Vec<Value>>(),
|
|
"type": "Set"
|
|
})], Stat{win_rate:0.0,games:1,kda:0.0,patch:"".to_string()}));
|
|
|
|
builds
|
|
}
|
|
}
|