from __future__ import annotations

import sys
from pathlib import Path


PROJECT_ROOT = Path(__file__).resolve().parent.parent
if str(PROJECT_ROOT) not in sys.path:
    sys.path.insert(0, str(PROJECT_ROOT))

from step_editor.relation_formulas import (  # noqa: E402
    RelationFormulaError,
    parse_relation_formula,
    validate_relation_formula_graph,
)


def _parse_many(texts: list[str]):
    return [parse_relation_formula(text) for text in texts]


def _assert_ok(texts: list[str]) -> None:
    validate_relation_formula_graph(_parse_many(texts))


def _assert_fails(texts: list[str], expected_fragment: str) -> None:
    try:
        validate_relation_formula_graph(_parse_many(texts))
    except RelationFormulaError as exc:
        message = str(exc)
        if expected_fragment not in message:
            raise AssertionError(f"expected {expected_fragment!r} in error message, got {message!r}") from exc
        return
    raise AssertionError(f"expected relation formulas to fail: {texts!r}")


def main() -> int:
    _assert_ok(["Face87.直径 = Face87.半径 + 0.1"])
    _assert_ok(["Face87.位置 = Face85.位置 + (0, 0, -3.5)"])
    _assert_ok(["Face85.直径 = Face87.半径", "Face11.直径 = Face85.半径"])

    _assert_fails(["Face87.直径 = Face87.直径 + 0.1"], "不能引用自身")
    _assert_fails(["Face85.直径 = Face87.半径", "Face85.直径 = Face11.半径"], "同一目标参数")
    _assert_fails(["Face85.直径 = Face87.直径", "Face87.直径 = Face85.直径"], "循环依赖")
    _assert_fails(
        [
            "Face1.位置 = Face2.位置",
            "Face2.位置 = Face3.位置",
            "Face3.位置 = Face1.位置",
        ],
        "循环依赖",
    )

    print("relation formula rules ok")
    return 0


if __name__ == "__main__":
    raise SystemExit(main())