import unreal


MAP_PATH = "/Game/Agrarian/Maps/L_GroundZeroTerrain_Test"
MATERIALS = {
    "wood": "/Game/Agrarian/Materials/M_AGR_GZ_Wood_Resource",
    "fiber": "/Game/Agrarian/Materials/M_AGR_GZ_Fiber_Resource",
    "food": "/Game/Agrarian/Materials/M_AGR_GZ_EdiblePlant_Resource",
    "stone": "/Game/Agrarian/Materials/M_AGR_GZ_Stone_Sandstone",
}

EXPECTED_RESOURCE_LABELS = {
    "wood": [
        "AGR_DemoWoodResource_01",
        "AGR_GZ_Wood_CoastalScrub_01",
        "AGR_GZ_Wood_CoastalScrub_02",
        "AGR_GZ_Wood_Hillside_03",
    ],
    "fiber": [
        "AGR_DemoFiberResource_01",
        "AGR_GZ_Fiber_Grassland_01",
        "AGR_GZ_Fiber_Grassland_02",
        "AGR_GZ_Fiber_Scrub_03",
        "AGR_GZ_Fiber_DrainageCandidate_04",
    ],
    "food": [
        "AGR_GZ_EdiblePlant_CoastalScrub_01",
        "AGR_GZ_EdiblePlant_Grassland_02",
        "AGR_GZ_EdiblePlant_DrainageCandidate_03",
    ],
    "stone": [
        "AGR_GZ_Stone_Slope_01",
        "AGR_GZ_Stone_Slope_02",
        "AGR_GZ_Stone_ExposedTerrain_03",
        "AGR_GZ_Stone_ValleyEdge_04",
    ],
}


def material_path(material):
    if not material:
        return ""
    return material.get_path_name().split(".", 1)[0]


def get_actor_label(actor):
    try:
        return actor.get_actor_label()
    except Exception:
        return actor.get_name()


def resource_item_id(actor):
    item_asset = actor.get_editor_property("yield_item_definition")
    definition = item_asset.get_editor_property("definition") if item_asset else None
    return str(definition.get_editor_property("item_id")) if definition else ""


def main():
    if not unreal.EditorLevelLibrary.load_level(MAP_PATH):
        raise RuntimeError(f"Could not load map: {MAP_PATH}")

    for path in MATERIALS.values():
        if not unreal.EditorAssetLibrary.load_asset(path):
            raise RuntimeError(f"Missing required resource material: {path}")

    actors_by_label = {get_actor_label(actor): actor for actor in unreal.EditorLevelLibrary.get_all_level_actors()}
    failures = []

    for expected_item_id, labels in EXPECTED_RESOURCE_LABELS.items():
        for label in labels:
            actor = actors_by_label.get(label)
            if not actor:
                failures.append(f"{label} missing")
                continue

            actual_item_id = resource_item_id(actor)
            if actual_item_id != expected_item_id:
                failures.append(f"{label} expected {expected_item_id}, got {actual_item_id}")

            if actor.get_editor_property("remaining_harvests") <= 0:
                failures.append(f"{label} has no remaining harvests")

            persistence_node_id = str(actor.get_editor_property("persistence_node_id"))
            if persistence_node_id != label:
                failures.append(f"{label} persistence node id expected {label}, got {persistence_node_id}")

            mesh_components = actor.get_components_by_class(unreal.StaticMeshComponent)
            if not mesh_components:
                failures.append(f"{label} has no static mesh component for MVP resource readability")
                continue
            expected_material = MATERIALS[expected_item_id]
            if not any(material_path(component.get_material(0)) == expected_material for component in mesh_components):
                failures.append(f"{label} expected readable material {expected_material}")

    if failures:
        raise RuntimeError("Ground Zero resource verification failed: " + "; ".join(failures))

    unreal.log(
        "Ground Zero resource verification complete: "
        f"{len(EXPECTED_RESOURCE_LABELS['wood'])} wood, "
        f"{len(EXPECTED_RESOURCE_LABELS['fiber'])} fiber, "
        f"{len(EXPECTED_RESOURCE_LABELS['food'])} edible plant, "
        f"{len(EXPECTED_RESOURCE_LABELS['stone'])} stone nodes."
    )


main()