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

_logger = logging.getLogger(__name__)

def get_allowed_ids(env, table_name, col_name, user_id):
    # Use SQL to avoid ORM recursion or self-filtering issues
    query = f"SELECT {col_name} FROM {table_name} WHERE user_id = %s"
    env.cr.execute(query, (user_id,))
    return [r[0] for r in env.cr.fetchall()]

class StockWarehouse(models.Model):
    _inherit = 'stock.warehouse'

    @api.model
    def _search(self, domain, offset=0, limit=None, order=None, **kwargs):
        if self.env.context.get('bypass_user_restriction'):
            return super()._search(domain, offset=offset, limit=limit, order=order, **kwargs)
        if not self.env.su and not self.env.user.has_group('base.group_system'):
            allowed_ids = get_allowed_ids(self.env, 'res_users_stock_warehouse_rel', 'warehouse_id', self.env.user.id)
            if allowed_ids:
                domain = Domain(domain or []) & Domain([('id', 'in', allowed_ids)])
        return super()._search(domain, offset=offset, limit=limit, order=order, **kwargs)

class StockPickingType(models.Model):
    _inherit = 'stock.picking.type'

    @api.model
    def _search(self, domain, offset=0, limit=None, order=None, **kwargs):
        if self.env.context.get('bypass_user_restriction'):
            return super()._search(domain, offset=offset, limit=limit, order=order, **kwargs)
        if not self.env.su and not self.env.user.has_group('base.group_system'):
            allowed_ids = get_allowed_ids(self.env, 'res_users_stock_picking_type_rel', 'picking_type_id', self.env.user.id)
            if allowed_ids:
                domain = Domain(domain or []) & Domain([('id', 'in', allowed_ids)])
        return super()._search(domain, offset=offset, limit=limit, order=order, **kwargs)

class StockLocation(models.Model):
    _inherit = 'stock.location'

    @api.model
    def _search(self, domain, offset=0, limit=None, order=None, **kwargs):
        if self.env.context.get('bypass_user_restriction'):
            return super()._search(domain, offset=offset, limit=limit, order=order, **kwargs)
        if not self.env.su and not self.env.user.has_group('base.group_system'):
            allowed_ids = get_allowed_ids(self.env, 'res_users_stock_location_rel', 'location_id', self.env.user.id)
            if allowed_ids:
                restrict_domain = [
                    '|', '|',
                    ('id', 'parent_of', allowed_ids),
                    ('id', 'child_of', allowed_ids),
                    ('usage', 'not in', ['internal', 'transit'])
                ]
                domain = Domain(domain or []) & Domain(restrict_domain)
        return super()._search(domain, offset=offset, limit=limit, order=order, **kwargs)

class MrpWorkcenter(models.Model):
    _inherit = 'mrp.workcenter'

    @api.model
    def _search(self, domain, offset=0, limit=None, order=None, **kwargs):
        if self.env.context.get('bypass_user_restriction'):
            return super()._search(domain, offset=offset, limit=limit, order=order, **kwargs)
        if not self.env.su and not self.env.user.has_group('base.group_system'):
            allowed_ids = get_allowed_ids(self.env, 'res_users_mrp_workcenter_rel', 'workcenter_id', self.env.user.id)
            if allowed_ids:
                domain = Domain(domain or []) & Domain([('id', 'in', allowed_ids)])
        return super()._search(domain, offset=offset, limit=limit, order=order, **kwargs)