# Part of Odoo. See LICENSE file for full copyright and licensing details.

from odoo import api, models
from odoo.fields import Domain


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

    @api.model
    def _get_frontend_writable_fields(self):
        """Define the fields a portal/public user can change on their contact and address records.

        :rtype: set
        """
        return {
            'name', 'phone', 'email', 'street', 'street2', 'city', 'state_id', 'country_id', 'zip',
            'zipcode', 'vat', 'parent_name',
        }

    def _can_edit_country(self):
        self.ensure_one()
        return True

    # TODO: Rename me in master
    def can_edit_vat(self):
        """Return whether the commercial fields of the partner can be edited by the customer.

        Commercial fields are synced between the parent (commercial entity) and the children. Only
        the commercial entity should be able to edit it (as in backend).
        """
        self.ensure_one()
        return self._is_individual_contact()

    def _is_individual_contact(self):
        """Return whether contact is part of actual company or not."""
        if not self.parent_id:
            return True

        contact_childs = self.commercial_partner_id.child_ids.filtered(
            lambda child: child.type == "contact"
        )
        return (
            len(contact_childs) == 1
            and contact_childs == self
            and not self.commercial_partner_id.user_ids
        )

    def _can_be_edited_by_current_customer(self, **kwargs):
        """Return whether partner can be edited by current user."""
        self.ensure_one()
        current_partner = self._get_current_partner(**kwargs)
        if self == current_partner:
            return True
        children_partner_ids = self.env['res.partner']._search([
            ('id', 'child_of', current_partner.commercial_partner_id.id),
            ('type', 'in', ('invoice', 'delivery', 'other')),
        ])
        return self.id in children_partner_ids

    @api.model
    def _get_current_partner(self, **kwargs):
        """ Get main partner of the current user base on logged in user and kwargs. """
        if self.env.user._is_public():
            return self.env['res.partner']
        return self.env.user.partner_id

    def _get_delivery_address_domain(self):
        return Domain([
            ('id', 'child_of', self.ids),
            '|', ('type', 'in', ['delivery', 'other']), ('id', '=', self.id),
        ])