"""Graphique des pièces rares par set."""

from pathlib import Path
from typing import List, Tuple

import matplotlib.pyplot as plt
from matplotlib.patches import Patch

from lib.filesystem import ensure_parent_dir
from lib.rebrickable.stats import read_rows


def load_top_sets(path: Path, limit: int = 15) -> List[dict]:
    """Charge les sets triés par nombre de pièces rares et limite le top."""
    rows = read_rows(path)
    sorted_rows = sorted(
        rows,
        key=lambda row: (
            -int(row["rare_parts_distinct"]),
            -int(row["rare_parts_quantity"]),
            row["set_num"],
        ),
    )
    return sorted_rows[:limit]


def split_counts(rows: List[dict]) -> Tuple[List[int], List[int]]:
    """Sépare les comptages minifig vs hors minifig."""
    non_minifig: List[int] = []
    minifig: List[int] = []
    for row in rows:
        total = int(row["rare_parts_distinct"])
        minifig_count = int(row["rare_minifig_parts_distinct"])
        non_minifig.append(total - minifig_count)
        minifig.append(minifig_count)
    return non_minifig, minifig


def plot_rare_parts_per_set(rare_by_set_path: Path, destination_path: Path) -> None:
    """Trace le top des sets contenant des pièces exclusives."""
    rows = load_top_sets(rare_by_set_path)
    if not rows:
        return
    non_minifig, minifig = split_counts(rows)
    y_positions = list(range(len(rows)))
    labels = [f"{row['set_num']} · {row['name']} ({row['year']})" for row in rows]
    owned_mask = [row["in_collection"] == "true" for row in rows]

    base_color = "#1f77b4"
    accent_color = "#f28e2b"
    fig, ax = plt.subplots(figsize=(11, 8))
    for y, value, is_owned in zip(y_positions, non_minifig, owned_mask):
        alpha = 0.92 if is_owned else 0.45
        ax.barh(y, value, color=base_color, alpha=alpha, label=None)
    for y, value, offset, is_owned in zip(y_positions, minifig, non_minifig, owned_mask):
        alpha = 0.92 if is_owned else 0.45
        ax.barh(y, value, left=offset, color=accent_color, alpha=alpha, label=None)

    ax.set_yticks(y_positions)
    ax.set_yticklabels(labels)
    ax.invert_yaxis()
    ax.set_xlabel("Variantes de pièces exclusives (hors rechanges)")
    ax.set_title("Pièces rares par set (top)")
    ax.grid(axis="x", linestyle="--", alpha=0.35)

    handles = [
        Patch(facecolor=base_color, edgecolor="none", label="Pièces hors minifigs"),
        Patch(facecolor=accent_color, edgecolor="none", label="Pièces de minifigs"),
        Patch(facecolor="#000000", edgecolor="none", alpha=0.92, label="Set possédé"),
        Patch(facecolor="#000000", edgecolor="none", alpha=0.45, label="Set manquant"),
    ]
    ax.legend(handles=handles, loc="lower right", frameon=False)

    ensure_parent_dir(destination_path)
    fig.tight_layout()
    fig.savefig(destination_path, dpi=170)
    plt.close(fig)