import hashlib
import json
import os

import FreeCAD as App
import Part


FREECAD_COMMIT = "0108fd4b4850cc46e625b60e53cea7a7bbe69f8d"
OUTPUT_PATH = os.environ.get("FREECAD_PROPERTY_LINKSUBHIDDEN_MUTATION_OUTPUT", "")


def shape_snapshot(obj):
    if "Shape" not in obj.PropertiesList:
        return {"applicable": False, "reason": "host-has-no-shape-property"}
    shape = obj.Shape
    if shape.isNull():
        return {"applicable": True, "isNull": True, "solids": 0, "faces": 0, "edges": 0, "vertices": 0, "brepSha256": None}
    return {
        "applicable": True,
        "isNull": False,
        "solids": len(shape.Solids),
        "faces": len(shape.Faces),
        "edges": len(shape.Edges),
        "vertices": len(shape.Vertexes),
        "brepSha256": hashlib.sha256(shape.exportBrepToString().encode("utf-8")).hexdigest(),
    }


def colored_value(owner):
    value = owner.ColoredElements
    if value is None:
        return None
    target, subs = value
    return {"object": target.Name if target is not None else None, "subElements": list(subs)}


def snapshot(document, owner, targets):
    return {
        "value": colored_value(owner),
        "propertyTypeId": owner.getTypeIdOfProperty("ColoredElements"),
        "group": owner.getGroupOfProperty("ColoredElements"),
        "propertyStatus": [str(item) for item in owner.getPropertyStatus("ColoredElements")],
        "editorMode": [str(item) for item in owner.getEditorMode("ColoredElements")],
        "ownerState": [str(item) for item in owner.State],
        "statusString": str(owner.getStatusString()),
        "shape": shape_snapshot(owner),
        "ownerOutList": [candidate.Name for candidate in owner.OutList],
        "targetInLists": {target.Name: [candidate.Name for candidate in target.InList] for target in targets},
        "targetShapes": {target.Name: shape_snapshot(target) for target in targets},
        "objectSet": [{"name": candidate.Name, "typeId": candidate.TypeId} for candidate in document.Objects],
    }


def run_case(object_type_id, object_name):
    document = App.newDocument("PropertyLinkSubHiddenMutation" + object_name)
    try:
        owner = document.addObject(object_type_id, object_name)
        first = document.addObject("Part::Feature", "ColorSourceA")
        first.Shape = Part.makeBox(2.0, 3.0, 4.0)
        second = document.addObject("Part::Feature", "ColorSourceB")
        second.Shape = Part.makeCylinder(1.5, 5.0)
        targets = [first, second]
        if "LinkedObject" in owner.PropertiesList:
            owner.LinkedObject = first

        owner.ColoredElements = (first, ["Face1", "Face3"])
        document.recompute()
        before = snapshot(document, owner, targets)

        owner.ColoredElements = (second, ["Face5"])
        touched_after_edit = snapshot(document, owner, targets)
        edit_recompute_result = bool(document.recompute())
        edited = snapshot(document, owner, targets)

        owner.ColoredElements = (first, ["Face1", "Face3"])
        touched_after_restore = snapshot(document, owner, targets)
        restore_recompute_result = bool(document.recompute())
        restored = snapshot(document, owner, targets)

        return {
            "objectTypeId": object_type_id,
            "objectName": object_name,
            "propertyName": "ColoredElements",
            "before": before,
            "touchedAfterEdit": touched_after_edit,
            "edited": edited,
            "touchedAfterRestore": touched_after_restore,
            "restored": restored,
            "editRecomputeResult": edit_recompute_result,
            "restoreRecomputeResult": restore_recompute_result,
            "valueRestored": before["value"] == restored["value"],
            "semanticStateRestored": before == restored,
        }
    finally:
        App.closeDocument(document.Name)


cases = [
    run_case("App::Link", "LinkProbe"),
    run_case("App::LinkGroup", "LinkGroupProbe"),
    run_case("App::LinkGroupPython", "LinkGroupPythonProbe"),
    run_case("App::LinkPython", "LinkPythonProbe"),
]
report = {
    "schemaVersion": 1,
    "status": "pass",
    "baselineId": "freecad-1.1.1-property-linksubhidden-mutation",
    "freecadVersion": ".".join(str(value) for value in App.Version()[:3]),
    "gitCommit": FREECAD_COMMIT,
    "propertyType": "App::PropertyLinkSubHidden",
    "caseCount": len(cases),
    "cases": cases,
}
if not OUTPUT_PATH:
    raise RuntimeError("FREECAD_PROPERTY_LINKSUBHIDDEN_MUTATION_OUTPUT is required")
with open(OUTPUT_PATH, "w", encoding="utf-8") as handle:
    json.dump(report, handle, indent=2, sort_keys=True)
    handle.write("\n")
print("FREECAD_PROPERTY_LINKSUBHIDDEN_MUTATION_RESULT=" + json.dumps({"status": report["status"], "caseCount": report["caseCount"]}, sort_keys=True))