from odoo import api, fields, models, Command


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

    equipment_count = fields.Integer(compute='_compute_equipment_count', string='Equipment Count')

    owned_equipment_ids = fields.One2many(
        comodel_name="bemade_fsm.equipment",
        compute="_compute_owned_equipment_ids",
        string="Owned Equipments"
    )

    equipment_ids = fields.One2many(
        comodel_name='bemade_fsm.equipment',
        inverse_name='partner_location_id',
        string='Site Equipment'
    )

    is_site_contact = fields.Boolean(
        string='Is a site contact',
        compute="_compute_is_site_contact",
        search="_search_is_site_contact",
    )

    site_ids = fields.Many2many(
        string='Work Sites',
        comodel_name='res.partner',
        relation='res_partner_site_contact_rel',
        column1='site_contact_id',
        column2='site_id',
        tracking=True
    )

    site_contacts = fields.Many2many(
        string='Site Contacts',
        comodel_name='res.partner',
        relation='res_partner_site_contact_rel',
        column1='site_id',
        column2='site_contact_id',
        domain=[('is_company', '=', False)],
        tracking=True
    )

    work_order_contacts = fields.Many2many(
        string='Work Order Recipients',
        comodel_name='res.partner',
        relation='res_partner_work_order_contacts_rel',
        column1='res_partner_id',
        column2='work_order_contact_id',
        domain=[('is_company', '=', False)],
        tracking=True
    )

    @api.depends(
        'equipment_ids',
        'child_ids.company_type',
        'child_ids.equipment_ids'
    )
    def _compute_owned_equipment_ids(self):
        for rec in self:
            ids = rec.equipment_ids | rec.child_ids.filtered(
                lambda l: l.company_type == 'company').mapped('equipment_ids')
            rec.owned_equipment_ids = ids or False

    @api.depends('site_ids')
    def _compute_is_site_contact(self):
        for rec in self:
            rec.is_site_contact = rec.site_ids is not False

    @api.depends('equipment_ids')
    def _compute_equipment_count(self):
        for rec in self:
            all_equipment_ids = self.env['bemade_fsm.equipment'].search(
                [('partner_location_id', '=', rec.id)])
            rec.equipment_count = len(all_equipment_ids)

    @api.model
    def _search_is_site_contact(self, operator, value):
        return [('site_contacts', '!=', False)]