# Copyright 2020 Creu Blanca
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
from uuid import uuid4

from jinja2.sandbox import SandboxedEnvironment

from odoo import fields, models
from odoo.tools.safe_eval import safe_eval

mako_template_env = SandboxedEnvironment(
    block_start_string="<%",
    block_end_string="%>",
    variable_start_string="${",
    variable_end_string="}",
    comment_start_string="<%doc>",
    comment_end_string="</%doc>",
    line_statement_prefix="%",
    line_comment_prefix="##",
    trim_blocks=True,  # do not output newline after blocks
    autoescape=True,  # XML/HTML automatic escaping
)


class IotTemplate(models.Model):
    _name = "iot.template"
    _inherit = "image.mixin"
    _description = "IoT Template for Device"
    _parent_name = "parent_id"
    _parent_store = True
    _parent_order = "name"

    parent_path = fields.Char(index=True, unaccent=False)
    name = fields.Char(required=True)
    input_ids = fields.One2many(
        "iot.template.input",
        inverse_name="template_id",
    )
    output_ids = fields.One2many(
        "iot.template.output",
        inverse_name="template_id",
    )
    key_ids = fields.One2many("iot.template.key", inverse_name="template_id")
    parent_id = fields.Many2one("iot.template", ondelete="restrict")
    tag_ids = fields.Many2many("iot.device.tag")
    group_id = fields.Many2one("iot.device.group")
    icon = fields.Selection(
        selection=lambda r: r.env["iot.device"]._fields["icon"].selection
    )

    def _get_keys(self, serial):
        if self.parent_id:
            keys = self.parent_id._get_keys(serial)
        else:
            keys = {"serial": serial}
        keys.update({key.key: key._generate_value() for key in self.key_ids})
        return keys

    def apply_template(self, device, keys):
        self.ensure_one()
        new_vals = {}
        if self.group_id and not device.group_id:
            new_vals["group_id"] = self.group_id.id
        if self.tag_ids:
            new_vals["tag_ids"] = [(4, tag_id) for tag_id in self.tag_ids.ids]
        if self.image_1920 and not device.image_1920:
            new_vals["image_1920"] = self.image_1920
        if self.icon and not device.icon:
            new_vals["icon"] = self.icon
        if new_vals:
            device.write(new_vals)
        for element in self.input_ids:
            element._apply_template(device, keys)
        for element in self.output_ids:
            element._apply_template(device, keys)
        if self.parent_id:
            self.parent_id.apply_template(device, keys)


class IotTemplateInput(models.Model):
    _name = "iot.template.input"
    _description = "IoT Input for Template"

    template_id = fields.Many2one("iot.template", required=True)
    name = fields.Char(required=True)
    params = fields.Text()
    call_model_id = fields.Many2one("ir.model")
    call_function = fields.Char(required=True)

    def _apply_template(self, device, keys):
        real_vals = {
            "device_id": device.id,
            "name": self.name,
            "call_function": self.call_function,
            "call_model_id": self.call_model_id.id,
            "template_input_id": self.id,
            "serial": uuid4(),
            "passphrase": uuid4(),
        }
        vals = safe_eval(self.params)
        for key in vals:
            vals[key] = mako_template_env.from_string(vals[key]).render(keys)
        real_vals.update(vals)
        return self.env["iot.device.input"].create(real_vals)


class IotTemplateOutput(models.Model):
    _name = "iot.template.output"
    _description = "Output templates for IoT"

    template_id = fields.Many2one("iot.template", required=True)
    name = fields.Char(required=True)
    communication_system_id = fields.Many2one("iot.communication.system", required=True)
    params = fields.Text()

    def _apply_template(self, device, keys):
        real_vals = {
            "device_id": device.id,
            "name": self.name,
            "communication_system_id": self.communication_system_id.id,
            "template_output_id": self.id,
        }
        vals = safe_eval(self.params or "{}")
        for key in vals:
            vals[key] = mako_template_env.from_string(vals[key]).render(keys)
        real_vals.update(vals)
        return self.env["iot.device.output"].create(real_vals)


class IotTemplateKey(models.Model):
    _name = "iot.template.key"
    _description = "IoT Keys for configuration"

    template_id = fields.Many2one("iot.template", required=True)
    key = fields.Char(required=True)

    def _generate_value(self):
        return uuid4()