[incubator-ponymail-foal] 08/15: Add single email endpoint

[hu...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

humbedooh pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-ponymail-foal.git

commit d1dcb5d04679e7c8063cf1061021534c55e53c7d
Author: Daniel Gruno <hu...@apache.org>
AuthorDate: Sun Sep 6 19:20:42 2020 +0200

    Add single email endpoint
---
 server/endpoints/email.py | 79 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)

diff --git a/server/endpoints/email.py b/server/endpoints/email.py
new file mode 100644
index 0000000..c22bca1
--- /dev/null
+++ b/server/endpoints/email.py
@@ -0,0 +1,79 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements.  See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership.
+# The ASF licenses this file to You under the Apache License, Version 2.0
+# (the "License"); you may not use this file except in compliance with
+# the License.  You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Simple endpoint that returns an email or an attachment from one"""
+""" THIS ONLY DEALS WITH PUBLIC EMAILS FOR NOW - AAA IS BEING WORKED ON"""
+
+import plugins.server
+import plugins.session
+import plugins.mbox
+import aiohttp.web
+import plugins.aaa
+import base64
+
+
+async def process(
+    server: plugins.server.BaseServer,
+    session: plugins.session.SessionObject,
+    indata: dict,
+) -> dict:
+
+    email = await plugins.mbox.get_email(session, id=indata.get("id"))
+    if email and isinstance(email, dict):
+        if plugins.aaa.can_access_email(session, email):
+            # Are we fetching an attachment?
+            if not indata.get("attachment"):
+                email["gravatar"] = plugins.mbox.gravatar(email)
+                return email
+            else:
+                fid = indata.get("file")
+                for entry in email.get("attachments", []):
+                    if entry.get("hash") == fid:
+                        ct = entry.get("content_type") or "application/binary"
+                        headers = {
+                            "Content-Type": ct,
+                            "Content-Length": str(entry.get("size")),
+                        }
+                        if "image/" not in ct and "text/" not in ct:
+                            headers[
+                                "Content-Disposition"
+                            ] = f"attachment; filename=\"{entry.get('filename')}\""
+                        try:
+                            attachment = await session.database.get(
+                                itype="attachment", id=indata.get("file")
+                            )
+                            if attachment:
+                                blob = base64.decodebytes(
+                                    attachment["_source"].get("source").encode("utf-8")
+                                )
+                                return aiohttp.web.Response(
+                                    headers=headers, status=200, body=blob
+                                )
+                        except Exception as e:
+                            print(e)
+                return aiohttp.web.Response(
+                    headers={}, status=404, text="Attachment not found"
+                )
+
+        else:
+            return aiohttp.web.Response(headers={}, status=404, text="Email not found")
+    else:
+        return aiohttp.web.Response(headers={}, status=404, text="Email not found")
+
+
+def register(server: plugins.server.BaseServer):
+    return plugins.server.Endpoint(process)