Merge pull request #12 from OpenTabletDriver/feature/cmd/embed

Add embed slash command

Author: InfinityGhost

Cargo.lock

@@ -19,3 +19,4 @@ tokio = { version = "1.21.2", features = ["macros", "signal", "rt-multi-thread"]
 regex = "1.6.0"
 octocrab = "0.17.0"
 reqwest = "0.11.12"
+hex = "0.4.3"

Cargo.toml

@@ -13,6 +13,7 @@ use std::collections::HashMap;
 use crate::structures::State;
 
 mod snippets;
+mod utils;
 
 pub async fn register(ctx: &Context) -> ApplicationCommandMap {
   println!("Registering slash commands...");
@@ -32,13 +33,24 @@ pub async fn register(ctx: &Context) -> ApplicationCommandMap {
 }
 
 pub async fn interact(ctx: &Context, interaction: &ApplicationCommandInteraction) {
-  match interaction.data.name.as_str() {
+  let name = &interaction.data.name;
+
+  match name.as_str() {
     "snippet" => snippets::snippet(ctx, interaction).await,
     "create-snippet" => snippets::create_snippet(ctx, interaction).await,
     "edit-snippet" => snippets::edit_snippet(ctx, interaction).await,
     "remove-snippet" => snippets::remove_snippet(ctx, interaction).await,
     "export-snippet" => snippets::export_snippet(ctx, interaction).await,
-    _ => println!("WARNING: Received invalid application command interaction!: {}", interaction.data.name)
+    "embed" => utils::embed(ctx, interaction).await,
+    _ => {
+      println!("WARNING: Received invalid application command interaction!: {}", name);
+
+      interaction.defer(ctx).await.expect("Failed to defer interaction");
+
+      let title = "Invalid application command";
+      let content = &format!("An invalid application command was recieved: {}", name);
+      respond_err(ctx, interaction, title, content).await;
+    }
   }
 }
 
@@ -98,13 +110,49 @@ impl ApplicationCommandMap {
       .description("Exports a snippet for user editing")
       .clone();
 
+    let embed = CreateApplicationCommand::default()
+      .description("Creates an embed in the current channel")
+      .create_option(|o| o
+        .name("title")
+        .description("The embed title")
+        .kind(CommandOptionType::String)
+      )
+      .create_option(|o| o
+        .name("description")
+        .description("The embed description")
+        .kind(CommandOptionType::String)
+      )
+      .create_option(|o| o
+        .name("color")
+        .description("The color of the embed in hexadecimal form. (ex: #ff00ff)")
+        .kind(CommandOptionType::String)
+      )
+      .create_option(|o| o
+        .name("url")
+        .description("The embed url")
+        .kind(CommandOptionType::String)
+      )
+      .create_option(|o| o
+        .name("footer")
+        .description("The embed footer text")
+        .kind(CommandOptionType::String)
+      )
+      .create_option(|o| o
+        .name("image")
+        .description("The image url for the embed")
+        .kind(CommandOptionType::String)
+      )
+      .clone();
+
+
     let mut commands = ApplicationCommandMap(CommandHashMap::new());
 
     commands.insert("snippet", snippet);
     commands.insert("create-snippet", create_snippet);
     commands.insert("edit-snippet", edit_snippet);
     commands.insert("remove-snippet", remove_snippet);
     commands.insert("export-snippet", export_snippet);
+    commands.insert("embed", embed);
 
     for (name, command) in commands.0.iter_mut() {
       match *name {

src/commands/mod.rs

@@ -0,0 +1,83 @@
+use serenity::builder::CreateEmbed;
+use serenity::model::prelude::interaction::application_command::{ApplicationCommandInteraction, CommandDataOptionValue};
+use serenity::prelude::Context;
+use serenity::utils::Colour;
+
+use super::{arg_opt, respond_err, respond_embed};
+
+pub(super) async fn embed(ctx: &Context, interaction: &ApplicationCommandInteraction) {
+  let title = arg_opt(interaction, "title");
+  let description = arg_opt(interaction, "description");
+  let color = arg_opt(interaction, "color");
+  let url = arg_opt(interaction, "url");
+  let footer_text = arg_opt(interaction, "footer");
+  let image = arg_opt(interaction, "image");
+
+  interaction.defer(ctx).await.expect("Failed to defer interaction");
+
+  let mut embed = CreateEmbed::default();
+
+  if let Some(CommandDataOptionValue::String(title)) = title {
+    embed.title(title);
+  }
+
+  if let Some(CommandDataOptionValue::String(description)) = description {
+    embed.description(description);
+  }
+
+  if let Some(CommandDataOptionValue::String(color)) = &color {
+    match hex::decode(color.to_ascii_lowercase().replace("#",  "")) {
+      Ok(hex_arr) => {
+        embed.color(Colour::from_rgb(hex_arr[0], hex_arr[1], hex_arr[2]));
+      },
+      Err(e) => {
+        let title = "Invalid color provided";
+        let content = &format!("The color '{}' is not a valid hexadecimal color: {}", &color, e);
+        return respond_err(ctx, interaction, title, content).await
+      }
+    }
+  }
+
+  if let Some(CommandDataOptionValue::String(url)) = url {
+    match url.parse::<reqwest::Url>() {
+      Ok(_) => {
+        if embed.0.contains_key("title") {
+          embed.url(url);
+        } else {
+          let title = "Invalid parameters";
+          let content = "A title is required for a url to function";
+          return respond_err(ctx, interaction, title, content).await
+        }
+      },
+      Err(e) => {
+        let title = "Invalid url provided";
+        let content = &format!("The url '{}' is not a valid url: {}", url, e);
+        return respond_err(ctx, interaction, title, content).await
+      }
+    }
+  }
+
+  if let Some(CommandDataOptionValue::String(footer_text)) = footer_text {
+    embed.footer(|f| f.text(footer_text));
+  }
+
+  if let Some(CommandDataOptionValue::String(image)) = image {
+    match image.parse::<reqwest::Url>() {
+      Ok(_) => {
+        embed.image(image);
+      },
+      Err(e) => {
+        let title = "Invalid image url provided";
+        let content = &format!("The image url '{}' is not a valid image url: {}", image, e);
+
+        return respond_err(ctx, interaction, title, content).await
+      }
+    }
+  }
+
+  if embed.0.contains_key("title") || embed.0.contains_key("description") || embed.0.contains_key("footer") {
+    respond_embed(ctx, interaction, &embed, false).await
+  } else {
+    respond_err(ctx, interaction, "Failed to respond with embed", "Embed does not have any content").await
+  }
+}