71 lines
2.2 KiB
Rust
71 lines
2.2 KiB
Rust
use async_trait::async_trait;
|
|
use reqwest::Method;
|
|
use serenity::builder::CreateApplicationCommand;
|
|
use serenity::model::interactions::application_command::{
|
|
ApplicationCommandInteraction, ApplicationCommandOptionType,
|
|
};
|
|
|
|
use geoffrey_models::models::locations::Location;
|
|
use geoffrey_models::models::parameters::info_params::InfoParams;
|
|
use geoffrey_models::models::response::api_error::GeoffreyAPIError;
|
|
|
|
use crate::bot::arg_parse::option_to_string;
|
|
use crate::bot::commands::BotCommand;
|
|
use crate::bot::formatters::GeoffreyFormatter;
|
|
use crate::bot::lang::NO_LOCATION_FOUND;
|
|
use crate::context::GeoffreyContext;
|
|
use crate::error::BotError;
|
|
|
|
pub struct InfoCommand;
|
|
|
|
#[async_trait]
|
|
impl BotCommand for InfoCommand {
|
|
type ApiParams = InfoParams;
|
|
type ApiResp = Location;
|
|
|
|
fn command_name() -> String {
|
|
"info".to_string()
|
|
}
|
|
|
|
fn request_type() -> Method {
|
|
Method::GET
|
|
}
|
|
|
|
fn custom_err_resp(err: &BotError) -> Option<String> {
|
|
match err {
|
|
BotError::GeoffreyApi(GeoffreyAPIError::EntryNotFound) => {
|
|
Some(NO_LOCATION_FOUND.to_string())
|
|
}
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
fn create_app_command(command: &mut CreateApplicationCommand) -> &mut CreateApplicationCommand {
|
|
command
|
|
.name(Self::command_name())
|
|
.description("Get info on a location in Geoffrey.")
|
|
.create_option(|option| {
|
|
option
|
|
.name("loc_name")
|
|
.description("The location to get info on")
|
|
.kind(ApplicationCommandOptionType::String)
|
|
.required(true)
|
|
})
|
|
}
|
|
|
|
async fn process_arguments(
|
|
command_interaction: ApplicationCommandInteraction,
|
|
) -> Result<Self::ApiParams, BotError> {
|
|
let options = command_interaction.data.options;
|
|
let location_name = option_to_string(options.get(0), "loc_name")?;
|
|
|
|
Ok(Self::ApiParams::new(location_name))
|
|
}
|
|
|
|
fn build_response(ctx: &GeoffreyContext, resp: Self::ApiResp, _: Self::ApiParams) -> String {
|
|
GeoffreyFormatter::new(ctx.settings.clone())
|
|
.push_loc_full(&resp)
|
|
.build()
|
|
}
|
|
}
|