nushell/src/commands/to_json.rs

67 lines
2.7 KiB
Rust
Raw Normal View History

2019-05-28 06:00:00 +02:00
use crate::object::{Primitive, Value};
use crate::prelude::*;
2019-06-30 08:46:49 +02:00
pub fn value_to_json_value(v: &Value) -> serde_json::Value {
match v {
Value::Primitive(Primitive::Boolean(b)) => serde_json::Value::Bool(*b),
Value::Primitive(Primitive::Bytes(b)) => {
serde_json::Value::Number(serde_json::Number::from(*b as u64))
}
Value::Primitive(Primitive::Date(d)) => serde_json::Value::String(d.to_string()),
Value::Primitive(Primitive::EndOfStream) => serde_json::Value::Null,
Value::Primitive(Primitive::BeginningOfStream) => serde_json::Value::Null,
2019-06-30 08:46:49 +02:00
Value::Primitive(Primitive::Float(f)) => {
serde_json::Value::Number(serde_json::Number::from_f64(f.into_inner()).unwrap())
}
Value::Primitive(Primitive::Int(i)) => {
serde_json::Value::Number(serde_json::Number::from(*i))
}
Value::Primitive(Primitive::Nothing) => serde_json::Value::Null,
Value::Primitive(Primitive::String(s)) => serde_json::Value::String(s.clone()),
2019-07-15 23:16:27 +02:00
Value::Primitive(Primitive::Path(s)) => serde_json::Value::String(s.display().to_string()),
2019-06-30 08:46:49 +02:00
Value::List(l) => {
serde_json::Value::Array(l.iter().map(|x| value_to_json_value(x)).collect())
}
Value::Block(_) => serde_json::Value::Null,
2019-07-04 07:11:56 +02:00
Value::Binary(b) => serde_json::Value::Array(
b.iter()
.map(|x| {
serde_json::Value::Number(serde_json::Number::from_f64(*x as f64).unwrap())
})
.collect(),
),
2019-06-30 08:46:49 +02:00
Value::Object(o) => {
let mut m = serde_json::Map::new();
for (k, v) in o.entries.iter() {
2019-07-03 19:37:09 +02:00
m.insert(k.clone(), value_to_json_value(v));
2019-06-30 08:46:49 +02:00
}
serde_json::Value::Object(m)
}
}
}
2019-07-24 00:22:11 +02:00
pub fn to_json(args: CommandArgs, registry: &CommandRegistry) -> Result<OutputStream, ShellError> {
let args = args.evaluate_once(registry)?;
let name_span = args.name_span();
2019-05-28 06:00:00 +02:00
let out = args.input;
2019-07-24 00:22:11 +02:00
2019-05-28 06:00:00 +02:00
Ok(out
2019-07-08 18:44:53 +02:00
.values
.map(
move |a| match serde_json::to_string(&value_to_json_value(&a)) {
Ok(x) => ReturnSuccess::value(
Value::Primitive(Primitive::String(x)).simple_spanned(name_span),
),
2019-08-06 05:03:13 +02:00
_ => Err(ShellError::labeled_error_with_secondary(
"Expected an object with JSON-compatible structure from pipeline",
"requires JSON-compatible input",
name_span,
2019-08-06 05:03:13 +02:00
format!("{} originates from here", a.item.type_name()),
a.span(),
)),
},
)
2019-07-08 18:44:53 +02:00
.to_output_stream())
2019-05-28 06:00:00 +02:00
}