diff options
| author | Benedikt Peetz <benedikt.peetz@b-peetz.de> | 2024-08-24 11:24:17 +0200 |
|---|---|---|
| committer | Benedikt Peetz <benedikt.peetz@b-peetz.de> | 2024-08-24 11:45:25 +0200 |
| commit | 4eeb519ae5398b5b5f5c7d9938e3887cac2c9faf (patch) | |
| tree | 57340831cbb5d2f1842cf5c8a7c8fad4a00ba29d /src/videos | |
| parent | fix(contrib/config.toml): Correct typo in config key (diff) | |
| download | yt-4eeb519ae5398b5b5f5c7d9938e3887cac2c9faf.zip | |
feat(videos): Allow limiting the number of videos to show
Diffstat (limited to 'src/videos')
| -rw-r--r-- | src/videos/mod.rs | 7 |
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() |