From b07e1ccbf78d011d9b86bd82900d00907e806814 Mon Sep 17 00:00:00 2001 From: Jonathan Turner Date: Tue, 16 Jul 2019 16:05:38 +1200 Subject: [PATCH] Add missing file --- src/commands/to_yaml.rs | 60 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 60 insertions(+) create mode 100644 src/commands/to_yaml.rs diff --git a/src/commands/to_yaml.rs b/src/commands/to_yaml.rs new file mode 100644 index 0000000000..b9e3d56d77 --- /dev/null +++ b/src/commands/to_yaml.rs @@ -0,0 +1,60 @@ +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, + 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::Filesystem => serde_yaml::Value::Null, + 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) + } + } +} + +pub fn to_yaml(args: CommandArgs) -> Result { + let out = args.input; + let name_span = args.name_span; + Ok(out + .values + .map( + move |a| match serde_yaml::to_string(&value_to_yaml_value(&a)) { + Ok(x) => { + ReturnSuccess::value(Value::Primitive(Primitive::String(x)).spanned(name_span)) + } + Err(_) => Err(ShellError::maybe_labeled_error( + "Can not convert to YAML string", + "can not convert piped data to YAML string", + name_span, + )), + }, + ) + .to_output_stream()) +}