# Copyright 2019 Alexandre Díaz
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

from odoo import models, api


class MailResendMessage(models.TransientModel):
    _inherit = "mail.resend.message"

    @api.model
    def default_get(self, fields):
        rec = super().default_get(fields)
        message_id = self._context.get('mail_message_to_resend')
        if message_id:
            MailMessageObj = self.env['mail.message']
            mail_message_id = MailMessageObj.browse(message_id)
            failed_states = MailMessageObj.get_failed_states()
            tracking_ids = mail_message_id.mail_tracking_ids.filtered(
                lambda x: x.state in failed_states)
            if any(tracking_ids):
                partner_ids = [(0, 0, {
                    "partner_id": tracking.partner_id.id,
                    "name": tracking.partner_id.name,
                    "email": tracking.partner_id.email,
                    "resend": True,
                    "message": tracking.error_description,
                }) for tracking in tracking_ids]
                rec['partner_ids'].extend(partner_ids)
        return rec

    @api.multi
    def resend_mail_action(self):
        for wizard in self:
            to_send = wizard.partner_ids.filtered("resend").mapped(
                "partner_id")
            if to_send:
                # Set as reviewed
                wizard.mail_message_id.mail_tracking_needs_action = False
                # Reset mail.tracking.email state
                tracking_ids = wizard.mail_message_id.mail_tracking_ids\
                    .filtered(lambda x: x.partner_id in to_send)
                tracking_ids.write({'state': False})
                # Send bus notifications to update Discuss and
                # mail_failed_messages widget
                notification = {
                    'type': 'toggle_tracking_status',
                    'message_ids': [self.mail_message_id.id],
                    'needs_actions': False
                }
                self.env['bus.bus'].sendone(
                    (self._cr.dbname, 'res.partner',
                     self.env.user.partner_id.id),
                    notification)
        super().resend_mail_action()