#!/usr/bin/env python

# -*- coding: utf-8 -*-

# Milter calls methods of your class at milter events.

# Return REJECT,TEMPFAIL,ACCEPT to short circuit processing for a message.

# You can also add/del recipients, replacebody, add/del headers, etc.

from __future__ import print_function, unicode_literals

import base64

import email

import hashlib

import os

import sys

import time

from io import BytesIO

from multiprocessing import Process as Thread, Queue

import Milter

import requests

from Milter.utils import parse_addr

import pagure.config

import pagure.lib.query

if 'PAGURE_CONFIG' not in os.environ \

        and os.path.exists('/etc/pagure/pagure.cfg'):

    os.environ['PAGURE_CONFIG'] = '/etc/pagure/pagure.cfg'

logq = Queue(maxsize=4)

_config = pagure.config.reload_config()

def get_email_body(emailobj):

    ''' Return the body of the email, preferably in text.

    '''

    def _get_body(emailobj):

        """ Return the first text/plain body found if the email is multipart

        or just the regular payload otherwise.

        """

        if emailobj.is_multipart():

            for payload in emailobj.get_payload():

                # If the message comes with a signature it can be that this

                # payload itself has multiple parts, so just return the

                # first one

                if payload.is_multipart():

                    return _get_body(payload)

                body = payload.get_payload()

                if payload.get_content_type() == 'text/plain':

                    return body

        else:

            return emailobj.get_payload()

    body = _get_body(emailobj)

    enc = emailobj['Content-Transfer-Encoding']

    if enc == 'base64':

        body = base64.decodestring(body)

    return body

def clean_item(item):

    ''' For an item provided as <item> return the content, if there are no</item>

    <> then return the string.

    '''

    if '<' in item:

        item = item.split('<')[1]

    if '>' in item:

        item = item.split('>')[0]

    return item

class PagureMilter(Milter.Base):

    def __init__(self):  # A new instance with each new connection.

        self.id = Milter.uniqueID()  # Integer incremented with each call.

        self.fp = None

    def log(self, message):

        print(message)

        sys.stdout.flush()

    def envfrom(self, mailfrom, *str):

        self.log("mail from: %s  -  %s" % (mailfrom, str))

        self.fromparms = Milter.dictfromlist(str)

        # NOTE: self.fp is only an *internal* copy of message data.  You

        # must use addheader, chgheader, replacebody to change the message

        # on the MTA.

        self.fp = BytesIO()

        self.canon_from = '@'.join(parse_addr(mailfrom))

        from_txt = 'From %s %s\n' % (self.canon_from, time.ctime())

        self.fp.write(from_txt.encode('utf-8'))

        return Milter.CONTINUE

    @Milter.noreply

    def header(self, name, hval):

        ''' Headers '''

        # add header to buffer

        header_txt = "%s: %s\n" % (name, hval)

        self.fp.write(header_txt.encode('utf-8'))

        return Milter.CONTINUE

    @Milter.noreply

    def eoh(self):

        ''' End of Headers '''

        self.fp.write(b"\n")

        return Milter.CONTINUE

    @Milter.noreply

    def body(self, chunk):

        ''' Body '''

        self.fp.write(chunk)

        return Milter.CONTINUE

    @Milter.noreply

    def envrcpt(self, to, *str):

        rcptinfo = to, Milter.dictfromlist(str)

        print(rcptinfo)

        return Milter.CONTINUE

    def eom(self):

        ''' End of Message '''

        self.fp.seek(0)

        msg = email.message_from_file(self.fp)

        msg_id = msg.get('In-Reply-To', None)

        if msg_id is None:

            self.log('No In-Reply-To, keep going')

            return Milter.CONTINUE

        # Ensure we don't get extra lines in the message-id

        msg_id = msg_id.split('\n')[0].strip()

        self.log('msg-ig %s' % msg_id)

        self.log('To %s' % msg['to'])

        self.log('Cc %s' % msg.get('cc'))

        self.log('From %s' % msg['From'])

        # Check the email was sent to the right address

        email_address = msg['to']

        if 'reply+' in msg.get('cc', ''):

            email_address = msg['cc']

        if 'reply+' not in email_address:

            self.log(

                'No valid recipient email found in To/Cc: %s'

                % email_address)

            return Milter.CONTINUE

        # Ensure the user replied to his/her own notification, not that

        # they are trying to forge their ID into someone else's

        salt = _config.get('SALT_EMAIL')

        from_email = clean_item(msg['From'])

        session = pagure.lib.query.create_session(_config['DB_URL'])

        try:

            user = pagure.lib.query.get_user(session, from_email)

        except:

            self.log(

                "Could not find an user in the DB associated with %s" %

                from_email)

            session.remove()

            return Milter.CONTINUE

        hashes = []

        for email_obj in user.emails:

            m = hashlib.sha512('%s%s%s' % (msg_id, salt, email_obj.email))

            hashes.append(m.hexdigest())

        tohash = email_address.split('@')[0].split('+')[-1]

        if tohash not in hashes:

            self.log('hash list: %s' % hashes)

            self.log('tohash:    %s' % tohash)

            self.log('Hash does not correspond to the destination')

            session.remove()

            return Milter.CONTINUE

        if msg['From'] and msg['From'] == _config.get('FROM_EMAIL'):

            self.log("Let's not process the email we send")

            session.remove()

            return Milter.CONTINUE

        msg_id = clean_item(msg_id)

        if msg_id and '-ticket-' in msg_id:

            self.log('Processing issue')

            session.remove()

            return self.handle_ticket_email(msg, msg_id)

        elif msg_id and '-pull-request-' in msg_id:

            self.log('Processing pull-request')

            session.remove()

            return self.handle_request_email(msg, msg_id)

        else:

            self.log('Not a pagure ticket or pull-request email, let it go')

            session.remove()

            return Milter.CONTINUE

    def handle_ticket_email(self, emailobj, msg_id):

        ''' Add the email as a comment on a ticket. '''

        uid = msg_id.split('-ticket-')[-1].split('@')[0]

        parent_id = None

        if '-' in uid:

            uid, parent_id = uid.rsplit('-', 1)

        if '/' in uid:

            uid = uid.split('/')[0]

        self.log('uid %s' % uid)

        self.log('parent_id %s' % parent_id)

        data = {

            'objid': uid,

            'comment': get_email_body(emailobj),

            'useremail': clean_item(emailobj['From']),

        }

        url = _config.get('APP_URL')

        if url.endswith('/'):

            url = url[:-1]

        url = '%s/pv/ticket/comment/' % url

        self.log('Calling URL: %s' % url)

        req = requests.put(url, data=data)

        if req.status_code == 200:

            self.log('Comment added')

            return Milter.ACCEPT

        self.log('Could not add the comment to ticket to pagure')

        self.log(req.text)

        return Milter.CONTINUE

    def handle_request_email(self, emailobj, msg_id):

        ''' Add the email as a comment on a request. '''

        uid = msg_id.split('-pull-request-')[-1].split('@')[0]

        parent_id = None

        if '-' in uid:

            uid, parent_id = uid.rsplit('-', 1)

        if '/' in uid:

            uid = uid.split('/')[0]

        self.log('uid %s' % uid)

        self.log('parent_id %s' % parent_id)

        data = {

            'objid': uid,

            'comment': get_email_body(emailobj),

            'useremail': clean_item(emailobj['From']),

        }

        url = _config.get('APP_URL')

        if url.endswith('/'):

            url = url[:-1]

        url = '%s/pv/pull-request/comment/' % url

        self.log('Calling URL: %s' % url)

        req = requests.put(url, data=data)

        if req.status_code == 200:

            self.log('Comment added on PR')

            return Milter.ACCEPT

        self.log('Could not add the comment to PR to pagure')

        self.log(req.text)

        return Milter.CONTINUE

def background():

    while True:

        t = logq.get()

        if not t:

            break

        msg, id, ts = t

        print("%s [%d]" % (time.strftime(

            '%Y%b%d %H:%M:%S', time.localtime(ts)), id))

        # 2005Oct13 02:34:11 [1] msg1 msg2 msg3 ...

        for i in msg:

            print(i,)

        print

def main():

    bt = Thread(target=background)

    bt.start()

    socketname = "/var/run/pagure/paguresock"

    timeout = 600

    # Register to have the Milter factory create instances of your class:

    Milter.factory = PagureMilter

    print("%s pagure milter startup" % time.strftime('%Y%b%d %H:%M:%S'))

    sys.stdout.flush()

    Milter.runmilter("paguremilter", socketname, timeout)

    logq.put(None)

    bt.join()

    print("%s pagure milter shutdown" % time.strftime('%Y%b%d %H:%M:%S'))

if __name__ == "__main__":

    main()