from odoo import _, fields, models
from odoo.exceptions import UserError


class SaleOrder(models.Model):
    _inherit = "sale.order"

    batch_id = fields.Many2one("sale.order.batch", copy=False)

    def action_add_to_batch(self):
        """
        TODO: Add Wizard to choose between batches if more than one existis. Picking the first one for now.
        """
        batch = self.env["sale.order.batch"].search([("state", "=", "open")], limit=1)
        if not batch:
            batch = self.env["sale.order.batch"].create()
        for sale_order in self:
            if not sale_order.batch_id and sale_order.state in ["draft", "sent"]:
                sale_order.batch_id = batch

    def action_view_sale_order_batch(self):
        return {
            "name": _("Sale Order Batch"),
            "type": "ir.actions.act_window",
            "view_type": "form",
            "view_mode": "form",
            "res_model": "sale.order.batch",
            "res_id": self.batch_id.id,
            # 'target': '',
        }

    def action_confirm(self):
        invalid_orders = []
        if not self.env.context.get("bypass_batch", False):
            for order in self:
                if order.batch_id:
                    invalid_orders.append(order.name)
            if invalid_orders:
                raise UserError(_(f"Sale Order belongs to a Batch: {', '.join(invalid_orders)}"))
        return super().action_confirm()

    def write(self, vals):
        if "batch_id" in vals:
            invalid_orders = []
            for order in self:
                if order.state not in ["draft", "sent"]:
                    invalid_orders.append(order.name)
            if invalid_orders:
                raise UserError(_(f"Sale Order not in State Draft or Sent: {', '.join(invalid_orders)}"))
        res = super().write(vals)
        if "batch_id" in vals:
            self.order_line._update_batch_products()
        return res