fixed anonymous + authenticated imap logic

1 files changed, 97 insertions, 0 deletions

diff --git a/src/imap/response.rs b/src/imap/response.rs
new file mode 100644
index 0000000..22e91f3
--- /dev/null
+++ b/src/imap/response.rs
@@ -0,0 +1,97 @@
+use anyhow::Result;
+use imap_codec::imap_types::command::Command;
+use imap_codec::imap_types::core::Tag;
+use imap_codec::imap_types::response::{Code, Data, Status, StatusKind};
+
+pub struct ResponseBuilder {
+    status: StatusKind,
+    tag: Option<Tag<'static>>,
+    code: Option<Code<'static>>,
+    text: String,
+    data: Vec<Data<'static>>,
+}
+
+impl<'a> Default for ResponseBuilder {
+    fn default() -> ResponseBuilder {
+        ResponseBuilder {
+            status: StatusKind::Bad,
+            tag: None,
+            code: None,
+            text: "".to_string(),
+            data: vec![],
+        }
+    }
+}
+
+impl ResponseBuilder {
+    pub fn to_req(mut self, cmd: &Command) -> Self {
+        self.tag = Some(cmd.tag);
+        self
+    }
+    pub fn tag(mut self, tag: Tag) -> Self {
+        self.tag = Some(tag);
+        self
+    }
+
+    pub fn message(mut self, txt: impl Into<String>) -> Self {
+        self.text = txt.into();
+        self
+    }
+
+    pub fn code(mut self, code: Code) -> Self {
+        self.code = Some(code);
+        self
+    }
+
+    pub fn data(mut self, data: Data) -> Self {
+        self.data.push(data);
+        self
+    }
+
+    pub fn set_data(mut self, data: Vec<Data>) -> Self {
+        self.data = data;
+        self
+    }
+
+    pub fn build(self) -> Result<Response> {
+        Ok(Response {
+            status: Status::new(self.tag, self.status, self.code, self.text)?,
+            data: self.data,
+        })
+    }
+}
+
+pub struct Response {
+    data: Vec<Data<'static>>,
+    status: Status<'static>,
+}
+
+impl Response {
+    pub fn ok() -> ResponseBuilder {
+        ResponseBuilder {
+            status: StatusKind::Ok,
+            ..ResponseBuilder::default()
+        }
+    }
+
+    pub fn no() -> ResponseBuilder {
+        ResponseBuilder {
+            status: StatusKind::No,
+            ..ResponseBuilder::default()
+        }
+    }
+
+    pub fn bad() -> ResponseBuilder {
+        ResponseBuilder {
+            status: StatusKind::Bad,
+            ..ResponseBuilder::default()
+        }
+    }
+
+    pub fn bye() -> Result<Response> {
+        Ok(Response {
+            status: Status::bye(None, "bye")?,
+            data: vec![],
+        })
+    }
+}