from odoo import fields, models


class PosPreparationDisplayOrderline(models.Model):
    _name = 'pos_preparation_display.orderline'
    _description = "Point of Sale preparation order line"

    todo = fields.Boolean("Status of the orderline", help="The status of a command line, todo or not")
    internal_note = fields.Char(help="Internal notes written at the time of the order")
    attribute_value_ids = fields.Many2many('product.template.attribute.value', 'pos_pdis_orderline_product_template_attribute_value_rel', string="Selected Attributes")
    product_id = fields.Many2one('product.product', string="Product ID")
    product_quantity = fields.Integer("Quantity of ordered product")
    product_cancelled = fields.Integer("Quantity of cancelled product")
    preparation_display_order_id = fields.Many2one(
        'pos_preparation_display.order', required=True, index=True, ondelete='cascade')

    def change_line_status(self, status):
        orderlines_status = []

        categories = self.mapped('product_id.pos_categ_ids')
        preparation_displays = self.env['pos_preparation_display.display'].search(['|', ('category_ids', 'in', categories.ids), ('category_ids', '=', False)])

        for orderline in self:
            orderline.todo = status[str(orderline.id)]
            orderlines_status.append({
                'id': orderline.id,
                'todo': orderline.todo
            })

        self.env['bus.bus']._sendmany([
            [
                f'preparation_display-{preparation_display.access_token}',
                'change_orderline_status',
                {
                    'preparation_display_id': preparation_display.id,
                    'status': orderlines_status
                }
            ]
            for preparation_display in preparation_displays
        ])

        return True

    def send_stricked_line_to_next_stage(self, preparation_display_id):
        order = self.preparation_display_order_id
        preparation_display = self.env['pos_preparation_display.display'].browse(preparation_display_id)

        stage_ids = preparation_display.stage_ids
        current_stage_id = order.order_stage_ids.filtered(lambda x: x.stage_id in stage_ids)[-1]
        current_stage_index = stage_ids.ids.index(current_stage_id.stage_id.id)
        next_stage_id = stage_ids.ids[current_stage_index + 1]

        new_order = self.env['pos_preparation_display.order'].create({
            'displayed': True,
            'pos_order_id': order.pos_order_id.id,
            'pos_config_id': order.pos_config_id.id,
        })

        new_order.order_stage_ids.create({
            'preparation_display_id': preparation_display_id,
            'stage_id': next_stage_id,
            'order_id': new_order.id,
            'done': False
        })

        category_ids = set()
        for record in self:
            record.todo = True
            record.preparation_display_order_id = new_order.id
            category_ids.update(record.product_id.pos_categ_ids.ids)

        preparation_displays = self.env['pos_preparation_display.display'].search([
            '&',
            '|', ('pos_config_ids', '=', False),
            ('pos_config_ids', 'in', [order.pos_config_id.id]),
            '|', ('category_ids', 'in', list(category_ids)),
            ('category_ids', '=', False)])

        for p_dis in preparation_displays:
            p_dis._send_load_orders_message()
        return new_order.id