summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/commands.rs12
-rw-r--r--src/config.rs1
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 {