search

apps/master/src/commands/mod.rs

@@ -1,3 +1,2 @@
-pub mod modal;
 pub mod ping;
 pub mod testnats;

apps/master/src/commands/modal.rs

@@ -1,28 +0,0 @@
-use serenity::{builder::*, model::prelude::*, prelude::*, utils::CreateQuickModal};
-
-pub async fn run(ctx: &Context, interaction: &CommandInteraction) -> Result<(), serenity::Error> {
-    let modal = CreateQuickModal::new("About you")
-        .timeout(std::time::Duration::from_secs(600))
-        .short_field("First name")
-        .short_field("Last name")
-        .paragraph_field("Hobbies and interests");
-    let response = interaction.quick_modal(ctx, modal).await?.unwrap();
-
-    let inputs = response.inputs;
-    let (first_name, last_name, hobbies) = (&inputs[0], &inputs[1], &inputs[2]);
-
-    response
-        .interaction
-        .create_response(
-            ctx,
-            CreateInteractionResponse::Message(CreateInteractionResponseMessage::new().content(
-                format!("**Name**: {first_name} {last_name}\n\nHobbies and interests: {hobbies}"),
-            )),
-        )
-        .await?;
-    Ok(())
-}
-
-pub fn register() -> CreateCommand {
-    CreateCommand::new("modal").description("Asks some details about you")
-}

apps/master/src/commands/ping.rs

@@ -1,7 +1,7 @@
 use serenity::{builder::CreateCommand, model::application::ResolvedOption};
 
 pub fn run(_options: &[ResolvedOption]) -> String {
-    "Hey, I'm alive!".to_string()
+    "Pong!".to_string()
 }
 
 pub fn register() -> CreateCommand {

apps/master/src/commands/testnats.rs

@@ -14,7 +14,7 @@ use serenity::{
     model::application::{CommandOptionType, ResolvedOption, ResolvedValue},
 };
 use types::{
-    jobs::{DownloadJob, JobResponse, Jobs, PlayJob},
+    jobs::{DownloadJob, JobResponse, Jobs, PlayJob, SearchJob},
     misc::new_uuid_v4,
 };
 
@@ -40,9 +40,44 @@ pub async fn run(
             )
             .await?;
 
+        let url: String;
+        let is_url = value.starts_with("https://");
+        if !is_url {
+            let search_job = SearchJob {
+                uuid: new_uuid_v4(),
+                query: value.to_string(),
+            };
+
+            let response = match nats_client
+                .request("corro-dj.search", to_stdvec(&search_job).unwrap().into())
+                .await
+            {
+                Ok(resp) => resp,
+                Err(_why) => return Err(serenity::Error::Other("send error")),
+            };
+
+            let search_response: JobResponse = from_bytes(&response.payload).unwrap();
+
+            if let Some(error) = search_response.error {
+                interaction
+                    .edit_response(ctx, EditInteractionResponse::new().content(error))
+                    .await?;
+                return Err(serenity::Error::Other("Search error"));
+            } else if let Some(Jobs::Search(content)) = search_response.content {
+                url = content.url;
+            } else {
+                interaction
+                    .edit_response(ctx, EditInteractionResponse::new().content("unknown error"))
+                    .await?;
+                return Err(serenity::Error::Other("unknown error"));
+            }
+        } else {
+            url = value.to_string();
+        }
+
         let download_job = DownloadJob {
             uuid: new_uuid_v4(),
-            url: value.to_string(),
+            url,
         };
 
         println!("job {:?}", download_job);
@@ -58,7 +93,7 @@ pub async fn run(
             Err(_why) => return Err(serenity::Error::Other("send error")),
         };
 
-        let job_response: JobResponse<Jobs> = from_bytes(&response.payload).unwrap();
+        let job_response: JobResponse = from_bytes(&response.payload).unwrap();
 
         println!("response: {:?}", job_response);
 

apps/master/src/main.rs

@@ -31,10 +31,6 @@ impl EventHandler for Handler {
                     None
                 }
                 "ping" => Some(commands::ping::run(&command.data.options())),
-                "modal" => {
-                    commands::modal::run(&ctx, &command).await.unwrap();
-                    None
-                }
                 _ => Some("not implemented :(".to_string()),
             };
 
@@ -53,11 +49,7 @@ impl EventHandler for Handler {
 
         if let Err(why) = Command::set_global_commands(
             &ctx.http,
-            vec![
-                commands::ping::register(),
-                commands::modal::register(),
-                commands::testnats::register(),
-            ],
+            vec![commands::ping::register(), commands::testnats::register()],
         )
         .await
         {

apps/worker/Cargo.toml

@@ -8,7 +8,7 @@ async-nats = { version = "0.46.0" }
 futures = { version = "0.3.32" }
 futures-executor = { version = "0.3.32" }
 postcard = { version = "1.1.3", features = ["use-std"] }
-rustls = { version = "0.23.37", default-features = false, features = ["ring"] }
+rustls = { version = "0.23.37", default-features = false, features = ["aws-lc-rs"] }
 serenity = { version = "0.12.5", default-features = false, features = [
     "cache",
     "rustls_backend",
@@ -18,5 +18,5 @@ songbird = { git = "https://github.com/beerpsi-forks/songbird.git", branch = "da
 symphonia = { version = "0.5.5" }
 tokio = { version = "1.50.0", features = ["macros", "rt-multi-thread"] }
 types = { path = "../../libs/types" }
-which = { version = "8.0.1" }
-yt-dlp = { version = "2.4.0" }
+which = { version = "8.0.2" }
+yt-dlp = { version = "2.5.0" }

apps/worker/src/main.rs

@@ -16,7 +16,7 @@ use yt_dlp::{Downloader, client::Libraries};
 
 struct Handler {
     nats_client: async_nats::Client,
-    yt_downloader: Downloader,
+    downloader: Downloader,
 }
 
 #[async_trait]
@@ -40,7 +40,7 @@ impl EventHandler for Handler {
 
             let result = match subject {
                 "download" => workers::download::download(
-                    &self.yt_downloader,
+                    &self.downloader,
                     from_bytes(&message.payload).unwrap(),
                 )
                 .await
@@ -56,6 +56,14 @@ impl EventHandler for Handler {
                             error: res.error,
                         })
                 }
+                "search" => {
+                    workers::search::search(&self.downloader, from_bytes(&message.payload).unwrap())
+                        .await
+                        .map(|res| JobResponse {
+                            content: res.content.map(Jobs::Search),
+                            error: res.error,
+                        })
+                }
                 _ => Err(format!("subject {subject} does not exists")),
             };
 
@@ -81,7 +89,7 @@ impl EventHandler for Handler {
 
 #[tokio::main]
 async fn main() {
-    rustls::crypto::ring::default_provider()
+    rustls::crypto::aws_lc_rs::default_provider()
         .install_default()
         .expect("Failed to install rustls crypto provider");
 
@@ -91,7 +99,7 @@ async fn main() {
         .await
         .expect("Error creating nats client");
 
-    let yt_downloader = Downloader::builder(
+    let downloader = Downloader::builder(
         Libraries::new(which("yt-dlp").unwrap(), which("ffmpeg").unwrap()),
         "output",
     )
@@ -101,7 +109,7 @@ async fn main() {
 
     let handler = Handler {
         nats_client,
-        yt_downloader,
+        downloader,
     };
 
     let mut discord_client = Client::builder(&discord_token, GatewayIntents::non_privileged())

apps/worker/src/workers/download.rs

@@ -1,5 +1,8 @@
 use types::jobs::{DownloadJob, DownloadResponse, JobResponse};
-use yt_dlp::Downloader;
+use yt_dlp::{
+    Downloader,
+    model::{AudioCodecPreference, AudioQuality},
+};
 
 pub async fn download(
     downloader: &Downloader,
@@ -16,8 +19,8 @@ pub async fn download(
         .download_audio_stream_with_quality(
             &video,
             format!("{}.ogg", video.id),
-            yt_dlp::model::AudioQuality::Best,
-            yt_dlp::model::AudioCodecPreference::Opus,
+            AudioQuality::Best,
+            AudioCodecPreference::Opus,
         )
         .await
     {

apps/worker/src/workers/mod.rs

@@ -1,2 +1,3 @@
 pub mod download;
 pub mod play;
+pub mod search;

apps/worker/src/workers/search.rs

@@ -0,0 +1,34 @@
+use types::jobs::{JobResponse, SearchJob, SearchResponse};
+use yt_dlp::Downloader;
+
+pub async fn search(
+    downloader: &Downloader,
+    job: SearchJob,
+) -> Result<JobResponse<SearchResponse>, String> {
+    println!("job: {:?}", job);
+
+    let result = match downloader
+        .youtube_extractor()
+        .search_first(&job.query)
+        .await
+    {
+        Ok(result) => match result.webpage_url {
+            Some(url) => JobResponse {
+                content: Some(SearchResponse { url }),
+                error: None,
+            },
+            None => JobResponse {
+                content: None,
+                error: Some("url is not defined".to_string()),
+            },
+        },
+        Err(why) => JobResponse {
+            content: None,
+            error: Some(format!("{why}")),
+        },
+    };
+
+    println!("reply: {:?}", result.content);
+
+    Ok(result)
+}