search

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())