1# -*- coding: utf-8 -*-
2# Part of Odoo. See LICENSE file for full copyright and licensing details.
3
4import re
5import werkzeug.urls
6
7from odoo import api, fields, models, tools
8
9
10class MailMail(models.Model):
11    """Add the mass mailing campaign data to mail"""
12    _inherit = ['mail.mail']
13
14    mailing_id = fields.Many2one('mailing.mailing', string='Mass Mailing')
15    mailing_trace_ids = fields.One2many('mailing.trace', 'mail_mail_id', string='Statistics')
16
17    @api.model_create_multi
18    def create(self, values_list):
19        """ Override mail_mail creation to create an entry in mail.mail.statistics """
20        # TDE note: should be after 'all values computed', to have values (FIXME after merging other branch holding create refactoring)
21        mails = super(MailMail, self).create(values_list)
22        for mail, values in zip(mails, values_list):
23            if values.get('mailing_trace_ids'):
24                mail.mailing_trace_ids.write({'message_id': mail.message_id})
25        return mails
26
27    def _get_tracking_url(self):
28        base_url = self.env['ir.config_parameter'].sudo().get_param('web.base.url')
29        token = tools.hmac(self.env(su=True), 'mass_mailing-mail_mail-open', self.id)
30        return werkzeug.urls.url_join(base_url, 'mail/track/%s/%s/blank.gif' % (self.id, token))
31
32    def _send_prepare_body(self):
33        """ Override to add the tracking URL to the body and to add
34        trace ID in shortened urls """
35        # TDE: temporary addition (mail was parameter) due to semi-new-API
36        self.ensure_one()
37        body = super(MailMail, self)._send_prepare_body()
38
39        if self.mailing_id and body and self.mailing_trace_ids:
40            for match in re.findall(tools.URL_REGEX, self.body_html):
41                href = match[0]
42                url = match[1]
43
44                parsed = werkzeug.urls.url_parse(url, scheme='http')
45
46                if parsed.scheme.startswith('http') and parsed.path.startswith('/r/'):
47                    new_href = href.replace(url, url + '/m/' + str(self.mailing_trace_ids[0].id))
48                    body = body.replace(href, new_href)
49
50            # generate tracking URL
51            tracking_url = self._get_tracking_url()
52            body = tools.append_content_to_html(
53                body,
54                '<img src="%s"/>' % tracking_url,
55                plaintext=False,
56            )
57
58        body = self.env['mail.render.mixin']._replace_local_links(body)
59
60        return body
61
62    def _send_prepare_values(self, partner=None):
63        # TDE: temporary addition (mail was parameter) due to semi-new-API
64        res = super(MailMail, self)._send_prepare_values(partner)
65        base_url = self.env['ir.config_parameter'].sudo().get_param('web.base.url').rstrip('/')
66        if self.mailing_id and res.get('body') and res.get('email_to'):
67            emails = tools.email_split(res.get('email_to')[0])
68            email_to = emails and emails[0] or False
69
70            urls_to_replace = [
71               (base_url + '/unsubscribe_from_list', self.mailing_id._get_unsubscribe_url(email_to, self.res_id)),
72               (base_url + '/view', self.mailing_id._get_view_url(email_to, self.res_id))
73            ]
74
75            for url_to_replace, new_url in urls_to_replace:
76                if url_to_replace in res['body']:
77                    res['body'] = res['body'].replace(url_to_replace, new_url if new_url else '#')
78        return res
79
80    def _postprocess_sent_message(self, success_pids, failure_reason=False, failure_type=None):
81        mail_sent = not failure_type  # we consider that a recipient error is a failure with mass mailling and show them as failed
82        for mail in self:
83            if mail.mailing_id:
84                if mail_sent is True and mail.mailing_trace_ids:
85                    mail.mailing_trace_ids.write({'sent': fields.Datetime.now(), 'exception': False})
86                elif mail_sent is False and mail.mailing_trace_ids:
87                    mail.mailing_trace_ids.write({'exception': fields.Datetime.now(), 'failure_type': failure_type})
88        return super(MailMail, self)._postprocess_sent_message(success_pids, failure_reason=failure_reason, failure_type=failure_type)
89