from odoo import models, fields, api, _, Command


class TaskTemplate(models.Model):
    _name = 'project.task.template'
    _description = "Template for new project tasks"

    @api.model
    def _current_company(self):
        return self.env.company

    name = fields.Char(string="Task Title", required=True)

    description = fields.Html(string="Description")

    assignees = fields.Many2many(
        comodel_name="res.users",
        string="Default Assignees",
        help="Employees assigned to tasks created from this template."
    )

    customer = fields.Many2one(
        comodel_name="res.partner",
        string="Default Customer",
        help="Default customer for tasks created from this template."
    )

    project = fields.Many2one(
        comodel_name="project.project",
        string="Default Project",
        help="Default project for tasks created from this template."
    )

    tags = fields.Many2many(
        comodel_name="project.tags",
        string="Default Tags",
        help="Default tags for tasks created from this template."
    )

    parent = fields.Many2one(
        comodel_name="project.task.template",
        string="Parent Task Template",
        ondelete='cascade'
    )

    subtasks = fields.One2many(
        comodel_name="project.task.template",
        inverse_name="parent",
        string="Subtask Templates"
    )
    sequence = fields.Integer(string="Sequence")

    company_id = fields.Many2one(
        comodel_name="res.company",
        string="Company",
        index=1,
        default=_current_company
    )

    planned_hours = fields.Float("Initially Planned Hours")

    equipment_ids = fields.Many2many(
        comodel_name="bemade_fsm.equipment",
        relation="bemade_fsm_task_template_equipment_rel",
        column1="task_template_id",
        column2="equipment_id",
        string="Equipment to Service",
    )

    def action_open_task(self):
        return {
            'view_mode': 'form',
            'res_model': 'project.task.template',
            'res_id': self.id,
            'type': 'ir.actions.act_window',
            'context': self._context
        }

    @api.onchange('customer')
    def _onchange_customer(self):
        for rec in self:
            new_equipment_ids = [eq.id for eq in rec.equipment_ids if eq.partner_location_id == rec.customer]
            rec.write({'equipment_ids': [Command.set(new_equipment_ids)]})

    def _prepare_new_task_values_from_self(self, project, name=False, parent_id=False):
        vals = {
            'project_id': project.id,
            'name': name or self.name,
            'description': self.description,
            'parent_id': parent_id,
            'user_ids': self.assignees.ids,
            'tag_ids': self.tags.ids,
            'allocated_hours': self.planned_hours,
            'sequence': self.sequence,
            'equipment_ids': [Command.set(self.equipment_ids.ids)] if self.equipment_ids else False,
            'partner_id': project.partner_id and project.partner_id.id,
            'company_id': self.company_id.id,
        }
        return vals

    def create_task_from_self(self, project, name=False, parent_id=False):
        """ Create a project.task from this template and return it. Can be called on a RecordSet of multiple templates.

        :param project: project.project record the task should be added to
        :param name: name for the new task (defaults to template name)
        :param parent_id: parent task for the new task (none by default)
        :return: project.task record created from this template
        """
        tasks = self.env['project.task']
        for rec in self:
            vals = rec._prepare_new_task_values_from_self(project, name, parent_id)
            task = rec.env['project.task'].create(vals)
            rec.subtasks.create_task_from_self(project, name=False, parent_id=task.id)
            tasks |= task
        return tasks