jmarya/hoard
jmarya/hoard
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

add limit option

Browse source
This commit is contained in:
JMARyA 2024-03-07 19:14:50 +01:00
parent e62cce11f2
commit 2f2ac4fb70
Signed by: jmarya
GPG key ID: 901B2ADDF27C2263
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/youtube/mod.rs
View file

@ -14,6 +14,8 @@ use crate::{ensure_dir_exists, Module};
pub struct YouTubeConfig { pub struct YouTubeConfig {
// Interval in minutes between checks // Interval in minutes between checks
interval: u64, interval: u64,
/// Amount of videos to query
limit: Option<u64>,
// Channels to check // Channels to check
channels: HashMap<String, String>, channels: HashMap<String, String>,
// Format of the Thumbnail // Format of the Thumbnail
@ -58,7 +60,10 @@ impl Module for YouTubeModule {
let download_options = self.config.download_options(); let download_options = self.config.download_options();
for (channel, channel_url) in &self.config.channels { for (channel, channel_url) in &self.config.channels {
log::info!("Fetching {channel} videos"); log::info!("Fetching {channel} videos");
match Self::get_latest_channel_videos(channel_url) { match Self::get_latest_channel_videos(
channel_url,
&self.config.limit.unwrap_or(10).to_string(),
) {
Ok(latest_videos) => { Ok(latest_videos) => {
for (video_title, video_url) in latest_videos { for (video_title, video_url) in latest_videos {
if self.db.check_for_url(&video_url).unwrap() { if self.db.check_for_url(&video_url).unwrap() {
@ -92,7 +97,10 @@ impl Module for YouTubeModule {
} }
impl YouTubeModule { impl YouTubeModule {
fn get_latest_channel_videos(channel: &str) -> Result<Vec<(String, String)>, String> { fn get_latest_channel_videos(
channel: &str,
limit: &str,
) -> Result<Vec<(String, String)>, String> {
let output = Command::new("yt-dlp") let output = Command::new("yt-dlp")
.arg("--no-warnings") .arg("--no-warnings")
.arg("--flat-playlist") .arg("--flat-playlist")
@ -100,7 +108,7 @@ impl YouTubeModule {
.arg("--print") .arg("--print")
.arg("title,webpage_url") .arg("title,webpage_url")
.arg("--playlist-end") .arg("--playlist-end")
.arg("10") .arg(limit)
.arg(channel) .arg(channel)
.output() .output()
.expect("Failed to execute yt-dlp"); .expect("Failed to execute yt-dlp");