# 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/>.

import ifcopenshell
import ifcopenshell.api.context
import ifcopenshell.api.geometry
import ifcopenshell.util.element


def remove_context(file: ifcopenshell.file, context: ifcopenshell.entity_instance) -> None:
    """Removes an IfcGeometricRepresentationContext

    Any representation geometry that is assigned to the context is also
    removed. If a context is removed, then any subcontexts are also removed.

    :param context: The IfcGeometricRepresentationContext entity to remove
    :return: None

    Example:

    .. code:: python

        model = ifcopenshell.api.context.add_context(model, context_type="Model")
        # Revit had a bug where they incorrectly called the body representation a "Facetation"
        body = ifcopenshell.api.context.add_context(model,
            context_type="Model", context_identifier="Facetation", target_view="MODEL_VIEW", parent=model
        )

        # Let's just get rid of it completely
        ifcopenshell.api.context.remove_context(model, context=body)
    """
    for subcontext in context.HasSubContexts:
        ifcopenshell.api.context.remove_context(file, context=subcontext)

    if getattr(context, "ParentContext", None):
        new = context.ParentContext
        for inverse in file.get_inverse(context):
            if inverse.is_a("IfcCoordinateOperation"):
                # Trick to make sure the coordinate operation is not referenced
                # by a context so we can delete it safely
                inverse.SourceCRS = inverse.TargetCRS
                ifcopenshell.util.element.remove_deep2(file, inverse)
            else:
                ifcopenshell.util.element.replace_attribute(inverse, context, new)
        file.remove(context)
    else:
        representations_in_context = context.RepresentationsInContext
        file.remove(context)
        for rep in representations_in_context:
            for element in ifcopenshell.util.element.get_elements_by_representation(file, rep):
                ifcopenshell.api.geometry.unassign_representation(file, product=element, representation=rep)
            ifcopenshell.api.geometry.remove_representation(file, representation=rep)
