mdqy 0.1.4

//! JSON output. Node attrs sit at the top level alongside `kind`,
//! `text`, `children` so downstream `jq` reads them directly.

use std::io;

use serde_json::{Map, Number, Value as J};

use crate::ast::Node;
use crate::error::RunError;
use crate::events::plain_text;
use crate::value::Value;

/// Flags from `--output json`. `compact` packs each result on one
/// line; `include_spans` adds the byte `span` object to every node.
#[derive(Debug, Clone, Copy, Default)]
pub struct JsonOptions {
    pub compact: bool,
    pub include_spans: bool,
}

impl JsonOptions {
    /// Compact, no spans. Used by `tostring`, `tojson`, and fallback
    /// serialisation in the md and tty emitters.
    pub const COMPACT: Self = Self {
        compact: true,
        include_spans: false,
    };
}

/// Emit one value as JSON, followed by a newline.
pub fn emit<W: io::Write>(
    writer: &mut W,
    value: &Value,
    opts: JsonOptions,
) -> Result<(), RunError> {
    let json = value_to_json(value, opts);
    let result = if opts.compact {
        serde_json::to_writer(&mut *writer, &json)
    } else {
        serde_json::to_writer_pretty(&mut *writer, &json)
    };
    result.map_err(|e| RunError::Io(e.to_string()))?;
    writer.write_all(b"\n")?;
    Ok(())
}

/// Convert a `serde_json::Value` into an mdqy `Value`. Used when a
/// builtin (`fromjson`, `--argjson`) or metadata parser hands us
/// JSON that the rest of the pipeline expects as `Value`.
#[must_use]
pub fn value_from_json(j: J) -> Value {
    use std::sync::Arc;
    match j {
        J::Null => Value::Null,
        J::Bool(b) => Value::Bool(b),
        J::Number(n) => Value::Number(n.as_f64().unwrap_or(f64::NAN)),
        J::String(s) => Value::from(s),
        J::Array(a) => Value::Array(Arc::new(a.into_iter().map(value_from_json).collect())),
        J::Object(m) => {
            let converted = m
                .into_iter()
                .map(|(k, v)| (k, value_from_json(v)))
                .collect();
            Value::Object(Arc::new(converted))
        }
    }
}

/// Convert a `Value` to a `serde_json::Value` in the published shape.
#[must_use]
pub fn value_to_json(value: &Value, opts: JsonOptions) -> J {
    match value {
        Value::Null => J::Null,
        Value::Bool(b) => J::Bool(*b),
        Value::Number(n) => number_to_json(*n),
        Value::String(s) => J::String(s.to_string()),
        Value::Array(arr) => J::Array(arr.iter().map(|v| value_to_json(v, opts)).collect()),
        Value::Object(map) => J::Object(
            map.iter()
                .map(|(k, v)| (k.clone(), value_to_json(v, opts)))
                .collect(),
        ),
        Value::Node(node) => node_to_json(node, opts),
    }
}

fn node_to_json(node: &Node, opts: JsonOptions) -> J {
    let mut obj = Map::new();
    obj.insert("kind".into(), J::String(node.kind.as_str().into()));
    for (k, v) in &node.attrs {
        obj.insert((*k).to_string(), value_to_json(v, opts));
    }
    if !node.children.is_empty() {
        let text = plain_text(&node.children);
        if !text.is_empty() {
            obj.insert("text".into(), J::String(text));
        }
        obj.insert(
            "children".into(),
            J::Array(
                node.children
                    .iter()
                    .map(|v| value_to_json(v, opts))
                    .collect(),
            ),
        );
    }
    if opts.include_spans {
        if let Some(span) = node.span {
            let s: Map<String, J> = [("start", span.start), ("end", span.end)]
                .into_iter()
                .map(|(k, v)| (k.into(), J::Number(Number::from(v as u64))))
                .collect();
            obj.insert("span".into(), J::Object(s));
        }
    }
    J::Object(obj)
}

/// Integer-valued floats emit as JSON integers, so `.level == 1`
/// works downstream without writing `1.0`.
fn number_to_json(n: f64) -> J {
    if n.is_finite() && n.fract() == 0.0 && n >= i64::MIN as f64 && n <= i64::MAX as f64 {
        J::Number(Number::from(n as i64))
    } else {
        Number::from_f64(n).map_or(J::Null, J::Number)
    }
}