# 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 api, fields, models


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_type = fields.Selection(
        selection=lambda self: self._get_variation_type(),
        required=True,
        default=lambda self: self._get_default_variation_type(),
    )
    variation_percent = fields.Float(
        digits="Product Price",
        string="Variation %",
    )
    fixed_price = fields.Float(digits="Product Price")

    @api.model
    def _get_variation_type(self):
        return [
            ("percentage", "Percentage"),
            ("fixed", "Fixed Price"),
        ]

    @api.model
    def _get_default_variation_type(self):
        return "percentage"

    def _get_new_price(self, line):
        """Get the price depending the change type chosen"""
        if self.variation_type == "percentage":
            return line.price_unit * (1.0 + self.variation_percent / 100.0)
        elif self.variation_type == "fixed":
            return self.fixed_price
        return line.price_unit

    def _get_new_line_value(self, line):
        self.ensure_one()
        return line._prepare_value_for_plan_successor_price(
            self.date_start,
            self.date_end,
            line.is_auto_renew,
            self._get_new_price(line),
            False,
        )

    def _get_old_line_date_end(self, line):
        return self.date_start - relativedelta(days=1)

    def action_apply(self):
        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):
            date_end = self._get_old_line_date_end(line)
            line.stop(date_end)
            new_line = line.copy(self._get_new_line_value(line))
            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")
            .with_context(date_start=self.date_start)
            .filtered("price_can_be_revised")
        )
        return to_revise