Fix /sql command a lot.

src/commands.rs

@@ -1,6 +1,7 @@
+#![allow(unused_variables)]
 use crate::db;
 use crate::errors::Error;
-use crate::errors::Error::{SQLInvalidCommand, SQLITE3Error};
+use crate::errors::Error::{SQLITE3Error, SQLInvalidCommand};
 use async_trait::async_trait;
 use html_escape::encode_text;
 use markov::Chain;
@@ -39,8 +40,9 @@ pub struct Sql {
 
 #[async_trait]
 pub trait Execute {
-    async fn run(&self, api: &Api, message: &Message) -> Result<(), Error>;
+    async fn exec(&self, api: &Api, message: &Message) -> Result<(), Error>;
-    async fn run_mystem(
+    async fn exec_with_result(&self, api: &Api, message: &Message) -> Result<String, Error>;
+    async fn exec_mystem(
         &self,
         api: &Api,
         message: &Message,
@@ -50,8 +52,12 @@ pub trait Execute {
 
 #[async_trait]
 impl Execute for Sql {
-    async fn run(&self, api: &Api, message: &Message) -> Result<(), Error> {
+    async fn exec(&self, api: &Api, message: &Message) -> Result<(), Error> {
-        let mut sql = self.data.to_uppercase();
+        unimplemented!()
+    }
+
+    async fn exec_with_result(&self, api: &Api, message: &Message) -> Result<String, Error> {
+        let mut sql = self.data.clone();
         let is_head = if sql.starts_with('-') {
             sql = sql.replacen("-", "", 1);
             false
@@ -88,7 +94,7 @@ impl Execute for Sql {
 
         let mut rows = match stmt.query(rusqlite::NO_PARAMS) {
             Err(e) => return Err(SQLITE3Error(e)),
-            Ok(mut rows) => rows,
+            Ok(rows) => rows,
         };
 
         let mut res: Vec<Vec<String>> = match rows.column_names() {
@@ -156,11 +162,11 @@ impl Execute for Sql {
         } else {
             msg
         };
-        Ok(())
+        Ok(msg)
     }
 
     #[allow(unused_variables)]
-    async fn run_mystem(
+    async fn exec_mystem(
         &self,
         api: &Api,
         message: &Message,
@@ -172,7 +178,7 @@ impl Execute for Sql {
 
 #[async_trait]
 impl Execute for Here {
-    async fn run(&self, api: &Api, message: &Message) -> Result<(), Error> {
+    async fn exec(&self, api: &Api, message: &Message) -> Result<(), Error> {
         let members: Vec<telegram_bot::User> = db::get_members(message.chat.id()).unwrap();
         for u in &members {
             debug!("Found user {:?} in chat {}", u, message.chat.id());
@@ -200,8 +206,12 @@ impl Execute for Here {
         Ok(())
     }
 
+    async fn exec_with_result(&self, api: &Api, message: &Message) -> Result<String, Error> {
+        unimplemented!()
+    }
+
     #[allow(unused_variables)]
-    async fn run_mystem(
+    async fn exec_mystem(
         &self,
         api: &Api,
         message: &Message,
@@ -213,7 +223,7 @@ impl Execute for Here {
 
 #[async_trait]
 impl Execute for Top {
-    async fn run(&self, api: &Api, message: &Message) -> Result<(), Error> {
+    async fn exec(&self, api: &Api, message: &Message) -> Result<(), Error> {
         let top = db::get_top(&message).await?;
         let mut msg = "<b>Your top using words:</b>\n<pre>".to_string();
         let mut counter = 1;
@@ -235,8 +245,12 @@ impl Execute for Top {
         Ok(())
     }
 
+    async fn exec_with_result(&self, api: &Api, message: &Message) -> Result<String, Error> {
+        unimplemented!()
+    }
+
     #[allow(unused_variables)]
-    async fn run_mystem(
+    async fn exec_mystem(
         &self,
         api: &Api,
         message: &Message,
@@ -248,7 +262,7 @@ impl Execute for Top {
 
 #[async_trait]
 impl Execute for MarkovAll {
-    async fn run(&self, api: &Api, message: &Message) -> Result<(), Error> {
+    async fn exec(&self, api: &Api, message: &Message) -> Result<(), Error> {
         let messages = db::get_messages_random_all().await?;
         let mut chain = Chain::new();
         chain.feed(messages);
@@ -269,8 +283,12 @@ impl Execute for MarkovAll {
         Ok(())
     }
 
+    async fn exec_with_result(&self, api: &Api, message: &Message) -> Result<String, Error> {
+        unimplemented!()
+    }
+
     #[allow(unused_variables)]
-    async fn run_mystem(
+    async fn exec_mystem(
         &self,
         api: &Api,
         message: &Message,
@@ -282,7 +300,7 @@ impl Execute for MarkovAll {
 
 #[async_trait]
 impl Execute for Markov {
-    async fn run(&self, api: &Api, message: &Message) -> Result<(), Error> {
+    async fn exec(&self, api: &Api, message: &Message) -> Result<(), Error> {
         let messages = db::get_messages_random_group(&message).await?;
         let mut chain = Chain::new();
         chain.feed(messages);
@@ -303,8 +321,12 @@ impl Execute for Markov {
         Ok(())
     }
 
+    async fn exec_with_result(&self, api: &Api, message: &Message) -> Result<String, Error> {
+        unimplemented!()
+    }
+
     #[allow(unused_variables)]
-    async fn run_mystem(
+    async fn exec_mystem(
         &self,
         api: &Api,
         message: &Message,
@@ -317,12 +339,16 @@ impl Execute for Markov {
 #[async_trait]
 impl Execute for Omedeto {
     #[allow(unused_variables)]
-    async fn run(&self, api: &Api, message: &Message) -> Result<(), Error> {
+    async fn exec(&self, api: &Api, message: &Message) -> Result<(), Error> {
+        unimplemented!()
+    }
+
+    async fn exec_with_result(&self, api: &Api, message: &Message) -> Result<String, Error> {
         unimplemented!()
     }
 
     #[warn(unused_must_use)]
-    async fn run_mystem(
+    async fn exec_mystem(
         &self,
         api: &Api,
         message: &Message,

src/handlers.rs

@@ -31,60 +31,68 @@ pub async fn handler(
                     Here {
                         data: "".to_string(),
                     }
-                    .run(&api, &message)
+                    .exec(&api, &message)
                     .await?
                 }
                 s if s.to_string().starts_with("/sql") => match {
                     Sql {
                         data: s.replace("/sql ", ""),
                     }
-                    .run(&api, &message)
+                    .exec_with_result(&api, &message)
                     .await
                 } {
-                    Ok(_) => debug!("/sql command sent to {}", message.chat.id()),
+                    Ok(msg) => {
+                        let _ = api
+                            .send(
+                                message
+                                    .text_reply(msg)
+                                    .parse_mode(ParseMode::Html),
+                            )
+                            .await?;
+                    },
                     Err(e) => {
-                        api.send(
+                        let _ = api
-                            message
+                            .send(
-                                .text_reply(format!("Error: {:#?}", e))
+                                message
-                                .parse_mode(ParseMode::Html),
+                                    .text_reply(format!("Error: {:#?}", e))
-                        )
+                                    .parse_mode(ParseMode::Html),
-                        .await?;
+                            )
-                        ()
+                            .await?;
                     }
                 },
                 "/top" => {
                     Top {
                         data: "".to_string(),
                     }
-                    .run(&api, &message)
+                    .exec(&api, &message)
                     .await?
                 }
                 "/stat" => {
                     Top {
                         data: "".to_string(),
                     }
-                    .run(&api, &message)
+                    .exec(&api, &message)
                     .await?
                 }
                 "/markov_all" => {
                     MarkovAll {
                         data: "".to_string(),
                     }
-                    .run(&api, &message)
+                    .exec(&api, &message)
                     .await?
                 }
                 "/markov" => {
                     Markov {
                         data: "".to_string(),
                     }
-                    .run(&api, &message)
+                    .exec(&api, &message)
                     .await?
                 }
                 "/omedeto" => {
                     Omedeto {
                         data: "".to_string(),
                     }
-                    .run_mystem(&api, &message, mystem)
+                    .exec_mystem(&api, &message, mystem)
                     .await?
                 }
                 _ => (),