Clean up define command

This commit is contained in:
charlotte ✨ 2021-07-08 00:18:53 +01:00
parent 6a615bea42
commit 82d017b00f
1 changed files with 26 additions and 30 deletions

View File

@ -1,5 +1,5 @@
use serenity::{
framework::standard::{macros::command, Args, CommandResult},
framework::standard::{macros::command, Args, CommandError, CommandResult},
model::channel::Message,
prelude::*,
};
@ -8,16 +8,18 @@ use serenity::{
#[command]
#[aliases("what's this")]
async fn define(ctx: &Context, message: &Message, args: Args) -> CommandResult {
let text: String = args.rest().trim().to_string();
let defs = &urbandict::get_definitions(&text);
if !args.is_empty() {
match defs {
Err(_e) => {
return Err("Invalid query >w<".into());
if args.is_empty() {
return Err("No arguments!".into());
}
Ok(v) => {
if !v.is_empty() {
let def = &v[0];
let text: String = args.rest().trim().to_string();
let defs =
urbandict::get_definitions(&text).map_err(|_| CommandError::from("Invalid query >w<"))?;
let def = defs
.first()
.ok_or_else(|| CommandError::from("No results!"))?;
let _ = message
.channel_id
.send_message(&ctx.http, |m| {
@ -32,12 +34,6 @@ async fn define(ctx: &Context, message: &Message, args: Args) -> CommandResult {
})
})
.await;
} else {
return Err("No results!".into());
}
}
}
}
Ok(())
}