Adding play command

src/commands/audio/mod.rs

@@ -1,6 +1,84 @@
+use log::debug;
+
+use serenity::model::application::interaction::{InteractionResponseType, application_command::ApplicationCommandInteraction};
+use serenity::model::prelude::{GuildId, ChannelId};
+use serenity::model::user::User;
+use serenity::prelude::*;
+use songbird::id::GuildId as SongbirdGuildId;
+
 pub mod pause;
 pub mod play;
 pub mod resume;
 pub mod skip;
 pub mod stop;
-pub mod volume;
+pub mod volume;
+
+pub async fn join(ctx: &Context, guild_id_option: &Option<GuildId>, user: &User) -> Result<(), String> {
+  let guild_id = match guild_id_option {
+    Some(g) => g,
+    None => {
+      return Err(format!("{}", "No guild ID set"));
+    }
+  };
+
+  let channel_id = match find_voice_channel(&ctx, &guild_id, &user) {
+    Ok(channel) => channel,
+    Err(err) => return Err(format!("{}", err))
+  };
+
+  debug!("<{}> Joining channel {}", guild_id.0, channel_id);
+  let manager = songbird::get(ctx).await.expect("Songbird Voice client placed in at initialization").clone();
+  let (_handle_lock, success) = manager.join(guild_id.to_owned(), channel_id.to_owned()).await;
+  match success {
+    Ok(s) => Ok(s),
+    Err(err) => Err(format!("{}", err))
+  }
+}
+
+pub async fn leave(ctx: &Context, guild_id_option: &Option<GuildId>) -> Result<(), String> {
+  let guild_id = match guild_id_option {
+    Some(g) => g,
+    None => {
+      return Err(format!("{}", "No guild ID set"));
+    }
+  };
+
+  let songbird_guild_id = SongbirdGuildId {
+    0: guild_id.0
+  };
+
+  let manager = songbird::get(ctx).await.expect("Songbird Voice client placed in at initialization").clone();
+  if manager.get(songbird_guild_id).is_some() {
+    debug!("<{}> Disconnecting from channel", guild_id.0);
+    if let Err(e) = manager.remove(songbird_guild_id).await {
+      return Err(format!("{}", e))
+    }
+  }
+  Ok(())
+}
+
+fn find_voice_channel(ctx: &Context, guild_id: &GuildId, user: &User) -> Result<ChannelId, String> {
+  let guild = match guild_id.to_guild_cached(ctx.cache.to_owned()) {
+    Some(g) => g,
+    None => return Err(format!("Guild not found"))
+  };
+
+  match guild.voice_states.get(&user.id).and_then(|voice_state| voice_state.channel_id) {
+    Some(channel) => Ok(channel),
+    None => return Err(format!("User is not in a voice channel"))
+  }
+}
+
+pub async fn create_response(ctx: &Context, command: &ApplicationCommandInteraction, content: String) -> Result<(), SerenityError> {
+  command.create_interaction_response(&ctx.http, |response: &mut serenity::builder::CreateInteractionResponse<'_>| {
+      response
+        .kind(InteractionResponseType::ChannelMessageWithSource)
+        .interaction_response_data(|message: &mut serenity::builder::CreateInteractionResponseData<'_>| message.content(content))
+    }).await
+}
+
+pub async fn edit_response(ctx: &Context, command: &ApplicationCommandInteraction, content: String) -> Result<serenity::model::channel::Message, SerenityError> {
+  command.edit_original_interaction_response(&ctx.http, |response: &mut serenity::builder::EditInteractionResponse| {
+    response.content(content)
+  }).await
+}