2019-08-19 07:16:39 +02:00
|
|
|
use crate::commands::WholeStreamCommand;
|
2019-09-05 18:23:42 +02:00
|
|
|
use crate::data::{Primitive, Value};
|
2019-07-16 06:05:38 +02:00
|
|
|
use crate::prelude::*;
|
|
|
|
|
2019-08-19 07:16:39 +02:00
|
|
|
pub struct ToYAML;
|
|
|
|
|
|
|
|
impl WholeStreamCommand for ToYAML {
|
|
|
|
fn name(&self) -> &str {
|
|
|
|
"to-yaml"
|
|
|
|
}
|
|
|
|
|
|
|
|
fn signature(&self) -> Signature {
|
|
|
|
Signature::build("to-yaml")
|
|
|
|
}
|
2019-08-30 00:52:32 +02:00
|
|
|
|
|
|
|
fn usage(&self) -> &str {
|
|
|
|
"Convert table into .yaml/.yml text"
|
|
|
|
}
|
|
|
|
|
|
|
|
fn run(
|
|
|
|
&self,
|
|
|
|
args: CommandArgs,
|
|
|
|
registry: &CommandRegistry,
|
|
|
|
) -> Result<OutputStream, ShellError> {
|
|
|
|
to_yaml(args, registry)
|
|
|
|
}
|
2019-08-19 07:16:39 +02:00
|
|
|
}
|
|
|
|
|
2019-09-01 18:20:31 +02:00
|
|
|
pub fn value_to_yaml_value(v: &Tagged<Value>) -> Result<serde_yaml::Value, ShellError> {
|
|
|
|
Ok(match v.item() {
|
2019-07-16 06:05:38 +02:00
|
|
|
Value::Primitive(Primitive::Boolean(b)) => serde_yaml::Value::Bool(*b),
|
|
|
|
Value::Primitive(Primitive::Bytes(b)) => {
|
2019-08-30 19:29:04 +02:00
|
|
|
serde_yaml::Value::Number(serde_yaml::Number::from(b.to_f64().unwrap()))
|
2019-07-16 06:05:38 +02:00
|
|
|
}
|
|
|
|
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-08-30 19:29:04 +02:00
|
|
|
Value::Primitive(Primitive::Decimal(f)) => {
|
|
|
|
serde_yaml::Value::Number(serde_yaml::Number::from(f.to_f64().unwrap()))
|
2019-07-16 06:05:38 +02:00
|
|
|
}
|
2019-09-01 18:20:31 +02:00
|
|
|
Value::Primitive(Primitive::Int(i)) => serde_yaml::Value::Number(serde_yaml::Number::from(
|
|
|
|
CoerceInto::<i64>::coerce_into(i.tagged(v.tag), "converting to YAML number")?,
|
|
|
|
)),
|
2019-07-16 06:05:38 +02:00
|
|
|
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()),
|
|
|
|
|
2019-09-05 18:23:42 +02:00
|
|
|
Value::Table(l) => {
|
2019-09-01 18:20:31 +02:00
|
|
|
let mut out = vec![];
|
|
|
|
|
|
|
|
for value in l {
|
|
|
|
out.push(value_to_yaml_value(value)?);
|
|
|
|
}
|
|
|
|
|
|
|
|
serde_yaml::Value::Sequence(out)
|
2019-07-16 06:05:38 +02:00
|
|
|
}
|
|
|
|
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(),
|
|
|
|
),
|
2019-09-05 18:23:42 +02:00
|
|
|
Value::Row(o) => {
|
2019-07-16 06:05:38 +02:00
|
|
|
let mut m = serde_yaml::Mapping::new();
|
|
|
|
for (k, v) in o.entries.iter() {
|
2019-09-01 18:20:31 +02:00
|
|
|
m.insert(
|
|
|
|
serde_yaml::Value::String(k.clone()),
|
|
|
|
value_to_yaml_value(v)?,
|
|
|
|
);
|
2019-07-16 06:05:38 +02:00
|
|
|
}
|
|
|
|
serde_yaml::Value::Mapping(m)
|
|
|
|
}
|
2019-09-01 18:20:31 +02:00
|
|
|
})
|
2019-07-16 06:05:38 +02:00
|
|
|
}
|
|
|
|
|
2019-08-19 07:16:39 +02:00
|
|
|
fn to_yaml(args: CommandArgs, registry: &CommandRegistry) -> Result<OutputStream, ShellError> {
|
2019-07-24 00:22:11 +02:00
|
|
|
let args = args.evaluate_once(registry)?;
|
|
|
|
let name_span = args.name_span();
|
2019-09-04 08:48:40 +02:00
|
|
|
let stream = async_stream_block! {
|
|
|
|
let input: Vec<Tagged<Value>> = args.input.values.collect().await;
|
|
|
|
|
|
|
|
let to_process_input = if input.len() > 1 {
|
|
|
|
let tag = input[0].tag;
|
2019-09-05 18:23:42 +02:00
|
|
|
vec![Tagged { item: Value::Table(input), tag } ]
|
2019-09-04 08:48:40 +02:00
|
|
|
} else if input.len() == 1 {
|
|
|
|
input
|
|
|
|
} else {
|
|
|
|
vec![]
|
|
|
|
};
|
|
|
|
|
|
|
|
for value in to_process_input {
|
|
|
|
match value_to_yaml_value(&value) {
|
|
|
|
Ok(yaml_value) => {
|
|
|
|
match serde_yaml::to_string(&yaml_value) {
|
|
|
|
Ok(x) => yield ReturnSuccess::value(
|
|
|
|
Value::Primitive(Primitive::String(x)).simple_spanned(name_span),
|
|
|
|
),
|
|
|
|
_ => yield Err(ShellError::labeled_error_with_secondary(
|
2019-09-04 18:29:49 +02:00
|
|
|
"Expected a table with YAML-compatible structure.span() from pipeline",
|
2019-09-04 08:48:40 +02:00
|
|
|
"requires YAML-compatible input",
|
|
|
|
name_span,
|
|
|
|
"originates from here".to_string(),
|
|
|
|
value.span(),
|
|
|
|
)),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
_ => yield Err(ShellError::labeled_error(
|
2019-09-04 18:29:49 +02:00
|
|
|
"Expected a table with YAML-compatible structure from pipeline",
|
2019-08-06 05:03:13 +02:00
|
|
|
"requires YAML-compatible input",
|
2019-09-04 08:48:40 +02:00
|
|
|
name_span))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
Ok(stream.to_output_stream())
|
2019-07-16 06:05:38 +02:00
|
|
|
}
|