use std::sync::Arc; use anyhow::Result; use tokio::sync::Notify; use imap_codec::imap_types::command::Command; use imap_codec::imap_types::core::Tag; use imap_codec::imap_types::response::{Code, Data, Status}; #[derive(Debug)] pub enum Body<'a> { Data(Data<'a>), Status(Status<'a>), } pub struct ResponseBuilder<'a> { tag: Option>, code: Option>, text: String, body: Vec>, } impl<'a> ResponseBuilder<'a> { pub fn to_req(mut self, cmd: &Command<'a>) -> Self { self.tag = Some(cmd.tag.clone()); self } pub fn tag(mut self, tag: Tag<'a>) -> Self { self.tag = Some(tag); self } pub fn message(mut self, txt: impl Into) -> Self { self.text = txt.into(); self } pub fn code(mut self, code: Code<'a>) -> Self { self.code = Some(code); self } pub fn data(mut self, data: Data<'a>) -> Self { self.body.push(Body::Data(data)); self } pub fn many_data(mut self, data: Vec>) -> Self { for d in data.into_iter() { self = self.data(d); } self } #[allow(dead_code)] pub fn info(mut self, status: Status<'a>) -> Self { self.body.push(Body::Status(status)); self } #[allow(dead_code)] pub fn many_info(mut self, status: Vec>) -> Self { for d in status.into_iter() { self = self.info(d); } self } pub fn set_body(mut self, body: Vec>) -> Self { self.body = body; self } pub fn ok(self) -> Result> { Ok(Response { completion: Status::ok(self.tag, self.code, self.text)?, body: self.body, }) } pub fn no(self) -> Result> { Ok(Response { completion: Status::no(self.tag, self.code, self.text)?, body: self.body, }) } pub fn bad(self) -> Result> { Ok(Response { completion: Status::bad(self.tag, self.code, self.text)?, body: self.body, }) } } #[derive(Debug)] pub struct Response<'a> { pub body: Vec>, pub completion: Status<'a>, } impl<'a> Response<'a> { pub fn build() -> ResponseBuilder<'a> { ResponseBuilder { tag: None, code: None, text: "".to_string(), body: vec![], } } pub fn bye() -> Result> { Ok(Response { completion: Status::bye(None, "bye")?, body: vec![], }) } } #[derive(Debug)] pub enum ResponseOrIdle { Response(Response<'static>), StartIdle(Arc), IdleEvent(Vec>), }