import logging
from odoo import api, fields, models, _
from odoo.osv import expression

_logger = logging.getLogger(__name__)

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

    @api.model
    def get_allowed_locations_for_mo(self, mo_id=None, picking_type_id=None):
        """
        Public helper for JS to fetch allowed locations. 
        Renamed to be public (no leading underscore) for Odoo 19 RPC accessibility.
        """
        allowed_ids = []
        source_name = "None"
        
        # 1. MO Lookup (Direct Database Query)
        if mo_id:
            mo = self.env['mrp.production'].sudo().browse(mo_id)
            if mo.exists() and mo.allowed_source_location_ids:
                allowed_ids = mo.allowed_source_location_ids.ids
                source_name = f"MO {mo.name}"

        # 2. Picking Type Fallback
        if not allowed_ids and picking_type_id:
            pt = self.env['stock.picking.type'].sudo().browse(picking_type_id)
            if pt.exists():
                if pt.default_location_src_ids:
                    allowed_ids = pt.default_location_src_ids.ids
                    source_name = f"PT {pt.display_name}"
                elif pt.default_location_src_id:
                    allowed_ids = [pt.default_location_src_id.id]
                    source_name = f"PT {pt.display_name} (M21)"

        if allowed_ids:
            # log as ERROR to ensure it appears in the remote server console regardless of level settings
            _logger.error(f"DEBUG_RESTRICT: Identified {len(allowed_ids)} Allowed Locations for {source_name}: {allowed_ids}")
            return allowed_ids

        return []

class StockLot(models.Model):
    _inherit = 'stock.lot'

    @api.model
    def name_search(self, name='', args=None, operator='ilike', limit=100):
        # We KEEP the Lot search override as it is necessary for the lot dropdown selection.
        ctx = self.env.context
        if not ctx.get('skip_location_restriction') and ctx.get('uid'):
            mo_id = (ctx.get('active_mo_id') or ctx.get('default_production_id') or ctx.get('production_id'))
            allowed_ids = self.env['stock.location'].sudo().get_allowed_locations_for_mo(mo_id=mo_id, picking_type_id=ctx.get('default_picking_type_id'))
            if allowed_ids:
                quant_domain = [('location_id', 'child_of', allowed_ids), ('quantity', '>', 0), ('lot_id', '!=', False)]
                if ctx.get('default_product_id'):
                    quant_domain.append(('product_id', '=', ctx.get('default_product_id')))
                
                quants = self.env['stock.quant'].with_context(skip_location_restriction=True).sudo().search(quant_domain)
                args = expression.AND([args or [], [('id', 'in', quants.mapped('lot_id').ids)]])
        return super().name_search(name, args=args, operator=operator, limit=limit)

    @api.model
    def web_search_read(self, domain, specification, offset=0, limit=None, order=None, count_limit=None):
        # We KEEP the Lot search override for the search catalogs as well.
        ctx = self.env.context
        if not ctx.get('skip_location_restriction') and ctx.get('uid'):
            mo_id = (ctx.get('active_mo_id') or ctx.get('default_production_id') or ctx.get('production_id'))
            allowed_ids = self.env['stock.location'].sudo().get_allowed_locations_for_mo(mo_id=mo_id, picking_type_id=ctx.get('default_picking_type_id'))
            if allowed_ids:
                quant_domain = [('location_id', 'child_of', allowed_ids), ('quantity', '>', 0), ('lot_id', '!=', False)]
                if ctx.get('default_product_id'):
                    quant_domain.append(('product_id', '=', ctx.get('default_product_id')))
                
                quants = self.env['stock.quant'].with_context(skip_location_restriction=True).sudo().search(quant_domain)
                domain = expression.AND([domain, [('id', 'in', quants.mapped('lot_id').ids)]])
        return super().web_search_read(domain, specification, offset=offset, limit=limit, order=order, count_limit=count_limit)