From 1ca6cd5de0656910213425e1d8f05256af820f21 Mon Sep 17 00:00:00 2001
From: Quentin Dufour <quentin@deuxfleurs.fr>
Date: Sat, 6 Jan 2024 11:33:40 +0100
Subject: [PATCH] search is re-enabled

---
 src/imap/mailbox_view.rs | 57 ++++++++++++++++++++++++----------------
 src/mail/query.rs        |  8 +++---
 2 files changed, 38 insertions(+), 27 deletions(-)

diff --git a/src/imap/mailbox_view.rs b/src/imap/mailbox_view.rs
index d83fa3e..6a4724d 100644
--- a/src/imap/mailbox_view.rs
+++ b/src/imap/mailbox_view.rs
@@ -21,7 +21,7 @@ use crate::imap::attributes::AttributesProxy;
 use crate::imap::flags;
 use crate::imap::mail_view::{MailView, SeenFlag};
 use crate::imap::response::Body;
-//use crate::imap::search;
+use crate::imap::search;
 use crate::imap::index::Index;
 
 
@@ -259,8 +259,7 @@ impl MailboxView {
 
         // [2/6] Fetch the emails
         let uuids = mail_idx_list.iter().map(|midx| midx.uuid).collect::<Vec<_>>();
-        let query = self.0.query(&uuids, query_scope);
-        let query_result = query.fetch().await?;
+        let query_result = self.0.query(&uuids, query_scope).fetch().await?;
             
         // [3/6] Derive an IMAP-specific view from the results, apply the filters
         let views = query_result.iter()
@@ -268,13 +267,18 @@ impl MailboxView {
             .map(|(qr, midx)| MailView::new(qr, midx))
             .collect::<Result<Vec<_>, _>>()?;
 
-        // [4/6] Apply the IMAP transformation to keep only relevant fields
+        // [4/6] Apply the IMAP transformation, bubble up any error
+        // We get 2 results:
+        //   - The one we send to the client
+        //   - The \Seen flags we must set internally
         let (flag_mgmt, imap_ret): (Vec<_>, Vec<_>) = views
             .iter()
-            .filter_map(|mv| mv.filter(&ap).ok().map(|(body, seen)| ((mv, seen), body)))
+            .map(|mv| mv.filter(&ap).map(|(body, seen)| ((mv, seen), body)))
+            .collect::<Result<Vec<_>, _>>()?
+            .into_iter()
             .unzip();
 
-        // [5/6] Register seen flags
+        // [5/6] Register the \Seen flags
         flag_mgmt
             .iter()
             .filter(|(_mv, seen)| matches!(seen, SeenFlag::MustAdd))
@@ -293,30 +297,37 @@ impl MailboxView {
         Ok(imap_ret)
     }
 
-    /// A very naive search implementation...
+    /// A naive search implementation...
     pub async fn search<'a>(
         &self,
         _charset: &Option<Charset<'a>>,
-        _search_key: &SearchKey<'a>,
-        _uid: bool,
+        search_key: &SearchKey<'a>,
+        uid: bool,
     ) -> Result<Vec<Body<'static>>> {
-        /*
         // 1. Compute the subset of sequence identifiers we need to fetch
-        let query = search::Criteria(search_key);
-        let (seq_set, seq_type) = query.to_sequence_set();
-        let mailids = MailIdentifiersList(self.get_mail_ids(&seq_set, seq_type.is_uid())?);
-        let mail_u32 = match uid {
-            true => mailids.uids(),
-            _ => mailids.ids(),
+        // based on the search query
+        let crit = search::Criteria(search_key);
+        let (seq_set, seq_type) = crit.to_sequence_set();
+
+        // 2. Get the selection
+        let selection = self.index().fetch(&seq_set, seq_type.is_uid())?;
+
+        // 3. Filter the selection based on the ID / UID / Flags
+
+        // 4. If needed, filter the selection based on the metadata
+        let _need_meta = crit.need_meta();
+
+        // 5. If needed, filter the selection based on the body
+        let _need_body = crit.need_body();
+
+        // 6. Format the result according to the client's taste:
+        // either return UID or ID.
+        let selection_fmt = match uid {
+            true => selection.into_iter().map(|in_idx| in_idx.uid).collect(),
+            _ => selection.into_iter().map(|in_idx| in_idx.i).collect(),
         };
 
-        // 2. Compute wether we will need to fetch the mail meta and/or the body
-        let _need_meta = query.need_meta();
-        let _need_body = query.need_body();
-
-        Ok(vec![Body::Data(Data::Search(mail_u32))])
-        */
-        unimplemented!()
+        Ok(vec![Body::Data(Data::Search(selection_fmt))])
     }
 
     // ----
diff --git a/src/mail/query.rs b/src/mail/query.rs
index 70feb89..7b26cb9 100644
--- a/src/mail/query.rs
+++ b/src/mail/query.rs
@@ -21,7 +21,7 @@ pub enum QueryScope {
 }
 
 impl<'a,'b> Query<'a,'b> {
-    pub async fn fetch(&self) -> Result<Vec<QueryResult>> {
+    pub async fn fetch(&self) -> Result<Vec<QueryResult<'a>>> {
         match self.scope {
             QueryScope::Index => self.index(),
             QueryScope::Partial => self.partial().await,
@@ -31,7 +31,7 @@ impl<'a,'b> Query<'a,'b> {
 
     // --- functions below are private *for reasons*
 
-    fn index(&self) -> Result<Vec<QueryResult>> {
+    fn index(&self) -> Result<Vec<QueryResult<'a>>> {
         self
             .emails
             .iter()
@@ -47,7 +47,7 @@ impl<'a,'b> Query<'a,'b> {
             .collect::<Result<Vec<_>, _>>()
     }
 
-    async fn partial(&self) -> Result<Vec<QueryResult>> {
+    async fn partial(&self) -> Result<Vec<QueryResult<'a>>> {
         let meta = self.frozen.mailbox.fetch_meta(self.emails).await?;
         let result = meta
             .into_iter()
@@ -61,7 +61,7 @@ impl<'a,'b> Query<'a,'b> {
     /// AND GENERATE SO MUCH NETWORK TRAFFIC.
     /// THIS FUNCTION SHOULD BE REWRITTEN, FOR EXAMPLE WITH
     /// SOMETHING LIKE AN ITERATOR
-    async fn full(&self) -> Result<Vec<QueryResult>> {
+    async fn full(&self) -> Result<Vec<QueryResult<'a>>> {
         let meta_list = self.partial().await?;
         meta_list
             .into_iter()