# Copyright 2016 Therp BV <http://therp.nl>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from odoo import fields, models, tools


class MailTemplate(models.Model):
    _inherit = "mail.template"

    body_type = fields.Selection(
        [("jinja2", "Jinja2"), ("qweb", "QWeb")],
        "Body templating engine",
        default="jinja2",
        required=True,
    )
    body_view_id = fields.Many2one("ir.ui.view", domain=[("type", "=", "qweb")])
    body_view_arch = fields.Text(related="body_view_id.arch")

    def generate_email(self, res_ids, fields=None):
        multi_mode = True
        if isinstance(res_ids, int):
            res_ids = [res_ids]
            multi_mode = False
        result = super(MailTemplate, self).generate_email(res_ids, fields=fields)
        for res_id, template in self.get_email_template(res_ids).items():
            if template.body_type == "qweb" and (not fields or "body_html" in fields):
                for record in self.env[template.model].browse(res_id):
                    body_html = template.body_view_id.render(
                        {"object": record, "email_template": template}
                    )
                    # Some wizards, like when sending a sales order, need this
                    # fix to display accents correctly
                    body_html = tools.ustr(body_html)
                    result[res_id]["body_html"] = self.render_post_process(body_html)
                    result[res_id]["body"] = tools.html_sanitize(
                        result[res_id]["body_html"]
                    )
        return result if multi_mode else result[res_ids[0]]