feat: run bot only if authorized

2 files changed, 13 insertions, 0 deletions

diff --git a/src/commands.rs b/src/commands.rs
index e498916..a3a3694 100644
--- a/src/commands.rs
+++ b/src/commands.rs
@@ -1,3 +1,4 @@
+use crate::config::Config;
 use std::error::Error;
 use teloxide::payloads::SendMessageSetters;
 use teloxide::prelude::{AutoSend, Bot, Message, UpdateWithCx};
@@ -16,6 +17,17 @@ pub enum Command {
 pub async fn handler(
     cx: UpdateWithCx<AutoSend<Bot>, Message>,
 ) -> Result<(), Box<dyn Error + Send + Sync>> {
+    let username = Config::from_env().unwrap().username;
+
+    if let Some(author) = &cx.update.from().unwrap().username {
+        if *author != username {
+            cx.reply_to("You are not allowed to do this action!")
+                .await?;
+
+            return Ok(());
+        }
+    }
+
     let txt = cx.update.text();
     if txt.is_none() {
         return Ok(());
diff --git a/src/config.rs b/src/config.rs
index 3199c62..668dda7 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -6,6 +6,7 @@ pub struct Config {
     pub cf: String,
     pub password: String,
     pub driver_url: String,
+    pub username: String,
 }
 
 impl Config {