feature(#98): implemented read emails api

Brain/src/model/requests/request_model.py

@@ -215,3 +215,15 @@ class AutoTaskDelete(BasicReq):
         reference_link: str
 
     data: Body
+
+
+"""endpoint /read_emails"""
+
+
+class EmailReader(BasicReq):
+    class Body(BaseModel):
+        sender: str
+        pwd: str
+        imap_folder: str
+
+    data: Body

Brain/src/router/api.py

@@ -19,6 +19,7 @@ from Brain.src.model.requests.request_model import (
     ClientInfo,
     get_client_info,
     AutoTaskDelete,
+    EmailReader,
 )
 from Brain.src.rising_plugin.risingplugin import (
     getCompletion,
@@ -27,6 +28,7 @@ from Brain.src.rising_plugin.risingplugin import (
 )
 from Brain.src.firebase.cloudmessage import CloudMessage
 from Brain.src.rising_plugin.image_embedding import embed_image_text, query_image_text
+from Brain.src.rising_plugin.gmail.manage_gmail import EmailManager
 
 from Brain.src.logs import logger
 from Brain.src.model.basic_model import BasicModel
@@ -492,4 +494,60 @@ def construct_blueprint_api() -> APIRouter:
             200, "Deleted data from real-time database of firebase", ""
         )
 
+    """@generator.request_body(
+            {
+                "token": "String",
+                "uuid": "String",
+                "data": {
+                    "sender": "test@gmail.com",
+                    "pwd": "use app password of your google account",
+                    "imap_folder": "inbox or drafts"
+                },
+            }
+        )
+
+        @generator.response(
+            status_code=200, schema={"message": "message", "result": [{
+                "From": "testfrom@test.com",
+                "To": "test@gmail.com",
+                "Date": "Tue, 04 Jul 2023 12:55:19 +0000",
+                "CC": "",
+                "Subject": "subject",
+                "Message Body": "message"
+            }]}
+        )
+
+    """
+
+    @router.post("/read_emails")
+    def read_emails(data: EmailReader):
+        # firebase admin init
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return assembler.to_response(ex.code, ex.message, "")
+        try:
+            token = setting.token
+            uuid = setting.uuid
+
+            # if imap_folder is drafts, then search is ALL
+            imap_search_cmd = "UNSEEN"
+            if data.data.imap_folder not in "inbox":
+                imap_search_cmd = "(ALL)"
+
+            # read emails
+            email_manager = EmailManager()
+            result = email_manager.read_emails(
+                sender=data.data.sender,
+                pwd=data.data.pwd,
+                imap_folder=data.data.imap_folder,
+                imap_search_command=imap_search_cmd,
+            )
+            result = json.loads(result)
+        except Exception as e:
+            if isinstance(e, BrainException):
+                return e.get_response_exp()
+            return assembler.to_response(400, "Failed to read emails", "")
+        return assembler.to_response(200, "", result)
+
     return router