kemitix/podal
1
0
Fork 0
Code Issues 5 Pull requests Releases 1 Packages Activity Actions

extract site as a parameter

Browse source
This commit is contained in:
Paul Campbell 2023-07-24 07:40:20 +01:00
parent 2cd5b20840
commit 2c953b5997
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/main.rs
View file

@ -55,10 +55,11 @@ fn main() -> Result<()> {
println!("Podal"); println!("Podal");
let subscriptions = "subscriptions.txt"; let subscriptions = "subscriptions.txt";
let history = "downloaded.txt"; let history = "downloaded.txt";
let site = "https://www.youtube.com/";
for channel_name in lines_from(subscriptions)? { for channel_name in lines_from(subscriptions)? {
let channel_name = channel_name?; let channel_name = channel_name?;
let feed_url = get_feed_url(channel_name)?; let feed_url = get_feed_url(site, channel_name)?;
for entry in get_feed(feed_url)?.entries() { for entry in get_feed(feed_url)?.entries() {
if let Some(link) = get_link(entry) { if let Some(link) = get_link(entry) {
if !is_already_downloaded(&link, history)? { if !is_already_downloaded(&link, history)? {
@ -73,13 +74,13 @@ fn main() -> Result<()> {
Ok(()) Ok(())
} }
fn get_feed_url(channel_name: String) -> Result<String> { fn get_feed_url(site: &str, channel_name: String) -> Result<String> {
if let Some(channel_prefix) = channel_name.chars().next() { if let Some(channel_prefix) = channel_name.chars().next() {
if channel_prefix != '@' { if channel_prefix != '@' {
return Err(format!("Channel Name must begin with an '@': {}", channel_name).into()); return Err(format!("Channel Name must begin with an '@': {}", channel_name).into());
} }
} }
let channel_url = format!("https://www.youtube.com/{}", channel_name); let channel_url = format!("{}{}", site, channel_name);
let response = reqwest::blocking::get(channel_url)?; let response = reqwest::blocking::get(channel_url)?;
let rss_url = scraper::Html::parse_document(&response.text()?) let rss_url = scraper::Html::parse_document(&response.text()?)
.select(&scraper::Selector::parse("link[title='RSS']").unwrap()) .select(&scraper::Selector::parse("link[title='RSS']").unwrap())