import hashlib
import json
import os

import FreeCAD as App


FREECAD_COMMIT = "0108fd4b4850cc46e625b60e53cea7a7bbe69f8d"


def mesh_snapshot(obj):
    mesh = obj.Mesh
    bounds = mesh.BoundBox
    return {
        "isEmpty": int(mesh.CountFacets) == 0,
        "facets": int(mesh.CountFacets),
        "points": int(mesh.CountPoints),
        "bounds": [round(float(value), 9) for value in (bounds.XMin, bounds.YMin, bounds.ZMin, bounds.XMax, bounds.YMax, bounds.ZMax)],
    }


def object_snapshot(document):
    obj = document.getObject("FileProbe")
    path = str(obj.FileName)
    asset_bytes = open(path, "rb").read() if os.path.isfile(path) else b""
    return {
        "objectSet": [{"name": item.Name, "typeId": item.TypeId} for item in document.Objects],
        "object": {
            "name": obj.Name,
            "typeId": obj.TypeId,
            "propertyTypeId": obj.getTypeIdOfProperty("FileName"),
            "value": path,
            "propertyStatus": [str(item) for item in obj.getPropertyStatus("FileName")],
            "editorMode": [str(item) for item in obj.getEditorMode("FileName")],
            "state": [str(item) for item in obj.State],
            "statusString": str(obj.getStatusString()),
            "hasShapeProperty": "Shape" in obj.PropertiesList,
            "hasMeshProperty": "Mesh" in obj.PropertiesList,
            "asset": {
                "exists": bool(asset_bytes),
                "bytes": len(asset_bytes),
                "sha256": hashlib.sha256(asset_bytes).hexdigest() if asset_bytes else "",
            },
            "mesh": mesh_snapshot(obj),
        },
    }


def create_native(path, initial_value):
    document = App.newDocument("PropertyFileRoundtrip")
    try:
        obj = document.addObject("Mesh::Import", "FileProbe")
        obj.FileName = initial_value
        document.recompute()
        document.saveAs(path)
    finally:
        App.closeDocument(document.Name)
    reopened = App.openDocument(path)
    try:
        reopened.recompute()
        return object_snapshot(reopened)
    finally:
        App.closeDocument(reopened.Name)


def verify_native(path, resaved_path):
    document = App.openDocument(path)
    try:
        document.recompute()
        reopened = object_snapshot(document)
        document.saveAs(resaved_path)
    finally:
        App.closeDocument(document.Name)
    resaved_document = App.openDocument(resaved_path)
    try:
        resaved_document.recompute()
        resaved = object_snapshot(resaved_document)
    finally:
        App.closeDocument(resaved_document.Name)
    return {"reopened": reopened, "resaved": resaved}


mode = os.environ.get("FREECAD_PROPERTY_FILE_ROUNDTRIP_MODE", "")
path = os.environ.get("FREECAD_PROPERTY_FILE_ROUNDTRIP_PATH", "")
resaved_path = os.environ.get("FREECAD_PROPERTY_FILE_ROUNDTRIP_RESAVED_PATH", "")
initial_value = os.environ.get("FREECAD_PROPERTY_FILE_ROUNDTRIP_INITIAL", "")
if mode == "create":
    result = create_native(path, initial_value)
elif mode == "verify":
    result = verify_native(path, resaved_path)
else:
    raise RuntimeError("FREECAD_PROPERTY_FILE_ROUNDTRIP_MODE must be create or verify")

print("FREECAD_PROPERTY_FILE_ROUNDTRIP_RESULT=" + json.dumps({
    "schemaVersion": 1,
    "status": "pass",
    "baselineId": "freecad-1.1.1-property-file-roundtrip",
    "freecadVersion": ".".join(str(value) for value in App.Version()[:3]),
    "gitCommit": FREECAD_COMMIT,
    "mode": mode,
    "result": result,
}, sort_keys=True))
