from odoo import models


class MailThread(models.AbstractModel):
    _inherit = 'mail.thread'

    def _message_get_suggested_recipients(self):
        """ Returns suggested recipients for ids. Those are a list of
        tuple (partner_id, partner_name, reason), to be managed by Chatter.

        This Studio override adds the field 'x_studio_partner_id' in the auto-suggested
        list."""
        result = super(MailThread, self)._message_get_suggested_recipients()
        # TODO: also support x_studio_user_id?
        field = self._fields.get('x_studio_partner_id')
        if field and field.type == 'many2one' and field.comodel_name == 'res.partner':
            for obj in self:
                if not obj.x_studio_partner_id:
                    continue
                obj._message_add_suggested_recipient(result, partner=obj.x_studio_partner_id, reason=self._fields['x_studio_partner_id'].string)
        return result

    def _sms_get_partner_fields(self):
        """Include partner field set automatically by studio as an SMS recipient."""
        fields = super()._sms_get_partner_fields()
        field = self._fields.get('x_studio_partner_id')
        if field and field.type == 'many2one' and field.comodel_name == 'res.partner':
            fields.append('x_studio_partner_id')
        return fields