wip: init commit

.gitignore

@@ -0,0 +1 @@
+/target

Cargo.toml

@@ -0,0 +1,16 @@
+[package]
+name = "shana-fetcher"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+clap = { version = "4.5.4", features = ["derive"] }
+rss = { version = "2.0.8", default-features = false }
+slog = "2.7.0"
+slog-async = "2.8.0"
+slog-term = "2.9.1"
+tokio = { version = "1.37.0", features = ["rt", "macros", "rt-multi-thread"] }
+transmission-rpc = "0.4.2"
+ureq = "2.9.7"

src/main.rs

@@ -0,0 +1,94 @@
+#[macro_use]
+extern crate slog;
+extern crate slog_async;
+extern crate slog_term;
+
+use clap::Parser;
+use slog::{Drain, Logger};
+use transmission_rpc::TransClient;
+use transmission_rpc::types::{
+    BasicAuth, Result as TransResult, RpcResponse, TorrentAddArgs, TorrentAddedOrDuplicate
+};
+
+use std::io::BufReader;
+
+const DEFAULT_FEED_URL: &str = "https://www.shanaproject.com/feeds/secure/user/42421/4P7Y3T5U2E/";
+
+#[derive(Parser)]
+#[command(version, about, long_about = None)]
+struct Cli {
+    #[arg(short, long, default_value="localhost:9091")]
+    rpc_url: String,
+
+    #[arg(short, long, default_value=DEFAULT_FEED_URL)]
+    feed_url: String,
+
+    #[arg(long, default_value="")]
+    user: String,
+
+    #[arg(long, default_value="")]
+    password: String,
+}
+
+fn setup_logger() -> slog::Logger{
+    let decorator = slog_term::TermDecorator::new().stderr().build();
+    let drain = slog_term::CompactFormat::new(decorator).build().fuse();
+    let drain = slog_async::Async::new(drain).build().fuse();
+
+    slog::Logger::root(drain, o!())
+}
+
+async fn add_torrent(_: &Logger, url: &str, user: &str, password: &str)
+    -> Result<(), String>
+{
+    let auth = BasicAuth {
+        user: user.to_string(), password: password.to_string()
+    };
+    let mut client = TransClient::with_auth(url.parse().unwrap(), auth);
+
+    let add = TorrentAddArgs {
+        filename: Some(url.into()),
+        ..TorrentAddArgs::default()
+    };
+
+    match client.torrent_add(add).await {
+        Ok(_)  => Ok(()),
+        Err(err) => Err((*err).to_string())
+    }
+}
+
+#[tokio::main]
+async fn main() -> Result<(), ureq::Error> {
+    let log = setup_logger();
+
+    let args = Cli::parse();
+
+    // TODO replace ureq with reqwest as transmission-rpc depends on it anyway
+    let feed_url = args.feed_url;
+    info!(log, "fetching {feed_url}");
+    let resp = ureq::get(&feed_url).call()?;
+    info!(log, "fetch ok");
+
+    info!(log, "parsing feed");
+    let resp_reader = BufReader::new(resp.into_reader());
+    let channel = rss::Channel::read_from(resp_reader).expect("valid stream");
+    info!(log, "parsing ok");
+
+    for e in channel.into_items().into_iter().rev() {
+        let title = e.title().unwrap_or("<no title>").to_string();
+        let log = log.new(o!("title" => title));
+
+        info!(log, "new entry");
+
+        if let Some(url) = e.link() {
+            info!(log, "fetching entry");
+            if let Err(e) = add_torrent(&log, url, &args.user, &args.password).await {
+                error!(log, "error adding torrent: {e}");
+            }
+        } else {
+            error!(log, "no url included, skipping");
+        }
+    }
+
+    Ok(())
+}