fulgur-chart 0.3.0

//! chart.js v4 spec のデータ専用・静的サブセットを IR へ変換する。

use crate::color::parse_color;
use crate::ir::*;
use serde::Deserialize;
use std::collections::HashMap;

#[derive(Deserialize)]
struct RawSpec {
    #[serde(rename = "type")]
    chart_type: String,
    data: RawData,
    #[serde(default)]
    options: RawOptions,
}

#[derive(Deserialize, Default)]
struct RawOptions {
    #[serde(rename = "indexAxis")]
    index_axis: Option<String>,
    #[serde(default)]
    plugins: RawPlugins,
    #[serde(default)]
    theme: Option<RawTheme>,
    // scales.{x,y}.stacked のみ navigate する(積み上げ判定)。それ以外は未マップ。
    #[serde(default)]
    scales: Option<serde_json::Value>,
}

/// `options.theme`: 視覚トークンの上書き。各フィールドは任意。
#[derive(Deserialize)]
struct RawTheme {
    #[serde(default)]
    palette: Option<Vec<String>>,
    #[serde(rename = "gridColor", default)]
    grid_color: Option<String>,
    #[serde(rename = "textColor", default)]
    text_color: Option<String>,
    #[serde(rename = "backgroundColor", default)]
    background_color: Option<String>,
    #[serde(rename = "fontSize", default)]
    font_size: Option<f64>,
}

#[derive(Deserialize, Default)]
struct RawPlugins {
    title: Option<RawTitle>,
    legend: Option<RawLegend>,
    datalabels: Option<RawDataLabels>,
}

#[derive(Deserialize)]
struct RawDataLabels {
    #[serde(default)]
    display: Option<bool>,
}

#[derive(Deserialize)]
struct RawTitle {
    #[serde(default)]
    display: bool,
    #[serde(default)]
    text: String,
}

#[derive(Deserialize)]
struct RawLegend {
    #[serde(default = "default_true")]
    display: bool,
    position: Option<String>,
}

fn default_true() -> bool {
    true
}

#[derive(Deserialize)]
struct RawData {
    #[serde(default)]
    labels: Vec<String>,
    datasets: Vec<RawDataset>,
}

#[derive(Deserialize)]
struct RawDataset {
    #[serde(default)]
    label: String,
    /// dataset 別の描画種別("bar"/"line")。混合チャートで使う。未指定なら chart 基本型に従う。
    #[serde(rename = "type", default)]
    dataset_type: Option<String>,
    data: DataField,
    #[serde(rename = "backgroundColor")]
    background_color: Option<ScalarOrArray<String>>,
    #[serde(rename = "borderColor")]
    border_color: Option<ScalarOrArray<String>>,
    #[serde(rename = "borderWidth")]
    border_width: Option<f64>,
    #[serde(default)]
    fill: FillSpec,
    #[serde(default)]
    tension: f64,
    // scatter のマーカー半径。Series.point_radius へマップする。
    #[serde(rename = "pointRadius", default)]
    point_radius: Option<f64>,
}

/// `data`: 数値配列(カテゴリ系)、ネスト配列(boxplot)、または点オブジェクト配列(scatter/bubble)。
/// untagged は順に試す: `Nums` → `[1,2]`、`Boxes` → `[[1,2,3,4,5]]`、`Points` → `[{x,y}]`。
#[derive(Deserialize)]
#[serde(untagged)]
enum DataField {
    Nums(Vec<f64>),
    Boxes(Vec<Vec<f64>>),
    Points(Vec<RawPoint>),
}

#[derive(Deserialize, Clone)]
struct RawPoint {
    x: f64,
    y: f64,
    #[serde(default)]
    r: Option<f64>,
}

impl DataField {
    /// 数値配列なら採用、それ以外は空。カテゴリ系チャートの `values` 用。
    fn into_values(self) -> Vec<f64> {
        match self {
            DataField::Nums(v) => v,
            _ => vec![],
        }
    }

    /// 点配列なら IR の `Point` へ、数値配列なら空。scatter/bubble の `points` 用。
    fn into_points(self) -> Vec<Point> {
        match self {
            DataField::Points(ps) => ps
                .into_iter()
                .map(|p| Point {
                    x: p.x,
                    y: p.y,
                    r: p.r,
                })
                .collect(),
            _ => vec![],
        }
    }

    /// ネスト配列なら IR の `BoxPoint` へ変換する。boxplot の `box_points` 用。
    /// 各行は厳密に [min, q1, median, q3, max] の 5 要素でなければならない。
    /// 5 要素以外の行はバイト数ガードをバイパスできるため拒否し NaN 行として扱う。
    fn into_box_points(self) -> Vec<crate::ir::BoxPoint> {
        match self {
            DataField::Boxes(rows) => rows
                .into_iter()
                .map(|row| {
                    if row.len() != 5 {
                        return crate::ir::BoxPoint {
                            min: f64::NAN,
                            q1: f64::NAN,
                            median: f64::NAN,
                            q3: f64::NAN,
                            max: f64::NAN,
                        };
                    }
                    crate::ir::BoxPoint {
                        min: row[0],
                        q1: row[1],
                        median: row[2],
                        q3: row[3],
                        max: row[4],
                    }
                })
                .collect(),
            _ => vec![],
        }
    }
}

/// chart.js の「スカラ or 配列」を許容する untagged ヘルパ。
#[derive(Deserialize)]
#[serde(untagged)]
enum ScalarOrArray<T> {
    One(T),
    Many(Vec<T>),
}

impl<T: Clone> ScalarOrArray<T> {
    fn into_vec(self) -> Vec<T> {
        match self {
            ScalarOrArray::One(v) => vec![v],
            ScalarOrArray::Many(v) => v,
        }
    }
}

/// `fill`: bool / 文字列("origin"等) を受ける。v1 は「塗るか否か」だけ解釈。
#[derive(Deserialize, Default)]
#[serde(untagged)]
enum FillSpec {
    Bool(bool),
    // v1 はモード文字列("origin"等)の中身を解釈せず「塗る」とだけ扱う。
    // 文字列を bool と区別して受理するためにペイロードは必要だが、値は未使用。
    Mode(#[allow(dead_code)] String),
    #[default]
    Absent,
}

impl FillSpec {
    fn is_filled(&self) -> bool {
        match self {
            FillSpec::Bool(b) => *b,
            FillSpec::Mode(_) => true,
            FillSpec::Absent => false,
        }
    }
}

pub fn parse(json: &str, strict: bool) -> Result<ChartSpec, String> {
    // matrix は専用パスで処理する（data 形式が {x,y,v} で他と異なるため）。
    // check_unknown_keys より先に捕捉することで、matrix の "v" キーを未知キーと
    // 誤判定するのを防ぐ。
    {
        let chart_type = serde_json::from_str::<serde_json::Value>(json)
            .ok()
            .and_then(|v| {
                v.get("type")
                    .and_then(|t| t.as_str())
                    .map(|s| s.to_string())
            });
        if chart_type.as_deref() == Some("matrix") {
            if strict {
                check_unknown_keys_matrix(json)?;
            }
            return parse_matrix(json);
        }
    }

    if strict {
        check_unknown_keys(json)?;
    }

    let raw: RawSpec = serde_json::from_str(json).map_err(|e| e.to_string())?;

    // 積み上げ判定: options.scales.x.stacked または options.scales.y.stacked が true。
    // scales は緩く型付けされた serde_json::Value のまま navigate する(深い検証はしない)。
    let stacked = raw
        .options
        .scales
        .as_ref()
        .map(|s| {
            let f = |axis: &str| {
                s.get(axis)
                    .and_then(|a| a.get("stacked"))
                    .and_then(|v| v.as_bool())
                    .unwrap_or(false)
            };
            f("x") || f("y")
        })
        .unwrap_or(false);

    // chart 基本型。bar/line のときだけ dataset 別 type による混合が起こりうる。
    // 基本型は SeriesType のフォールバックにも使う(bar→Bar, line→Line, それ以外→Bar(未使用))。
    let base_series_type = match raw.chart_type.as_str() {
        "line" => SeriesType::Line,
        _ => SeriesType::Bar,
    };

    // dataset の実効描画種別を解決する。
    // type が "bar"/"line" ならそれを優先し、無ければ chart 基本型に従う。
    let resolve_series_type = |dt: &Option<String>| -> SeriesType {
        match dt.as_deref() {
            Some("bar") => SeriesType::Bar,
            Some("line") => SeriesType::Line,
            _ => base_series_type,
        }
    };

    // dataset 別 type は「基本 type が bar/line」かつ値が bar/line のときのみ有効。
    // pie/scatter 等への type 指定や bar/line 以外の値は、黙って無視せず明示エラーにする。
    let is_mixable_base = matches!(raw.chart_type.as_str(), "bar" | "line");
    for ds in &raw.data.datasets {
        if let Some(t) = &ds.dataset_type {
            if !is_mixable_base {
                return Err(format!(
                    "dataset の type は基本 type が bar/line のときのみ指定できます(基本 type={})",
                    raw.chart_type
                ));
            }
            if t != "bar" && t != "line" {
                return Err(format!("未対応の dataset type: {t}"));
            }
        }
    }

    // 各 dataset の実効種別。Mixed 判定と Series.series_type の双方に使う。
    let series_types: Vec<SeriesType> = raw
        .data
        .datasets
        .iter()
        .map(|ds| resolve_series_type(&ds.dataset_type))
        .collect();

    // 描画 kind は、基本型が bar/line のとき「解決後の dataset 種別」で決める:
    // 両方含む→Mixed、全 Line→Line、全 Bar→Bar。dataset 別 type の単独上書き
    // (例 {"type":"bar","datasets":[{"type":"line"}]})も正しく反映される。
    let has_bar = series_types.contains(&SeriesType::Bar);
    let has_line = series_types.contains(&SeriesType::Line);
    let bar_kind = || ChartKind::Bar {
        horizontal: raw.options.index_axis.as_deref() == Some("y"),
        stacked,
    };

    // 混合(bar+line)は縦・非積み上げのみ対応。横棒(indexAxis:y)や積み上げと併用すると
    // それらが黙って失われるため、受理せず明示エラーにする(mixed.rs は縦・非積み上げ前提)。
    if is_mixable_base && has_bar && has_line {
        let horizontal = raw.options.index_axis.as_deref() == Some("y");
        if horizontal || stacked {
            return Err(
                "混合チャート(bar+line)は横棒(indexAxis:y)・積み上げ(stacked)と併用できません"
                    .to_string(),
            );
        }
    }

    let kind = if is_mixable_base && has_bar && has_line {
        ChartKind::Mixed
    } else if is_mixable_base && has_line && !has_bar {
        ChartKind::Line
    } else if is_mixable_base && has_bar && !has_line {
        bar_kind()
    } else {
        // dataset 空(種別未確定)、または mixable でない型。基本 type で決める。
        match raw.chart_type.as_str() {
            "bar" => bar_kind(),
            "line" => ChartKind::Line,
            "pie" => ChartKind::Pie { donut_ratio: 0.0 },
            "doughnut" => ChartKind::Pie { donut_ratio: 0.5 },
            "scatter" => ChartKind::Scatter,
            "bubble" => ChartKind::Bubble,
            "radar" => ChartKind::Radar,
            // QuickChart の正式名は "progressBar"。互換のため "progress" も受理する。
            "progress" | "progressBar" => ChartKind::Progress,
            "boxplot" => ChartKind::BoxPlot,
            other => return Err(format!("未対応の type: {other}")),
        }
    };

    // datalabels: 既存は「キーが存在し display!=false なら有効」。
    // progress のみ既定 ON（QuickChart 準拠）。明示 display:false は尊重する。
    let data_labels = match (&raw.options.plugins.datalabels, &kind) {
        (Some(dl), _) => dl.display != Some(false),
        (None, ChartKind::Progress) => true,
        (None, _) => false,
    };

    // テーマ解決(配色に使うため色解決より先に行う)。
    let theme = build_theme(raw.options.theme);

    let is_pie = matches!(kind, ChartKind::Pie { .. });
    // progress も pie 同様に前景をソリッド(alpha=1.0)で塗る。
    let is_progress = matches!(kind, ChartKind::Progress);
    // scatter/bubble はどちらも点データ(Series.points)を使う線形×線形チャート。
    let is_point_based = matches!(kind, ChartKind::Scatter | ChartKind::Bubble);
    let is_boxplot = matches!(kind, ChartKind::BoxPlot);

    // データ形状とチャート種の整合を検査する。点ベース(scatter/bubble)は {x,y(,r)}
    // 配列、カテゴリ系は数値配列を要する。非空の不一致は空チャート化せず明示エラーに。
    for ds in &raw.data.datasets {
        let mismatched = match &ds.data {
            DataField::Nums(v) => (is_point_based || is_boxplot) && !v.is_empty(),
            DataField::Boxes(v) => !is_boxplot && !v.is_empty(),
            DataField::Points(v) => !is_point_based && !v.is_empty(),
        };
        if mismatched {
            return Err(format!(
                "チャート種 {} とデータ形状が一致しません",
                raw.chart_type
            ));
        }
    }

    let series: Vec<Series> = raw
        .data
        .datasets
        .into_iter()
        .enumerate()
        .map(|(i, ds)| {
            // 点ベースは点データ、boxplot はボックスデータ、それ以外は数値配列を採る。`data` は一度だけ消費する。
            let (values, points, box_points) = if is_point_based {
                (vec![], ds.data.into_points(), vec![])
            } else if is_boxplot {
                (vec![], vec![], ds.data.into_box_points())
            } else {
                (ds.data.into_values(), vec![], vec![])
            };
            let n = if is_point_based {
                points.len()
            } else if is_boxplot {
                box_points.len()
            } else {
                values.len()
            };
            let fill_alpha = if is_pie || is_progress {
                1.0_f32
            } else {
                0.5_f32
            };
            let fill = resolve_colors(
                ds.background_color,
                is_pie,
                i,
                n,
                &theme.palette,
                fill_alpha,
                theme.is_custom_palette,
            );
            let stroke = resolve_colors(
                ds.border_color,
                is_pie,
                i,
                n,
                &theme.palette,
                1.0,
                theme.is_custom_palette,
            );
            // 実効描画種別。線の既定線幅(3.0)を chart 基本型でなく系列種別で決めるため、
            // 単一種別(全 Line→3.0 / 全 Bar→1.0)では従来と byte 一致し、混合では line だけ太くなる。
            let series_type = series_types[i];
            Series {
                name: ds.label,
                values,
                points,
                fill,
                stroke,
                stroke_width: ds.border_width.unwrap_or(default_border_width(series_type)),
                area: ds.fill.is_filled(),
                tension: ds.tension,
                series_type,
                point_radius: ds.point_radius,
                box_points,
            }
        })
        .collect();

    // レーダーは負値に未対応。半径が負になると頂点が反対スポークへ反転し、
    // 実データと異なる多角形になるため、parse 時に明示的に拒否する。
    if matches!(kind, ChartKind::Radar)
        && series
            .iter()
            .any(|s: &Series| s.values.iter().any(|v| v.is_finite() && *v < 0.0))
    {
        return Err("レーダーチャートは負の値に未対応です".to_string());
    }

    // scatter/bubble は線形×線形軸でゼロ起点を強制しない(データ由来のドメインを使う)。
    // 縦棒/ライン: 値軸が Y → y_axis.begin_at_zero=true（デフォルト）。
    // 横棒: 値軸が X → x_axis.begin_at_zero=true（デフォルト）。
    // ユーザーが options.scales.{x,y}.beginAtZero を明示した場合はそちらを優先する。
    let is_horizontal = matches!(
        kind,
        ChartKind::Bar {
            horizontal: true,
            ..
        }
    );
    let value_begin_at_zero = !is_point_based;

    // suggestedMin/suggestedMax および beginAtZero: options.scales.{x,y} から取得する。
    let scales_val = raw.options.scales.as_ref();
    let x_baz_json = scales_val
        .and_then(|s| s.get("x"))
        .and_then(|a| a.get("beginAtZero"))
        .and_then(|v| v.as_bool());
    let y_baz_json = scales_val
        .and_then(|s| s.get("y"))
        .and_then(|a| a.get("beginAtZero"))
        .and_then(|v| v.as_bool());
    let x_begin_at_zero = x_baz_json.unwrap_or(is_horizontal && value_begin_at_zero);
    let y_begin_at_zero =
        y_baz_json.unwrap_or(!is_horizontal && value_begin_at_zero && !is_boxplot);
    let suggested_min_y = scales_val
        .and_then(|s| s.get("y"))
        .and_then(|a| a.get("suggestedMin"))
        .and_then(|v| v.as_f64());
    let suggested_max_y = scales_val
        .and_then(|s| s.get("y"))
        .and_then(|a| a.get("suggestedMax"))
        .and_then(|v| v.as_f64());
    let suggested_min_x = scales_val
        .and_then(|s| s.get("x"))
        .and_then(|a| a.get("suggestedMin"))
        .and_then(|v| v.as_f64());
    let suggested_max_x = scales_val
        .and_then(|s| s.get("x"))
        .and_then(|a| a.get("suggestedMax"))
        .and_then(|v| v.as_f64());

    Ok(ChartSpec {
        kind,
        series,
        categories: raw.data.labels,
        x_axis: AxisSpec {
            title: None,
            min: None,
            max: None,
            suggested_min: suggested_min_x,
            suggested_max: suggested_max_x,
            begin_at_zero: x_begin_at_zero,
            grid: true,
        },
        y_axis: AxisSpec {
            title: None,
            min: None,
            max: None,
            suggested_min: suggested_min_y,
            suggested_max: suggested_max_y,
            begin_at_zero: y_begin_at_zero,
            grid: true,
        },
        legend: legend_pos(&raw.options.plugins.legend),
        title: raw
            .options
            .plugins
            .title
            .filter(|t| t.display)
            .map(|t| t.text),
        width: 800.0,
        height: 450.0,
        data_labels,
        theme,
    })
}

/// `options.theme` を [`Theme`] へ解決する。各トークンは「指定 + 妥当」なら上書き、
/// それ以外はデフォルト値を保つ。色は `parse_color` を通し、不正値はそのトークンの
/// デフォルトにフォールバックする。パレットは妥当な要素を入力順で採り、空または
/// 全要素不正ならデフォルトパレットを使う。
fn build_theme(raw: Option<RawTheme>) -> Theme {
    let mut theme = Theme::default();
    let Some(raw) = raw else {
        return theme;
    };

    if let Some(entries) = raw.palette {
        let parsed: Vec<Color> = entries.iter().filter_map(|c| parse_color(c)).collect();
        if !parsed.is_empty() {
            theme.palette = parsed;
            theme.is_custom_palette = true;
        }
    }
    if let Some(c) = raw.grid_color.as_deref().and_then(parse_color) {
        theme.grid_color = c;
    }
    if let Some(c) = raw.text_color.as_deref().and_then(parse_color) {
        theme.text_color = c;
    }
    if let Some(c) = raw.background_color.as_deref().and_then(parse_color) {
        theme.background = Some(c);
    }
    if let Some(sz) = raw.font_size {
        if sz.is_finite() && sz > 0.0 {
            theme.font_size = sz;
        }
    }
    theme
}

/// 系列の既定線幅。line 系列は太く(3.0)、bar 系列は細い(1.0)。
/// chart 基本型でなく系列種別で決めることで、混合チャートの line 系列も正しく太くなる。
/// 単一種別では従来挙動(全 Line→3.0 / 非 Line→1.0)と byte 一致する。
fn default_border_width(series_type: SeriesType) -> f64 {
    match series_type {
        SeriesType::Line => 3.0,
        SeriesType::Bar => 1.0,
    }
}

/// 指定色(スカラ/配列)を点ごとの Vec<Color> に解決する。
/// 未指定: pie はスライス別パレット(n色)、それ以外は系列インデックスの 1 色。
/// 不正色のフォールバックも pie はスライス位置色・非pieは系列色で、未指定時と一貫させる。
/// 自動配色はテーマの `palette`(空でないことが保証済み)を巡回する。
fn resolve_colors(
    spec: Option<ScalarOrArray<String>>,
    is_pie: bool,
    series_index: usize,
    n: usize,
    palette: &[Color],
    default_alpha: f32,
    is_custom_palette: bool,
) -> Vec<Color> {
    let pick = |i: usize| {
        let c = palette[i % palette.len()];
        Color {
            a: if is_custom_palette && c.a < 1.0 {
                c.a
            } else {
                default_alpha
            },
            ..c
        }
    };
    match spec {
        Some(s) => s
            .into_vec()
            .iter()
            .enumerate()
            .map(|(idx, c)| {
                parse_color(c).unwrap_or_else(|| {
                    if is_pie {
                        pick(idx)
                    } else {
                        pick(series_index)
                    }
                })
            })
            .collect(),
        None if is_pie => (0..n).map(pick).collect(),
        None => vec![pick(series_index)],
    }
}

fn legend_pos(l: &Option<RawLegend>) -> LegendPos {
    match l {
        Some(l) if !l.display => LegendPos::None,
        Some(l) => match l.position.as_deref() {
            Some("bottom") => LegendPos::Bottom,
            Some("left") => LegendPos::Left,
            Some("right") => LegendPos::Right,
            _ => LegendPos::Top,
        },
        None => LegendPos::Top,
    }
}

/// strict モード用: 既知キーのホワイトリストに照らし、未知キーを検出する。
///
/// 防御的に走査し、ノードが欠落/想定外の形なら `Ok(())` を返す（後段の通常パースが
/// 適切な Err を出す）。最初に見つけた未知キーのパスを `Err` で返す。
// strict 用ホワイトリスト。chart.js v4 サブセットとして「認識済み」のキーを並べる。
// IR へ未マップでも、設計で v1 サポート対象に挙げたキーは strict でも受理する
// （strict が弾くのは未知キーであり、認識済み・未完成キーではない）:
//   datalabels=Task16(最小データラベル) / scales=Task9 / pointRadius=Task13。
fn check_unknown_keys(json: &str) -> Result<(), String> {
    let value: serde_json::Value = match serde_json::from_str(json) {
        Ok(v) => v,
        Err(_) => return Ok(()), // 不正 JSON は後段パースに委ねる
    };
    let Some(top) = value.as_object() else {
        return Ok(()); // object でなければ後段パースに委ねる
    };

    check_object(top, &["type", "data", "options"], "")?;

    if let Some(data) = top.get("data").and_then(|v| v.as_object()) {
        check_object(data, &["labels", "datasets"], "data")?;
        if let Some(datasets) = data.get("datasets").and_then(|v| v.as_array()) {
            for (i, ds) in datasets.iter().enumerate() {
                if let Some(ds) = ds.as_object() {
                    check_object(
                        ds,
                        &[
                            "label",
                            "type",
                            "data",
                            "backgroundColor",
                            "borderColor",
                            "borderWidth",
                            "fill",
                            "tension",
                            "pointRadius",
                        ],
                        &format!("data.datasets[{i}]"),
                    )?;
                    // scatter/bubble の点データ {x,y,r} 各オブジェクト内のキーも検査する。
                    // RawPoint は未知キーを無視するため、ここで typo(例 radius)を検出する。
                    if let Some(points) = ds.get("data").and_then(|v| v.as_array()) {
                        for (j, pt) in points.iter().enumerate() {
                            if let Some(pt) = pt.as_object() {
                                check_object(
                                    pt,
                                    &["x", "y", "r"],
                                    &format!("data.datasets[{i}].data[{j}]"),
                                )?;
                            }
                        }
                    }
                }
            }
        }
    }

    if let Some(options) = top.get("options").and_then(|v| v.as_object()) {
        check_object(
            options,
            &["indexAxis", "plugins", "scales", "theme"],
            "options",
        )?;
        if let Some(plugins) = options.get("plugins").and_then(|v| v.as_object()) {
            check_object(
                plugins,
                &["title", "legend", "datalabels"],
                "options.plugins",
            )?;
            if let Some(dl) = plugins.get("datalabels").and_then(|v| v.as_object()) {
                check_object(dl, &["display"], "options.plugins.datalabels")?;
            }
        }
        if let Some(theme) = options.get("theme").and_then(|v| v.as_object()) {
            check_object(
                theme,
                &[
                    "palette",
                    "gridColor",
                    "textColor",
                    "backgroundColor",
                    "fontSize",
                ],
                "options.theme",
            )?;
        }
        // scales 配下も検査する。stacked は描画に効く load-bearing キーなので、
        // typo(例 stakced)を strict で取りこぼさないようにする。各軸は設計が認める
        // サブセットのみ許可(stacked のみ実装、他は認識済み・未実装)。
        if let Some(scales) = options.get("scales").and_then(|v| v.as_object()) {
            check_object(scales, &["x", "y"], "options.scales")?;
            for axis in ["x", "y"] {
                if let Some(ax) = scales.get(axis).and_then(|v| v.as_object()) {
                    check_object(
                        ax,
                        &[
                            "stacked",
                            "min",
                            "max",
                            "title",
                            "grid",
                            "beginAtZero",
                            "suggestedMin",
                            "suggestedMax",
                        ],
                        &format!("options.scales.{axis}"),
                    )?;
                }
            }
        }
    }

    Ok(())
}

fn check_unknown_keys_matrix(json: &str) -> Result<(), String> {
    let value: serde_json::Value = match serde_json::from_str(json) {
        Ok(v) => v,
        Err(_) => return Ok(()),
    };
    let Some(top) = value.as_object() else {
        return Ok(());
    };
    check_object(top, &["type", "data", "options"], "")?;
    if let Some(data) = top.get("data").and_then(|v| v.as_object()) {
        check_object(data, &["datasets"], "data")?;
        if let Some(datasets) = data.get("datasets").and_then(|v| v.as_array()) {
            for (i, ds) in datasets.iter().enumerate() {
                if let Some(ds) = ds.as_object() {
                    check_object(
                        ds,
                        &[
                            "label",
                            "data",
                            "backgroundColor",
                            "borderColor",
                            "borderWidth",
                        ],
                        &format!("data.datasets[{i}]"),
                    )?;
                    if let Some(points) = ds.get("data").and_then(|v| v.as_array()) {
                        for (j, pt) in points.iter().enumerate() {
                            if let Some(pt) = pt.as_object() {
                                check_object(
                                    pt,
                                    &["x", "y", "v"],
                                    &format!("data.datasets[{i}].data[{j}]"),
                                )?;
                            }
                        }
                    }
                }
            }
        }
    }
    if let Some(options) = top.get("options").and_then(|v| v.as_object()) {
        check_object(options, &["plugins", "theme"], "options")?;
        if let Some(plugins) = options.get("plugins").and_then(|v| v.as_object()) {
            check_object(plugins, &["title", "legend"], "options.plugins")?;
        }
        if let Some(theme) = options.get("theme").and_then(|v| v.as_object()) {
            check_object(
                theme,
                &[
                    "palette",
                    "gridColor",
                    "textColor",
                    "backgroundColor",
                    "fontSize",
                ],
                "options.theme",
            )?;
        }
    }
    Ok(())
}

fn parse_matrix(json: &str) -> Result<ChartSpec, String> {
    #[derive(Deserialize)]
    struct MatrixWrapper {
        data: MatrixRawData,
        #[serde(default)]
        options: RawOptions,
    }

    #[derive(Deserialize)]
    struct MatrixRawData {
        datasets: Vec<MatrixRawDataset>,
    }

    #[derive(Deserialize)]
    struct MatrixRawDataset {
        #[allow(dead_code)]
        #[serde(default)]
        label: String,
        data: Vec<MatrixRawCell>,
        #[serde(rename = "backgroundColor", default)]
        background_color: Option<ScalarOrArray<String>>,
        #[serde(rename = "borderColor", default)]
        border_color: Option<ScalarOrArray<String>>,
        #[serde(rename = "borderWidth", default)]
        border_width: Option<f64>,
    }

    #[derive(Deserialize)]
    struct MatrixRawCell {
        x: String,
        y: String,
        v: f64,
    }

    let raw: MatrixWrapper = serde_json::from_str(json).map_err(|e| e.to_string())?;

    if raw.data.datasets.len() > 1 {
        return Err("matrix チャートは dataset が 1 つのみサポートされます".to_string());
    }
    if raw.data.datasets.is_empty() {
        return Err("matrix チャートには dataset が 1 つ必要です".to_string());
    }

    let ds = raw.data.datasets.into_iter().next().unwrap();

    // x/y カテゴリを出現順に収集（重複除去）— HashMap で O(n) ルックアップ
    let mut x_cats: Vec<String> = Vec::new();
    let mut x_idx: HashMap<String, usize> = HashMap::new();
    let mut y_cats: Vec<String> = Vec::new();
    let mut y_idx: HashMap<String, usize> = HashMap::new();
    for cell in &ds.data {
        if !x_idx.contains_key(&cell.x) {
            x_idx.insert(cell.x.clone(), x_cats.len());
            x_cats.push(cell.x.clone());
        }
        if !y_idx.contains_key(&cell.y) {
            y_idx.insert(cell.y.clone(), y_cats.len());
            y_cats.push(cell.y.clone());
        }
    }

    let n_cols = x_cats.len();
    let n_rows = y_cats.len();

    // グリッドサイズ上限チェック
    if n_cols.saturating_mul(n_rows) > 10_000 {
        return Err(format!(
            "matrix grid too large: {}×{} = {} cells (limit 10000)",
            n_cols,
            n_rows,
            n_cols * n_rows
        ));
    }

    // NaN で初期化したグリッドを構築
    let mut grid: Vec<Vec<f64>> = vec![vec![f64::NAN; n_cols]; n_rows];
    for cell in &ds.data {
        let ci = x_idx[&cell.x];
        let ri = y_idx[&cell.y];
        grid[ri][ci] = cell.v;
    }

    let theme = build_theme(raw.options.theme);

    let color_hi = ds
        .background_color
        .as_ref()
        .and_then(|c| match c {
            ScalarOrArray::One(v) => parse_color(v),
            ScalarOrArray::Many(vs) => vs.first().and_then(|v| parse_color(v)),
        })
        .unwrap_or(theme.palette[0]);
    let color_lo = Color {
        r: 255,
        g: 255,
        b: 255,
        a: 1.0,
    };

    let stroke_color: Vec<Color> = ds
        .border_color
        .as_ref()
        .and_then(|c| match c {
            ScalarOrArray::One(v) => parse_color(v),
            ScalarOrArray::Many(vs) => vs.first().and_then(|v| parse_color(v)),
        })
        .map(|c| vec![c])
        .unwrap_or_default();

    let series: Vec<Series> = y_cats
        .iter()
        .enumerate()
        .map(|(i, name)| Series {
            name: name.clone(),
            values: grid[i].clone(),
            points: vec![],
            fill: vec![color_hi],
            stroke: stroke_color.clone(),
            stroke_width: ds.border_width.unwrap_or(0.0),
            area: false,
            tension: 0.0,
            series_type: SeriesType::Bar,
            point_radius: None,
            box_points: vec![],
        })
        .collect();

    Ok(ChartSpec {
        kind: ChartKind::Matrix { color_lo, color_hi },
        series,
        categories: x_cats,
        x_axis: AxisSpec {
            title: None,
            min: None,
            max: None,
            suggested_min: None,
            suggested_max: None,
            begin_at_zero: false,
            grid: false,
        },
        y_axis: AxisSpec {
            title: None,
            min: None,
            max: None,
            suggested_min: None,
            suggested_max: None,
            begin_at_zero: false,
            grid: false,
        },
        legend: legend_pos(&raw.options.plugins.legend),
        title: raw
            .options
            .plugins
            .title
            .filter(|t| t.display)
            .map(|t| t.text),
        width: 800.0,
        height: 450.0,
        data_labels: false,
        theme,
    })
}

/// `obj` のキーを `allowed` に照らし、最初の未知キーを `Err(パス)` で返す。
fn check_object(
    obj: &serde_json::Map<String, serde_json::Value>,
    allowed: &[&str],
    path: &str,
) -> Result<(), String> {
    for key in obj.keys() {
        if !allowed.contains(&key.as_str()) {
            let full = if path.is_empty() {
                key.clone()
            } else {
                format!("{path}.{key}")
            };
            return Err(format!("未知のキー: {full}"));
        }
    }
    Ok(())
}

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

    #[test]
    fn parse_boxplot_basic() {
        let json = r#"{
            "type": "boxplot",
            "data": {
                "labels": ["Mon", "Tue"],
                "datasets": [{
                    "label": "Values",
                    "data": [
                        [10, 25, 50, 75, 90],
                        [5, 20, 45, 70, 95]
                    ]
                }]
            }
        }"#;
        let spec = parse(json, false).expect("parse error");
        assert!(matches!(spec.kind, crate::ir::ChartKind::BoxPlot));
        assert_eq!(spec.series.len(), 1);
        let bp = &spec.series[0].box_points;
        assert_eq!(bp.len(), 2);
        assert_eq!(bp[0].min, 10.0);
        assert_eq!(bp[0].q1, 25.0);
        assert_eq!(bp[0].median, 50.0);
        assert_eq!(bp[0].q3, 75.0);
        assert_eq!(bp[0].max, 90.0);
        assert_eq!(bp[1].min, 5.0);
        assert_eq!(bp[1].median, 45.0);
    }

    #[test]
    fn parse_boxplot_rejects_flat_nums() {
        let json = r#"{
            "type": "boxplot",
            "data": {
                "labels": ["A"],
                "datasets": [{"data": [10, 25, 50, 75, 90]}]
            }
        }"#;
        assert!(
            parse(json, false).is_err(),
            "boxplot with flat numbers should fail"
        );
    }

    #[test]
    fn auto_fill_gets_half_alpha() {
        // backgroundColor 未指定のバーチャートで fill が alpha=0.5 になること。
        let json = r#"{
            "type": "bar",
            "data": {
                "labels": ["A", "B"],
                "datasets": [{"label": "S1", "data": [1, 2]}]
            }
        }"#;
        let spec = parse(json, false).expect("parse error");
        let fill_alpha = spec.series[0].fill[0].a;
        assert!(
            (fill_alpha - 0.5).abs() < 1e-6,
            "fill alpha は 0.5 であるべき、実際は {}",
            fill_alpha
        );
    }

    #[test]
    fn auto_stroke_gets_full_alpha() {
        // borderColor 未指定のバーチャートで stroke が alpha=1.0 になること。
        let json = r#"{
            "type": "bar",
            "data": {
                "labels": ["A", "B"],
                "datasets": [{"label": "S1", "data": [1, 2]}]
            }
        }"#;
        let spec = parse(json, false).expect("parse error");
        let stroke_alpha = spec.series[0].stroke[0].a;
        assert!(
            (stroke_alpha - 1.0).abs() < 1e-6,
            "stroke alpha は 1.0 であるべき、実際は {}",
            stroke_alpha
        );
    }

    #[test]
    fn pie_auto_fill_gets_full_alpha() {
        // pie チャートは chart.js v4 の colorizeDoughnutDataset が BORDER_COLORS を使うため alpha=1.0。
        let json =
            r#"{"type": "pie", "data": {"labels": ["A", "B"], "datasets": [{"data": [1, 2]}]}}"#;
        let spec = parse(json, false).expect("parse error");
        let fill_alpha = spec.series[0].fill[0].a;
        assert!(
            (fill_alpha - 1.0).abs() < 1e-6,
            "pie の fill alpha は 1.0 であるべき、実際は {}",
            fill_alpha
        );
    }
}