Refactoring: Split into submodules.

1 files changed, 5 insertions, 123 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 0ebc364..43ae727 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -17,133 +17,15 @@
 extern crate reqwest;
 extern crate serde;
 
-use reqwest::{
-    header::{Accept, ContentType, qitem},
-    mime,
-    StatusCode,
-};
-use serde::Serialize;
-use std::io::Read;
+mod client;
+mod error;
+mod item;
 
-const ZOTAPI_CHANNEL_STREAM_PATH : &str = "/api/z/1.0/channel/stream";
-const ZOTAPI_NETWORK_STREAM_PATH : &str = "/api/z/1.0/network/stream";
-const ZOTAPI_ITEM_UPDATE_PATH : &str = "/api/z/1.0/item/update";
-
-#[derive(Debug)]
-pub enum Error {
-    Http(reqwest::Error),
-    Io(std::io::Error),
-    Unauthorized,
-    Unknown,
-}
-
-impl From<reqwest::Error> for Error {
-    fn from(e: reqwest::Error) -> Error {
-        Error::Http(e)
-    }
-}
-
-impl From<std::io::Error> for Error {
-    fn from(e: std::io::Error) -> Error {
-        Error::Io(e)
-    }
-}
-
-pub struct ItemBuilder<'a> {
-    client : &'a Client,
-    body: &'a str,
-}
-
-impl<'a> ItemBuilder<'a> {
-    fn new(client: &'a Client) -> ItemBuilder<'a> {
-        ItemBuilder {
-            client: client,
-            body: "",
-        }
-    }
-    pub fn body(&mut self, text: &'a str) -> &mut ItemBuilder<'a> {
-        self.body = text;
-        self
-    }
-
-    pub fn create(&self) -> Result<String, Error> {
-        self.client.post_data(ZOTAPI_ITEM_UPDATE_PATH, &[("body", self.body)])
-    }
-}
-
-pub struct Client {
-    inner: reqwest::Client,
-    base_url: String,
-    user: String,
-    pw: String,
-}
-
-impl Client {
-    pub fn channel_stream(&self) -> Result<String, Error> {
-        self.fetch_stream(ZOTAPI_CHANNEL_STREAM_PATH)
-    }
-
-    pub fn network_stream(&self) -> Result<String, Error> {
-        self.fetch_stream(ZOTAPI_NETWORK_STREAM_PATH)
-    }
-
-    pub fn item(&self) -> ItemBuilder {
-        ItemBuilder::new(self)
-    }
-
-    fn url(&self, path: &str) -> String {
-        self.base_url.clone() + path
-    }
-
-    fn fetch_stream(&self, path: &str) -> Result<String, Error> {
-        let url = self.url(path);
-        let mut res = self.inner.get(&url)
-            .header(Accept(vec![qitem(mime::APPLICATION_JSON)]))
-            .basic_auth(self.user.clone(), Some(self.pw.clone()))
-            .send()?;
-
-        match res.status() {
-            StatusCode::Unauthorized => Err(Error::Unauthorized),
-            StatusCode::Ok => {
-                let mut body = String::new();
-                res.read_to_string(&mut body)?;
-                Ok(body)
-            },
-            _ => Err(Error::Unknown)
-        }
-    }
-
-    fn post_data<T>(&self, path: &str, data: &T) -> Result<String, Error>
-        where T: Serialize,
-    {
-        let url = self.url(path);
-        let mut res = self.inner.post(&url)
-            .header(Accept(vec![qitem(mime::APPLICATION_JSON)]))
-            .header(ContentType::form_url_encoded())
-            .basic_auth(self.user.clone(), Some(self.pw.clone()))
-            .form(&data)
-            .send()?;
-
-        match res.status() {
-            StatusCode::Unauthorized => Err(Error::Unauthorized),
-            StatusCode::Ok => {
-                let mut body = String::new();
-                res.read_to_string(&mut body)?;
-                Ok(body)
-            },
-            _ => Err(Error::Unknown)
-        }
-    }
-}
+use client::Client;
 
 pub fn client(url: &str, user: &str, pw: &str) -> Client
 {
-    Client {
-        inner: reqwest::Client::new(),
-        base_url: String::from(url),
-        user: String::from(user),
-        pw: String::from(pw),
-    }
+    Client::new(url, user, pw)
 }
 
 #[cfg(test)]