# Copyright 2015 ACSONE SA/NV
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from odoo import api, fields, models


class MailComposeMessage(models.TransientModel):
    _inherit = "mail.compose.message"

    @api.model
    def default_get(self, fields_list):
        res = super().default_get(fields_list)
        if (
            "can_attach_attachment" not in res
            and res.get("model")
            and res.get("res_ids")
            and res.get("composition_mode", "") != "mass_mail"
        ):
            res["can_attach_attachment"] = True  # pragma: no cover
        return res

    can_attach_attachment = fields.Boolean()
    object_attachment_ids = fields.Many2many(
        comodel_name="ir.attachment",
        relation="mail_compose_message_ir_attachments_object_rel",
        column1="wizard_id",
        column2="attachment_id",
        string="Object Attachments",
    )
    display_object_attachment_ids = fields.One2many(
        comodel_name="ir.attachment",
        compute="_compute_display_object_attachment_ids",
    )

    @api.depends("res_ids", "model")
    def _compute_display_object_attachment_ids(self):
        for composer in self:
            res_ids = self._evaluate_res_ids()
            model = self.model
            if model and res_ids:
                attachments = self.env["ir.attachment"].search(
                    [
                        ("res_model", "=", model),
                        ("res_id", "in", res_ids),
                    ]
                )
                composer.display_object_attachment_ids = attachments
            else:
                composer.display_object_attachment_ids = False

    def _prepare_mail_values(self, res_ids):
        res = super()._prepare_mail_values(res_ids)
        if self.object_attachment_ids.ids and self.model and len(res_ids) == 1:
            res[res_ids[0]].setdefault("attachment_ids", []).extend(
                self.object_attachment_ids.ids
            )
        return res