# storylines.py (Modernized & GUI-Compatible)

from analysis.team_identity_v2 import generate_team_identity_block
from analysis.team_identity_v2 import summarize_team_tags

__all__ = ["matchup_storyline", "generate_storyline"]


# ---------------------------------------------------------
# Team Tag Aggregation
# ---------------------------------------------------------

def summarize_team_tags(team_players):
    """
    Aggregates tag tiers + percentiles for a team.
    """

    tags = [
        "slayer",
        "objective_payload",
        "objective_domination",
        "sharpshooter",
        "consistency",
        "clutch",
    ]

    summary = {}

    for tag in tags:
        pcts = []
        tiers = []

        for p in team_players:
            tag_data = p.get(tag, {})
            pcts.append(tag_data.get("pct", 0))
            tiers.append(tag_data.get("tier", "D"))

        if not pcts:
            summary[tag] = {
                "avg_pct": 0,
                "top_tier": "D",
                "count_S": 0,
                "count_A": 0,
                "count_B": 0,
                "count_C": 0,
                "count_D": 0,
            }
            continue

        summary[tag] = {
            "avg_pct": sum(pcts) / len(pcts),
            "top_tier": max(tiers, key=lambda t: "SABCD".index(t)),
            "count_S": tiers.count("S"),
            "count_A": tiers.count("A"),
            "count_B": tiers.count("B"),
            "count_C": tiers.count("C"),
            "count_D": tiers.count("D"),
        }

    return summary


# ---------------------------------------------------------
# Team Strength Comparison
# ---------------------------------------------------------

def compare_team_strengths(teamA, teamB, team_a_name, team_b_name):
    """
    Compares two team tag summaries and returns storyline points.
    """

    storyline = []

    tags = {
        "slayer": "Slayer (Elimination Pressure)",
        "objective_payload": "Payload Objective",
        "objective_domination": "Domination Objective",
        "sharpshooter": "Sharpshooter (Accuracy)",
        "consistency": "Consistency (Stability)",
        "clutch": "Clutch (High-Pressure Performance)",
    }

    for tag, label in tags.items():
        a = teamA[tag]["avg_pct"]
        b = teamB[tag]["avg_pct"]

        diff = a - b

        if abs(diff) < 5:
            storyline.append(f"Both teams are evenly matched in {label}.")
        elif diff > 0:
            storyline.append(f"{team_a_name} hold an advantage in {label}.")
        else:
            storyline.append(f"{team_b_name} hold an advantage in {label}.")

    return storyline


# ---------------------------------------------------------
# Main Storyline Generator
# ---------------------------------------------------------

def generate_storyline(teamA_players, teamB_players):
    """
    Generates a full caster-ready storyline block.
    """

    # Extract real team names
    team_a_name = teamA_players[0].get("team", "Team A")
    team_b_name = teamB_players[0].get("team", "Team B")

    # 1. Summaries
    A = summarize_team_tags(teamA_players)
    B = summarize_team_tags(teamB_players)

    # 2. Strength comparison
    matchup_points = compare_team_strengths(A, B, team_a_name, team_b_name)

    # 3. Identify elite players
    elite_A = [p["name"] for p in teamA_players if p.get("slayer", {}).get("tier") == "S"]
    elite_B = [p["name"] for p in teamB_players if p.get("slayer", {}).get("tier") == "S"]

    # 4. Build storyline text
    lines = []

    lines.append(f"MATCH STORYLINE — {team_a_name} vs {team_b_name}\n")

    # Team A identity
    lines.append(f"{team_a_name} Identity:")
    for tag, data in A.items():
        lines.append(
            f"  - {tag.replace('_', ' ').title()}: "
            f"{data['top_tier']} Tier (avg {data['avg_pct']:.1f} percentile)"
        )

    lines.append("")

    # Team B identity
    lines.append(f"{team_b_name} Identity:")
    for tag, data in B.items():
        lines.append(
            f"  - {tag.replace('_', ' ').title()}: "
            f"{data['top_tier']} Tier (avg {data['avg_pct']:.1f} percentile)"
        )

    lines.append("")

    # Elite players
    if elite_A:
        lines.append(f"{team_a_name} Elite Players: {', '.join(elite_A)}")
    if elite_B:
        lines.append(f"{team_b_name} Elite Players: {', '.join(elite_B)}")

    lines.append("")

    # Matchup points
    lines.append("Matchup Breakdown:")
    for point in matchup_points:
        lines.append(f"  - {point}")

    return "\n".join(lines)


# ---------------------------------------------------------
# Legacy Wrapper (GUI expects this exact name)
# ---------------------------------------------------------

def matchup_storyline(teamA_players, teamB_players):
    """
    Legacy wrapper for GUI compatibility.
    """
    return generate_storyline(teamA_players, teamB_players)