'How hard could it be'

hammond-downloader/src/downloader.rs

@@ -39,7 +39,7 @@ fn download_into(dir: &str, file_title: &str, url: &str) -> Result<String> {
     ct_len.map(|x| info!("File Lenght: {}", x));
     ct_type.map(|x| info!("Content Type: {}", x));
 
-    let ext = get_ext(ct_type.cloned()).unwrap_or(String::from("unkown"));
+    let ext = get_ext(ct_type.cloned()).unwrap_or(String::from("unknown"));
     info!("Extension: {}", ext);
 
     // Construct a temp file to save desired content.

hammond-downloader/src/lib.rs

@@ -16,3 +16,4 @@ extern crate tempdir;
 
 pub mod downloader;
 pub mod errors;
+pub mod manager;

hammond-downloader/src/manager.rs

@@ -0,0 +1,121 @@
+use hammond_data::Episode;
+use hammond_data::dbqueries;
+
+use downloader::get_episode;
+
+use std::collections::HashSet;
+use std::sync::{Arc, Mutex};
+use std::path::PathBuf;
+use std::thread;
+
+struct DonwloadInstance {
+    uri: String,
+    // FIXME: MAKE ME A PATHBUF
+    local_uri: Option<String>,
+    downloaded_bytes: u64,
+    total_bytes: u64,
+}
+
+impl DonwloadInstance {
+    fn new(url: &str, total_bytes: u64) -> Self {
+        DonwloadInstance {
+            uri: url.into(),
+            local_uri: None,
+            downloaded_bytes: 0,
+            total_bytes,
+        }
+    }
+}
+
+struct Manager {
+    active: Arc<Mutex<HashSet<i32>>>,
+}
+
+impl Default for Manager {
+    fn default() -> Self {
+        Manager {
+            active: Arc::new(Mutex::new(HashSet::new())),
+        }
+    }
+}
+
+impl Manager {
+    fn new() -> Self {
+        Manager::default()
+    }
+
+    fn add(&self, id: i32, directory: &str) {
+        {
+            let mut m = self.active.lock().unwrap();
+            m.insert(id);
+        }
+
+        let dir = directory.to_owned();
+        let list = self.active.clone();
+        thread::spawn(move || {
+            let episode = dbqueries::get_episode_from_rowid(id).unwrap();
+            let e = get_episode(&mut episode.into(), dir.as_str());
+            if let Err(err) = e {
+                error!("Error: {}", err);
+            };
+
+            let mut m = list.lock().unwrap();
+            m.remove(&id);
+        });
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use downloader;
+
+    use diesel::Identifiable;
+
+    use hammond_data::database;
+    use hammond_data::feed::*;
+    use hammond_data::{Episode, Source};
+    use hammond_data::dbqueries;
+
+    use std::path::Path;
+    use std::{thread, time};
+
+    #[test]
+    // This test inserts an rss feed to your `XDG_DATA/hammond/hammond.db` so we make it explicit
+    // to run it.
+    #[ignore]
+    // THIS IS NOT A RELIABLE TEST
+    // Just quick sanity check
+    fn test_start_dl() {
+        let url = "http://www.newrustacean.com/feed.xml";
+
+        // Create and index a source
+        let source = Source::from_url(url).unwrap();
+        // Copy it's id
+        let sid = source.id().clone();
+
+        // Convert Source it into a Feed and index it
+        let feed = source.into_feed(true).unwrap();
+        index(&feed);
+
+        // Get the Podcast
+        let pd = dbqueries::get_podcast_from_source_id(sid).unwrap();
+        // Get an episode
+        let episode: Episode = {
+            let con = database::connection();
+            dbqueries::get_episode_from_pk(&*con.get().unwrap(), "e000: Hello, world!", *pd.id())
+                .unwrap()
+        };
+
+        let manager = Manager::new();
+        let download_fold = downloader::get_download_folder(&pd.title()).unwrap();
+        manager.add(episode.rowid(), download_fold.as_str());
+
+        // Give it soem time to download the file
+        thread::sleep(time::Duration::from_secs(20));
+
+        let final_path = format!("{}/{}.unknown", &download_fold, episode.rowid());
+        println!("{}", &final_path);
+        assert!(Path::new(&final_path).exists());
+    }
+}