# IfcOpenShell - IFC toolkit and geometry engine
# Copyright (C) 2021 Dion Moult <dion@thinkmoult.com>
#
# This file is part of IfcOpenShell.
#
# IfcOpenShell is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# IfcOpenShell is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with IfcOpenShell.  If not, see <http://www.gnu.org/licenses/>.

from typing import Any

import ifcopenshell
import ifcopenshell.api.sequence
import ifcopenshell.util.constraint
import ifcopenshell.util.date
import ifcopenshell.util.resource


def edit_resource_time(
    file: ifcopenshell.file, resource_time: ifcopenshell.entity_instance, attributes: dict[str, Any]
) -> None:
    """Edits the attributes of an IfcResourceTime

    For more information about the attributes and data types of an
    IfcResourceTime, consult the IFC documentation.

    :param resource_time: The IfcResourceTime entity you want to edit
    :param attributes: a dictionary of attribute names and values.
    :return: None

    Example:

    .. code:: python

        # Add our own crew
        crew = ifcopenshell.api.resource.add_resource(model, ifc_class="IfcCrewResource")

        # Add some labour to our crew.
        labour = ifcopenshell.api.resource.add_resource(model,
            parent_resource=crew, ifc_class="IfcLaborResource")

        # Labour resource is quantified in terms of time.
        ifcopenshell.api.resource.add_resource_quantity(model,
            resource=labour, ifc_class="IfcQuantityTime")

        # Store the unit time used in hours
        ifcopenshell.api.resource.edit_resource_quantity(model,
            physical_quantity=time, attributes={"TimeValue": 8.0})

        # Let's imagine we've used the resource for 2 days.
        time = ifcopenshell.api.resource.add_resource_time(model, resource=labour)
        ifcopenshell.api.resource.edit_resource_time(model,
            resource_time=time, attributes={"ScheduleWork": "P16H"})
    """
    usecase = Usecase()
    usecase.file = file
    return usecase.execute(resource_time, attributes)


class Usecase:
    file: ifcopenshell.file

    def execute(self, resource_time: ifcopenshell.entity_instance, attributes: dict[str, Any]) -> None:
        resource = self.get_resource(resource_time)

        # If the user specifies both an end date and a duration, the duration takes priority
        if attributes.get("ScheduleWork", None) and "ScheduleFinish" in attributes.keys():
            del attributes["ScheduleFinish"]
        if attributes.get("ActualWork", None) and "ActualFinish" in attributes.keys():
            del attributes["ActualFinish"]

        for name, value in attributes.items():
            metrics = ifcopenshell.util.constraint.get_metric_constraints(resource, "Usage." + name)
            if metrics and ifcopenshell.util.constraint.is_hard_constraint(metrics[0]):
                continue
            if value:
                if "Start" in name or "Finish" in name or name == "StatusTime":
                    value = ifcopenshell.util.date.datetime2ifc(value, "IfcDateTime")
                elif name == "ScheduleWork" or name == "ActualWork" or name == "RemainingTime":
                    value = ifcopenshell.util.date.datetime2ifc(value, "IfcDuration")
            setattr(resource_time, name, value)
            if name == "ScheduleUsage" and ifcopenshell.util.constraint.get_metric_constraints(
                resource, "Usage.ScheduleWork"
            ):
                task = ifcopenshell.util.resource.get_task_assignments(resource)
                if task:
                    ifcopenshell.api.sequence.calculate_task_duration(self.file, task=task)

    def get_resource(self, resource_time: ifcopenshell.entity_instance) -> ifcopenshell.entity_instance:
        return next(e for e in self.file.get_inverse(resource_time) if e.is_a("IfcResource"))