# Copyright 2019 Tecnativa - Vicent Cubells
# Copyright 2019 Tecnativa - Carlos Dauden
# Copyright 2020 ACSONE SA/NV
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).

from dateutil.relativedelta import relativedelta

from odoo import fields, models

import odoo.addons.decimal_precision as dp


class ContractPriceRevisionWizard(models.TransientModel):
    """ Update contract price based on percentage variation """

    _name = "contract.price.revision.wizard"
    _description = "Wizard to update price based on percentage variation"

    date_start = fields.Date(required=True,)
    date_end = fields.Date()
    variation_percent = fields.Float(
        digits=dp.get_precision("Product Price"), required=True, string="Variation %",
    )

    def _get_new_line_value(self, line):
        self.ensure_one()
        return {
            "date_start": self.date_start,
            "last_date_invoiced": False,
            "date_end": self.date_end,
            "predecessor_contract_line_id": line.id,
            "price_unit": line.price_unit * (1.0 + self.variation_percent / 100.0),
        }

    def action_apply(self):
        ContractLine = self.env["contract.line"]
        active_ids = self.env.context.get("active_ids")
        contracts = self.env["contract.contract"].browse(active_ids)
        for line in self._get_contract_lines_to_revise(contracts):
            line.update({"date_end": self.date_start - relativedelta(days=1)})
            # As copy or copy_data are trigerring constraints, don't use them
            new_line = line.new(line._cache)
            new_line.update(self._get_new_line_value(line))
            new_line._onchange_date_start()
            new_line = ContractLine.create(new_line._convert_to_write(new_line._cache))
            line.update({"successor_contract_line_id": new_line.id})
        action = self.env["ir.actions.act_window"].for_xml_id(
            "contract", "action_customer_contract"
        )
        action["domain"] = [("id", "in", active_ids)]
        return action

    def _get_contract_lines_to_revise(self, contracts):
        self.ensure_one()
        to_revise = contracts.mapped("contract_line_ids").filtered(
            lambda x: not x.automatic_price
            and not x.successor_contract_line_id
            and x.recurring_next_date
            and (not x.date_end or x.date_end >= self.date_start)
        )
        return to_revise