2019-07-16 06:05:38 +02:00
|
|
|
use crate::object::{Primitive, Value};
|
|
|
|
use crate::prelude::*;
|
|
|
|
|
|
|
|
pub fn value_to_yaml_value(v: &Value) -> serde_yaml::Value {
|
|
|
|
match v {
|
|
|
|
Value::Primitive(Primitive::Boolean(b)) => serde_yaml::Value::Bool(*b),
|
|
|
|
Value::Primitive(Primitive::Bytes(b)) => {
|
|
|
|
serde_yaml::Value::Number(serde_yaml::Number::from(*b as u64))
|
|
|
|
}
|
|
|
|
Value::Primitive(Primitive::Date(d)) => serde_yaml::Value::String(d.to_string()),
|
|
|
|
Value::Primitive(Primitive::EndOfStream) => serde_yaml::Value::Null,
|
2019-07-27 09:45:00 +02:00
|
|
|
Value::Primitive(Primitive::BeginningOfStream) => serde_yaml::Value::Null,
|
2019-07-16 06:05:38 +02:00
|
|
|
Value::Primitive(Primitive::Float(f)) => {
|
|
|
|
serde_yaml::Value::Number(serde_yaml::Number::from(f.into_inner()))
|
|
|
|
}
|
|
|
|
Value::Primitive(Primitive::Int(i)) => {
|
|
|
|
serde_yaml::Value::Number(serde_yaml::Number::from(*i))
|
|
|
|
}
|
|
|
|
Value::Primitive(Primitive::Nothing) => serde_yaml::Value::Null,
|
|
|
|
Value::Primitive(Primitive::String(s)) => serde_yaml::Value::String(s.clone()),
|
|
|
|
Value::Primitive(Primitive::Path(s)) => serde_yaml::Value::String(s.display().to_string()),
|
|
|
|
|
|
|
|
Value::List(l) => {
|
|
|
|
serde_yaml::Value::Sequence(l.iter().map(|x| value_to_yaml_value(x)).collect())
|
|
|
|
}
|
|
|
|
Value::Block(_) => serde_yaml::Value::Null,
|
|
|
|
Value::Binary(b) => serde_yaml::Value::Sequence(
|
|
|
|
b.iter()
|
|
|
|
.map(|x| serde_yaml::Value::Number(serde_yaml::Number::from(*x)))
|
|
|
|
.collect(),
|
|
|
|
),
|
|
|
|
Value::Object(o) => {
|
|
|
|
let mut m = serde_yaml::Mapping::new();
|
|
|
|
for (k, v) in o.entries.iter() {
|
|
|
|
m.insert(serde_yaml::Value::String(k.clone()), value_to_yaml_value(v));
|
|
|
|
}
|
|
|
|
serde_yaml::Value::Mapping(m)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-07-24 00:22:11 +02:00
|
|
|
pub fn to_yaml(args: CommandArgs, registry: &CommandRegistry) -> Result<OutputStream, ShellError> {
|
|
|
|
let args = args.evaluate_once(registry)?;
|
|
|
|
let name_span = args.name_span();
|
2019-07-16 06:05:38 +02:00
|
|
|
let out = args.input;
|
|
|
|
Ok(out
|
|
|
|
.values
|
|
|
|
.map(
|
|
|
|
move |a| match serde_yaml::to_string(&value_to_yaml_value(&a)) {
|
2019-08-05 10:54:29 +02:00
|
|
|
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 YAML-compatible structure from pipeline",
|
|
|
|
"requires YAML-compatible input",
|
2019-07-16 06:05:38 +02:00
|
|
|
name_span,
|
2019-08-06 05:03:13 +02:00
|
|
|
format!("{} originates from here", a.item.type_name()),
|
|
|
|
a.span(),
|
2019-07-16 06:05:38 +02:00
|
|
|
)),
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.to_output_stream())
|
|
|
|
}
|