Compare commits

...

7 Commits

15 changed files with 89 additions and 64 deletions

View File

@ -0,0 +1,12 @@
<b>Команды (для админов)</b>
<code>/mute {ID | REPLY} &lt;DURATION&gt; [TIME METRIC]</code> - выдаёт мут на заданное время;
<code>/unmute {ID | REPLY}</code> - снимает мут;
Для более подробного описания, введите команды без аргументов.
<b>Эмодзи (для админов)</b>
🎲 - выдаёт мут пользователю на случайно заданное время (от 1 до 6 дней);
🎰 - выдаёт мут пользователю на случайно заданное время (от 1 до 63 дней), в случае джекпота участник отправляется в бан.

View File

@ -3,6 +3,7 @@ use std::include_str;
pub mod files { pub mod files {
use super::*; use super::*;
pub const HELP_COMMAND_TEXT: &str = include_str!("help_command.html");
pub const MUTE_COMMAND_HELP: &str = include_str!("mute_command_help.html"); pub const MUTE_COMMAND_HELP: &str = include_str!("mute_command_help.html");
pub const UNMUTE_COMMAND_HELP: &str = include_str!("unmute_command_help.html"); pub const UNMUTE_COMMAND_HELP: &str = include_str!("unmute_command_help.html");
} }

View File

@ -1,10 +1,12 @@
<code>/mute [ID | REPLY] &lt;DURATION&gt; [TIME METRIC]</code> <code>/mute {ID | REPLY} &lt;DURATION&gt; [TIME METRIC]</code>
<b>Выдать мут участнику чата.</b> <b>Выдаёт мут.</b>
Использовать при ответе на сообщение или указать <code>ID</code> пользователя. Использовать при ответе на сообщение или указать ID пользователя.
<i>DURATION<i> - длительность мута. <b><em>1. ID | REPLY.</em></b> ID или ответ на сообщение пользователя;
<i>TIME METRIC<i> - метрика времени в секундах, минутах, часах и днях ([s|min|h|d|w|m]/[с|мин|ч|д|н|м]); по умолчанию используются дни. <b><em>2. DURATION.</em></b> Продолжительность мута (DURATION &gt; 0);
<b><em>3. TIME METRIC.</em></b> Временная метрика. Измеряется в минутах/часах/днях/неделях/месяцах. По умолчанию применяется в днях.
<b><u>Команда доступна только администраторам.</u></b>
Доступно для пользователей с административными привилегиями в чате.

View File

@ -1,7 +1,8 @@
<code>/unmute [ID | UNMUTE]</code> <code>/unmute {ID | REPLY}</code>
<b>Снять мут с участника чата.</b> <b>Cнимает мут с пользователя.</b>
Использовать при ответе на сообщение или указать <code>ID</code> пользователя. <b><em>1. ID | REPLY.</em></b> ID или ответ на сообщение пользователя.
<b><u>Команда доступна только администраторам.</u></b>
Доступно для пользователей с административными привилегиями в чате.

View File

@ -25,7 +25,7 @@ pub async fn ban_member(
Some(id) => id, Some(id) => id,
None => { None => {
sender_builder sender_builder
.text("ban: нет ответа на сообщение пользователя.") .text("Нет ID или ответа на сообщение пользователя.")
.reply_to(message.id()) .reply_to(message.id())
.build() .build()
.send(&bot) .send(&bot)
@ -39,21 +39,19 @@ pub async fn ban_member(
let callback = || async { ban_chat_member(&bot, user_id, chat_id).await }; let callback = || async { ban_chat_member(&bot, user_id, chat_id).await };
let demote_args: (&Bot, i64, i64) = (&bot, user_id, chat_id); let demote_args: (&Bot, i64, i64) = (&bot, user_id, chat_id);
sender_builder sender_builder.set_text("Невозможно забанить пользователя.");
.set_text("ban: невозможно забанить пользователя.");
if try_restrict(callback, demote_args, sender_builder.clone().build()) if try_restrict(callback, demote_args, sender_builder.clone().build())
.await .await
.is_err() .is_err()
{ {
sender_builder.build().send(&bot).await?;
Ok(EventReturn::Cancel) Ok(EventReturn::Cancel)
} else { } else {
let banned_user_name: String = user.get_user_name(&bot, &message).await.unwrap(); let banned_user_name: String = user.get_user_name(&bot, &message).await.unwrap();
sender_builder sender_builder
.reply_to(message.id()) .reply_to(message.id())
.text(format!("ban: пользователь {} забанен.", banned_user_name)) .text(format!("Пользователь {banned_user_name} забанен."))
.build() .build()
.send(&bot) .send(&bot)
.await?; .await?;

View File

@ -25,19 +25,7 @@ pub async fn mute_member(
) -> HandlerResult { ) -> HandlerResult {
let (bot, message, mut sender_builder): ExtractedEntityData = handler_entity.extract(); let (bot, message, mut sender_builder): ExtractedEntityData = handler_entity.extract();
let user_id: i64 = match user.get_id() { let user_id: i64 = user.get_id().unwrap();
Some(id) => id,
None => {
sender_builder
.text("mute: нет ответа на сообщение пользователя.")
.reply_to(message.id())
.build()
.send(&bot)
.await
.unwrap();
return Ok(EventReturn::Cancel);
}
};
sleep(Duration::from_millis(time.1)).await; sleep(Duration::from_millis(time.1)).await;
@ -46,13 +34,12 @@ pub async fn mute_member(
let callback = || async { restrict(&bot, user_id, unmute_date, chat_id).await }; let callback = || async { restrict(&bot, user_id, unmute_date, chat_id).await };
sender_builder.set_text("mute: невозможно выдать мут."); sender_builder.set_text("Невозможно выдать мут.");
if try_restrict(callback, demote_args, sender_builder.clone().build()) if try_restrict(callback, demote_args, sender_builder.clone().build())
.await .await
.is_err() .is_err()
{ {
sender_builder.build().send(&bot).await?;
Ok(EventReturn::Cancel) Ok(EventReturn::Cancel)
} else { } else {
let muted_user_name: String = user.get_user_name(&bot, &message).await.unwrap(); let muted_user_name: String = user.get_user_name(&bot, &message).await.unwrap();
@ -60,8 +47,7 @@ pub async fn mute_member(
sender_builder sender_builder
.reply_to(message.id()) .reply_to(message.id())
.text(format!( .text(format!(
"mute: пользователю {} выдан мут на {:?} {}.", "Пользователю {muted_user_name} выдан мут на {mute_duration} {postfix}."
muted_user_name, mute_duration, postfix
)) ))
.build() .build()
.send(&bot) .send(&bot)

View File

@ -28,9 +28,7 @@ pub async fn unmute_member(
if let Err(error) = bot.send(bot_action).await { if let Err(error) = bot.send(bot_action).await {
sender_builder sender_builder
.text(format!( .text(format!("Невозможно снять мут с пользователя: {error:?}."))
"unmute: невозможно снять мут с пользователя: {error:?}"
))
.build() .build()
.send(&bot) .send(&bot)
.await?; .await?;
@ -40,7 +38,7 @@ pub async fn unmute_member(
sender_builder sender_builder
.reply_to(message.id()) .reply_to(message.id())
.text(format!("unmute: пользователю {} удален мут.", muted_user_name)) .text(format!("С пользователя {muted_user_name} снят мут."))
.build() .build()
.send(&bot) .send(&bot)
.await .await

View File

@ -1,28 +1,16 @@
use telers::{ use telers::{
enums::parse_mode::ParseMode,
event::{telegram::HandlerResult, EventReturn}, event::{telegram::HandlerResult, EventReturn},
types::Message, types::Message,
Bot, Bot,
}; };
use crate::types::structs::message_sender::MessageSender; use crate::{assets::files::HELP_COMMAND_TEXT, types::structs::message_sender::MessageSender};
const HELP_TEXT: &str = "\
Команды:\n\
/unmute [ID] - снять мут с участника чата.\n\
/mute [ID] - выдать мут участнику чата.\n\
\n\
Для получения подробной помощи по командам использовать их без аргументов.\n\
\n\
Эмодзи:\n\
Отправлять в чат в ответ на сообщение другого пользователя.\n\
\n\
🎲 - мут участнику чата на рандомный интервал времени.\n\
🎰 - мут участнику чата на рандомный интервал времени; выпадание джекпота \
отправляет в бан.";
pub async fn help(bot: Bot, message: Message) -> HandlerResult { pub async fn help(bot: Bot, message: Message) -> HandlerResult {
MessageSender::builder(message.chat().id()) MessageSender::builder(message.chat().id())
.text(HELP_TEXT) .text(HELP_COMMAND_TEXT)
.parse_mode(ParseMode::HTML)
.build() .build()
.send(&bot) .send(&bot)
.await .await

View File

@ -35,8 +35,9 @@ pub async fn mute(bot: Bot, message: Message, command: CommandObject) -> Handler
if let Ok(id) = raw_id.parse::<i64>() { if let Ok(id) = raw_id.parse::<i64>() {
TargetUser::Id(id) TargetUser::Id(id)
} else { } else {
handler_entity.message_sender_builder handler_entity
.text("mute: нет ответа на сообщение или не указано ID пользователя.") .message_sender_builder
.text("Нет ID или ответа на сообщение пользователя.")
.build() .build()
.send(&handler_entity.bot_instance) .send(&handler_entity.bot_instance)
.await?; .await?;
@ -59,7 +60,7 @@ pub async fn mute(bot: Bot, message: Message, command: CommandObject) -> Handler
handler_entity handler_entity
.message_sender_builder .message_sender_builder
.set_text("mute: не указана длительность (число)."); .set_text("Не указана длительность мута.");
match args.get(duration_argument_position).cloned() { match args.get(duration_argument_position).cloned() {
Some(duration_str) => { Some(duration_str) => {

View File

@ -27,7 +27,7 @@ pub async fn unmute(bot: Bot, message: Message, command: CommandObject) -> Handl
Some(raw_id) => { Some(raw_id) => {
handler_entity handler_entity
.message_sender_builder .message_sender_builder
.set_text("unmute: ID пользователя не указан."); .set_text("Нет ID или ответа на сообщение пользователя.");
if let Ok(parsed_id) = raw_id.parse::<i64>() { if let Ok(parsed_id) = raw_id.parse::<i64>() {
let on_id: TargetUser = TargetUser::Id(parsed_id); let on_id: TargetUser = TargetUser::Id(parsed_id);

View File

@ -22,8 +22,20 @@ pub async fn dice_handler(bot: Bot, message: Message) -> HandlerResult {
let handler_entity: HandlerEntity = HandlerEntity::new(bot, message, sender); let handler_entity: HandlerEntity = HandlerEntity::new(bot, message, sender);
let (mute_time, emoji): (TimeMetrics, &str) = (TimeMetrics::Days(dice.value), &dice.emoji); let (mute_time, emoji): (TimeMetrics, &str) = (TimeMetrics::Days(dice.value), &dice.emoji);
let target: TargetUser = TargetUser::Reply(handler_entity.message_reciever.clone()); let target: TargetUser = TargetUser::Reply(handler_entity.message_reciever.clone());
if !target.exist() {
handler_entity
.message_sender_builder
.text("Нет ответа на сообщение пользователя.")
.build()
.send(&handler_entity.bot_instance)
.await?;
return Ok(EventReturn::Cancel);
}
match emoji { match emoji {
"🎲" => { "🎲" => {
mute_member(handler_entity, chat_id, target, (mute_time, DICE_DELAY_MS)).await?; mute_member(handler_entity, chat_id, target, (mute_time, DICE_DELAY_MS)).await?;

View File

@ -30,11 +30,10 @@ impl InnerMiddleware for AdminCheck {
let chat_id: i64 = message.chat_id().unwrap(); let chat_id: i64 = message.chat_id().unwrap();
if is_admin(&admins_list, message.from().unwrap()) { if is_admin(&admins_list, message.from().unwrap().id) {
let response = next(request).await?; let response = next(request).await?;
return Ok(response); return Ok(response);
} else { } else {
println!("lol");
MessageSender::builder(chat_id) MessageSender::builder(chat_id)
.text("Недостаточно прав для использования данной команды.") .text("Недостаточно прав для использования данной команды.")
.build() .build()

View File

@ -15,6 +15,21 @@ pub enum TargetUser {
} }
impl TargetUser { impl TargetUser {
pub fn exist(&self) -> bool {
match self {
Self::Id(_id) => true,
Self::Reply(msg) => {
if let Some(replied_msg) = msg.reply_to_message() {
return replied_msg.from().map(|user| user.id).is_some();
} else {
false
}
}
Self::CurrentMessage(msg) => msg.from().map(|user| user.id).is_some(),
Self::None => false,
}
}
pub fn get_id(&self) -> Option<i64> { pub fn get_id(&self) -> Option<i64> {
match self { match self {
Self::Id(id) => Some(*id), Self::Id(id) => Some(*id),

View File

@ -1,11 +1,11 @@
use telers::types::{chat_member::ChatMember, User}; use telers::types::chat_member::ChatMember;
pub fn is_admin(all_admin_members: &Vec<ChatMember>, user: &User) -> bool { pub fn is_admin(all_admin_members: &Vec<ChatMember>, user_id: i64) -> bool {
all_admin_members all_admin_members
.iter() .iter()
.any(|admin: &ChatMember| match admin { .any(|admin: &ChatMember| match admin {
ChatMember::Administrator(admin) => &admin.user == user, ChatMember::Administrator(admin) => admin.user.id == user_id,
ChatMember::Owner(owner) => &owner.user == user, ChatMember::Owner(owner) => owner.user.id == user_id,
_ => false, _ => false,
}) })
} }

View File

@ -1,9 +1,9 @@
use telers::{errors::SessionErrorKind as ErrorKind, event::EventReturn, Bot}; use telers::{errors::SessionErrorKind as ErrorKind, event::EventReturn, types::ChatMember, Bot};
use crate::types::structs::message_sender::MessageSender; use crate::types::structs::message_sender::MessageSender;
use std::future::Future; use std::future::Future;
use super::demote::demote_user; use super::{admin_check::is_admin, demote::demote_user, get_all_admins::get_all_admins};
const DEMOTE_FAILURE_MESSAGE: &str = "Команда не может быть выполнена: \ const DEMOTE_FAILURE_MESSAGE: &str = "Команда не может быть выполнена: \
не удалось удалить административные привилегии пользователя."; не удалось удалить административные привилегии пользователя.";
@ -19,6 +19,18 @@ where
{ {
let (bot, user_id, chat_id): (&Bot, i64, i64) = demote_args; let (bot, user_id, chat_id): (&Bot, i64, i64) = demote_args;
let admins: Vec<ChatMember> = get_all_admins(bot, chat_id).await.unwrap();
if is_admin(&admins, user_id) {
MessageSender::builder(chat_id)
.text("Нельзя применить эту команду на администраторе или бота.")
.build()
.send(bot)
.await
.unwrap();
return Err(EventReturn::Cancel);
}
if future_callback().await.is_err() { if future_callback().await.is_err() {
if demote_user(bot, user_id, chat_id).await.is_err() { if demote_user(bot, user_id, chat_id).await.is_err() {
MessageSender::builder(chat_id) MessageSender::builder(chat_id)