from pathlib import Path
import json
import os
import sys

from sqlalchemy.exc import SQLAlchemyError

ROOT = Path(__file__).resolve().parents[1]
os.environ.setdefault("DATABASE_URL", "postgresql+psycopg://user:pass@localhost:5432/ai_ict_test")
if str(ROOT) not in sys.path:
    sys.path.insert(0, str(ROOT))

from src.services.broker import LiveBrokerService


def main(env=None) -> int:
    env = os.environ if env is None else env
    output_json = _get_bool(env, "LIVE_ADAPTER_OUTPUT_JSON", False)
    operation = str(env.get("LIVE_ADAPTER_OPERATION") or "reconcile").strip().lower()
    adapter_name = str(env.get("LIVE_ADAPTER_NAME") or "okx").strip().lower()
    requested_by = str(env.get("LIVE_REQUESTED_BY") or "local_operator")
    explicit_live_confirmation = _get_bool(env, "LIVE_EXPLICIT_CONFIRMATION", False)
    instrument_id = _get_optional_int(env, "LIVE_INSTRUMENT_ID")
    execution_ticket_id = _get_optional_int(env, "LIVE_EXECUTION_TICKET_ID")
    kill_switch_reason = str(env.get("LIVE_KILL_SWITCH_REASON") or "manual_kill_switch")
    service = LiveBrokerService()
    try:
        if operation == "submit":
            if execution_ticket_id is None:
                raise ValueError("missing_required_env:LIVE_EXECUTION_TICKET_ID")
            result = service.submit_confirmed_ticket(
                execution_ticket_id,
                requested_by=requested_by,
                explicit_live_confirmation=explicit_live_confirmation,
                adapter_name=adapter_name,
            )
        elif operation == "reconcile":
            result = service.reconcile_orders(
                requested_by=requested_by,
                instrument_id=instrument_id,
                adapter_name=adapter_name,
            )
        elif operation == "cancel_all":
            result = service.cancel_all(
                requested_by=requested_by,
                instrument_id=instrument_id,
                explicit_live_confirmation=explicit_live_confirmation,
                adapter_name=adapter_name,
            )
        elif operation == "kill_switch":
            result = service.activate_kill_switch(
                requested_by=requested_by,
                reason=kill_switch_reason,
                explicit_live_confirmation=explicit_live_confirmation,
                adapter_name=adapter_name,
            )
        else:
            raise ValueError(f"unsupported_live_adapter_operation:{operation}")
    except SQLAlchemyError as exc:
        payload = {
            "passed": False,
            "operation": operation,
            "adapter_name": adapter_name,
            "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 live adapter sample: {exc.__class__.__name__}")
        return 2

    payload = {
        "passed": True,
        "operation": operation,
        "adapter_name": adapter_name,
        "result": result.to_dict() if result is not None else None,
    }
    if output_json:
        print(json.dumps(payload, default=str, ensure_ascii=False, sort_keys=True))
    else:
        print(
            f"Live adapter operation: {operation}; "
            f"status: {(result.status if result else 'missing')}; "
            f"reasons: {(result.reason_codes if result else [])}"
        )
    return 0


def _get_optional_int(env, key: str):
    value = env.get(key)
    if value is None or value == "":
        return None
    return int(value)


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