flibusta-apps/book_bot
1
0
Fork 0
mirror of https://github.com/flibusta-apps/book_bot.git synced 2025-12-06 07:25:36 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-09-24 04:09:24 +02:00
parent 95edba5e34
commit 0afe3acfcd
5 changed files with 54 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/bots/approved_bot/modules/annotations/commands.rs
View File

@@ -13,9 +13,8 @@ impl CommandParse<Self> for AnnotationCommand {
fn parse(s: &str, bot_name: &str) -> Result<Self, CommandParseError> {
Regex::new(r"^/(?P<an_type>[ab])_an_(?P<id>\d+)$")
.unwrap_or_else(|_| panic!("Broken AnnotationCommand regexp!"))
.captures(
&s.replace(&format!("@{bot_name}"), "")
).ok_or(CommandParseError)
.captures(&s.replace(&format!("@{bot_name}"), ""))
.ok_or(CommandParseError)
.map(|caps| (
caps["an_type"].to_string(),
caps["id"].parse::<u32>().unwrap_or_else(|_| panic!("Can't get id from AnnotationCommand!"))

3
src/bots/approved_bot/modules/annotations/mod.rs
View File

@@ -32,6 +32,7 @@ async fn download_image(
Ok(reqwest::get(file).await?.error_for_status()?)
}
pub async fn send_annotation_handler<T, Fut>(
message: Message,
bot: CacheMe<Throttle<Bot>>,
@@ -108,6 +109,7 @@ where
Ok(())
}
pub async fn annotation_pagination_handler<T, Fut>(
cq: CallbackQuery,
bot: CacheMe<Throttle<Bot>>,
@@ -158,6 +160,7 @@ where
Ok(())
}
pub fn get_annotations_handler() -> crate::bots::BotHandler {
dptree::entry()
.branch(

31
src/bots/approved_bot/modules/book/callback_data.rs
View File

@@ -2,7 +2,7 @@ use std::str::FromStr;
use regex::Regex;
use crate::bots::approved_bot::modules::utils::pagination::GetPaginationCallbackData;
use crate::bots::approved_bot::modules::utils::{pagination::GetPaginationCallbackData, errors::CallbackQueryParseError};
#[derive(Clone)]
@@ -13,27 +13,26 @@ pub enum BookCallbackData {
}
impl FromStr for BookCallbackData {
type Err = strum::ParseError;
type Err = CallbackQueryParseError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let re = Regex::new(r"^b(?P<an_type>[ats])_(?P<id>\d+)_(?P<page>\d+)$").unwrap();
let caps = re.captures(s);
let caps = match caps {
Some(v) => v,
None => return Err(strum::ParseError::VariantNotFound),
};
let annotation_type = &caps["an_type"];
let id = caps["id"].parse::<u32>().unwrap();
let page = caps["page"].parse::<u32>().unwrap();
match annotation_type {
Regex::new(r"^b(?P<an_type>[ats])_(?P<id>\d+)_(?P<page>\d+)$")
.unwrap_or_else(|_| panic!("Broken BookCallbackData regex pattern!"))
.captures(s)
.ok_or(CallbackQueryParseError)
.map(|caps| (
caps["an_type"].to_string(),
caps["id"].parse::<u32>().unwrap(),
caps["page"].parse::<u32>().unwrap()
))
.map(|(annotation_type, id, page)|
match annotation_type.as_str() {
"a" => Ok(BookCallbackData::Author { id, page }),
"t" => Ok(BookCallbackData::Translator { id, page }),
"s" => Ok(BookCallbackData::Sequence { id, page }),
_ => Err(strum::ParseError::VariantNotFound),
_ => panic!("Unknown BookCallbackData type: {}!", annotation_type),
}
)?
}
}

28
src/bots/approved_bot/modules/book/commands.rs
View File

@@ -12,25 +12,21 @@ pub enum BookCommand {
impl CommandParse<Self> for BookCommand {
fn parse(s: &str, bot_name: &str) -> Result<Self, CommandParseError> {
let re = Regex::new(r"^/(?P<an_type>[ats])_(?P<id>\d+)$").unwrap();
let full_bot_name = format!("@{bot_name}");
let after_replace = s.replace(&full_bot_name, "");
let caps = re.captures(&after_replace);
let caps = match caps {
Some(v) => v,
None => return Err(CommandParseError),
};
let annotation_type = &caps["an_type"];
let id: u32 = caps["id"].parse().unwrap();
match annotation_type {
Regex::new(r"^/(?P<an_type>[ats])_(?P<id>\d+)$")
.unwrap_or_else(|_| panic!("Broken BookCommand regexp!"))
.captures(&s.replace(&format!("@{bot_name}"), ""))
.ok_or(CommandParseError)
.map(|caps| (
caps["an_type"].to_string(),
caps["id"].parse().unwrap()
))
.map(|(annotation_type, id)| {
match annotation_type.as_str() {
"a" => Ok(BookCommand::Author { id }),
"t" => Ok(BookCommand::Translator { id }),
"s" => Ok(BookCommand::Sequence { id }),
_ => Err(CommandParseError),
_ => panic!("Unknown BookCommand type: {}!", annotation_type),
}
})?
}
}

27
src/bots/approved_bot/modules/download/callback_data.rs
View File

@@ -3,6 +3,8 @@ use std::str::FromStr;
use regex::Regex;
use strum_macros::EnumIter;
use crate::bots::approved_bot::modules::utils::errors::CommandParseError;
#[derive(Clone, EnumIter)]
pub enum DownloadQueryData {
@@ -20,21 +22,20 @@ impl ToString for DownloadQueryData {
}
impl FromStr for DownloadQueryData {
type Err = strum::ParseError;
type Err = CommandParseError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let re = Regex::new(r"^d_(?P<book_id>\d+)_(?P<file_type>\w+)$").unwrap();
let caps = re.captures(s);
let caps = match caps {
Some(v) => v,
None => return Err(strum::ParseError::VariantNotFound),
};
let book_id: u32 = caps["book_id"].parse().unwrap();
let file_type: String = caps["file_type"].to_string();
Ok(DownloadQueryData::DownloadData { book_id, file_type })
Regex::new(r"^d_(?P<book_id>\d+)_(?P<file_type>\w+)$")
.unwrap_or_else(|_| panic!("Broken DownloadQueryData regexp!"))
.captures(s)
.ok_or(CommandParseError)
.map(|caps| (
caps["book_id"].parse().unwrap(),
caps["file_type"].to_string()
))
.map(|(book_id, file_type)| {
DownloadQueryData::DownloadData { book_id, file_type }
})
}
}