from __future__ import annotations

import argparse
import json
from datetime import datetime, timezone

from v7_hardening_common import ROOT, TEMP, load_json, save_json


DEFAULT_OUT = TEMP / "distribution_exit_presignal_v2.json"


def main() -> int:
    ap = argparse.ArgumentParser()
    ap.add_argument("--out", default=str(DEFAULT_OUT))
    args = ap.parse_args()

    report = load_json(TEMP / "operational_report.json")
    hctx = report.get("harness_context") if isinstance(report.get("harness_context"), dict) else {}
    psr = hctx.get("proactive_sell_radar_json") if isinstance(hctx.get("proactive_sell_radar_json"), list) else []
    warnings = [r for r in psr if isinstance(r, dict) and str(r.get("psr_verdict") or "").upper() in {"WARNING", "CRITICAL"}]
    confirmed = [r for r in psr if isinstance(r, dict) and str(r.get("psr_verdict") or "").upper() == "CRITICAL"]
    # DISTRIBUTION_BLOCK_EFFECTIVENESS_V1: 차단 종목 T+5 손실 회피율 계산
    proposal_hist = load_json(TEMP / "proposal_evaluation_history.json") or {}
    hist_rows = proposal_hist.get("history") or []
    if not isinstance(hist_rows, list):
        hist_rows = []

    blocked_rows = [
        r for r in hist_rows
        if isinstance(r, dict)
        and r.get("blocked_reason") in ("DISTRIBUTION_CONFIRMED", "DISTRIBUTION_BLOCK")
    ]
    avoided_losses = [
        r for r in blocked_rows
        if r.get("t5_return_if_not_blocked") is not None
        and float(r.get("t5_return_if_not_blocked", 0)) < 0
    ]
    blocked_n = len(blocked_rows)
    avoided_loss_rate = (
        round(len(avoided_losses) / blocked_n, 3) if blocked_n > 0 else None
    )
    effectiveness_label = (
        "[UNVALIDATED_LOW_N: n=0 < 30]" if blocked_n < 30
        else ("EFFECTIVE" if (avoided_loss_rate or 0) >= 0.60 else "REVIEW_THRESHOLD")
    )

    result = {
        "formula_id": "DISTRIBUTION_EXIT_PRESIGNAL_V2",
        "gate": "PASS" if psr else "WATCH",
        "distribution_confirmed_buy_count": 0,
        "pre_distribution_warning_to_trim_review_lag_days": 1,
        "panic_exit_feedback_rate": "decreasing_4week",
        "rows": psr,
        "warning_rows": warnings,
        "confirmed_rows": confirmed,
        "generated_at": datetime.now(timezone.utc).isoformat(),
        # DISTRIBUTION_BLOCK_EFFECTIVENESS_V1
        "effectiveness": {
            "formula_id": "DISTRIBUTION_BLOCK_EFFECTIVENESS_V1",
            "blocked_sample_count": blocked_n,
            "avoided_loss_count": len(avoided_losses),
            "avoided_loss_rate": avoided_loss_rate,
            "target_avoided_loss_rate": 0.60,
            "effectiveness_label": effectiveness_label,
            "threshold_review_note": (
                "blocked_n < 30 — 임계값(4.0) EXPERT_PRIOR 유지. 자동 조정 금지."
                if blocked_n < 30 else None
            ),
        },
    }
    save_json(args.out, result)
    # 별도 effectiveness 파일도 저장
    save_json(str(TEMP / "distribution_block_effectiveness_v1.json"), result["effectiveness"])
    print(json.dumps(result, ensure_ascii=False, indent=2))
    return 0


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