Moving mails_resouce render_GET logic to mail_service

1 files changed, 6 insertions, 3 deletions

diff --git a/service/pixelated/adapter/services/mail_service.py b/service/pixelated/adapter/services/mail_service.py
index cb844ddf..ab0b16bd 100644
--- a/service/pixelated/adapter/services/mail_service.py
+++ b/service/pixelated/adapter/services/mail_service.py
@@ -18,17 +18,20 @@
 class MailService:
     __slots__ = ['leap_session', 'account', 'mailbox_name']
 
-    def __init__(self, mailboxes, mail_sender, tag_service, soledad_querier):
+    def __init__(self, mailboxes, mail_sender, tag_service, soledad_querier, search_engine):
         self.tag_service = tag_service
         self.mailboxes = mailboxes
         self.querier = soledad_querier
+        self.search_engine = search_engine
         self.mail_sender = mail_sender
 
     def all_mails(self):
         return self.querier.all_mails()
 
-    def mails(self, ids):
-        return self.querier.mails(ids)
+    def mails(self, query, window_size, page):
+        mail_ids, total = self.search_engine.search(query, window_size, page)
+
+        return self.querier.mails(mail_ids), total
 
     def update_tags(self, mail_id, new_tags):
         reserved_words = self.tag_service.extract_reserved(new_tags)