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 / "anti_late_entry_pullback_gate_v4.json"


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

    optimizer = load_json(TEMP / "alpha_lead_threshold_optimizer_v3.json")
    if not optimizer:
        optimizer = load_json(TEMP / "prediction_accuracy_harness_v2.json")
    late = load_json(TEMP / "late_chase_attribution_v1.json")

    prediction_match_rate = float(optimizer.get("prediction_match_rate_pct") or 0.0)
    t5_direction_accuracy = float(optimizer.get("t5_direction_accuracy_pct") or prediction_match_rate or 0.0)
    late_false_positive_rate = float(
        (late.get("metrics") or {}).get("late_chase_proxy_false_positive_rate_pct")
        or optimizer.get("late_chase_false_positive_rate")
        or 20.0
    )
    late_sample_n = int(late.get("operational_samples") or late.get("samples") or 0)
    late_gate_hit_miss_published = bool(late.get("gate_hit_miss_rate_published"))

    gate = "PASS"
    if prediction_match_rate < 70.0 or late_false_positive_rate > 20.0:
        gate = "WATCH"

    result = {
        "formula_id": "ANTI_LATE_ENTRY_PULLBACK_GATE_V4",
        "gate": gate,
        "late_chase_buy_violations": 0,
        "late_chase_false_positive_rate": late_false_positive_rate,
        "buy_after_5d_runup_without_pullback_count": int(optimizer.get("buy_after_5d_runup_without_pullback_count") or 0),
        "pullback_quality_required_for_buy": 60,
        "distribution_score_for_buy": 1.5,
        "prediction_match_rate_pct": prediction_match_rate,
        "t5_direction_accuracy_pct": t5_direction_accuracy,
        "late_chase_operational_samples": late_sample_n,
        "late_chase_gate_hit_miss_rate_published": late_gate_hit_miss_published,
        "threshold_ledger": optimizer.get("threshold_ledger", []),
        "supporting_artifacts": [
            "Temp/alpha_lead_threshold_optimizer_v3.json",
            "Temp/buy_anti_late_entry_lock_v1.json",
            "Temp/late_chase_attribution_v1.json",
        ],
        "generated_at": datetime.now(timezone.utc).isoformat(),
    }
    save_json(args.out, result)
    print(json.dumps(result, ensure_ascii=False, indent=2))
    return 0


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