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 <[email protected]> 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)
