# Copyright 2021 ForgeFlow S.L. (https://forgeflow.com)
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from odoo import api, fields, models


class MaintenanceEquipmentCategory(models.Model):
    _inherit = "maintenance.equipment.category"

    sequence_prefix = fields.Char(
        help="The equipment's sequence will be created using this prefix.",
    )
    sequence_id = fields.Many2one(
        "ir.sequence",
        string="Entry Sequence",
        help="This field contains the information related to the "
        "numbering of the equipments belonging to this category.",
        copy=False,
    )
    sequence_number_next = fields.Integer(
        string="Next Number",
        help="The next sequence number will be used for the next equipment.",
        compute="_compute_seq_number_next",
        inverse="_inverse_seq_number_next",
    )

    @api.model
    def _create_sequence(self, vals):
        """Create new no_gap entry sequence"""
        seq = {
            "name": vals.get("name", False) or self.name,
            "implementation": "no_gap",
            "prefix": (vals.get("sequence_prefix", False) or self.sequence_prefix),
            "padding": 4,
            "number_increment": 1,
            "use_date_range": False,
        }
        seq = self.env["ir.sequence"].create(seq)
        seq_date_range = seq._get_current_sequence()
        seq_date_range.number_next = vals.get("sequence_number_next", 1)
        return seq

    # do not depend on 'sequence_id.date_range_ids', because
    # sequence_id._get_current_sequence() may invalidate it!
    @api.depends("sequence_id.use_date_range", "sequence_id.number_next_actual")
    def _compute_seq_number_next(self):
        """Compute 'sequence_number_next' according to the current sequence
        in use, an ir.sequence or an ir.sequence.date_range.
        """
        for category in self:
            if category.sequence_id:
                sequence = category.sequence_id._get_current_sequence()
                category.sequence_number_next = sequence.number_next_actual
            else:
                category.sequence_number_next = 1

    def _inverse_seq_number_next(self):
        """
        Inverse 'sequence_number_next' to edit the current sequence next number
        """
        for category in self:
            if category.sequence_id and category.sequence_number_next:
                sequence = category.sequence_id._get_current_sequence()
                sequence.sudo().number_next = category.sequence_number_next

    @api.model_create_multi
    def create(self, vals_list):
        for vals in vals_list:
            if not vals.get("sequence_id", False):
                if vals.get("sequence_prefix", False):
                    vals["sequence_id"] = self.sudo()._create_sequence(vals).id
            else:
                vals["sequence_prefix"] = (
                    self.env["ir.sequence"].browse(vals["sequence_id"]).prefix
                )
        result = super().create(vals_list)
        result._compute_equipment_code()
        return result

    def write(self, vals):
        if not vals.get("sequence_id", False):
            if vals.get("sequence_prefix", False):
                vals["sequence_id"] = self.sudo()._create_sequence(vals).id
        else:
            vals["sequence_prefix"] = (
                self.env["ir.sequence"].browse(vals["sequence_id"]).prefix
            )
        result = super(MaintenanceEquipmentCategory, self).write(vals)
        self._compute_equipment_code()
        return result

    @api.onchange("sequence_id")
    def _onchange_sequence_id(self):
        if self.sequence_id:
            self.sequence_prefix = self.sequence_id.prefix

    def _compute_equipment_code(self):
        for category in self:
            if category.sequence_id:
                category_equipments = category.env["maintenance.equipment"].search(
                    [("category_id", "=", category.id)]
                )
                for equipment in category_equipments:
                    if not equipment.serial_no and equipment.category_id.sequence_id:
                        equipment.serial_no = equipment.category_id.sequence_id._next()


class MaintenanceEquipment(models.Model):
    _inherit = "maintenance.equipment"

    @api.model
    def create(self, vals):
        equipment = super(MaintenanceEquipment, self).create(vals)
        if equipment.category_id and not equipment.serial_no:
            sequence_id = (
                self.env["maintenance.equipment.category"]
                .browse(vals["category_id"])
                .sequence_id
            )
            if sequence_id:
                equipment.serial_no = sequence_id._next()
        return equipment

    def write(self, vals):
        result = super(MaintenanceEquipment, self).write(vals)
        for rec in self:
            if rec.category_id and not rec.serial_no and rec.category_id.sequence_id:
                rec.serial_no = rec.category_id.sequence_id._next()
        return result