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

Refactor book callback data parsers

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-09-26 18:51:02 +02:00
parent 452040e83a
commit eb275b9dec
1 changed files with 28 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,7 +3,9 @@ use std::str::FromStr;
use regex::Regex;
use strum_macros::EnumIter;
use crate::bots::approved_bot::modules::utils::errors::CommandParseError;
use crate::bots::approved_bot::modules::utils::errors::{
CallbackQueryParseError, CommandParseError,
};
#[derive(Clone, EnumIter)]
pub enum DownloadQueryData {
@@ -60,25 +62,25 @@ impl ToString for DownloadArchiveQueryData {
}
impl FromStr for DownloadArchiveQueryData {
type Err = strum::ParseError;
type Err = CallbackQueryParseError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let re = Regex::new(r"^da_(?P<obj_type>[sat])_(?P<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 id: u32 = caps["id"].parse().unwrap();
let file_type: String = caps["file_type"].to_string();
Ok(match caps["obj_type"].to_string().as_str() {
Regex::new(r"^da_(?P<obj_type>[sat])_(?P<id>\d+)_(?P<file_type>\w+)$")
.unwrap_or_else(|_| panic!("Broken BookCallbackData regex pattern!"))
.captures(s)
.ok_or(CallbackQueryParseError)
.map(|caps| {
(
caps["id"].parse().unwrap(),
caps["file_type"].to_string(),
caps["obj_type"].to_string(),
)
})
.map(|(id, file_type, obj_type)| match obj_type.as_str() {
"s" => DownloadArchiveQueryData::Sequence { id, file_type },
"a" => DownloadArchiveQueryData::Author { id, file_type },
"t" => DownloadArchiveQueryData::Translator { id, file_type },
_ => return Err(strum::ParseError::VariantNotFound),
_ => panic!("Unknown DownloadArchiveQueryData type: {}!", obj_type),
})
}
}
@@ -95,19 +97,14 @@ impl ToString for CheckArchiveStatus {
}
impl FromStr for CheckArchiveStatus {
type Err = strum::ParseError;
type Err = CallbackQueryParseError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let re = Regex::new(r"^check_da_(?P<task_id>\w+)$").unwrap();
let caps = re.captures(s);
let caps = match caps {
Some(v) => v,
None => return Err(strum::ParseError::VariantNotFound),
};
let task_id: String = caps["task_id"].parse().unwrap();
Ok(CheckArchiveStatus { task_id })
Regex::new(r"^check_da_(?P<task_id>\w+)$")
.unwrap_or_else(|_| panic!("Broken CheckArchiveStatus regex pattern!"))
.captures(s)
.ok_or(CallbackQueryParseError)
.map(|caps| caps["task_id"].parse().unwrap())
.map(|task_id| CheckArchiveStatus { task_id })
}
}