# Copyright 2020 Tecnativa - Carlos Dauden
# Copyright 2020 Tecnativa - Sergio Teruel
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
from flectra import fields, models


class ResPartner(models.Model):
    _inherit = "res.partner"

    partner_delivery_id = fields.Many2one(
        comodel_name="res.partner",
        string="Shipping address",
    )
    partner_invoice_id = fields.Many2one(
        comodel_name="res.partner",
        string="Invoice address",
    )

    def get_address_default_type(self):
        """This will be the extension method for other contact types"""
        return ["delivery", "invoice"]

    def address_get(self, adr_pref=None):
        """Force the delivery or invoice addresses. It will try to default
        to the one set in the commercial partner if any"""
        res = super().address_get(adr_pref)
        adr_pref = adr_pref or []
        default_address_type_list = {
            x for x in adr_pref if x in self.get_address_default_type()
        }
        for partner in self:
            for addr_type in default_address_type_list:
                default_address_id = (
                    partner["partner_{}_id".format(addr_type)]
                    or partner.commercial_partner_id["partner_{}_id".format(addr_type)]
                )
                if default_address_id:
                    res[addr_type] = default_address_id.id
        return res

    def write(self, vals):
        """We want to prevent archived contacts as default addresses"""
        if vals.get("active") is False:
            self.search([("partner_delivery_id", "in", self.ids)]).write(
                {"partner_delivery_id": False}
            )
            self.search([("partner_invoice_id", "in", self.ids)]).write(
                {"partner_invoice_id": False}
            )
        return super().write(vals)