From b71d9678892814a62d422af11eb91e61943836a2 Mon Sep 17 00:00:00 2001
From: Alex Auvolat <alex@adnab.me>
Date: Wed, 29 Jun 2022 20:00:38 +0200
Subject: Fetch ENVELOPE works \o/

---
 src/imap/command/selected.rs | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

(limited to 'src/imap/command/selected.rs')

diff --git a/src/imap/command/selected.rs b/src/imap/command/selected.rs
index e10eab6..b3a2ffd 100644
--- a/src/imap/command/selected.rs
+++ b/src/imap/command/selected.rs
@@ -46,12 +46,13 @@ impl<'a> SelectedContext<'a> {
         attributes: &MacroOrFetchAttributes,
         uid: &bool,
     ) -> Result<(Response, flow::Transition)> {
-        let resp = self.mailbox.fetch(sequence_set, attributes, uid).await?;
-
-        Ok((
-            Response::ok("FETCH completed")?.with_body(resp),
-            flow::Transition::None,
-        ))
+        match self.mailbox.fetch(sequence_set, attributes, uid).await {
+            Ok(resp) => Ok((
+                Response::ok("FETCH completed")?.with_body(resp),
+                flow::Transition::None,
+            )),
+            Err(e) => Ok((Response::no(&e.to_string())?, flow::Transition::None)),
+        }
     }
 
     pub async fn noop(self) -> Result<(Response, flow::Transition)> {
-- 
cgit v1.2.3