# This file was generated with the assistance of an AI coding tool.
import ifcopenshell
import ifcopenshell.api.aggregate
import ifcopenshell.api.owner.settings
import ifcopenshell.api.project
import ifcopenshell.api.root
import ifcopenshell.api.spatial
import ifcopenshell.api.unit
import pytest

import ifcmcp.server as server_mod


@pytest.fixture
def model():
    """Create an IFC4 model with a spatial hierarchy, a wall, and a slab."""
    f = ifcopenshell.api.project.create_file()
    ifcopenshell.api.owner.settings.get_user = lambda ifc: (ifc.by_type("IfcPersonAndOrganization") or [None])[0]
    ifcopenshell.api.owner.settings.get_application = lambda ifc: (ifc.by_type("IfcApplication") or [None])[0]

    project = ifcopenshell.api.root.create_entity(f, ifc_class="IfcProject", name="TestProject")
    ifcopenshell.api.unit.assign_unit(f)

    site = ifcopenshell.api.root.create_entity(f, ifc_class="IfcSite", name="TestSite")
    building = ifcopenshell.api.root.create_entity(f, ifc_class="IfcBuilding", name="TestBuilding")
    storey = ifcopenshell.api.root.create_entity(f, ifc_class="IfcBuildingStorey", name="Ground Floor")

    ifcopenshell.api.aggregate.assign_object(f, products=[site], relating_object=project)
    ifcopenshell.api.aggregate.assign_object(f, products=[building], relating_object=site)
    ifcopenshell.api.aggregate.assign_object(f, products=[storey], relating_object=building)

    wall = ifcopenshell.api.root.create_entity(f, ifc_class="IfcWall", name="Wall001")
    ifcopenshell.api.spatial.assign_container(f, products=[wall], relating_structure=storey)

    slab = ifcopenshell.api.root.create_entity(f, ifc_class="IfcSlab", name="Slab001")
    ifcopenshell.api.spatial.assign_container(f, products=[slab], relating_structure=storey)

    return f


@pytest.fixture
def model_file(model, tmp_path):
    """Write the model fixture to a temp file and return the path."""
    path = tmp_path / "test.ifc"
    model.write(str(path))
    return str(path)


@pytest.fixture(autouse=True)
def reset_server_state():
    """Reset module-level state before each test."""
    server_mod._model = None
    server_mod._model_path = None
    yield
    server_mod._model = None
    server_mod._model_path = None


@pytest.fixture
def loaded_model(model):
    """Set the server module state to an in-memory model (no file path)."""
    server_mod._model = model
    server_mod._model_path = None
    return model