From 0e10cdbdf03f6d7e220ced8824b103002743cf3c Mon Sep 17 00:00:00 2001
From: Akulij <aakulij@gmail.com>
Date: Tue, 20 May 2025 01:06:16 +0500
Subject: [PATCH] create command module

defines BotCommand struct
---
 src/commands.rs | 99 +++++++++++++++++++++++++++++++++++++++++++++++++
 src/main.rs     |  1 +
 2 files changed, 100 insertions(+)
 create mode 100644 src/commands.rs

diff --git a/src/commands.rs b/src/commands.rs
new file mode 100644
index 0000000..d6bcdb6
--- /dev/null
+++ b/src/commands.rs
@@ -0,0 +1,99 @@
+use std::str::FromStr;
+
+use teloxide::utils::command::ParseError;
+
+#[derive(thiserror::Error, Debug)]
+pub enum CommandError {
+    #[error("parse error: {0:?}")]
+    ParseError(#[from] ParseError),
+    #[error("failed to validate command: {0:?}")]
+    ValidationError(String),
+}
+
+pub struct BotCommand {
+    command: String,
+    args: Option<String>,
+}
+
+impl FromStr for BotCommand {
+    type Err = CommandError;
+
+    fn from_str(s: &str) -> Result<Self, Self::Err> {
+        let (command, args) = s.split_once(" ").map_or((s, None), |s| (s.0, Some(s.1)));
+
+        match command.strip_prefix("/") {
+            Some(command) => Ok(Self {
+                command: command.to_string(),
+                args: args.map(str::to_string),
+            }),
+            None => Err(CommandError::ParseError(ParseError::IncorrectFormat(
+                "Not a command".into(),
+            ))),
+        }
+    }
+}
+
+impl BotCommand {
+    pub fn from_validate(s: &str, cmds: &[&str]) -> Result<Self, CommandError> {
+        let bc = Self::from_str(s)?;
+
+        if !cmds.contains(&bc.command.as_str()) {
+            return Err(CommandError::ValidationError(format!(
+                "invalid command {}",
+                bc.command
+            )));
+        };
+
+        Ok(bc)
+    }
+
+    pub fn command(&self) -> &str {
+        &self.command
+    }
+
+    pub fn args(&self) -> Option<&str> {
+        self.args.as_deref()
+    }
+
+    pub fn args_list(&self) -> Vec<&str> {
+        let args = match self.args {
+            Some(ref args) => args.as_str(),
+            None => "",
+        };
+
+        args.split_whitespace().collect()
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_botcommand_from_str_simple() {
+        let cmdstr = "/start";
+
+        let bc = BotCommand::from_str(cmdstr).unwrap();
+        assert_eq!(bc.command(), "start");
+        assert_eq!(bc.args(), None);
+    }
+
+    #[test]
+    fn test_botcommand_from_str_with_args() {
+        let cmdstr = "/some_long_cmd arg1 arg2";
+
+        let bc = BotCommand::from_str(cmdstr).unwrap();
+        assert_eq!(bc.command(), "some_long_cmd");
+        assert_eq!(bc.args(), Some("arg1 arg2"));
+    }
+
+    #[test]
+    fn test_botcommand_arg_list() {
+        let cmdstr = "/some_long_cmd arg1 arg2";
+
+        let bc = BotCommand::from_str(cmdstr).unwrap();
+        assert_eq!(bc.command(), "some_long_cmd");
+        assert_eq!(bc.args(), Some("arg1 arg2"));
+        assert_eq!(bc.args_list(), vec!["arg1", "arg2"]);
+    }
+}
diff --git a/src/main.rs b/src/main.rs
index 6191c95..d42103c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,5 +1,6 @@
 pub mod admin;
 pub mod botscript;
+pub mod commands;
 pub mod db;
 pub mod mongodb_storage;
 pub mod utils;