From 6a7bc688498cc07e7f13093abfc30e1428a7585e Mon Sep 17 00:00:00 2001
From: Simon <simon.huenecke@gmail.com>
Date: Thu, 15 Jan 2026 19:17:46 +0000
Subject: [PATCH] improved all provider

---
 src/providers/all.rs | 87 ++++++++++++++++++++++++++------------------
 src/util/discord.rs  |  8 ++--
 2 files changed, 56 insertions(+), 39 deletions(-)

diff --git a/src/providers/all.rs b/src/providers/all.rs
index 0855532..1a00722 100644
--- a/src/providers/all.rs
+++ b/src/providers/all.rs
@@ -1,7 +1,9 @@
 use std::fs;
+use std::time::Duration;
 use async_trait::async_trait;
 use error_chain::error_chain;
-use futures::future::join_all;
+use futures::StreamExt;
+use futures::stream::FuturesUnordered;
 use crate::api::{get_provider, ClientVersion};
 use crate::providers::{DynProvider, Provider};
 use crate::status::Channel;
@@ -41,7 +43,7 @@ impl Provider for AllProvider {
         per_page: String,
         options: ServerOptions,
     ) -> Vec<VideoItem> {
-        let mut sites_str = options.clone().sites.unwrap();
+        let mut sites_str = options.clone().sites.unwrap_or_default();
         if sites_str.is_empty() {
             let files = fs::read_dir("./src/providers").unwrap();
             let providers = files.map(|entry| entry.unwrap().file_name())
@@ -51,14 +53,16 @@ impl Provider for AllProvider {
                 .collect::<Vec<String>>();
             sites_str = providers.join(",");
         }
+
         let providers: Vec<DynProvider> = sites_str
             .split(',')
             .filter(|s| !s.is_empty())
-            .filter_map(|s| get_provider(s)) // assumes get_provider -> Option<DynProvider>
+            .filter_map(|s| get_provider(s))
             .collect();
-       
-        let futures = providers.iter().map(|provider| {
-            let provider = provider.clone();
+
+        let mut futures = FuturesUnordered::new();
+
+        for provider in providers {
             let cache = cache.clone();
             let pool = pool.clone();
             let sort = sort.clone();
@@ -66,39 +70,52 @@ impl Provider for AllProvider {
             let page = page.clone();
             let per_page = per_page.clone();
             let options = options.clone();
-            async move {
-            match tokio::time::timeout(
-                std::time::Duration::from_secs(55),
-                provider.get_videos(
-                cache,
-                pool,
-                sort,
-                query,
-                page,
-                per_page,
-                options,
-                ),
-            )
-            .await
-            {
-                Ok(v) => v,
-                Err(_) => {
-                // timed out -> return empty result for this provider
-                vec![]
-                }
+
+            // Spawn the task so it lives independently of this function
+            futures.push(tokio::spawn(async move {
+                provider.get_videos(cache, pool, sort, query, page, per_page, options).await
+            }));
+        }
+
+        let mut all_results = Vec::new();
+        let timeout_timer = tokio::time::sleep(Duration::from_secs(10));
+        tokio::pin!(timeout_timer);
+
+        // Collect what we can within 55 seconds
+        loop {
+            tokio::select! {
+                Some(result) = futures.next() => {
+                    // Ignore errors (panics or task cancellations)
+                    if let Ok(videos) = result {
+                        all_results.push(videos);
+                    }
+                },
+                _ = &mut timeout_timer => {
+                    // 55 seconds passed. Stop waiting and return what we have.
+                    // The tasks remaining in 'futures' will continue running in the 
+                    // background because they were 'tokio::spawn'ed.
+                    break;
+                },
+                else => break, // All tasks finished before the timeout
             }
-            }
-        }).collect::<Vec<_>>();
-        let results:Vec<Vec<VideoItem>> = join_all(futures).await;
-        let video_items: Vec<VideoItem> = interleave(&results);
-        return video_items;
+        }
+
+        interleave(&all_results)
     }
-    
-    fn get_channel(&self,clientversion:ClientVersion) -> Option<Channel> {
-        println!("Getting channel for placeholder with client version: {:?}",clientversion);
+
+    fn get_channel(&self, clientversion: ClientVersion) -> Option<Channel> {
         let _ = clientversion;
         Some(Channel {
-            id:"placeholder".to_string(),name:"PLACEHOLDER".to_string(),description:"PLACEHOLDER FOR PARENT CLASS".to_string(),premium:false,favicon:"https://www.google.com/s2/favicons?sz=64&domain=missav.ws".to_string(),status:"active".to_string(),categories:vec![],options:vec![],nsfw:true,cacheDuration:None,
+            id: "placeholder".to_string(),
+            name: "PLACEHOLDER".to_string(),
+            description: "PLACEHOLDER FOR PARENT CLASS".to_string(),
+            premium: false,
+            favicon: "https://hottub.spacemoehre.de/favicon.ico".to_string(),
+            status: "active".to_string(),
+            categories: vec![],
+            options: vec![],
+            nsfw: true,
+            cacheDuration: None,
         })
     }
 }
diff --git a/src/util/discord.rs b/src/util/discord.rs
index dd50ed5..091cfee 100644
--- a/src/util/discord.rs
+++ b/src/util/discord.rs
@@ -8,7 +8,7 @@ use crate::util::requester;
 
 // Global cache: Map<ErrorSignature, LastSentTimestamp>
 static ERROR_CACHE: Lazy<DashMap<String, u64>> = Lazy::new(DashMap::new);
-const COOLDOWN_SECONDS: u64 = 3600; // 1 Hour cooldown
+// const COOLDOWN_SECONDS: u64 = 3600; // 1 Hour cooldown
 
 pub fn format_error_chain(err: &dyn Error) -> String {
     let mut chain_str = String::new();
@@ -40,11 +40,11 @@ pub async fn send_discord_error_report(
     // Create a unique key based on error content and location
     let error_signature = format!("{}-{}-{}", error_msg, file, line);
 
-    if let Some(last_sent) = ERROR_CACHE.get(&error_signature) {
-        if now - *last_sent < COOLDOWN_SECONDS {
+    if let Some(_) = ERROR_CACHE.get(&error_signature) {
+        // if now - *last_sent < COOLDOWN_SECONDS {
             // Error is still in cooldown, skip sending
             return;
-        }
+        // }
     }
     
     // Update the cache with the current timestamp