from pathlib import Path
import json
import os
import sys

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.services.ranges import DealingRangeService


def main(env=None) -> int:
    env = os.environ if env is None else env
    output_json = _get_bool(env, "RANGES_OUTPUT_JSON", False)
    service = DealingRangeService()
    try:
        candidate = service.build_active_range(instrument_id=1, timeframe="1m")
        written = service.persist_active_range(instrument_id=1, timeframe="1m", candidate=candidate)
    except SQLAlchemyError as exc:
        payload = {
            "passed": False,
            "instrument_id": 1,
            "timeframe": "1m",
            "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"Database unavailable for ranges sample: {exc.__class__.__name__}")
        return 2
    payload = {"passed": True, "instrument_id": 1, "timeframe": "1m", "has_candidate": candidate is not None, "rows_written": written}
    if output_json:
        print(json.dumps(payload, default=str, ensure_ascii=False, sort_keys=True))
    else:
        print(f"Persisted {written} dealing range rows")
    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())