From 04b5b54212629f058bdab1ba55c89a3d417e0454 Mon Sep 17 00:00:00 2001
From: Paul Woolcock <paul@woolcock.us>
Date: Wed, 7 Oct 2020 05:44:48 -0400
Subject: Change builder style for requests

---
 src/requests/directory.rs | 80 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 src/requests/directory.rs

(limited to 'src/requests/directory.rs')

diff --git a/src/requests/directory.rs b/src/requests/directory.rs
new file mode 100644
index 0000000..14ff2c4
--- /dev/null
+++ b/src/requests/directory.rs
@@ -0,0 +1,80 @@
+use crate::errors::Error;
+use serde::Serialize;
+use std::borrow::Cow;
+
+mod bool_qs_serialize {
+    use serde::Serializer;
+
+    pub fn is_false(b: &bool) -> bool {
+        !*b
+    }
+
+    pub fn serialize<S: Serializer>(b: &bool, s: S) -> Result<S::Ok, S::Error> {
+        if *b {
+            s.serialize_i64(1)
+        } else {
+            s.serialize_i64(0)
+        }
+    }
+}
+
+/// Represents the options for the directory request
+#[derive(Debug, Clone, Default, PartialEq, Serialize)]
+pub struct DirectoryRequest<'a> {
+    offset: Option<usize>,
+    limit: Option<usize>,
+    order: Option<Cow<'a, str>>, // TODO enum
+    #[serde(skip_serializing_if = "bool_qs_serialize::is_false")]
+    #[serde(serialize_with = "bool_qs_serialize::serialize")]
+    local: bool,
+}
+impl<'a> DirectoryRequest<'a> {
+    /// make a new DirectoryRequest builder
+    pub fn new() -> Self {
+        DirectoryRequest::default()
+    }
+
+    /// sets the offset
+    pub fn offset(mut self, offset: usize) -> Self {
+        self.offset = Some(offset);
+        self
+    }
+
+    /// sets the limit
+    pub fn limit(mut self, limit: usize) -> Self {
+        self.limit = Some(limit);
+        self
+    }
+
+    /// sets the order
+    pub fn order<I: Into<Cow<'a, str>>>(mut self, order: I) -> Self {
+        self.order = Some(order.into());
+        self
+    }
+
+    /// sets the local
+    pub fn local(mut self) -> Self {
+        self.local = true;
+        self
+    }
+
+    /// Turns this builder into a querystring
+    ///
+    /// # Example
+    ///
+    /// ```
+    /// # extern crate elefren;
+    /// # use elefren::requests::DirectoryRequest;
+    /// let request = DirectoryRequest::new();
+    /// assert_eq!(
+    ///     &request
+    ///         .limit(10)
+    ///         .to_querystring()
+    ///         .expect("Couldn't serialize qs"),
+    ///     "limit=10"
+    /// );
+    /// ```
+    pub fn to_querystring(&self) -> Result<String, Error> {
+        Ok(format!("{}", serde_qs::to_string(&self)?))
+    }
+}
-- 
cgit v1.2.3