101 lines
2.4 KiB
Rust
101 lines
2.4 KiB
Rust
//! # Steam RSS
|
|
//!
|
|
//! > **Get RSS feeds for Steam games.**
|
|
//!
|
|
//! *AGPL-3.0-or-later*
|
|
|
|
#![forbid(unsafe_code)]
|
|
#![warn(missing_docs, clippy::missing_docs_in_private_items)]
|
|
|
|
use std::{thread::sleep, time::Duration};
|
|
|
|
use {
|
|
clap::Parser,
|
|
color_eyre::{install, Result},
|
|
indicatif::{ProgressBar, ProgressStyle},
|
|
};
|
|
|
|
/// CLI arguments struct using [`clap`]'s Derive API.
|
|
#[derive(Debug, Parser)]
|
|
#[clap(about, author, version)]
|
|
pub struct Args {
|
|
/// A game's AppID, can be used multiple times.
|
|
#[clap(short, long)]
|
|
pub appid: Vec<usize>,
|
|
|
|
/// Output the feeds as OPML.
|
|
#[clap(long)]
|
|
pub opml: bool,
|
|
|
|
/// The time in milliseconds to sleep between HTTP requests.
|
|
#[clap(short, long, default_value = "250")]
|
|
pub timeout: u64,
|
|
|
|
/// Verify potential feeds by downloading them and checking if they return XML.
|
|
#[clap(short, long)]
|
|
pub verify: bool,
|
|
}
|
|
|
|
/// A simple feed struct.
|
|
#[derive(Debug)]
|
|
pub struct Feed {
|
|
/// The text to use for the feed in the OPML output.
|
|
pub text: Option<String>,
|
|
|
|
/// The URL of the feed.
|
|
pub url: String,
|
|
}
|
|
|
|
fn main() -> Result<()> {
|
|
install()?;
|
|
|
|
let args = Args::parse();
|
|
let timeout = Duration::from_millis(args.timeout);
|
|
|
|
let ureq_agent = ureq::AgentBuilder::new()
|
|
.user_agent("Steam Feeds (https://git.bauke.xyz/Bauke/steam-rss)")
|
|
.build();
|
|
let mut potential_feeds = vec![];
|
|
let mut feeds_to_output = vec![];
|
|
|
|
for appid in args.appid {
|
|
potential_feeds.push(Feed {
|
|
text: Some(format!("Steam AppID {appid}")),
|
|
url: format!("https://steamcommunity.com/games/{appid}/rss/"),
|
|
});
|
|
}
|
|
|
|
if args.verify {
|
|
let progress = ProgressBar::new(potential_feeds.len().try_into()?)
|
|
.with_style(ProgressStyle::with_template("Verifying {pos}/{len} {bar}")?);
|
|
|
|
for potential_feed in potential_feeds {
|
|
let response = ureq_agent.get(&potential_feed.url).call()?;
|
|
if response.content_type() == "text/xml" {
|
|
feeds_to_output.push(potential_feed);
|
|
}
|
|
|
|
sleep(timeout);
|
|
progress.inc(1);
|
|
}
|
|
} else {
|
|
feeds_to_output = potential_feeds;
|
|
}
|
|
|
|
let mut opml_document = opml::OPML::default();
|
|
for feed in feeds_to_output {
|
|
if args.opml {
|
|
opml_document
|
|
.add_feed(&feed.text.unwrap_or_else(|| feed.url.clone()), &feed.url);
|
|
} else {
|
|
println!("{}", feed.url);
|
|
}
|
|
}
|
|
|
|
if args.opml {
|
|
println!("{}", opml_document.to_string()?);
|
|
}
|
|
|
|
Ok(())
|
|
}
|