#!/usr/bin/env python3
import hashlib
import json
import pathlib
import sys

import bpy


def modifier_report(obj):
    return [{
        "name": modifier.name,
        "type": modifier.type,
        "showViewport": modifier.show_viewport,
        "showRender": modifier.show_render,
        "showEditMode": modifier.show_in_editmode,
        "showOnCage": modifier.show_on_cage,
    } for modifier in obj.modifiers]


def main():
    arguments = sys.argv[sys.argv.index("--") + 1:]
    if len(arguments) != 2:
        raise SystemExit("usage: blender -b --python check-wbv2-p4-001-desktop.py -- FIXTURE REPORT")
    fixture, output = (pathlib.Path(value).resolve() for value in arguments)
    bpy.ops.wm.open_mainfile(filepath=str(fixture), load_ui=False)
    obj = bpy.data.objects.get("WBV2P4001ModifierObject")
    if obj is None:
        raise RuntimeError("modifier fixture object is missing")
    before = {"object": obj.name, "vertexCount": len(obj.data.vertices), "polygonCount": len(obj.data.polygons), "modifiers": modifier_report(obj)}
    if [item["type"] for item in before["modifiers"]] != ["BEVEL", "DECIMATE", "PARTICLE_SYSTEM"]:
        raise RuntimeError(f"unexpected modifier stack: {before}")
    temporary = fixture.with_name("wbv2-p4-001-reopen.blend")
    try:
        bpy.ops.wm.save_as_mainfile(filepath=str(temporary), check_existing=False, compress=True)
        bpy.ops.wm.open_mainfile(filepath=str(temporary), load_ui=False)
        reopened = bpy.data.objects.get("WBV2P4001ModifierObject")
        after = {"object": reopened.name, "vertexCount": len(reopened.data.vertices), "polygonCount": len(reopened.data.polygons), "modifiers": modifier_report(reopened)}
    finally:
        temporary.unlink(missing_ok=True)
    if before != after:
        raise RuntimeError(f"modifier save/reopen drift: {before} != {after}")
    report = {"schemaVersion": 1, "task": "WBV2-P4-001", "operation": "MODIFIER_DECLARED_SLICES_DESKTOP", "fixtureSha256": hashlib.sha256(fixture.read_bytes()).hexdigest(), "state": after, "saveReopen": "EXACT", "blenderVersion": bpy.app.version_string}
    output.parent.mkdir(parents=True, exist_ok=True)
    output.write_text(json.dumps(report, indent=2, sort_keys=True) + "\n", encoding="utf-8")
    print(f"wbv2-p4-001-desktop-ok object={after['object']} modifiers={len(after['modifiers'])} unknown=PARTICLE_SYSTEM saveReopen=exact")


if __name__ == "__main__":
    main()