v0.2.6 Working on audio commands from api

service/src/bot/api/routes.rs

@@ -1,14 +1,16 @@
-use std::sync::Arc;
+use std::{sync::Arc, pin::Pin};
 
 use actix_web::{get, post, put, delete, web, HttpResponse, HttpRequest, ResponseError};
 use log::warn;
 use serde::{Serialize, Deserialize};
-use serenity::{http::Http, model::prelude::{GuildChannel, ChannelType}};
+use serenity::model::prelude::{GuildChannel, ChannelType};
 use siren::ServiceError;
 
+use crate::AppState;
+
 #[get("/guilds")]
-async fn get_guilds(data: web::Data<Arc<Http>>) -> HttpResponse {
-  let guild_results = &data.get_guilds(None, None).await;
+async fn get_guilds(data: web::Data<Arc<AppState>>) -> HttpResponse {
+  let guild_results = &data.http.get_guilds(None, None).await;
   let guilds = match guild_results {
     Ok(guilds) => guilds,
     Err(err) => return ResponseError::error_response(&ServiceError {
@@ -20,8 +22,8 @@ async fn get_guilds(data: web::Data<Arc<Http>>) -> HttpResponse {
 }
 
 #[get("/{id}/text")]
-async fn get_text_channels(id: web::Path<String>, data: web::Data<Arc<Http>>) -> HttpResponse {
-  let channel_results = &data.get_channels(id.parse::<u64>().unwrap()).await;
+async fn get_text_channels(id: web::Path<String>, data: web::Data<Arc<AppState>>) -> HttpResponse {
+  let channel_results = &data.http.get_channels(id.parse::<u64>().unwrap()).await;
   let channels = match channel_results {
     Ok(channels) => channels.iter().filter(|c| c.kind == ChannelType::Text).collect::<Vec<&GuildChannel>>(),
     Err(err) => return ResponseError::error_response(&ServiceError {
@@ -33,8 +35,8 @@ async fn get_text_channels(id: web::Path<String>, data: web::Data<Arc<Http>>) ->
 }
 
 #[get("/{id}/voice")]
-async fn get_voice_channels(id: web::Path<String>, data: web::Data<Arc<Http>>) -> HttpResponse {
-  let channel_results = &data.get_channels(id.parse::<u64>().unwrap()).await;
+async fn get_voice_channels(id: web::Path<String>, data: web::Data<Arc<AppState>>) -> HttpResponse {
+  let channel_results = &data.http.get_channels(id.parse::<u64>().unwrap()).await;
   let channels = match channel_results {
     Ok(channels) => channels.iter().filter(|c| c.kind == ChannelType::Voice).collect::<Vec<&GuildChannel>>(),
     Err(err) => return ResponseError::error_response(&ServiceError {
@@ -51,7 +53,7 @@ struct ChannelMessage {
 }
 
 #[post("/{guild_id}/text/{channel_id}/message")]
-async fn send_message(path: web::Path<(String, String)>, text: web::Json<ChannelMessage>, data: web::Data<Arc<Http>>) -> HttpResponse {
+async fn send_message(path: web::Path<(String, String)>, text: web::Json<ChannelMessage>, data: web::Data<Arc<AppState>>) -> HttpResponse {
   let (guild_id, channel_id) = path.into_inner();
   let guild_id = match guild_id.parse::<u64>() {
     Ok(id) => id,
@@ -73,7 +75,7 @@ async fn send_message(path: web::Path<(String, String)>, text: web::Json<Channel
       })
     }
   };
-  let channel_results = &data.get_channels(guild_id).await;
+  let channel_results = &data.http.get_channels(guild_id).await;
   let channels = match channel_results {
     Ok(channels) => channels,
     Err(err) => {
@@ -96,7 +98,7 @@ async fn send_message(path: web::Path<(String, String)>, text: web::Json<Channel
     }
   };
 
-  if let Err(err) = channel.say(&data.get_ref(), &text.message).await {
+  if let Err(err) = channel.say(&Pin::new(&data.http).get_ref(), &text.message).await {
     warn!("Could not send message: {:?}", err);
     return ResponseError::error_response(&ServiceError {
       status: 422,
@@ -107,6 +109,62 @@ async fn send_message(path: web::Path<(String, String)>, text: web::Json<Channel
   HttpResponse::Ok().finish()
 }
 
+#[post("/{guild_id}/voice/play")]
+async fn play(path: web::Path<String>, data: web::Data<Arc<AppState>>) -> HttpResponse {
+  let guild_id = path.into_inner();
+  let guild_id = match guild_id.parse::<u64>() {
+    Ok(id) => id,
+    Err(err) => {
+      warn!("Could not parse guild id: {:?}", err);
+      return ResponseError::error_response(&ServiceError {
+        status: 422,
+        message: err.to_string()
+      })
+    }
+  };
+
+  if let Some(handler_lock) = data.songbird.get(guild_id) {
+    let handler = handler_lock.lock().await;
+    if let Err(err) = handler.queue().pause() {
+      warn!("Could not pause track: {:?}", err);
+      return ResponseError::error_response(&ServiceError {
+        status: 422,
+        message: err.to_string()
+      })
+    }
+  }
+
+  HttpResponse::Ok().finish()
+}
+
+#[post("/{guild_id}/voice/pause")]
+async fn pause(path: web::Path<String>, data: web::Data<Arc<AppState>>) -> HttpResponse {
+  let guild_id = path.into_inner();
+  let guild_id = match guild_id.parse::<u64>() {
+    Ok(id) => id,
+    Err(err) => {
+      warn!("Could not parse guild id: {:?}", err);
+      return ResponseError::error_response(&ServiceError {
+        status: 422,
+        message: err.to_string()
+      })
+    }
+  };
+
+  if let Some(handler_lock) = data.songbird.get(guild_id) {
+    let handler = handler_lock.lock().await;
+    if let Err(err) = handler.queue().pause() {
+      warn!("Could not pause track: {:?}", err);
+      return ResponseError::error_response(&ServiceError {
+        status: 422,
+        message: err.to_string()
+      })
+    }
+  }
+
+  HttpResponse::Ok().finish()
+}
+
 pub fn init_routes(config: &mut web::ServiceConfig) {
   config
     .service(get_guilds)
@@ -114,5 +172,6 @@ pub fn init_routes(config: &mut web::ServiceConfig) {
       .service(get_text_channels)
       .service(get_voice_channels)
       .service(send_message)
+      .service(pause)
   );
 }