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.sweeps import SweepService


def main(env=None) -> int:
    env = os.environ if env is None else env
    output_json = _get_bool(env, "SWEEPS_OUTPUT_JSON", False)
    service = SweepService()
    try:
        candidates = service.detect_sweeps(instrument_id=1, timeframe="1m")
        written = service.persist_sweeps(instrument_id=1, timeframe="1m", candidates=candidates)
    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 sweeps sample: {exc.__class__.__name__}")
        return 2
    payload = {"passed": True, "instrument_id": 1, "timeframe": "1m", "candidates": len(candidates), "events_written": written}
    if output_json:
        print(json.dumps(payload, default=str, ensure_ascii=False, sort_keys=True))
    else:
        print(f"Detected {len(candidates)} sweeps; wrote {written} events")
    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())