backend: correspondents

backend/src/routes/folders.rs

@@ -19,7 +19,8 @@ use crate::{
 };
 
 use super::documents::{
-    load_primary_assets, load_tags_for_documents, to_document_response, to_iso, DocumentResponse,
+    load_correspondents_for_documents, load_primary_assets, load_tags_for_documents,
+    to_document_response, to_iso, DocumentResponse,
 };
 
 const QUICKWIT_MAX_HITS: usize = 200;
@@ -213,6 +214,7 @@ pub async fn list_folder_contents(
 
     let doc_ids: Vec<Uuid> = docs.iter().map(|doc| doc.id).collect();
     let tags_map = load_tags_for_documents(&mut conn, &doc_ids)?;
+    let mut correspondents_map = load_correspondents_for_documents(&mut conn, &doc_ids)?;
     drop(conn);
 
     let primary_versions = load_primary_assets(&state, &docs).await?;
@@ -220,12 +222,14 @@ pub async fn list_folder_contents(
     let mut documents = Vec::with_capacity(doc_ids.len());
     for doc in docs {
         let tags = tags_map.get(&doc.id).cloned();
+        let correspondents = correspondents_map.remove(&doc.id).unwrap_or_default();
         let current_version = primary_versions.get(&doc.id).cloned();
         documents.push(to_document_response(
             &state,
             user.user_id,
             doc,
             tags,
+            correspondents,
             current_version,
         )?);
     }
@@ -397,18 +401,21 @@ pub async fn search_documents(
 
     let doc_ids: Vec<Uuid> = docs.iter().map(|doc| doc.id).collect();
     let tags_map = load_tags_for_documents(&mut conn, &doc_ids)?;
+    let mut correspondents_map = load_correspondents_for_documents(&mut conn, &doc_ids)?;
     drop(conn);
 
     let primary_versions = load_primary_assets(&state, &docs).await?;
     let mut response = Vec::with_capacity(doc_ids.len());
     for doc in docs {
         let tags = tags_map.get(&doc.id).cloned();
+        let correspondents = correspondents_map.remove(&doc.id).unwrap_or_default();
         let current_version = primary_versions.get(&doc.id).cloned();
         response.push(to_document_response(
             &state,
             user.user_id,
             doc,
             tags,
+            correspondents,
             current_version,
         )?);
     }