cleaner job send function

2026-03-14 00:07:06 +01:00
parent 8ed01f2049
commit b63982714b
13 changed files with 186 additions and 89 deletions
@@ -1,16 +1,18 @@
-use std::path::PathBuf;
-
-use postcard::{from_bytes, to_stdvec};
+use nats::functions::JobClient;
 use serenity::{
    all::{
-        CommandInteraction, Context, CreateCommandOption, CreateInteractionResponse,
-        CreateInteractionResponseMessage, EditInteractionResponse,
+        CommandInteraction,
+        Context,
+        CreateCommandOption,
+        CreateInteractionResponse,
+        CreateInteractionResponseMessage,
+        EditInteractionResponse,
    },
    builder::CreateCommand,
    model::application::{CommandOptionType, ResolvedOption, ResolvedValue},
 };
 use types::{
-    jobs::{DownloadJob, JobResponse, Jobs, PlayJob, SearchJob},
+    jobs::{DownloadJob, JobsMap, JobsResponseMap, PlayJob, SearchJob},
    misc::{new_uuid_v4, parse_url_or_default},
 };
 use url::Url;
@@ -27,26 +29,85 @@ pub async fn run(
        ..
    }) = options.first()
    {
+        interaction
+            .create_response(
+                ctx,
+                CreateInteractionResponse::Message(
+                    CreateInteractionResponseMessage::new()
+                        .content(format!("Searching: {value}...")),
+                ),
+            )
+            .await?;
+
        let url: Url;
        let is_url = value.starts_with("https://") || value.starts_with("http://");

        if !is_url {
-            let response = match nats_client
-                .request(
-                    "corro-dj.search",
-                    to_stdvec(&SearchJob {
-                        uuid: new_uuid_v4(),
-                        query: value.to_string(),
-                    })
-                    .unwrap()
-                    .into(),
-                )
+            let search_response = match nats_client
+                .send_job(JobsMap::Search(SearchJob {
+                    uuid: new_uuid_v4(),
+                    query: value.to_string(),
+                }))
                .await
            {
-                Ok(resp) => resp,
+                Ok(resp) => match resp {
+                    JobsResponseMap::Search(resp) => resp,
+                    _ => return Err(serenity::Error::Other("Unexpected return type")),
+                },
                Err(_why) => return Err(serenity::Error::Other("send error")),
            };
+
+            println!("{:?}", &search_response);
+
+            url = search_response.song.url;
+        } else {
+            url = parse_url_or_default(value.to_string());
        }
+
+        let download_response = match nats_client
+            .send_job(JobsMap::Download(DownloadJob {
+                uuid: new_uuid_v4(),
+                url,
+            }))
+            .await
+        {
+            Ok(resp) => match resp {
+                JobsResponseMap::Download(resp) => resp,
+                _ => return Err(serenity::Error::Other("Unexpected return type")),
+            },
+            Err(_why) => return Err(serenity::Error::Other("send error")),
+        };
+
+        println!("{:?}", &download_response);
+
+        let guild_id = interaction.guild_id.unwrap();
+
+        let channel_id = guild_id
+            .get_user_voice_state(&ctx.http, interaction.user.id)
+            .await
+            .unwrap()
+            .channel_id
+            .unwrap();
+
+        let _ = match nats_client
+            .send_job(JobsMap::Play(PlayJob {
+                uuid: new_uuid_v4(),
+                path: download_response.path,
+                channel_id: channel_id.into(),
+                guild_id: guild_id.into(),
+            }))
+            .await
+        {
+            Ok(resp) => match resp {
+                JobsResponseMap::Play(resp) => resp,
+                _ => return Err(serenity::Error::Other("Unexpected return type")),
+            },
+            Err(_why) => return Err(serenity::Error::Other("send error")),
+        };
+
+        interaction
+            .edit_response(ctx, EditInteractionResponse::new().content("Playing..."))
+            .await?;
    }
    Ok(())
 }
@@ -60,6 +121,6 @@ pub fn register() -> CreateCommand {
                "song",
                "Name or url of the song to play",
            )
-            .required(false),
+            .required(true),
        )
 }