from pathlib import Path
import json
import os
import sys

ROOT = Path(__file__).resolve().parents[1]
if str(ROOT) not in sys.path:
    sys.path.insert(0, str(ROOT))

MODELS_DIR = ROOT / "src" / "models"


def main(env=None) -> int:
    env = os.environ if env is None else env
    output_json = _get_bool(env, "SHOW_MODELS_OUTPUT_JSON", False)
    model_files = sorted(
        path.stem
        for path in MODELS_DIR.glob("*.py")
        if path.is_file() and path.stem != "__init__"
    )
    if output_json:
        print(
            json.dumps(
                {"passed": True, "model_count": len(model_files), "models": model_files},
                ensure_ascii=False,
                sort_keys=True,
            )
        )
        return 0
    print(f"model exports: {len(model_files)}")
    for name in model_files:
        print(name)
    return 0


def _get_bool(env, key: str, default: bool) -> bool:
    value = env.get(key)
    if value is None or value == "":
        return default
    return str(value).strip().lower() in {"1", "true", "yes", "on"}


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