OpenSource/teloxide
OpenSource/teloxide
1
0
Fork 0
mirror of https://github.com/teloxide/teloxide.git synced 2025-03-22 06:45:37 +01:00
Code Issues Projects Releases Packages Wiki Activity

fixed example admin_bot

Browse source
This commit is contained in:
p0lunin 2020-02-12 16:08:15 +02:00
parent ce48e211a2
commit 4058126665
1 changed files with 25 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
examples/admin_bot/src/main.rs
View file

@ -5,7 +5,7 @@ use teloxide::types::ChatPermissions;
type Ctx = DispatcherHandlerCtx<Message>; type Ctx = DispatcherHandlerCtx<Message>;
#[derive(BotCommand)] #[derive(BotCommand)]
#[command(rename = "lowercase", description = "use command in format /%command% %num% %unit%")] #[command(rename = "lowercase", description = "Use commands in format /%command% %num% %unit%")]
enum Command { enum Command {
#[command(description = "kick user from chat.")] #[command(description = "kick user from chat.")]
Kick, Kick,
@ -22,7 +22,7 @@ fn calc_restrict_time(num: i32, unit: &str) -> Result<i32, &str> {
"h"|"hours" => Ok(num * 3600), "h"|"hours" => Ok(num * 3600),
"m"|"minutes" => Ok(num * 60), "m"|"minutes" => Ok(num * 60),
"s"|"seconds" => Ok(num), "s"|"seconds" => Ok(num),
_ => Err("allowed units: *h*, *m*, *s*") _ => Err("allowed units: h, m, s")
} }
} }
@ -49,7 +49,7 @@ fn parse_time_restrict(args: Vec<&str>) -> Result<i32, &str> {
async fn handle_command(ctx: Ctx) -> Result<(), ()> { async fn handle_command(ctx: Ctx) -> Result<(), ()> {
if let Some(text) = ctx.update.text() { if let Some(text) = ctx.update.text() {
let (command, args): (Command, Vec<&str>) = Command::parse(text).unwrap_or((Command::Help, vec![])); let (command, args): (Command, Vec<&str>) = Command::parse(text).ok_or(())?;
match command { match command {
Command::Help => { Command::Help => {
@ -58,10 +58,12 @@ async fn handle_command(ctx: Ctx) -> Result<(), ()> {
Command::Kick => { Command::Kick => {
match ctx.update.reply_to_message() { match ctx.update.reply_to_message() {
Some(mes) => { Some(mes) => {
ctx.bot.unban_chat_member( if let Some(user) = mes.from() {
mes.chat_id(), ctx.bot.unban_chat_member(
mes.from().unwrap().id ctx.update.chat_id(),
).send().await; user.id
).send().await;
}
}, },
None => { None => {
ctx.reply_to("Use this command in reply to another message").send().await; ctx.reply_to("Use this command in reply to another message").send().await;
@ -72,14 +74,15 @@ async fn handle_command(ctx: Ctx) -> Result<(), ()> {
match ctx.update.reply_to_message() { match ctx.update.reply_to_message() {
Some(mes) => match parse_time_restrict(args) { Some(mes) => match parse_time_restrict(args) {
Ok(time) => { Ok(time) => {
dbg!(&ctx.update); if let Some(user) = mes.from() {
ctx.bot.kick_chat_member( ctx.bot.kick_chat_member(
mes.chat_id(), ctx.update.chat_id(),
mes.from().unwrap().id user.id
) )
.until_date(time) .until_date(ctx.update.date + time)
.send() .send()
.await; .await;
}
} }
Err(msg) => { Err(msg) => {
ctx.answer(msg).send().await; ctx.answer(msg).send().await;
@ -94,14 +97,16 @@ async fn handle_command(ctx: Ctx) -> Result<(), ()> {
match ctx.update.reply_to_message() { match ctx.update.reply_to_message() {
Some(mes) => match parse_time_restrict(args) { Some(mes) => match parse_time_restrict(args) {
Ok(time) => { Ok(time) => {
ctx.bot.restrict_chat_member( if let Some(user) = mes.from() {
mes.chat_id(), ctx.bot.restrict_chat_member(
mes.from().unwrap().id, ctx.update.chat_id(),
user.id,
ChatPermissions::default() ChatPermissions::default()
) )
.until_date(time) .until_date(ctx.update.date + time)
.send() .send()
.await; .await;
}
} }
Err(msg) => { Err(msg) => {
ctx.answer(msg).send().await; ctx.answer(msg).send().await;
@ -122,7 +127,7 @@ async fn handle_command(ctx: Ctx) -> Result<(), ()> {
async fn main() { async fn main() {
pretty_env_logger::init(); pretty_env_logger::init();
let bot = Bot::new("865293832:AAHD-ox6hi6Ws_pxBFb8VIp1uymHoMab2MM"); let bot = Bot::new("YourAwesomeToken");
Dispatcher::new(bot) Dispatcher::new(bot)
.message_handler(&handle_command) .message_handler(&handle_command)
.dispatch() .dispatch()