from pathlib import Path
import json
import os
import sys

from sqlalchemy import text
from sqlalchemy.exc import SQLAlchemyError

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

from src.db.engine import engine


def main(env=None) -> int:
    env = os.environ if env is None else env
    output_json = _get_bool(env, "CHECK_DB_CONNECTION_OUTPUT_JSON", False)
    try:
        with engine.connect() as conn:
            result = conn.execute(text("select 1"))
            value = result.scalar_one()
    except SQLAlchemyError as exc:
        payload = {"passed": False, "available": False, "error_type": exc.__class__.__name__, "error": str(exc)}
        if output_json:
            print(json.dumps(payload, default=str, ensure_ascii=False, sort_keys=True))
        else:
            print(f"DB connection failed: {exc.__class__.__name__}")
        return 2
    payload = {"passed": value == 1, "available": value == 1, "value": value}
    if output_json:
        print(json.dumps(payload, default=str, ensure_ascii=False, sort_keys=True))
    else:
        print("DB connection OK" if value == 1 else "DB connection unexpected result")
    return 0 if value == 1 else 2


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())