"""
core/pitch_classifier.py — 투구/타자 결과 분류

네이버 리포트 데이터를 기반으로 관리자 사이트에서 선택해야 할
라디오 버튼의 라벨(eventName)을 결정합니다.
Playwright 의존성 없이 순수 파이썬 로직만 포함합니다.
"""
from __future__ import annotations

import re
from typing import Any

from core.config_loader import (
    pitch_type_map,
    pitch_result_map,
    batter_result_map,
)


# ──────────────────────────────────────────────
# 구종 분류
# ──────────────────────────────────────────────

def classify_pitch_type(pitch_type_text: str) -> str | None:
    """네이버 구종 텍스트 → 사이트 구종 라벨

    예: '직구' → '패스트볼', '포크' → '포크볼'
    """
    return pitch_type_map().get(pitch_type_text or "")


# ──────────────────────────────────────────────
# 투구 결과 분류
# ──────────────────────────────────────────────

def classify_pitch_result(pitch_result_text: str) -> str | None:
    """네이버 투구결과 텍스트 → 사이트 투구결과 라벨

    예: '볼' → '볼', '스트라이크' → '스트라이크(루킹)'
    """
    return pitch_result_map().get(pitch_result_text or "")


def normalize_pitch_result_code(pitch: dict[str, Any], event: dict[str, Any] | None = None) -> str:
    """투구의 pitchResult 코드를 정규화

    피치클락, 번트헛스윙, 폭투/포일 등 특수 상황 처리
    """
    pitch_result = (pitch.get("pitchResult") or "").strip()
    pitch_result_text = (pitch.get("pitchResultText") or "").strip()
    normalized_text = pitch_result_text.replace(" ", "")

    # 피치클락 투수위반 → 볼
    if "피치클락" in pitch_result_text and "투수위반" in pitch_result_text:
        return "B"

    # 번트 헛스윙 → BS
    if "번트" in normalized_text and "헛스윙" in normalized_text:
        return "BS"

    # 폭투/포일 진루 시 → 볼
    runner_events = _get_pitch_runner_events(pitch, event)
    if any(
        re_.get("type") == "wild_pitch_advance" or "폭투" in (re_.get("text") or "")
        for re_ in runner_events
    ):
        return "B"
    if any(
        re_.get("type") == "passed_ball_advance" or "포일" in (re_.get("text") or "")
        for re_ in runner_events
    ):
        return "B"

    return pitch_result


# ──────────────────────────────────────────────
# 타자 결과 분류
# ──────────────────────────────────────────────

def classify_batter_result(result_type: str) -> str | None:
    """결과 타입 코드 → 사이트 타자결과 라벨 (기본 매핑)

    더 복잡한 추론이 필요한 경우 infer_batter_result_label 사용.
    """
    return batter_result_map().get(result_type or "")


def infer_batter_result_label(
    result: dict[str, Any],
    event: dict[str, Any] | None = None,
) -> str | None:
    """타석 결과를 종합적으로 추론하여 사이트 라벨 반환

    result.type, result.text, 주루이벤트, 마지막 투구 등을 모두 분석.
    """
    result_type = result.get("type") or ""
    result_text = (result.get("text") or "").strip()
    runner_events = (event or {}).get("runnerEvents") or []
    last_pitch_result_text = get_last_pitch_result_text(event)

    # 낫아웃
    if result_type == "strikeout_not_out" or "낫아웃" in result_text:
        if "폭투" in result_text:
            return "폭투 낫아웃 진루"
        if "포일" in result_text:
            return "포일 낫아웃 진루"
        if "아웃" in result_text:
            return "스트라이크-낫아웃"
        return "낫아웃-출루"

    # 삼진
    if result_type == "strikeout":
        if "헛스윙" in last_pitch_result_text or "헛스윙" in result_text:
            return "스윙 스트라이크-아웃"
        return "루킹스트라이크-아웃"

    # 희생 번트
    if "희생 번트" in result_text or "희생번트" in result_text:
        return "희생 번트"

    # 번트 아웃
    if "번트 아웃" in result_text or "번트아웃" in result_text:
        return "번트-아웃"

    # 보크
    if any(
        "보크" in (re_.get("text") or "") and "진루" in (re_.get("text") or "")
        for re_ in runner_events
    ):
        if "볼" in last_pitch_result_text:
            return "보크-볼"
        return "보크"

    # 폭투-볼
    if any(re_.get("type") == "wild_pitch_advance" for re_ in runner_events):
        return "폭투-볼"

    # 포볼
    if result_type == "walk":
        if any(
            re_.get("type") == "wild_pitch_advance" or "폭투" in (re_.get("text") or "")
            for re_ in runner_events
        ):
            return "폭투-포볼"
        if any(
            re_.get("type") == "passed_ball_advance" or "포일" in (re_.get("text") or "")
            for re_ in runner_events
        ):
            return "포일-포볼"
        return "포볼"

    # 포일-볼/스트라이크
    if any(
        (re_.get("type") or "") == "passed_ball_advance"
        for re_ in runner_events
    ):
        if "볼" in last_pitch_result_text:
            return "포일-볼"
        return "포일-스트라이크"

    # 수비실책
    if result_type == "reach_on_error" or "실책" in result_text:
        return "수비실책"

    # 야수선택
    if result_type == "reach_on_fielder_choice":
        return "야수선택"

    # 땅볼출루
    if result_type == "reach_on_grounder":
        return "땅볼출루(무안타)"

    # 병살
    if result_type == "double_play":
        if "번트" in result_text:
            return "번트-병살"
        return "병살-아웃"

    # N루타 후 주루아웃
    if result_type == "single_runner_out":
        return "1루타 후 주루아웃"
    if result_type == "double_runner_out":
        return "2루타 후 주루아웃"
    if result_type == "triple_runner_out":
        return "3루타 후 주루아웃"

    # N루타 후 수비실책진루
    if result_type == "single_error_advance":
        return "1루타 후 수비실책진루"
    if result_type == "double_error_advance":
        return "2루타 후 수비실책진루"
    if result_type == "triple_error_advance":
        return "3루타 후 수비실책진루"

    # 파울희생플라이
    if "파울희생플라이" in result_text or "파울 희생플라이" in result_text:
        return "희생 플라이"

    # 아웃 상세
    if result_type == "out":
        if "병살" in result_text:
            if "번트" in result_text:
                return "번트-병살"
            return "병살-아웃"
        if "희생 플라이" in result_text or "희생플라이" in result_text:
            return "희생 플라이"
        if "인필드플라이" in result_text:
            return "인필드플라이"
        if "파울플라이" in result_text:
            return "파울플라이-아웃"
        return "아웃"

    # 번트안타
    if result_type == "bunt_hit":
        return "번트안타"

    # 내야안타
    if result_type == "single":
        if "번트안타" in result_text:
            return "번트안타"
        if "내야안타" in result_text:
            return "내야안타"

    # 몸에 맞는 볼
    if result_type == "hit_by_pitch" or "헤드샷" in result_text:
        return "몸에 맞는 볼"

    # 기본 매핑 폴백
    return classify_batter_result(result_type)


def is_simple_terminal_result_type(result_type: str) -> bool:
    """팝업 없이 즉시 완료되는 결과 타입인지 확인"""
    return result_type in {"strikeout", "strikeout_not_out", "walk", "intentional_walk", "hit_by_pitch"}


def is_ball_in_play_event(event: dict[str, Any]) -> bool:
    """인플레이 이벤트인지 확인 (마지막 투구가 H)"""
    pitches = event.get("pitches") or []
    result = event.get("result") or {}
    if not pitches or not result:
        return False
    return pitches[-1].get("pitchResult") == "H"


# ──────────────────────────────────────────────
# 내부 헬퍼
# ──────────────────────────────────────────────

def _get_pitch_runner_events(
    pitch: dict[str, Any],
    event: dict[str, Any] | None,
) -> list[dict[str, Any]]:
    """투구에 연결된 주루이벤트 반환"""
    if pitch.get("runnerEvents"):
        return pitch["runnerEvents"]
    if event:
        pitch_num = pitch.get("pitchNum")
        for re_ in (event.get("runnerEvents") or []):
            if re_.get("pitchNum") == pitch_num:
                return [re_]
    return []


def get_last_pitch_result_text(event: dict[str, Any] | None) -> str:
    """이벤트의 마지막 투구 결과 텍스트 반환"""
    if not event:
        return ""
    pitches = event.get("pitches") or []
    if not pitches:
        return ""
    return (pitches[-1].get("pitchResultText") or "").strip()