about summary refs log tree commit diff stats
path: root/src/videos/mod.rs
diff options
context:
space:
mode:
authorBenedikt Peetz <benedikt.peetz@b-peetz.de>2024-08-24 11:24:17 +0200
committerBenedikt Peetz <benedikt.peetz@b-peetz.de>2024-08-24 11:45:25 +0200
commit4eeb519ae5398b5b5f5c7d9938e3887cac2c9faf (patch)
tree57340831cbb5d2f1842cf5c8a7c8fad4a00ba29d /src/videos/mod.rs
parentfix(contrib/config.toml): Correct typo in config key (diff)
downloadyt-4eeb519ae5398b5b5f5c7d9938e3887cac2c9faf.tar.gz
yt-4eeb519ae5398b5b5f5c7d9938e3887cac2c9faf.zip
feat(videos): Allow limiting the number of videos to show
Diffstat (limited to '')
-rw-r--r--src/videos/mod.rs7
1 files changed, 5 insertions, 2 deletions
diff --git a/src/videos/mod.rs b/src/videos/mod.rs
index 5bf34e3..b51e469 100644
--- a/src/videos/mod.rs
+++ b/src/videos/mod.rs
@@ -20,16 +20,19 @@ use crate::{
     storage::video_database::{getters::get_videos, VideoStatus},
 };
 
-pub async fn query(app: &App, search_query: Option<String>) -> Result<()> {
-    let all_videos = get_videos(app, &VideoStatus::ALL, None).await?;
+pub async fn query(app: &App, limit: Option<usize>, search_query: Option<String>) -> Result<()> {
+    let all_videos = get_videos(app, VideoStatus::ALL, None).await?;
 
     // turn one video to a color display, to pre-warm the hash shrinking cache
     if let Some(val) = all_videos.get(0) {
         val.to_color_display(app).await?;
     }
 
+    let limit = limit.unwrap_or(all_videos.len());
+
     let all_video_strings: Vec<String> = all_videos
         .into_iter()
+        .take(limit)
         .map(|vid| vid.to_color_display_owned(app))
         .collect::<FuturesUnordered<_>>()
         .try_collect()