splashboard 2.10.2

use std::path::PathBuf;

use async_trait::async_trait;

use crate::payload::{
    Bar, BarsData, Body, CalendarData, EntriesData, HeatmapData, ImageData, ImageLinkedListData,
    NumberSeriesData, Payload, PixelArtData, PointSeriesData, RatioData, TextBlockData, TextData,
};
use crate::render::Shape;

use super::{FetchContext, FetchError, Fetcher, Safety};

/// Every payload shape ReadStore knows how to deserialize. Intentionally broad — ReadStore is
/// the escape hatch for user-defined widgets, so it supports the non-dynamic shapes exhaustively;
/// config picks which variant a specific file maps to.
const READ_STORE_SHAPES: &[Shape] = &[
    Shape::Text,
    Shape::TextBlock,
    Shape::Entries,
    Shape::Ratio,
    Shape::NumberSeries,
    Shape::PointSeries,
    Shape::Bars,
    Shape::Image,
    Shape::PixelArt,
    Shape::ImageLinkedList,
    Shape::Calendar,
    Shape::Heatmap,
];

/// Renders a local file the user wrote. The file lives at a fixed, sanitized path
/// `$HOME/.splashboard/store/<widget_id>.<ext>` — config cannot redirect the read, so a hostile
/// repo-local config can't traverse the filesystem. The value on disk is user-controlled;
/// splashboard just deserializes it into a `Payload` of the shape the config declares.
///
/// Fills the gap that #5 (plugin protocol) and #20 (command widget) were meant to cover:
/// arbitrary user-defined display surface, without any exec path. Users populate the file via
/// whatever they like (editor, cron, post-commit hook, CI step).
pub struct ReadStoreFetcher;

#[async_trait]
impl Fetcher for ReadStoreFetcher {
    fn name(&self) -> &str {
        "basic_read_store"
    }
    fn safety(&self) -> Safety {
        // Always Safe: the path is derived from the widget id under a fixed home subdir, so
        // there's no path-traversal vector even in an untrusted local config.
        Safety::Safe
    }
    fn description(&self) -> &'static str {
        "Reads a payload file the user maintains at `$HOME/.splashboard/store/<widget_id>.<ext>` (text / json / toml) and renders it as the declared shape. The escape hatch for ad-hoc widgets — populate the file from any cron, editor, or post-commit hook and splashboard surfaces whatever you wrote."
    }
    fn refresh_interval(&self) -> u64 {
        60
    }
    fn shapes(&self) -> &[Shape] {
        READ_STORE_SHAPES
    }
    fn cache_key(&self, ctx: &FetchContext) -> String {
        // Differentiate by widget id + declared shape so two widgets pointed at different
        // files (different ids) don't collide, and a single widget changing its shape gets
        // a fresh entry rather than reading an old-shape cached payload.
        let shape = ctx.shape.map(|s| s.as_str()).unwrap_or("");
        let file_format = ctx.file_format.as_deref().unwrap_or("");
        format!(
            "read_store-{}-{}-{}",
            sanitize(&ctx.widget_id),
            shape,
            file_format
        )
    }
    async fn fetch(&self, ctx: &FetchContext) -> Result<Payload, FetchError> {
        // Runtime always derives a shape (renderer's single accepted shape, or our
        // `default_shape`); `None` falls back defensively to TextBlock so direct callers still
        // behave.
        let shape = ctx.shape.unwrap_or(Shape::TextBlock);
        let file_format = ctx
            .file_format
            .as_deref()
            .unwrap_or(default_format_for_shape(shape));
        let path = resolve_path(&ctx.widget_id, file_format)
            .ok_or_else(|| FetchError::Failed("could not resolve splashboard home".into()))?;
        let body = load_body(&path, file_format, shape)?;
        Ok(Payload {
            icon: None,
            status: None,
            format: None,
            body,
        })
    }
}

/// `text` is the natural default for text shapes; everything else needs structure. Saves users
/// from writing `file_format = "text"` for every simple notes widget.
fn default_format_for_shape(shape: Shape) -> &'static str {
    match shape {
        Shape::Text | Shape::TextBlock => "text",
        _ => "json",
    }
}

fn resolve_path(widget_id: &str, file_format: &str) -> Option<PathBuf> {
    let ext = extension_for(file_format);
    let name = format!("{}.{ext}", sanitize(widget_id));
    crate::paths::read_store_dir().map(|d| d.join(name))
}

fn extension_for(file_format: &str) -> &'static str {
    match file_format {
        "toml" => "toml",
        "text" => "txt",
        _ => "json",
    }
}

/// Confine filenames to safe characters so widget ids never escape the store subdir or break
/// on case-insensitive filesystems. Same rule as the cache module.
fn sanitize(s: &str) -> String {
    s.chars()
        .map(|c| {
            if c.is_ascii_alphanumeric() || c == '-' || c == '_' {
                c
            } else {
                '_'
            }
        })
        .collect()
}

fn load_body(path: &std::path::Path, file_format: &str, shape: Shape) -> Result<Body, FetchError> {
    let bytes = match std::fs::read(path) {
        Ok(b) => b,
        Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(empty_body(shape)),
        Err(e) => return Err(FetchError::Failed(format!("read failed: {e}"))),
    };
    let text =
        std::str::from_utf8(&bytes).map_err(|e| FetchError::Failed(format!("utf-8: {e}")))?;
    parse_body(text, file_format, shape)
}

fn parse_body(text: &str, file_format: &str, shape: Shape) -> Result<Body, FetchError> {
    match (file_format, shape) {
        ("text", Shape::Text) => Ok(Body::Text(TextData {
            value: text.lines().collect::<Vec<_>>().join(" "),
        })),
        ("text", Shape::TextBlock) => Ok(Body::TextBlock(TextBlockData {
            lines: text.lines().map(str::to_string).collect(),
        })),
        ("text", _) => Err(FetchError::Failed(format!(
            "shape {:?} requires json or toml, not text",
            shape.as_str()
        ))),
        ("json", shape) => from_json(text, shape),
        ("toml", shape) => from_toml(text, shape),
        (other, _) => Err(FetchError::Failed(format!(
            "unknown file_format: {other:?}"
        ))),
    }
}

/// Deserialize the file as the inner `*Data` struct for the requested shape, and wrap in the
/// right `Body` variant. Users write the data payload directly (e.g. `{ "cells": [[...]] }`
/// for a heatmap) — no wrapping `{"shape":...}` envelope required.
fn from_json(text: &str, shape: Shape) -> Result<Body, FetchError> {
    macro_rules! parse_as {
        ($ty:ty, $variant:ident) => {
            serde_json::from_str::<$ty>(text)
                .map(Body::$variant)
                .map_err(|e| FetchError::Failed(format!("json parse: {e}")))
        };
    }
    match shape {
        Shape::Text => parse_as!(TextData, Text),
        Shape::TextBlock => parse_as!(TextBlockData, TextBlock),
        Shape::Entries => parse_as!(EntriesData, Entries),
        Shape::Ratio => parse_as!(RatioData, Ratio),
        Shape::NumberSeries => parse_as!(NumberSeriesData, NumberSeries),
        Shape::PointSeries => parse_as!(PointSeriesData, PointSeries),
        Shape::Bars => parse_as!(BarsData, Bars),
        Shape::Image => parse_as!(ImageData, Image),
        Shape::PixelArt => parse_as!(PixelArtData, PixelArt),
        Shape::ImageLinkedList => parse_as!(ImageLinkedListData, ImageLinkedList),
        Shape::Calendar => parse_as!(CalendarData, Calendar),
        Shape::Heatmap => parse_as!(HeatmapData, Heatmap),
        other => Err(FetchError::Failed(format!(
            "read_store doesn't support shape {:?}",
            other.as_str()
        ))),
    }
}

fn from_toml(text: &str, shape: Shape) -> Result<Body, FetchError> {
    macro_rules! parse_as {
        ($ty:ty, $variant:ident) => {
            toml::from_str::<$ty>(text)
                .map(Body::$variant)
                .map_err(|e| FetchError::Failed(format!("toml parse: {e}")))
        };
    }
    match shape {
        Shape::Text => parse_as!(TextData, Text),
        Shape::TextBlock => parse_as!(TextBlockData, TextBlock),
        Shape::Entries => parse_as!(EntriesData, Entries),
        Shape::Ratio => parse_as!(RatioData, Ratio),
        Shape::NumberSeries => parse_as!(NumberSeriesData, NumberSeries),
        Shape::PointSeries => parse_as!(PointSeriesData, PointSeries),
        Shape::Bars => parse_as!(BarsData, Bars),
        Shape::Image => parse_as!(ImageData, Image),
        Shape::PixelArt => parse_as!(PixelArtData, PixelArt),
        Shape::ImageLinkedList => parse_as!(ImageLinkedListData, ImageLinkedList),
        Shape::Calendar => parse_as!(CalendarData, Calendar),
        Shape::Heatmap => parse_as!(HeatmapData, Heatmap),
        other => Err(FetchError::Failed(format!(
            "read_store doesn't support shape {:?}",
            other.as_str()
        ))),
    }
}

/// Empty-but-valid body for the declared shape. Used when the file is missing so the splash
/// stays quiet rather than breaking — matches the "optional" flavor of ReadStore widgets.
fn empty_body(shape: Shape) -> Body {
    match shape {
        Shape::Text => Body::Text(TextData {
            value: String::new(),
        }),
        Shape::Entries => Body::Entries(EntriesData { items: Vec::new() }),
        Shape::Ratio => Body::Ratio(RatioData {
            value: 0.0,
            label: None,
            denominator: None,
        }),
        Shape::NumberSeries => Body::NumberSeries(NumberSeriesData { values: Vec::new() }),
        Shape::PointSeries => Body::PointSeries(PointSeriesData { series: Vec::new() }),
        Shape::Bars => Body::Bars(BarsData {
            bars: Vec::<Bar>::new(),
        }),
        Shape::Image => Body::Image(ImageData {
            path: String::new(),
        }),
        Shape::PixelArt => Body::PixelArt(PixelArtData {
            pixels: Vec::new(),
            label: None,
        }),
        Shape::ImageLinkedList => Body::ImageLinkedList(ImageLinkedListData { items: Vec::new() }),
        Shape::Calendar => Body::Calendar(CalendarData {
            year: 1970,
            month: 1,
            day: None,
            events: Vec::new(),
        }),
        Shape::Heatmap => Body::Heatmap(HeatmapData {
            cells: Vec::new(),
            thresholds: None,
            row_labels: None,
            col_labels: None,
        }),
        _ => Body::TextBlock(TextBlockData { lines: Vec::new() }),
    }
}

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

    fn ctx(id: &str, shape: Shape, file_format: &str) -> FetchContext {
        FetchContext {
            widget_id: id.into(),
            timeout: Duration::from_secs(1),
            shape: Some(shape),
            file_format: Some(file_format.into()),
            ..Default::default()
        }
    }

    /// Scope-guard sharing the crate-wide `paths::TEST_ENV_LOCK`, so any parallel test that
    /// also touches `SPLASHBOARD_HOME` serializes with us.
    struct ScopedHome {
        previous: Option<String>,
        _lock: std::sync::MutexGuard<'static, ()>,
    }
    impl ScopedHome {
        fn new(dir: &std::path::Path) -> Self {
            let lock = crate::paths::TEST_ENV_LOCK
                .lock()
                .unwrap_or_else(|e| e.into_inner());
            let previous = std::env::var("SPLASHBOARD_HOME").ok();
            unsafe {
                std::env::set_var("SPLASHBOARD_HOME", dir);
            }
            Self {
                previous,
                _lock: lock,
            }
        }
    }
    impl Drop for ScopedHome {
        fn drop(&mut self) {
            unsafe {
                match self.previous.take() {
                    Some(v) => std::env::set_var("SPLASHBOARD_HOME", v),
                    None => std::env::remove_var("SPLASHBOARD_HOME"),
                }
            }
        }
    }

    #[tokio::test]
    async fn reads_heatmap_json() {
        let tmp = tempfile::tempdir().unwrap();
        let store = tmp.path().join("store");
        std::fs::create_dir_all(&store).unwrap();
        std::fs::write(store.join("habit.json"), r#"{"cells":[[0,1,2,3,4]]}"#).unwrap();
        let _guard = ScopedHome::new(tmp.path());
        let p = ReadStoreFetcher
            .fetch(&ctx("habit", Shape::Heatmap, "json"))
            .await
            .unwrap();
        match p.body {
            Body::Heatmap(d) => assert_eq!(d.cells, vec![vec![0, 1, 2, 3, 4]]),
            other => panic!("expected heatmap body, got {other:?}"),
        }
    }

    #[tokio::test]
    async fn text_block_splits_on_newline() {
        let tmp = tempfile::tempdir().unwrap();
        let store = tmp.path().join("store");
        std::fs::create_dir_all(&store).unwrap();
        std::fs::write(store.join("notes.txt"), "one\ntwo\nthree").unwrap();
        let _guard = ScopedHome::new(tmp.path());
        let p = ReadStoreFetcher
            .fetch(&ctx("notes", Shape::TextBlock, "text"))
            .await
            .unwrap();
        match p.body {
            Body::TextBlock(d) => assert_eq!(d.lines, vec!["one", "two", "three"]),
            other => panic!("expected text_block body, got {other:?}"),
        }
    }

    #[tokio::test]
    async fn reads_image_linked_list_json() {
        let tmp = tempfile::tempdir().unwrap();
        let store = tmp.path().join("store");
        std::fs::create_dir_all(&store).unwrap();
        std::fs::write(
            store.join("cards.json"),
            r#"{"items":[
                {"title":"first","url":"https://example.com/1","thumbnail_path":"/tmp/a.png","subtitle":"src · 2h"},
                {"title":"second"}
            ]}"#,
        )
        .unwrap();
        let _guard = ScopedHome::new(tmp.path());
        let p = ReadStoreFetcher
            .fetch(&ctx("cards", Shape::ImageLinkedList, "json"))
            .await
            .unwrap();
        match p.body {
            Body::ImageLinkedList(d) => {
                assert_eq!(d.items.len(), 2);
                assert_eq!(d.items[0].title, "first");
                assert_eq!(d.items[0].url.as_deref(), Some("https://example.com/1"));
                assert_eq!(d.items[0].thumbnail_path.as_deref(), Some("/tmp/a.png"));
                assert_eq!(d.items[1].url, None);
                assert_eq!(d.items[1].thumbnail_path, None);
            }
            other => panic!("expected image_linked_list body, got {other:?}"),
        }
    }

    #[tokio::test]
    async fn missing_image_linked_list_file_renders_empty_for_declared_shape() {
        let tmp = tempfile::tempdir().unwrap();
        let _guard = ScopedHome::new(tmp.path());
        let p = ReadStoreFetcher
            .fetch(&ctx("absent_cards", Shape::ImageLinkedList, "json"))
            .await
            .unwrap();
        match p.body {
            Body::ImageLinkedList(d) => assert!(d.items.is_empty()),
            other => panic!("expected empty image_linked_list body, got {other:?}"),
        }
    }

    #[tokio::test]
    async fn missing_file_renders_empty_for_declared_shape() {
        let tmp = tempfile::tempdir().unwrap();
        let _guard = ScopedHome::new(tmp.path());
        let p = ReadStoreFetcher
            .fetch(&ctx("absent", Shape::Heatmap, "json"))
            .await
            .unwrap();
        match p.body {
            Body::Heatmap(d) => assert!(d.cells.is_empty()),
            other => panic!("expected empty heatmap, got {other:?}"),
        }
    }

    #[tokio::test]
    async fn missing_shape_falls_back_to_text_block() {
        // Runtime always supplies a shape now, but defensively treat `None` as TextBlock so a
        // stray caller doesn't get an error and the cache_key stays stable.
        let tmp = tempfile::tempdir().unwrap();
        let store = tmp.path().join("store");
        std::fs::create_dir_all(&store).unwrap();
        std::fs::write(store.join("x.txt"), "hello").unwrap();
        let _guard = ScopedHome::new(tmp.path());
        let c = FetchContext {
            widget_id: "x".into(),
            timeout: Duration::from_secs(1),
            file_format: Some("text".into()),
            ..Default::default()
        };
        let p = ReadStoreFetcher.fetch(&c).await.unwrap();
        match p.body {
            Body::TextBlock(d) => assert_eq!(d.lines, vec!["hello"]),
            other => panic!("expected text_block body, got {other:?}"),
        }
    }

    #[test]
    fn sanitize_drops_path_traversal() {
        assert_eq!(sanitize("../etc/passwd"), "___etc_passwd");
        assert_eq!(sanitize("../../secret"), "______secret");
        assert_eq!(sanitize("a b/c"), "a_b_c");
        assert_eq!(sanitize("habit-01_alt"), "habit-01_alt");
    }

    #[test]
    fn cache_key_differs_across_widget_ids() {
        let a = ReadStoreFetcher.cache_key(&ctx("habit", Shape::Heatmap, "json"));
        let b = ReadStoreFetcher.cache_key(&ctx("sleep", Shape::Heatmap, "json"));
        assert_ne!(a, b);
    }

    #[test]
    fn cache_key_differs_across_shapes() {
        let a = ReadStoreFetcher.cache_key(&ctx("habit", Shape::Heatmap, "json"));
        let b = ReadStoreFetcher.cache_key(&ctx("habit", Shape::TextBlock, "json"));
        assert_ne!(a, b);
    }

    fn structured_cases() -> Vec<(Shape, &'static str, &'static str, Body)> {
        vec![
            (
                Shape::Text,
                r#"{"value":"hello"}"#,
                r#"value = "hello""#,
                Body::Text(TextData {
                    value: "hello".into(),
                }),
            ),
            (
                Shape::TextBlock,
                r#"{"lines":["one","two"]}"#,
                r#"lines = ["one", "two"]"#,
                Body::TextBlock(TextBlockData {
                    lines: vec!["one".into(), "two".into()],
                }),
            ),
            (
                Shape::Entries,
                r#"{"items":[{"key":"branch","value":"main"}]}"#,
                r#"[[items]]
key = "branch"
value = "main""#,
                Body::Entries(EntriesData {
                    items: vec![crate::payload::Entry {
                        key: "branch".into(),
                        value: Some("main".into()),
                        status: None,
                    }],
                }),
            ),
            (
                Shape::Ratio,
                r#"{"value":0.25,"label":"done","denominator":4}"#,
                "value = 0.25\nlabel = \"done\"\ndenominator = 4",
                Body::Ratio(RatioData {
                    value: 0.25,
                    label: Some("done".into()),
                    denominator: Some(4),
                }),
            ),
            (
                Shape::NumberSeries,
                r#"{"values":[1,2,3]}"#,
                "values = [1, 2, 3]",
                Body::NumberSeries(NumberSeriesData {
                    values: vec![1, 2, 3],
                }),
            ),
            (
                Shape::PointSeries,
                r#"{"series":[{"name":"cpu","points":[[0.0,1.0],[1.0,2.0]]}]}"#,
                "[[series]]\nname = \"cpu\"\npoints = [[0.0, 1.0], [1.0, 2.0]]",
                Body::PointSeries(PointSeriesData {
                    series: vec![crate::payload::PointSeries {
                        name: "cpu".into(),
                        points: vec![(0.0, 1.0), (1.0, 2.0)],
                    }],
                }),
            ),
            (
                Shape::Bars,
                r#"{"bars":[{"label":"todo","value":3}]}"#,
                "[[bars]]\nlabel = \"todo\"\nvalue = 3",
                Body::Bars(BarsData {
                    bars: vec![Bar {
                        label: "todo".into(),
                        value: 3,
                        value_label: None,
                    }],
                }),
            ),
            (
                Shape::Image,
                r#"{"path":"/tmp/pic.png"}"#,
                r#"path = "/tmp/pic.png""#,
                Body::Image(ImageData {
                    path: "/tmp/pic.png".into(),
                }),
            ),
            (
                Shape::PixelArt,
                r#"{"pixels":[[{"r":255,"g":0,"b":0,"a":255},{"r":0,"g":0,"b":0,"a":0}]],"label":"hi"}"#,
                "label = \"hi\"\npixels = [[{r=255,g=0,b=0,a=255},{r=0,g=0,b=0,a=0}]]",
                Body::PixelArt(PixelArtData {
                    pixels: vec![vec![
                        crate::payload::PixelColor::opaque(255, 0, 0),
                        crate::payload::PixelColor::TRANSPARENT,
                    ]],
                    label: Some("hi".into()),
                }),
            ),
            (
                Shape::Calendar,
                r#"{"year":2026,"month":4,"day":30,"events":[1,15]}"#,
                "year = 2026\nmonth = 4\nday = 30\nevents = [1, 15]",
                Body::Calendar(CalendarData {
                    year: 2026,
                    month: 4,
                    day: Some(30),
                    events: vec![1, 15],
                }),
            ),
            (
                Shape::Heatmap,
                r#"{"cells":[[0,1],[2,3]],"thresholds":[1,2,3,4],"row_labels":["Mon","Tue"],"col_labels":["AM","PM"]}"#,
                "cells = [[0, 1], [2, 3]]\nthresholds = [1, 2, 3, 4]\nrow_labels = [\"Mon\", \"Tue\"]\ncol_labels = [\"AM\", \"PM\"]",
                Body::Heatmap(HeatmapData {
                    cells: vec![vec![0, 1], vec![2, 3]],
                    thresholds: Some(vec![1, 2, 3, 4]),
                    row_labels: Some(vec!["Mon".into(), "Tue".into()]),
                    col_labels: Some(vec!["AM".into(), "PM".into()]),
                }),
            ),
        ]
    }

    #[test]
    fn fetcher_contract_and_resolve_path_are_stable() {
        assert_eq!(ReadStoreFetcher.name(), "basic_read_store");
        assert_eq!(ReadStoreFetcher.safety(), Safety::Safe);
        assert_eq!(ReadStoreFetcher.shapes(), READ_STORE_SHAPES);
        assert!(
            ReadStoreFetcher
                .description()
                .contains("$HOME/.splashboard/store/")
        );
        assert_eq!(default_format_for_shape(Shape::Text), "text");
        assert_eq!(default_format_for_shape(Shape::TextBlock), "text");
        assert_eq!(default_format_for_shape(Shape::Heatmap), "json");

        let tmp = tempfile::tempdir().unwrap();
        let _guard = ScopedHome::new(tmp.path());
        assert_eq!(
            resolve_path("notes", "text"),
            Some(tmp.path().join("store").join("notes.txt"))
        );
        assert_eq!(
            resolve_path("chart?", "toml"),
            Some(tmp.path().join("store").join("chart_.toml"))
        );
    }

    #[tokio::test]
    async fn fetch_defaults_file_format_for_text_and_structured_shapes() {
        let tmp = tempfile::tempdir().unwrap();
        let store = tmp.path().join("store");
        std::fs::create_dir_all(&store).unwrap();
        std::fs::write(store.join("headline.txt"), "hello\nworld").unwrap();
        std::fs::write(store.join("habit.json"), r#"{"cells":[[1,2],[3,4]]}"#).unwrap();
        let _guard = ScopedHome::new(tmp.path());
        let text_ctx = FetchContext {
            widget_id: "headline".into(),
            timeout: Duration::from_secs(1),
            shape: Some(Shape::Text),
            ..Default::default()
        };
        let heatmap_ctx = FetchContext {
            widget_id: "habit".into(),
            timeout: Duration::from_secs(1),
            shape: Some(Shape::Heatmap),
            ..Default::default()
        };

        assert_eq!(
            ReadStoreFetcher.fetch(&text_ctx).await.unwrap().body,
            Body::Text(TextData {
                value: "hello world".into(),
            })
        );
        assert_eq!(
            ReadStoreFetcher.fetch(&heatmap_ctx).await.unwrap().body,
            Body::Heatmap(HeatmapData {
                cells: vec![vec![1, 2], vec![3, 4]],
                thresholds: None,
                row_labels: None,
                col_labels: None,
            })
        );
    }

    #[test]
    fn structured_json_and_toml_cover_all_supported_shapes() {
        structured_cases()
            .into_iter()
            .for_each(|(shape, json, toml, expected)| {
                assert_eq!(from_json(json, shape).unwrap(), expected.clone());
                assert_eq!(from_toml(toml, shape).unwrap(), expected);
            });
    }

    #[test]
    fn parse_and_load_errors_cover_fallback_branches() {
        assert_eq!(
            parse_body("one\ntwo", "text", Shape::Text).unwrap(),
            Body::Text(TextData {
                value: "one two".into(),
            })
        );
        assert_eq!(
            parse_body("one\ntwo", "text", Shape::TextBlock).unwrap(),
            Body::TextBlock(TextBlockData {
                lines: vec!["one".into(), "two".into()],
            })
        );
        assert!(matches!(
            parse_body("{}", "text", Shape::Bars),
            Err(FetchError::Failed(message)) if message.contains("requires json or toml")
        ));
        assert!(matches!(
            parse_body("{}", "yaml", Shape::TextBlock),
            Err(FetchError::Failed(message)) if message.contains("unknown file_format")
        ));
        assert!(matches!(
            from_json("{", Shape::Text),
            Err(FetchError::Failed(message)) if message.contains("json parse")
        ));
        assert!(matches!(
            from_json("{}", Shape::Badge),
            Err(FetchError::Failed(message)) if message.contains("doesn't support shape")
        ));
        assert!(matches!(
            from_toml("value = [", Shape::Text),
            Err(FetchError::Failed(message)) if message.contains("toml parse")
        ));
        assert!(matches!(
            from_toml("", Shape::Timeline),
            Err(FetchError::Failed(message)) if message.contains("doesn't support shape")
        ));

        let tmp = tempfile::tempdir().unwrap();
        let dir_error = load_body(tmp.path(), "json", Shape::Text).unwrap_err();
        assert!(
            matches!(dir_error, FetchError::Failed(message) if message.contains("read failed"))
        );

        let invalid = tmp.path().join("bad.json");
        std::fs::write(&invalid, [0xff, 0xfe]).unwrap();
        let utf8_error = load_body(&invalid, "json", Shape::Text).unwrap_err();
        assert!(matches!(utf8_error, FetchError::Failed(message) if message.contains("utf-8")));
    }

    #[test]
    fn empty_body_matches_declared_shape_defaults() {
        [
            (
                Shape::Text,
                Body::Text(TextData {
                    value: String::new(),
                }),
            ),
            (
                Shape::TextBlock,
                Body::TextBlock(TextBlockData { lines: Vec::new() }),
            ),
            (
                Shape::Entries,
                Body::Entries(EntriesData { items: Vec::new() }),
            ),
            (
                Shape::Ratio,
                Body::Ratio(RatioData {
                    value: 0.0,
                    label: None,
                    denominator: None,
                }),
            ),
            (
                Shape::NumberSeries,
                Body::NumberSeries(NumberSeriesData { values: Vec::new() }),
            ),
            (
                Shape::PointSeries,
                Body::PointSeries(PointSeriesData { series: Vec::new() }),
            ),
            (Shape::Bars, Body::Bars(BarsData { bars: Vec::new() })),
            (
                Shape::Image,
                Body::Image(ImageData {
                    path: String::new(),
                }),
            ),
            (
                Shape::PixelArt,
                Body::PixelArt(PixelArtData {
                    pixels: Vec::new(),
                    label: None,
                }),
            ),
            (
                Shape::Calendar,
                Body::Calendar(CalendarData {
                    year: 1970,
                    month: 1,
                    day: None,
                    events: Vec::new(),
                }),
            ),
            (
                Shape::Heatmap,
                Body::Heatmap(HeatmapData {
                    cells: Vec::new(),
                    thresholds: None,
                    row_labels: None,
                    col_labels: None,
                }),
            ),
            (
                Shape::Badge,
                Body::TextBlock(TextBlockData { lines: Vec::new() }),
            ),
        ]
        .into_iter()
        .for_each(|(shape, expected)| assert_eq!(empty_body(shape), expected));
    }
}