From d99083c47ddd31288ed1b688002388f6e68c5e04 Mon Sep 17 00:00:00 2001
From: Harald Eilertsen <haraldei@anduin.net>
Date: Thu, 13 Feb 2020 14:45:41 +0100
Subject: Refactor how ABConfigs are fetched.

Introduce a `z()` menber function that creates the request object
that we use for further configuring the requset. This eliminates the
need to two fetch functions, and is meant to provide a consistent
way of doing these requests.
---
 src/abconfig.rs | 36 ++++++++++++++++++++----------------
 src/lib.rs      |  3 ++-
 2 files changed, 22 insertions(+), 17 deletions(-)

(limited to 'src')

diff --git a/src/abconfig.rs b/src/abconfig.rs
index 723747c..6348030 100644
--- a/src/abconfig.rs
+++ b/src/abconfig.rs
@@ -20,7 +20,7 @@ use serde::Deserialize;
 /// A struct for storing a key value pair with a category in
 /// relation to a contact. Typically used to store permissions
 /// a given contact has on a given channel.
-#[derive(Debug, Deserialize)]
+#[derive(Debug, Default, Deserialize)]
 pub struct ABConfig {
     /// Database ID for this entry
     pub id: u32,
@@ -41,30 +41,34 @@ pub struct ABConfig {
     pub v: String,
 }
 
-/// Fetch ABConfig for all contacts
-pub fn fetch(client: &Client) -> Result<Vec<ABConfig>, Error> {
-    let payload = client.fetch_stream("abconfig", &())?;
-    Ok(serde_json::from_str(&payload)?)
+impl ABConfig {
+    pub fn z() -> ABConfigRequest {
+        ABConfigRequest::default()
+    }
 }
 
 #[derive(Default)]
 pub struct ABConfigRequest {
-    abook_id: u32,
+    abook_id: Option<u32>,
 }
 
 impl ABConfigRequest {
+    /// Limit to ABConfigs for a given contact
+    ///
+    /// `abook` is the abook id of the given contact
+    pub fn with_abook_id(mut self, abook: u32) -> ABConfigRequest {
+        self.abook_id = Some(abook);
+        self
+    }
+
     pub fn fetch(&self, client: &Client) -> Result<Vec<ABConfig>, Error> {
-        let mut res = client.get("abconfig")
-            .query(&[("abook_id", self.abook_id.to_string())])
-            .send()?;
+        let mut req = client.get("abconfig");
 
-        Ok(serde_json::from_str(&res.text()?)?)
+        if let Some(id) = self.abook_id {
+            req = req.query(&[("abook_id", id.to_string())]);
+        }
+
+        Ok(serde_json::from_str(&req.send()?.text()?)?)
     }
 }
 
-/// Fetch ABConfig for a given contact
-///
-/// `abook` is the abook id of the given contact
-pub fn with_abook_id(abook: u32) -> ABConfigRequest {
-    ABConfigRequest { abook_id: abook }
-}
diff --git a/src/lib.rs b/src/lib.rs
index 7a17269..1be4c92 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -14,7 +14,7 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
-pub mod abconfig;
+mod abconfig;
 mod abook;
 mod channel_stream;
 mod client;
@@ -24,6 +24,7 @@ mod item;
 mod network_stream;
 mod xchan;
 
+pub use abconfig::ABConfig;
 pub use abook::abook;
 pub use channel_stream::channel_stream;
 pub use client::*;
-- 
cgit v1.2.3