UPC/mailtoticket

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

import base64

import logging
logger = logging.getLogger(__name__)


class Filtre(object):

    def __init__(self, msg=None, tickets=None, identitat=None):
        self.msg = msg
        self.tickets = tickets
        self.identitat = identitat

    def set_mail(self, msg):
        self.msg = msg

    def set_tickets(self, tickets):
        self.tickets = tickets

    def set_identitat(self, identitat):
        self.identitat = identitat

    def es_aplicable(self):
        return False

    def filtrar(self):
        return

    def get_uid(self):
        # Comprovem primer el reply-to pel cas de Prisma en que tenim
        # un formulari que envia mails com un usuari genèric i amb reply-to
        # a l'adreça de l'usuari que ha fet la petició.
        # TODO: intentar fer aixo configurable
        uid = None
        if self.msg.get_reply_to() is not None:
            uid = self.identitat.obtenir_uid(self.msg.get_reply_to())
        if uid is None:
            uid = self.identitat.obtenir_uid(self.msg.get_from())
        return uid

    def get_uid_addicionals(self):
        return [self.identitat.obtenir_uid(mail) for mail in self.msg.get_cc()]

    def codificar_base_64_si_cal(self, attachment):
        if attachment['Content-Transfer-Encoding'] == 'base64':
            return attachment.get_payload()
        else:
            try:
                return base64.b64encode(attachment.get_payload())
            except TypeError:
                return base64.b64encode(attachment.get_payload().encode())
            except Exception:
                logger.error("Error codificant l'attachment: {}".format(
                             attachment.get_payload()))
                raise

    def afegir_attachments_canviant_body(self, ticket_id, username, body):
        (cids, ids) = self.afegir_attachments(ticket_id, username)
        body = self.tractar_attachments_inline(body, cids)
        body = self.afegir_links_attachments(body, ids)
        return body

    def afegir_attachments(self, ticket_id, username):
        logger.info("Tractem attachments del ticket %s" % ticket_id)
        i = 0
        cids = {}
        ids = {}
        for a in self.msg.get_attachments():
            ctype = a.get_content_type()
            fitxer = a.get_filename()
            cid = a.get('Content-ID')
            i += 1
            if fitxer is None:
                fitxer = 'attach%d.%s' % (i, ctype.split("/")[1])

            logger.info("Afegim attachment: %s / %s" % (fitxer, cid))
            codi_annex = self.tickets.annexar_fitxer_tiquet(
                ticket_id,
                username,
                fitxer,
                self.codificar_base_64_si_cal(a)
            )
            if cid is not None:
                cids[cid[1:-1]] = codi_annex
            else:
                ids[codi_annex] = a

        return (cids, ids)

    def tractar_attachments_inline(self, html, cids):
        for cid in cids:
            id_attachment = cids[cid]
            html = html.replace("cid:" + cid,
                                self.url_attachment(id_attachment))

        return html

    def afegir_links_attachments(self, html, ids):
        if len(ids) == 0:
            return html

        html += "<br><br>Attachments:<ul>"
        for id_attachment in ids:
            a = ids[id_attachment]
            url = self.url_attachment(id_attachment)
            html += "<li><a href=\"%s\">%s (%s)</a>" % (
                url,
                a.get_filename(),
                a.get_content_type()
            )

        html += "</ul>"
        return html

    def url_attachment(self, id_attachment):
        return "/tiquetsusuaris/control/file?fileId=%s" % id_attachment