umbral_admin/

widgets.rs

//! Dashboard widget system for umbral-admin.
//!
//! Plugins register widgets via `AdminPlugin::register_widget`. Each widget
//! has a `key`, `title`, `kind`, `default_span`, optional `permission`, and
//! an async data function. The admin dashboard renders a 12-column grid of
//! the user's saved layout (defaulting to all permitted widgets).
//!
//! ## Registration shape
//!
//! ```rust,ignore
//! admin.register_widget(Widget {
//!     key:          "umbral_total_models",
//!     title:        "Total Models".to_string(),
//!     kind:         WidgetKind::Kpi,
//!     default_span: Span { cols: 3, rows: 1 },
//!     permission:   None,
//!     data:         WidgetDataFn::new(|_user| async move {
//!         WidgetPayload::Kpi(KpiPayload {
//!             value:     "42".to_string(),
//!             unit:      None,
//!             delta:     None,
//!             sparkline: None,
//!         })
//!     }),
//! });
//! ```
//!
//! ## Endpoint contract
//!
//! - `GET /admin/api/dashboard/catalog` — `[{key, title, kind, default_span}]`
//! - `GET /admin/api/dashboard/layout`  — user's saved layout or default
//! - `PUT /admin/api/dashboard/layout`  — save user's layout
//! - `GET /admin/api/dashboard/widgets/{key}/data` — typed payload JSON

use std::future::Future;
use std::pin::Pin;
use std::sync::Arc;

use serde::{Deserialize, Serialize};
use umbral_auth::AuthUser;

// =========================================================================
// Span
// =========================================================================

/// Grid span in the 12-column dashboard grid.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Span {
    /// Number of columns to occupy (1–12).
    pub cols: u8,
    /// Number of rows to occupy (1–N).
    pub rows: u8,
}

impl Default for Span {
    fn default() -> Self {
        Self { cols: 3, rows: 1 }
    }
}

// =========================================================================
// WidgetKind
// =========================================================================

/// The visual kind of a dashboard widget. Drives how the payload is rendered.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "lowercase")]
pub enum WidgetKind {
    /// Simple single-value KPI (legacy, kept for backwards compat).
    Kpi,
    /// Shop-style summary card: title + icon + small unit / subtitle
    /// + large humanized value + optional growth-vs-previous-period.
    /// The everyday "Total sales / Orders / Customers" tile.
    Card,
    Line,
    Bar,
    /// Donut chart — labeled slices summing to 100%. Best for
    /// low-cardinality breakdowns (status distribution, top N
    /// regions, mode share) where a bar chart's axes are
    /// overkill. 3-6 slices reads cleanly; past that switch
    /// to a bar.
    Donut,
    /// Radial gauge — one or more 0–100% tracks rendered as
    /// concentric arcs (ApexCharts `radialBar`). The everyday
    /// "progress toward a goal" tile: quota attainment, capacity
    /// used, completion rate, SLA. A single track reads as one big
    /// ring with the percent in the centre; 2–4 tracks compare
    /// related ratios (e.g. per-plan conversion).
    Radial,
    /// Heatmap — a 2-D grid of cells colored by magnitude (ApexCharts
    /// `heatmap`). Best for "activity by time" patterns: day-of-week ×
    /// hour-of-day signups, cohort retention, per-region load. Each
    /// row is a series; each cell an `(x, value)` pair.
    Heatmap,
    /// Progress bars — a ranked list of labeled horizontal bars, each
    /// filled relative to the largest value (or an explicit target).
    /// The "top N by metric" tile: revenue by product, traffic by
    /// source, completion per category. Pure HTML; no chart library.
    Progress,
    Table,
    Feed,
}

impl WidgetKind {
    pub fn as_str(&self) -> &'static str {
        match self {
            WidgetKind::Kpi => "kpi",
            WidgetKind::Card => "card",
            WidgetKind::Line => "line",
            WidgetKind::Bar => "bar",
            WidgetKind::Donut => "donut",
            WidgetKind::Radial => "radial",
            WidgetKind::Heatmap => "heatmap",
            WidgetKind::Progress => "progress",
            WidgetKind::Table => "table",
            WidgetKind::Feed => "feed",
        }
    }
}

// =========================================================================
// Typed payloads
// =========================================================================

/// KPI card payload.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct KpiPayload {
    /// The primary metric value (displayed large).
    pub value: String,
    /// Optional unit label, e.g. "rows" or "MB".
    pub unit: Option<String>,
    /// Optional delta percentage; positive = up, negative = down.
    pub delta: Option<f64>,
    /// Optional sparkline data points (values only; x is implicit index).
    pub sparkline: Option<Vec<f64>>,
}

// =========================================================================
// Card payload — the everyday "summary tile" widget.
// =========================================================================

/// Summary card payload. Renders as:
///
/// ```text
/// ┌──────────────────────────────────────────┐
/// │ TITLE                          [icon]    │  ← title row (from Widget)
/// │                                          │
/// │ USD                       12,438.20      │  ← unit (sm, left) + value (lg, right)
/// │                                          │
/// │ This month        ↑ 12.3% vs last month  │  ← subtitle + growth
/// └──────────────────────────────────────────┘
/// ```
///
/// Build with [`CardPayload::new`] + the chained setters; pass to
/// [`WidgetPayload::Card`].
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CardPayload {
    /// Formatted primary value, e.g. "12,438.20" or "12.4K". Use
    /// [`humanize_number`] for the K/M/B/T compaction.
    pub value: String,
    /// Optional unit / context label shown on the left side of the
    /// value row, e.g. "USD", "rows", "today".
    pub unit: Option<String>,
    /// Optional Lucide icon name (e.g. "dollar-sign", "shopping-cart").
    /// Rendered via the data-lucide attribute the wrapper already
    /// initializes.
    pub icon: Option<String>,
    /// Optional caption below the value, e.g. "This month".
    pub subtitle: Option<String>,
    /// Percentage delta vs. the previous period, signed:
    /// `+12.3` = up 12.3%, `-4.1` = down 4.1%. The renderer picks
    /// the arrow + color from the sign.
    pub delta_percent: Option<f64>,
    /// Optional comparison label, e.g. "vs last month".
    pub delta_label: Option<String>,
    /// Optional trend trail — a flat series of N points the
    /// renderer plots as a fade-right sparkline under the value.
    /// X is implicit (evenly spaced); Y autoscales between
    /// min/max. Pair with `growth(...)` so the pill matches the
    /// trail visually. Keep the series small (7–30 points) —
    /// anything denser turns into noise at sparkline scale.
    pub sparkline: Option<Vec<f64>>,
}

impl CardPayload {
    /// New card with just a primary value. Caller picks the format
    /// — strings stay as-is, numbers should be pre-humanized with
    /// [`humanize_number`] / [`format_thousands`].
    pub fn new(value: impl Into<String>) -> Self {
        Self {
            value: value.into(),
            unit: None,
            icon: None,
            subtitle: None,
            delta_percent: None,
            delta_label: None,
            sparkline: None,
        }
    }

    pub fn unit(mut self, unit: impl Into<String>) -> Self {
        self.unit = Some(unit.into());
        self
    }

    pub fn icon(mut self, icon: impl Into<String>) -> Self {
        self.icon = Some(icon.into());
        self
    }

    pub fn subtitle(mut self, subtitle: impl Into<String>) -> Self {
        self.subtitle = Some(subtitle.into());
        self
    }

    /// Compute the delta automatically from current + previous raw
    /// numbers. Skips the delta when `previous` is zero (no baseline
    /// to grow from) or non-finite — the renderer just won't show
    /// the growth row in that case.
    pub fn growth(mut self, current: f64, previous: f64) -> Self {
        if previous.is_finite() && previous != 0.0 && current.is_finite() {
            self.delta_percent = Some(((current - previous) / previous.abs()) * 100.0);
        }
        self
    }

    /// Explicit delta percent (signed) + label. Use when you've
    /// computed the percentage yourself or want a custom label.
    pub fn delta(mut self, percent: f64, label: impl Into<String>) -> Self {
        self.delta_percent = Some(percent);
        self.delta_label = Some(label.into());
        self
    }

    /// Standalone label for the delta — pairs with [`Self::growth`]
    /// for the common case "auto-compute the percent but customize
    /// the comparison label" (e.g. `"vs prior 30d"`).
    pub fn delta_label(mut self, label: impl Into<String>) -> Self {
        self.delta_label = Some(label.into());
        self
    }

    /// Attach a trend trail rendered as a fade-right sparkline
    /// under the value. Pass 7–30 raw numbers (daily totals,
    /// hourly counts, etc.); the renderer autoscales and colors
    /// the stroke to match [`Self::delta_percent`]'s sign.
    pub fn sparkline(mut self, points: impl IntoIterator<Item = f64>) -> Self {
        self.sparkline = Some(points.into_iter().collect());
        self
    }
}

/// Humanize a number into a compact display string:
///
/// | input            | output     |
/// |------------------|------------|
/// | `42.0`           | `"42"`     |
/// | `1_234.5`        | `"1,234.50"` |
/// | `12_438.2`       | `"12.4K"`  |
/// | `1_500_000.0`    | `"1.50M"`  |
/// | `2_700_000_000.` | `"2.70B"`  |
///
/// Suitable for card values where horizontal space is scarce.
pub fn humanize_number(n: f64) -> String {
    if !n.is_finite() {
        return "—".to_string();
    }
    let abs = n.abs();
    let sign = if n < 0.0 { "-" } else { "" };
    if abs < 1000.0 {
        // Two decimals when there's a fractional part; integer otherwise.
        if (abs.fract() - 0.0).abs() < f64::EPSILON {
            return format!("{sign}{}", abs as i64);
        }
        return format!("{sign}{:.2}", abs);
    }
    if abs < 1_000_000.0 {
        if abs < 10_000.0 {
            // Keep the thousands separator at the low end of the K
            // range — "9,876" reads better than "9.9K" for amounts a
            // user is likely to mentally verify against the data.
            return format_thousands(n);
        }
        return format!("{sign}{:.1}K", abs / 1_000.0);
    }
    if abs < 1_000_000_000.0 {
        return format!("{sign}{:.2}M", abs / 1_000_000.0);
    }
    if abs < 1_000_000_000_000.0 {
        return format!("{sign}{:.2}B", abs / 1_000_000_000.0);
    }
    format!("{sign}{:.2}T", abs / 1_000_000_000_000.0)
}

/// Format a number with thousands separators and (when fractional)
/// two decimal places. Use for values where the full digits matter
/// (currency totals, audit counts) — for compact display use
/// [`humanize_number`].
pub fn format_thousands(n: f64) -> String {
    if !n.is_finite() {
        return "—".to_string();
    }
    let sign = if n < 0.0 { "-" } else { "" };
    let abs = n.abs();
    let int_part = abs.trunc() as u128;
    let frac_part = abs - abs.trunc();

    // Insert commas every 3 digits, right-to-left.
    let int_str = int_part.to_string();
    let bytes = int_str.as_bytes();
    let mut grouped = String::with_capacity(int_str.len() + int_str.len() / 3);
    for (i, b) in bytes.iter().enumerate() {
        if i > 0 && (bytes.len() - i) % 3 == 0 {
            grouped.push(',');
        }
        grouped.push(*b as char);
    }

    if frac_part > 0.0 {
        format!("{sign}{grouped}.{:02}", (frac_part * 100.0).round() as u64)
    } else {
        format!("{sign}{grouped}")
    }
}

/// One data series for Line or Bar charts.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Series {
    pub name: String,
    pub points: Vec<ChartPoint>,
}

/// X/Y data point. X is a string for flexible labeling.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ChartPoint {
    pub x: String,
    pub y: f64,
}

/// Line chart payload.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct LinePayload {
    pub series: Vec<Series>,
    /// Describes what `x` represents; e.g. "date", "category".
    pub x_type: String,
}

/// Bar chart payload (same shape as Line).
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct BarPayload {
    pub series: Vec<Series>,
    pub x_type: String,
}

/// One slice of a donut chart.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DonutSlice {
    pub label: String,
    pub value: f64,
    /// Optional explicit color (CSS hex / rgb / token name).
    /// `None` falls back to the chart's default palette.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub color: Option<String>,
}

/// Donut chart payload — categorical breakdown summing to 100%.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DonutPayload {
    pub slices: Vec<DonutSlice>,
}

impl DonutPayload {
    pub fn new(slices: Vec<DonutSlice>) -> Self {
        Self { slices }
    }

    /// Build slices from `(label, value)` tuples; the chart
    /// picks colors from its default palette.
    pub fn from_pairs<L: Into<String>>(pairs: impl IntoIterator<Item = (L, f64)>) -> Self {
        Self::new(
            pairs
                .into_iter()
                .map(|(label, value)| DonutSlice {
                    label: label.into(),
                    value,
                    color: None,
                })
                .collect(),
        )
    }
}

/// One arc of a [`RadialPayload`] gauge — a labeled 0–100% value.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RadialTrack {
    pub label: String,
    /// Percent in `[0, 100]`. The `RadialPayload` constructors clamp
    /// this so the arc never overruns the ring.
    pub value: f64,
    /// Optional explicit arc color (CSS hex / rgb / token name).
    /// `None` falls back to the chart's default palette.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub color: Option<String>,
}

/// Radial gauge payload — one or more 0–100% tracks rendered as
/// concentric arcs (ApexCharts `radialBar`). Use for "progress toward
/// a goal" metrics: quota attainment, capacity used, completion rate.
///
/// ```ignore
/// // One ring: 73% of the monthly sales goal.
/// RadialPayload::goal("Monthly goal", sales, target)
/// // Compare conversion across plans.
/// RadialPayload::from_pairs([("Free", 8.0), ("Pro", 21.5), ("Team", 34.0)])
/// ```
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RadialPayload {
    pub tracks: Vec<RadialTrack>,
}

impl RadialPayload {
    /// New payload from explicit tracks; each `value` is clamped to
    /// `[0, 100]` (non-finite -> 0).
    pub fn new(tracks: Vec<RadialTrack>) -> Self {
        Self {
            tracks: tracks
                .into_iter()
                .map(|t| RadialTrack {
                    value: clamp_percent(t.value),
                    ..t
                })
                .collect(),
        }
    }

    /// A single-track gauge — the common case (one big ring with the
    /// percent in the centre).
    pub fn single(label: impl Into<String>, percent: f64) -> Self {
        Self::new(vec![RadialTrack {
            label: label.into(),
            value: percent,
            color: None,
        }])
    }

    /// A single-track gauge whose percent is `current / target * 100`
    /// — the literal "progress toward a goal" shape. A non-positive
    /// `target` yields 0% (nothing to measure against).
    pub fn goal(label: impl Into<String>, current: f64, target: f64) -> Self {
        let pct = if target > 0.0 {
            current / target * 100.0
        } else {
            0.0
        };
        Self::single(label, pct)
    }

    /// Build tracks from `(label, percent)` tuples; the chart picks
    /// colors from its default palette. Each percent is clamped.
    pub fn from_pairs<L: Into<String>>(pairs: impl IntoIterator<Item = (L, f64)>) -> Self {
        Self::new(
            pairs
                .into_iter()
                .map(|(label, value)| RadialTrack {
                    label: label.into(),
                    value,
                    color: None,
                })
                .collect(),
        )
    }
}

/// Clamp a percentage into `[0, 100]`; non-finite -> 0.
fn clamp_percent(v: f64) -> f64 {
    if v.is_finite() {
        v.clamp(0.0, 100.0)
    } else {
        0.0
    }
}

/// One cell in a [`HeatmapRow`] — an x-axis bucket and its magnitude.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct HeatmapCell {
    /// X-axis label for this cell (e.g. an hour `"09"`, a month `"Mar"`).
    pub x: String,
    /// The value that colors the cell. Higher = hotter.
    pub y: f64,
}

/// One row (series) of a [`HeatmapPayload`] — a label plus its cells
/// across the shared x-axis.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct HeatmapRow {
    pub name: String,
    pub cells: Vec<HeatmapCell>,
}

/// Heatmap payload — a 2-D grid of cells colored by magnitude
/// (ApexCharts `heatmap`). Every row shares the same ordered x-axis.
/// Use for "activity by time" patterns (day-of-week × hour), cohort
/// retention, or per-region load.
///
/// ```ignore
/// HeatmapPayload::from_grid(
///     ["Mon", "Tue", "Wed"],
///     ["00-06", "06-12", "12-18", "18-24"],
///     vec![
///         vec![2.0, 9.0, 14.0, 6.0],
///         vec![1.0, 11.0, 17.0, 8.0],
///         vec![3.0, 13.0, 19.0, 7.0],
///     ],
/// )
/// ```
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct HeatmapPayload {
    pub rows: Vec<HeatmapRow>,
}

impl HeatmapPayload {
    /// New payload from explicit rows.
    pub fn new(rows: Vec<HeatmapRow>) -> Self {
        Self { rows }
    }

    /// Build a rectangular grid from row labels, shared column (x)
    /// labels, and a `values[row][col]` matrix. A short value row is
    /// padded with `0.0` and extra values past the columns are dropped,
    /// so the grid is always rectangular regardless of ragged input.
    pub fn from_grid<R, C>(
        row_labels: impl IntoIterator<Item = R>,
        col_labels: impl IntoIterator<Item = C>,
        values: Vec<Vec<f64>>,
    ) -> Self
    where
        R: Into<String>,
        C: Into<String>,
    {
        let cols: Vec<String> = col_labels.into_iter().map(Into::into).collect();
        let rows = row_labels
            .into_iter()
            .enumerate()
            .map(|(r, label)| {
                let row_vals = values.get(r);
                let cells = cols
                    .iter()
                    .enumerate()
                    .map(|(c, x)| HeatmapCell {
                        x: x.clone(),
                        y: row_vals.and_then(|v| v.get(c)).copied().unwrap_or(0.0),
                    })
                    .collect();
                HeatmapRow {
                    name: label.into(),
                    cells,
                }
            })
            .collect();
        Self { rows }
    }
}

/// One row of a [`ProgressPayload`] — a labeled horizontal bar.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ProgressItem {
    pub label: String,
    /// Pre-formatted value shown at the right of the row (the
    /// `from_pairs` constructors thousands-group it; `new` takes it
    /// verbatim).
    pub display: String,
    /// Bar fill width, `0–100`, relative to the payload's reference
    /// (the largest value, or an explicit target).
    pub percent: f64,
    /// Optional explicit bar color (CSS hex / rgb / token name).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub color: Option<String>,
}

/// Progress-bar list payload — a ranked set of labeled horizontal bars,
/// each filled relative to the largest value (or an explicit target).
/// The "top N by metric" tile: revenue by product, traffic by source,
/// completion per category. Rendered as pure HTML — no chart library.
///
/// ```ignore
/// // Revenue by product; the top product fills the bar.
/// ProgressPayload::from_pairs([("Pro", 48200.0), ("Team", 31000.0), ("Free", 9400.0)])
/// // Completion per team, each measured against a 100-task target.
/// ProgressPayload::from_pairs_of([("Web", 82.0), ("Mobile", 57.0)], 100.0)
/// ```
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ProgressPayload {
    pub items: Vec<ProgressItem>,
}

impl ProgressPayload {
    /// New payload from explicit items (you set `display` + `percent`).
    pub fn new(items: Vec<ProgressItem>) -> Self {
        Self { items }
    }

    /// From `(label, value)` pairs, with each bar sized relative to the
    /// LARGEST value (the top item fills the bar). `display` is the
    /// thousands-grouped value.
    pub fn from_pairs<L: Into<String>>(pairs: impl IntoIterator<Item = (L, f64)>) -> Self {
        let items: Vec<(String, f64)> = pairs.into_iter().map(|(l, v)| (l.into(), v)).collect();
        let reference = items
            .iter()
            .map(|(_, v)| *v)
            .filter(|v| v.is_finite())
            .fold(0.0_f64, f64::max);
        Self::build(items, reference)
    }

    /// From `(label, value)` pairs, with each bar sized against an
    /// explicit `target` (e.g. a per-row "% of goal"). A non-positive
    /// target falls back to sizing against the largest value.
    pub fn from_pairs_of<L: Into<String>>(
        pairs: impl IntoIterator<Item = (L, f64)>,
        target: f64,
    ) -> Self {
        let items: Vec<(String, f64)> = pairs.into_iter().map(|(l, v)| (l.into(), v)).collect();
        let reference = if target > 0.0 {
            target
        } else {
            items
                .iter()
                .map(|(_, v)| *v)
                .filter(|v| v.is_finite())
                .fold(0.0_f64, f64::max)
        };
        Self::build(items, reference)
    }

    /// Shared: turn `(label, value)` + a reference max into rendered
    /// items. `percent = value / reference * 100`, clamped to
    /// `[0, 100]`; a zero/non-finite reference yields empty bars.
    fn build(items: Vec<(String, f64)>, reference: f64) -> Self {
        let items = items
            .into_iter()
            .map(|(label, value)| {
                let percent = if reference > 0.0 && value.is_finite() {
                    (value / reference * 100.0).clamp(0.0, 100.0)
                } else {
                    0.0
                };
                ProgressItem {
                    label,
                    display: format_thousands(value),
                    percent,
                    color: None,
                }
            })
            .collect();
        Self { items }
    }
}

/// Table widget column descriptor.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct TableColumn {
    pub key: String,
    pub label: String,
}

/// Table widget payload.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct TablePayload {
    pub columns: Vec<TableColumn>,
    pub rows: Vec<serde_json::Value>,
    /// Optional "View all →" link in the widget header. Populated
    /// via [`Self::view_all_for`] (auto-resolves the admin URL
    /// from a `Model` type) or set explicitly when the target
    /// isn't a managed admin model.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub view_all_url: Option<String>,
}

impl TablePayload {
    /// New payload from columns + rows; no `view_all` link.
    pub fn new(columns: Vec<TableColumn>, rows: Vec<serde_json::Value>) -> Self {
        Self {
            columns,
            rows,
            view_all_url: None,
        }
    }

    /// Auto-resolve the "View all" link from a `Model` type — the
    /// admin's changelist URL for that table. Mirrors the pattern
    /// used by `models![T, U, V]`: rename the struct's
    /// `#[umbral(table = "...")]` and the link follows automatically.
    ///
    /// ```rust,ignore
    /// WidgetPayload::Table(
    ///     TablePayload::new(columns, rows)
    ///         .view_all_for::<Order>()
    /// )
    /// // → "View all →" links to {admin_base}/order/
    /// ```
    pub fn view_all_for<T: umbral::orm::Model>(mut self) -> Self {
        self.view_all_url = Some(format!(
            "{}/{}/",
            crate::branding::current().base_path,
            T::TABLE,
        ));
        self
    }

    /// Explicit URL override — use when the link target isn't a
    /// managed admin model (an external dashboard, a custom route).
    pub fn view_all_url(mut self, url: impl Into<String>) -> Self {
        self.view_all_url = Some(url.into());
        self
    }
}

/// One item in an activity feed.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct FeedItem {
    pub actor: String,
    pub verb: String,
    pub object: String,
    pub object_link: Option<String>,
    pub at: String,
}

/// Activity feed payload.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct FeedPayload {
    pub items: Vec<FeedItem>,
    /// Optional "View all →" link in the widget header. Same
    /// shape as [`TablePayload::view_all_url`] — auto-resolve
    /// from a `Model` via [`Self::view_all_for`].
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub view_all_url: Option<String>,
}

impl FeedPayload {
    /// New payload from items; no `view_all` link.
    pub fn new(items: Vec<FeedItem>) -> Self {
        Self {
            items,
            view_all_url: None,
        }
    }

    /// Auto-resolve the "View all" link from a `Model` type. The
    /// recent-signups feed for instance:
    ///
    /// ```rust,ignore
    /// WidgetPayload::Feed(
    ///     FeedPayload::new(items).view_all_for::<AuthUser>()
    /// )
    /// // → "View all →" links to {admin_base}/auth_user/
    /// ```
    pub fn view_all_for<T: umbral::orm::Model>(mut self) -> Self {
        self.view_all_url = Some(format!(
            "{}/{}/",
            crate::branding::current().base_path,
            T::TABLE,
        ));
        self
    }

    pub fn view_all_url(mut self, url: impl Into<String>) -> Self {
        self.view_all_url = Some(url.into());
        self
    }
}

/// Union of all widget payloads. The JSON discriminant is the variant name.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "kind", rename_all = "lowercase")]
pub enum WidgetPayload {
    Kpi(KpiPayload),
    Card(CardPayload),
    Line(LinePayload),
    Bar(BarPayload),
    Donut(DonutPayload),
    Radial(RadialPayload),
    Heatmap(HeatmapPayload),
    Progress(ProgressPayload),
    Table(TablePayload),
    Feed(FeedPayload),
}

/// Quote a CSV field per RFC 4180: wrap in `"` when it contains a comma, a
/// quote or a newline, and double any embedded quotes.
///
/// Also neutralises the spreadsheet-formula injection vector: a cell whose text
/// starts with `=`, `+`, `-` or `@` is executed as a formula when the file is
/// opened in Excel or Sheets, so a row whose value someone typed into your app
/// (`=HYPERLINK(...)`, `=cmd|...`) becomes code running on the machine of
/// whoever opened the export. Prefixing a tab keeps the text readable while
/// making it inert.
fn csv_field(value: &str) -> String {
    let dangerous = value
        .chars()
        .next()
        .is_some_and(|c| matches!(c, '=' | '+' | '-' | '@'));
    let value = if dangerous {
        format!("\t{value}")
    } else {
        value.to_string()
    };
    if value.contains(',') || value.contains('"') || value.contains('\n') || value.contains('\r') {
        format!("\"{}\"", value.replace('"', "\"\""))
    } else {
        value
    }
}

fn csv_row<I: IntoIterator<Item = String>>(cells: I) -> String {
    cells
        .into_iter()
        .map(|c| csv_field(&c))
        .collect::<Vec<_>>()
        .join(",")
}

fn num(v: f64) -> String {
    if v.fract() == 0.0 {
        format!("{}", v as i64)
    } else {
        format!("{v}")
    }
}

impl WidgetPayload {
    /// The payload as CSV, when it has rows worth exporting.
    ///
    /// `None` for the shapes that are a single number or a prose feed — a KPI
    /// tile is not a table, and pretending otherwise would hand the user a
    /// one-cell file. Everything with a series, slices, bars or rows exports.
    pub fn to_csv(&self) -> Option<String> {
        let mut out = String::new();
        match self {
            // A chart is (series, x, y) once flattened, which is exactly what a
            // spreadsheet wants — one tidy row per point, series as a column so
            // a multi-series chart round-trips into a pivot table.
            WidgetPayload::Line(LinePayload { series, .. })
            | WidgetPayload::Bar(BarPayload { series, .. }) => {
                out.push_str("series,x,y\n");
                for s in series {
                    for p in &s.points {
                        out.push_str(&csv_row([s.name.clone(), p.x.clone(), num(p.y)]));
                        out.push('\n');
                    }
                }
            }
            WidgetPayload::Donut(p) => {
                out.push_str("label,value\n");
                for s in &p.slices {
                    out.push_str(&csv_row([s.label.clone(), num(s.value)]));
                    out.push('\n');
                }
            }
            WidgetPayload::Radial(p) => {
                out.push_str("label,value\n");
                for t in &p.tracks {
                    out.push_str(&csv_row([t.label.clone(), num(t.value)]));
                    out.push('\n');
                }
            }
            WidgetPayload::Progress(p) => {
                for i in &p.items {
                    out.push_str(&csv_row([
                        i.label.clone(),
                        num(i.percent),
                        i.display.clone(),
                    ]));
                    out.push('\n');
                }
            }
            WidgetPayload::Heatmap(p) => {
                out.push_str("row,column,value\n");
                for r in &p.rows {
                    for c in &r.cells {
                        out.push_str(&csv_row([r.name.clone(), c.x.clone(), num(c.y)]));
                        out.push('\n');
                    }
                }
            }
            WidgetPayload::Table(p) => {
                out.push_str(&csv_row(p.columns.iter().map(|c| c.label.clone())));
                out.push('\n');
                for row in &p.rows {
                    let cells = p.columns.iter().map(|c| match row.get(&c.key) {
                        Some(serde_json::Value::String(s)) => s.clone(),
                        Some(serde_json::Value::Null) | None => String::new(),
                        Some(v) => v.to_string(),
                    });
                    out.push_str(&csv_row(cells));
                    out.push('\n');
                }
            }
            // A single number or a prose feed is not a table.
            WidgetPayload::Kpi(_) | WidgetPayload::Card(_) | WidgetPayload::Feed(_) => return None,
        }
        Some(out)
    }
}

// =========================================================================
// WidgetDataFn
// =========================================================================

/// Per-request parameters a widget's data closure can read.
/// Sourced from the query string on
/// `GET /admin/api/dashboard/widgets/<key>/data?<params>`.
///
/// Defaults are all `None` — closures that don't care can use
/// `WidgetDataFn::new(|user| ...)` and ignore params entirely.
/// Closures that DO care use `WidgetDataFn::with_params` and
/// branch on `params.period` / `params.start` / `params.end`.
#[derive(Debug, Clone, Default)]
pub struct WidgetParams {
    /// Period preset like `"7d"`, `"30d"`, `"90d"`. The
    /// rendering side emits chips that pass this through.
    pub period: Option<String>,
    /// Explicit ISO start date (`YYYY-MM-DD`) — overrides
    /// `period` when both are present.
    pub start: Option<String>,
    /// Explicit ISO end date (`YYYY-MM-DD`).
    pub end: Option<String>,
    /// Catch-all for any other widget-specific query params
    /// — `?model=order` for a future per-model filter, etc.
    /// Closures read by `params.raw.get("...")`.
    pub raw: std::collections::HashMap<String, String>,
}

impl WidgetParams {
    /// Build from a `?key=value&...` query string. Recognised
    /// keys (`period`, `start`, `end`) populate the typed
    /// fields; the rest land in `raw`.
    pub fn from_query<S: AsRef<str>>(query: S) -> Self {
        let mut out = Self::default();
        for pair in query.as_ref().split('&').filter(|s| !s.is_empty()) {
            let (k, v) = pair.split_once('=').unwrap_or((pair, ""));
            let value = urlencoding_decode(v);
            match k {
                "period" => out.period = Some(value),
                "start" => out.start = Some(value),
                "end" => out.end = Some(value),
                _ => {
                    out.raw.insert(k.to_string(), value);
                }
            }
        }
        out
    }

    /// Number of days the `period` preset represents. `"7d"`
    /// → 7, `"30d"` → 30, `"90d"` → 90. None for unrecognised /
    /// missing values so callers fall back to a default.
    pub fn period_days(&self) -> Option<i64> {
        let p = self.period.as_deref()?;
        let digits: String = p.chars().take_while(|c| c.is_ascii_digit()).collect();
        digits.parse().ok()
    }

    /// Value of a [`WidgetFilter::choice`] control, by its key.
    ///
    /// ```rust,ignore
    /// let status = params.choice("status").unwrap_or("open");
    /// ```
    pub fn choice(&self, key: &str) -> Option<&str> {
        self.raw
            .get(key)
            .map(String::as_str)
            .filter(|v| !v.is_empty())
    }

    /// The resolved window as `(start, end)` ISO dates, if a date-range filter
    /// supplied both. Takes precedence over `period` — a caller who picked
    /// explicit dates meant them.
    pub fn date_range(&self) -> Option<(&str, &str)> {
        match (self.start.as_deref(), self.end.as_deref()) {
            (Some(s), Some(e)) if !s.is_empty() && !e.is_empty() => Some((s, e)),
            _ => None,
        }
    }
}

/// Minimal `%XX` → byte decoder; avoids pulling a query-string
/// crate just for the four chars we need (`+` → space, `%2F` → `/`,
/// etc.). Anything malformed passes through unchanged.
fn urlencoding_decode(raw: &str) -> String {
    let mut out = String::with_capacity(raw.len());
    let bytes = raw.as_bytes();
    let mut i = 0;
    while i < bytes.len() {
        match bytes[i] {
            b'+' => {
                out.push(' ');
                i += 1;
            }
            b'%' if i + 2 < bytes.len() => {
                let hi = (bytes[i + 1] as char).to_digit(16);
                let lo = (bytes[i + 2] as char).to_digit(16);
                if let (Some(h), Some(l)) = (hi, lo) {
                    out.push(char::from((h as u8) * 16 + l as u8));
                    i += 3;
                } else {
                    out.push(bytes[i] as char);
                    i += 1;
                }
            }
            b => {
                out.push(b as char);
                i += 1;
            }
        }
    }
    out
}

// =========================================================================
// WidgetFilter — declarative, per-widget UI controls
// =========================================================================

/// What kind of control a [`WidgetFilter`] renders.
///
/// The variants exist because the three shapes reach the data closure by
/// different routes: a period lands in `WidgetParams::period`, a date range in
/// `start` / `end`, and a choice in `raw[key]`. Keeping that in the type means
/// the template never has to guess which query parameter a control writes to.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "kind", rename_all = "snake_case")]
pub enum WidgetFilterKind {
    /// A chip strip of presets — `7d`, `30d`, `90d`. Writes `?period=`.
    Period { presets: Vec<FilterOption> },
    /// Two `<input type="date">` boxes. Writes `?start=` and `?end=`.
    DateRange,
    /// A `<select>`. Writes `?<key>=`, readable via `params.choice(key)`.
    Choice { options: Vec<FilterOption> },
}

/// One selectable value in a period or choice filter.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct FilterOption {
    pub value: String,
    pub label: String,
}

impl FilterOption {
    pub fn new(value: impl Into<String>, label: impl Into<String>) -> Self {
        Self {
            value: value.into(),
            label: label.into(),
        }
    }
}

/// A control the admin renders in a widget's header, declared at registration
/// time and rendered for **every** widget kind.
///
/// Before this existed the only filter in the admin was a `[7d][30d][90d]` chip
/// strip hardcoded inside `line.html`, which meant a bar chart or a table could
/// not be filtered at all, and no widget could offer anything but a period. A
/// filter is now data: the widget declares it, the admin renders it, and the
/// value arrives in [`WidgetParams`].
///
/// ```rust,ignore
/// Widget::new("orders_by_status", "Orders", WidgetKind::Bar, data_fn)
///     .filter(WidgetFilter::period_default())
///     .filter(WidgetFilter::choice("status", "Status", [
///         ("open", "Open"), ("paid", "Paid"),
///     ]))
/// ```
///
/// The chosen value is **sticky per user**: picking one persists to the user's
/// admin preferences, so it survives a reload the same way a period chip does.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct WidgetFilter {
    /// Query-parameter name. `"period"` for a period filter, `"start"`/`"end"`
    /// are implied by a date range, anything else for a choice.
    pub key: String,
    /// Label shown beside the control.
    pub label: String,
    /// Which control to render.
    #[serde(flatten)]
    pub kind: WidgetFilterKind,
    /// Value selected when the user has never touched this filter.
    pub default: Option<String>,
    /// Resolved value for THIS request — URL, else the user's saved choice,
    /// else `default`. Filled in by the handler; ignored at registration.
    #[serde(default)]
    pub active: Option<String>,
    /// Resolved range ends, for a [`WidgetFilterKind::DateRange`] only.
    /// Filled in by the handler; ignored at registration.
    #[serde(default)]
    pub active_start: Option<String>,
    #[serde(default)]
    pub active_end: Option<String>,
    /// Query-string fragment carrying every OTHER filter's current value, e.g.
    /// `"&status=paid&period=7d"`. Filled in by the handler.
    ///
    /// It lives here, computed in Rust, because the alternative was assembling
    /// it in the template with `split`/`trim` gymnastics — and a control that
    /// gets this wrong silently resets its neighbours the moment you touch it.
    #[serde(default)]
    pub carry: String,
    /// The same fragment without the leading `&`, for a control that starts the
    /// query string rather than appending to one (a `<select>` whose own value
    /// htmx appends via `hx-include`).
    #[serde(default)]
    pub carry_lead: String,
}

impl WidgetFilter {
    /// A period chip strip with custom presets.
    pub fn period<I, V, L>(presets: I) -> Self
    where
        I: IntoIterator<Item = (V, L)>,
        V: Into<String>,
        L: Into<String>,
    {
        Self {
            key: "period".to_string(),
            label: "Period".to_string(),
            kind: WidgetFilterKind::Period {
                presets: presets
                    .into_iter()
                    .map(|(v, l)| FilterOption::new(v, l))
                    .collect(),
            },
            default: None,
            active: None,
            active_start: None,
            active_end: None,
            carry: String::new(),
            carry_lead: String::new(),
        }
    }

    /// The conventional `[7d] [30d] [90d]` strip — what `line.html` hardcoded.
    pub fn period_default() -> Self {
        Self::period([("7d", "7d"), ("30d", "30d"), ("90d", "90d")]).with_default("30d")
    }

    /// A start/end date pair. Reaches the closure as `params.start` / `params.end`,
    /// which take precedence over `period` when both are present.
    pub fn date_range() -> Self {
        Self {
            key: "range".to_string(),
            label: "Date range".to_string(),
            kind: WidgetFilterKind::DateRange,
            default: None,
            active: None,
            active_start: None,
            active_end: None,
            carry: String::new(),
            carry_lead: String::new(),
        }
    }

    /// A `<select>` writing `?<key>=`. Read it back with `params.choice(key)`.
    pub fn choice<I, V, L>(key: impl Into<String>, label: impl Into<String>, options: I) -> Self
    where
        I: IntoIterator<Item = (V, L)>,
        V: Into<String>,
        L: Into<String>,
    {
        Self {
            key: key.into(),
            label: label.into(),
            kind: WidgetFilterKind::Choice {
                options: options
                    .into_iter()
                    .map(|(v, l)| FilterOption::new(v, l))
                    .collect(),
            },
            default: None,
            active: None,
            active_start: None,
            active_end: None,
            carry: String::new(),
            carry_lead: String::new(),
        }
    }

    /// Pre-select a value on first paint.
    pub fn with_default(mut self, value: impl Into<String>) -> Self {
        self.default = Some(value.into());
        self
    }

    /// The value in force for this request.
    pub fn active_value(&self) -> Option<&str> {
        self.active.as_deref().or(self.default.as_deref())
    }
}

pub(crate) type DataFuture = Pin<Box<dyn Future<Output = WidgetPayload> + Send + 'static>>;
pub(crate) type DataFnInner =
    Arc<dyn Fn(AuthUser, WidgetParams) -> DataFuture + Send + Sync + 'static>;

/// Wrapper around the async data closure. Build via
/// [`WidgetDataFn::new`] (closure ignores per-request params) or
/// [`WidgetDataFn::with_params`] (closure reads `WidgetParams` to
/// honour period / date-range filters from the request URL).
#[derive(Clone)]
pub struct WidgetDataFn(pub(crate) DataFnInner);

impl WidgetDataFn {
    /// Create from any `async fn(AuthUser) -> WidgetPayload` —
    /// per-request params are dropped on the floor. Use when the
    /// widget renders the same thing regardless of UI controls
    /// (KPI counts, registry sizes, etc.).
    pub fn new<F, Fut>(f: F) -> Self
    where
        F: Fn(AuthUser) -> Fut + Send + Sync + 'static,
        Fut: Future<Output = WidgetPayload> + Send + 'static,
    {
        Self(Arc::new(move |user, _params| Box::pin(f(user))))
    }

    /// Create from `async fn(AuthUser, WidgetParams) ->
    /// WidgetPayload`. Use for filterable widgets — the line
    /// chart reads `params.period` to switch between 7d / 30d /
    /// 90d views, a future table widget might read
    /// `params.raw.get("status")` for status filtering, etc.
    pub fn with_params<F, Fut>(f: F) -> Self
    where
        F: Fn(AuthUser, WidgetParams) -> Fut + Send + Sync + 'static,
        Fut: Future<Output = WidgetPayload> + Send + 'static,
    {
        Self(Arc::new(move |user, params| Box::pin(f(user, params))))
    }
}

impl std::fmt::Debug for WidgetDataFn {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.write_str("WidgetDataFn(<fn>)")
    }
}

// =========================================================================
// Widget
// =========================================================================

/// A registered dashboard widget.
///
/// Register via `AdminPlugin::register_widget(...)`.
#[derive(Debug, Clone)]
pub struct Widget {
    /// URL-safe unique key, e.g. `"umbral_total_models"`.
    pub key: &'static str,
    /// Human-readable title shown in the widget card header.
    pub title: String,
    /// Determines which renderer (KPI card, chart, table, feed).
    pub kind: WidgetKind,
    /// Default grid span when the user hasn't customized.
    pub default_span: Span,
    /// Optional permission codename. `None` = any staff user may see.
    pub permission: Option<&'static str>,
    /// Async function that computes and returns the payload.
    pub data: WidgetDataFn,
    /// Default period preset used by line/bar/etc. widgets that
    /// carry a period-chip strip — `"7d"`, `"30d"`, `"90d"`. When
    /// `Some`, the handler pre-fills `WidgetParams.period` from
    /// this value on first load (no `?period=` in the URL), so
    /// the matching chip renders highlighted AND the data
    /// closure receives the same period via `params.period_days()`.
    /// `None` falls back to whatever the template / data closure
    /// chooses as its fallback.
    pub default_period: Option<&'static str>,
    /// Controls rendered in this widget's header — see [`WidgetFilter`].
    ///
    /// Empty means no controls, with one compatibility exception: a
    /// [`WidgetKind::Line`] with no declared filters still gets the historic
    /// `[7d][30d][90d]` strip, because that strip used to be hardcoded in
    /// `line.html` and removing it would silently strip controls from every
    /// line chart already in the wild.
    pub filters: Vec<WidgetFilter>,
}

impl Widget {
    /// Build a widget without naming every field.
    ///
    /// Prefer this over a struct literal. `Widget` grows fields as the
    /// dashboard grows (this is how `filters` arrived), and every new field
    /// breaks every literal that spells out the old ones. Going through the
    /// builder means the next field costs you nothing.
    ///
    /// ```rust,ignore
    /// Widget::new("orders", "Orders", WidgetKind::Bar, data_fn)
    ///     .with_span(6, 2)
    ///     .filter(WidgetFilter::period_default())
    /// ```
    pub fn new(
        key: &'static str,
        title: impl Into<String>,
        kind: WidgetKind,
        data: WidgetDataFn,
    ) -> Self {
        Self {
            key,
            title: title.into(),
            kind,
            default_span: Span::default(),
            permission: None,
            data,
            default_period: None,
            filters: Vec::new(),
        }
    }

    /// Gate the widget behind a permission codename.
    pub fn with_permission(mut self, codename: &'static str) -> Self {
        self.permission = Some(codename);
        self
    }

    /// Add one filter control to the widget header.
    pub fn filter(mut self, filter: WidgetFilter) -> Self {
        self.filters.push(filter);
        self
    }

    /// Replace the widget's filters wholesale.
    pub fn with_filters(mut self, filters: impl IntoIterator<Item = WidgetFilter>) -> Self {
        self.filters = filters.into_iter().collect();
        self
    }

    /// The filters to render for this request, including the line-chart
    /// compatibility strip described on [`Widget::filters`].
    pub(crate) fn effective_filters(&self) -> Vec<WidgetFilter> {
        if self.filters.is_empty() && matches!(self.kind, WidgetKind::Line) {
            let mut period = WidgetFilter::period_default();
            if let Some(d) = self.default_period {
                period.default = Some(d.to_string());
            }
            return vec![period];
        }
        self.filters.clone()
    }

    /// Override the default grid span. Lets a caller resize a
    /// builtin (or any pre-built widget) at registration time
    /// without having to re-construct the whole struct literal:
    ///
    /// ```rust,ignore
    /// .register_widget(builtin_total_models_widget().with_span(6, 2))
    /// .register_widget(builtin_recent_users_widget().with_span(6, 2))
    /// ```
    ///
    /// `cols` is clamped at the 12-col grid; `rows` is whatever
    /// the dashboard's `auto-rows-[...]` accepts (1 = 120px).
    pub fn with_span(mut self, cols: u8, rows: u8) -> Self {
        self.default_span = Span { cols, rows };
        self
    }

    /// Pre-select a period chip on the widget — `"7d"`, `"30d"`,
    /// `"90d"`. On first load (no `?period=` in the URL), the
    /// handler stamps this into `WidgetParams.period` before
    /// calling the data closure, so the chip strip highlights
    /// the right preset AND the data fn computes the right
    /// window. Override on a per-request basis happens via the
    /// chip clicks (which send their own `?period=` query).
    ///
    /// ```ignore
    /// shop_daily_sales_chart().with_default_period("7d")
    /// // → first paint shows 7d highlighted, 7 days of data;
    /// //   clicking "30d" hands control to the URL state.
    /// ```
    pub fn with_default_period(mut self, period: &'static str) -> Self {
        self.default_period = Some(period);
        self
    }
}

// =========================================================================
// WidgetInstance (user's saved layout entry)
// =========================================================================

/// One entry in a user's saved layout JSON.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct WidgetInstance {
    pub key: String,
    pub span: Span,
}

// =========================================================================
// Widget catalog entry (API response shape)
// =========================================================================

/// Serialized catalog entry returned by `GET /admin/api/dashboard/catalog`.
#[derive(Debug, Clone, Serialize)]
pub struct CatalogEntry {
    pub key: &'static str,
    pub title: String,
    pub kind: String,
    pub default_span: Span,
}

// =========================================================================
// Sections — grouped widgets
// =========================================================================

/// A named group of widgets on the dashboard. Each section renders
/// as its own heading + (optional) subtitle + widget grid, so a
/// dashboard with 20 widgets reads as themed clusters rather than
/// one mega-grid.
///
/// Build with the chainable API:
///
/// ```rust,ignore
/// use umbral_admin::WidgetSection;
///
/// let sales = WidgetSection::new("Sales overview")
///     .subtitle("Daily KPIs across the storefront")
///     .widget(shop_total_sales_widget())
///     .widget(shop_orders_widget())
///     .widget(shop_avg_order_value_widget());
///
/// AdminPlugin::default().dashboard_section(sales);
/// ```
///
/// Register multiple sections by chaining `.dashboard_section(...)`.
/// Widgets registered via the legacy `.register_widget(...)` end up
/// in an implicit final section titled "Widgets" — so existing apps
/// keep working without code changes.
#[derive(Debug, Clone)]
pub struct WidgetSection {
    /// Heading shown above the section (e.g. "Sales overview").
    pub title: String,
    /// Optional descriptive line under the title — keep it short,
    /// it's not a paragraph.
    pub subtitle: Option<String>,
    /// Widgets in this section, rendered in registration order.
    pub widgets: Vec<Widget>,
}

impl WidgetSection {
    /// New empty section with just a title.
    pub fn new(title: impl Into<String>) -> Self {
        Self {
            title: title.into(),
            subtitle: None,
            widgets: Vec::new(),
        }
    }

    /// Add a one-line subtitle under the heading.
    pub fn subtitle(mut self, subtitle: impl Into<String>) -> Self {
        self.subtitle = Some(subtitle.into());
        self
    }

    /// Append one widget to the section.
    pub fn widget(mut self, w: Widget) -> Self {
        self.widgets.push(w);
        self
    }

    /// Append many widgets at once (handy for splatting a Vec).
    pub fn widgets(mut self, ws: impl IntoIterator<Item = Widget>) -> Self {
        self.widgets.extend(ws);
        self
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn radial_kind_serializes_as_radial() {
        assert_eq!(WidgetKind::Radial.as_str(), "radial");
    }

    #[test]
    fn radial_single_builds_one_track() {
        let p = RadialPayload::single("Monthly goal", 73.0);
        assert_eq!(p.tracks.len(), 1);
        assert_eq!(p.tracks[0].label, "Monthly goal");
        assert_eq!(p.tracks[0].value, 73.0);
        assert!(p.tracks[0].color.is_none());
    }

    #[test]
    fn radial_clamps_out_of_range_and_non_finite_percents() {
        // Over 100, under 0, and non-finite all clamp into [0, 100].
        assert_eq!(RadialPayload::single("over", 150.0).tracks[0].value, 100.0);
        assert_eq!(RadialPayload::single("under", -20.0).tracks[0].value, 0.0);
        // Non-finite (NaN, ±∞) is meaningless as a percent -> 0.
        assert_eq!(RadialPayload::single("nan", f64::NAN).tracks[0].value, 0.0);
        assert_eq!(
            RadialPayload::single("inf", f64::INFINITY).tracks[0].value,
            0.0,
        );
    }

    #[test]
    fn radial_goal_is_current_over_target() {
        assert_eq!(RadialPayload::goal("g", 73.0, 100.0).tracks[0].value, 73.0);
        // current > target clamps to 100 (overachieved, full ring).
        assert_eq!(
            RadialPayload::goal("g", 120.0, 100.0).tracks[0].value,
            100.0
        );
        // A non-positive target has nothing to measure against -> 0%.
        assert_eq!(RadialPayload::goal("g", 5.0, 0.0).tracks[0].value, 0.0);
    }

    #[test]
    fn radial_from_pairs_keeps_order_and_clamps() {
        let p = RadialPayload::from_pairs([("Free", 8.0), ("Pro", 150.0), ("Team", 34.0)]);
        assert_eq!(p.tracks.len(), 3);
        assert_eq!(p.tracks[0].label, "Free");
        assert_eq!(p.tracks[1].value, 100.0); // clamped
        assert_eq!(p.tracks[2].label, "Team");
    }

    #[test]
    fn radial_payload_serializes_with_kind_tag() {
        let payload = WidgetPayload::Radial(RadialPayload::single("Quota", 42.0));
        let json = serde_json::to_value(&payload).expect("serialize");
        assert_eq!(json["kind"], "radial");
        assert_eq!(json["tracks"][0]["label"], "Quota");
        assert_eq!(json["tracks"][0]["value"], 42.0);
        // No explicit color -> the field is skipped entirely.
        assert!(json["tracks"][0].get("color").is_none());
    }

    #[test]
    fn heatmap_kind_serializes_as_heatmap() {
        assert_eq!(WidgetKind::Heatmap.as_str(), "heatmap");
    }

    #[test]
    fn heatmap_from_grid_is_rectangular_and_padded() {
        // A ragged matrix: row 0 short (padded with 0), row 1 long
        // (extra dropped), row 2 exact.
        let p = HeatmapPayload::from_grid(
            ["Mon", "Tue", "Wed"],
            ["AM", "PM"],
            vec![vec![3.0], vec![1.0, 2.0, 99.0], vec![4.0, 5.0]],
        );
        assert_eq!(p.rows.len(), 3);
        // Every row has exactly one cell per column label.
        for row in &p.rows {
            assert_eq!(row.cells.len(), 2, "row `{}` must be rectangular", row.name);
            assert_eq!(row.cells[0].x, "AM");
            assert_eq!(row.cells[1].x, "PM");
        }
        assert_eq!(p.rows[0].name, "Mon");
        assert_eq!(p.rows[0].cells[1].y, 0.0); // short row padded
        assert_eq!(p.rows[1].cells[1].y, 2.0); // extra `99.0` dropped
        assert_eq!(p.rows[2].cells[0].y, 4.0);
    }

    #[test]
    fn heatmap_payload_serializes_with_kind_tag() {
        let payload = WidgetPayload::Heatmap(HeatmapPayload::from_grid(
            ["Row"],
            ["a", "b"],
            vec![vec![7.0, 8.0]],
        ));
        let json = serde_json::to_value(&payload).expect("serialize");
        assert_eq!(json["kind"], "heatmap");
        assert_eq!(json["rows"][0]["name"], "Row");
        assert_eq!(json["rows"][0]["cells"][0]["x"], "a");
        assert_eq!(json["rows"][0]["cells"][1]["y"], 8.0);
    }

    #[test]
    fn progress_kind_serializes_as_progress() {
        assert_eq!(WidgetKind::Progress.as_str(), "progress");
    }

    #[test]
    fn progress_from_pairs_sizes_against_largest_value() {
        let p = ProgressPayload::from_pairs([("A", 100.0), ("B", 50.0), ("C", 25.0)]);
        assert_eq!(p.items.len(), 3);
        // The largest value fills the bar; the rest are proportional.
        assert_eq!(p.items[0].percent, 100.0);
        assert_eq!(p.items[1].percent, 50.0);
        assert_eq!(p.items[2].percent, 25.0);
        // `display` is the thousands-grouped value, order preserved.
        assert_eq!(p.items[0].label, "A");
        assert_eq!(p.items[0].display, "100");
    }

    #[test]
    fn progress_from_pairs_of_sizes_against_target_and_clamps() {
        let p = ProgressPayload::from_pairs_of([("Web", 82.0), ("Mobile", 150.0)], 100.0);
        assert_eq!(p.items[0].percent, 82.0);
        // Over target fills the bar rather than overrunning it.
        assert_eq!(p.items[1].percent, 100.0);
    }

    #[test]
    fn progress_non_positive_target_falls_back_to_max() {
        // target = 0 -> size against the largest value (40 -> 100%).
        let p = ProgressPayload::from_pairs_of([("A", 40.0), ("B", 10.0)], 0.0);
        assert_eq!(p.items[0].percent, 100.0);
        assert_eq!(p.items[1].percent, 25.0);
    }

    #[test]
    fn progress_payload_serializes_with_kind_tag() {
        let payload = WidgetPayload::Progress(ProgressPayload::from_pairs([("Pro", 48200.0)]));
        let json = serde_json::to_value(&payload).expect("serialize");
        assert_eq!(json["kind"], "progress");
        assert_eq!(json["items"][0]["label"], "Pro");
        assert_eq!(json["items"][0]["display"], "48,200");
        assert_eq!(json["items"][0]["percent"], 100.0);
        // No explicit color -> the field is skipped entirely.
        assert!(json["items"][0].get("color").is_none());
    }
}

#[cfg(test)]
mod csv_tests {
    use super::*;

    #[test]
    fn csv_quotes_commas_quotes_and_newlines() {
        assert_eq!(csv_field("plain"), "plain");
        assert_eq!(csv_field("a,b"), "\"a,b\"");
        assert_eq!(csv_field("say \"hi\""), "\"say \"\"hi\"\"\"");
        assert_eq!(csv_field("two\nlines"), "\"two\nlines\"");
    }

    /// A cell beginning `=`, `+`, `-` or `@` is executed as a FORMULA when the
    /// file is opened in Excel or Sheets. A product name someone typed into your
    /// app then becomes code running on the machine of whoever opened the export.
    /// The value must survive as readable text, but inert.
    #[test]
    fn csv_neutralises_spreadsheet_formulas() {
        for (payload, still_reads) in [
            ("=HYPERLINK(\"http://evil\")", "HYPERLINK"),
            ("+1+1", "+1+1"),
            ("-2+3", "-2+3"),
            ("@SUM(A1)", "@SUM(A1)"),
        ] {
            let out = csv_field(payload);
            // The tab may sit inside the RFC-4180 quoting, so accept either.
            assert!(
                out.starts_with('\t') || out.starts_with("\"\t"),
                "a formula-leading cell must be neutralised, got {out}"
            );
            assert!(
                out.contains(still_reads),
                "and must remain readable, got {out}"
            );
        }
    }

    #[test]
    fn a_bar_payload_exports_one_row_per_point() {
        let payload = WidgetPayload::Bar(BarPayload {
            series: vec![Series {
                name: "sales".into(),
                points: vec![
                    ChartPoint {
                        x: "Mon".into(),
                        y: 3.0,
                    },
                    ChartPoint {
                        x: "Tue".into(),
                        y: 4.5,
                    },
                ],
            }],
            x_type: "day".into(),
        });
        let csv = payload.to_csv().expect("a bar chart has rows");
        assert_eq!(csv, "series,x,y\nsales,Mon,3\nsales,Tue,4.5\n");
    }

    #[test]
    fn a_kpi_has_nothing_to_export() {
        let payload = WidgetPayload::Kpi(KpiPayload {
            value: "42".into(),
            unit: None,
            delta: None,
            sparkline: None,
        });
        assert!(payload.to_csv().is_none());
    }
}