Fixes

src/bots/approved_bot/modules/download/commands.rs

@@ -0,0 +1,77 @@
+use regex::Regex;
+use strum_macros::EnumIter;
+
+use crate::bots::approved_bot::modules::utils::CommandParse;
+
+
+#[derive(Clone)]
+pub struct StartDownloadCommand {
+    pub id: u32,
+}
+
+impl ToString for StartDownloadCommand {
+    fn to_string(&self) -> String {
+        let StartDownloadCommand { id } = self;
+        format!("/d_{id}")
+    }
+}
+
+impl CommandParse<Self> for StartDownloadCommand {
+    fn parse(s: &str, bot_name: &str) -> Result<Self, strum::ParseError> {
+        let re = Regex::new(r"^/d_(?P<book_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(strum::ParseError::VariantNotFound),
+        };
+
+        let book_id: u32 = caps["book_id"].parse().unwrap();
+
+        Ok(StartDownloadCommand { id: book_id })
+    }
+}
+
+#[derive(Clone, EnumIter)]
+pub enum DownloadArchiveCommand {
+    Sequence { id: u32},
+    Author { id: u32 },
+    Translator { id: u32 }
+}
+
+impl ToString for DownloadArchiveCommand {
+    fn to_string(&self) -> String {
+        match self {
+            DownloadArchiveCommand::Sequence { id } => format!("/da_s_{id}"),
+            DownloadArchiveCommand::Author { id } => format!("/da_a_{id}"),
+            DownloadArchiveCommand::Translator { id } => format!("/da_t_{id}"),
+        }
+    }
+}
+
+impl CommandParse<Self> for DownloadArchiveCommand {
+    fn parse(s: &str, bot_name: &str) -> Result<Self, strum::ParseError> {
+        let re = Regex::new(r"^/da_(?P<type>[sat])_(?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(strum::ParseError::VariantNotFound),
+        };
+
+        let obj_id: u32 = caps["id"].parse().unwrap();
+
+        match &caps["type"] {
+            "s" => Ok(DownloadArchiveCommand::Sequence { id: obj_id }),
+            "a" => Ok(DownloadArchiveCommand::Author { id: obj_id }),
+            "t" => Ok(DownloadArchiveCommand::Translator { id: obj_id }),
+            _ => Err(strum::ParseError::VariantNotFound)
+        }
+    }
+}