from odoo import models, fields, api


class FSMVisit(models.Model):
    _name = "bemade_fsm.visit"
    _description = "Represents a single visit by assigned service personnel."

    label = fields.Text(
        string="Label",
        required=True,
        related="so_section_id.name",
        readonly=False,
        copy=True,
    )

    approx_date = fields.Date(string="Approximate Date", copy=False)

    so_section_id = fields.Many2one(
        comodel_name="sale.order.line",
        string="Sale Order Section",
        help=(
            "The section on the sale order that represents the labour and parts for"
            " this visit"
        ),
        ondelete="cascade",
    )

    sale_order_id = fields.Many2one(
        comodel_name="sale.order",
        string="Sales Order",
        readonly=True,
    )

    is_completed = fields.Boolean(
        string="Completed", related="so_section_id.is_fully_delivered"
    )

    is_invoiced = fields.Boolean(
        string="Invoiced", related="so_section_id.is_fully_delivered_and_invoiced"
    )

    summarized_equipment_ids = fields.Many2many(
        comodel_name="fsm.equipment",
        string="Equipment to Service",
        compute="_compute_summarized_equipment_ids",
    )

    task_id = fields.Many2one(
        comodel_name="project.task",
        compute="_compute_task_id",
        string="Service Visit",
    )

    task_ids = fields.One2many(comodel_name="project.task", inverse_name="visit_id")

    visit_no = fields.Integer(
        compute="_compute_visit_no",
    )

    @api.depends("task_ids")
    def _compute_task_id(self):
        for rec in self:
            rec.task_id = rec.task_ids and rec.task_ids[0]

    @api.depends("so_section_id", "sale_order_id.summary_equipment_ids")
    def _compute_summarized_equipment_ids(self):
        for rec in self:
            lines = rec.so_section_id.get_section_line_ids()
            equipment_ids = []
            for line in lines:
                for equipment in line.equipment_ids:
                    equipment_ids.append(equipment)
            rec.summarized_equipment_ids = equipment_ids

    @api.model_create_multi
    def create(self, vals_list):
        recs = super().create(vals_list)
        for i, rec in enumerate(recs.filtered(lambda visit: not visit.so_section_id)):
            rec.so_section_id = rec.env["sale.order.line"].create(
                {
                    "order_id": rec.sale_order_id.id,
                    "display_type": "line_section",
                    "name": vals_list[i].get("label", False),
                }
            )
        return recs

    @api.depends(
        "so_section_id",
        "sale_order_id",
        "sale_order_id.visit_ids",
        "sale_order_id.visit_ids.so_section_id",
        "sale_order_id.visit_ids.so_section_id.sequence",
    )
    def _compute_visit_no(self):
        for rec in self:
            ordered_visit_lines = self.sale_order_id.visit_ids.so_section_id.sorted(
                "sequence"
            )
            # Just a straight O(n) search here since n will always be relatively small
            for index, line in enumerate(ordered_visit_lines):
                if line == rec.so_section_id:
                    rec.visit_no = index + 1
                    return