Implemented watchdog

This commit is contained in:
Lukas Wölfer
2025-07-22 01:39:41 +02:00
parent 8bae1fdfbb
commit 618a6c1d02
6 changed files with 373 additions and 211 deletions

62
src/wikiinfo.rs Normal file
View File

@@ -0,0 +1,62 @@
use mwbot::{
Bot, Page,
generators::{Generator, querypage::QueryPage, search::Search},
};
pub async fn wanted_ids(bot: Bot) -> Vec<(u32, Page)> {
let mut gene = QueryPage::new("Wantedpages").generate(&bot);
let mut result = vec![];
while let Some(x) = gene.recv().await {
let p = match x {
Ok(p) => p,
Err(e) => {
tracing::error!("Could not get search result: {e}");
continue;
}
};
if let Ok(n) = parse_wsdc_page_name(p.title()) {
result.push((n, p));
}
}
result
}
#[derive(thiserror::Error, Debug)]
enum TitleParseError {
#[error("Does not contain the correct prefix")]
NoPrefix,
#[error("Possibly incorrect page")]
Sketchy,
}
fn parse_wsdc_page_name(name: &str) -> Result<u32, TitleParseError> {
if !name.starts_with("WSDC/") {
return Err(TitleParseError::NoPrefix);
}
match name.trim_start_matches("WSDC/").parse::<u32>() {
Ok(n) => Ok(n),
Err(e) => {
tracing::error!("Page {name} does not fit: {e}");
Err(TitleParseError::Sketchy)
}
}
}
#[allow(dead_code)]
pub async fn index_wsdc_ids(bot: &Bot) -> Vec<u32> {
let mut gene = Search::new("WSDC/").generate(bot);
let mut result = vec![];
while let Some(x) = gene.recv().await {
let p = match x {
Ok(p) => p,
Err(e) => {
tracing::error!("Could not get search result: {e}");
continue;
}
};
if let Ok(n) = parse_wsdc_page_name(p.title()) {
result.push(n);
}
}
result
}