1 files changed, 120 insertions, 0 deletions
diff --git a/src/lastfm/mod.rs b/src/lastfm/mod.rs
new file mode 100644
index 0000000..96f4555
--- /dev/null
+++ b/src/lastfm/mod.rs
@@ -0,0 +1,120 @@
+use failure;
+use reqwest;
+
+use error::Result;
+
+mod api_types;
+
+const API_ROOT: &'static str = "https://ws.audioscrobbler.com/2.0/";
+
+pub struct LastFMClient {
+    client: reqwest::Client,
+    api_key: String,
+    user: String,
+}
+
+pub struct Track {
+    pub artist: String,
+    pub name: String,
+    pub album: String,
+    pub timestamp: i64,
+}
+
+pub struct Tracks<'a> {
+    client: &'a LastFMClient,
+    page: u64,
+    buf: Vec<Track>,
+}
+
+impl<'a> Tracks<'a> {
+    fn new(client: &LastFMClient) -> Tracks {
+        Tracks {
+            client,
+            page: 0,
+            buf: vec![],
+        }
+    }
+
+    fn get_next_page(&mut self) -> Result<()> {
+        self.page += 1;
+
+        let mut res = self.client.client
+            .get(API_ROOT)
+            .query(&[
+                ("method", "user.getrecenttracks"),
+                ("api_key", &self.client.api_key),
+                ("user", &self.client.user),
+                ("format", "json"),
+                ("page", &format!("{}", self.page)),
+                ("limit", "200"),
+            ])
+            .send()?;
+
+        if res.status().is_success() {
+            let data: api_types::recent_tracks = res.json()?;
+            self.buf = data.recenttracks.track
+                .iter()
+                .map(|t| {
+                    Ok(Track {
+                        artist: t.artist.text.clone(),
+                        name: t.name.clone(),
+                        album: t.album.text.clone(),
+                        timestamp: t.date.uts.parse()?,
+                    })
+                })
+                .collect::<Result<Vec<Track>>>()?;
+            Ok(())
+        }
+        else {
+            Err(failure::err_msg(res.status().as_str().to_string()))
+        }
+    }
+}
+
+impl<'a> Iterator for Tracks<'a> {
+    type Item = Track;
+
+    fn next(&mut self) -> Option<Track> {
+        if self.buf.len() == 0 {
+            let result = self.get_next_page();
+            if result.is_err() {
+                return None;
+            }
+        }
+        self.buf.pop()
+    }
+}
+
+impl LastFMClient {
+    pub fn new(api_key: &str, user: &str) -> LastFMClient {
+        LastFMClient {
+            client: reqwest::Client::new(),
+            api_key: api_key.to_string(),
+            user: user.to_string(),
+        }
+    }
+
+    pub fn track_count(&self) -> Result<u64> {
+        let mut res = self.client
+            .get(API_ROOT)
+            .query(&[
+                ("method", "user.getrecenttracks"),
+                ("api_key", &self.api_key),
+                ("user", &self.user),
+                ("format", "json"),
+            ])
+            .send()?;
+
+        if res.status().is_success() {
+            let data: api_types::recent_tracks = res.json()?;
+            Ok(data.recenttracks.attr.total.parse()?)
+        }
+        else {
+            Err(failure::err_msg(res.status().as_str().to_string()))
+        }
+    }
+
+    pub fn tracks(&self) -> Tracks {
+        Tracks::new(&self)
+    }
+}