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

from odoo import api, models


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

    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()