nushell/src/commands/to_toml.rs

72 lines
3.0 KiB
Rust
Raw Normal View History

2019-06-01 20:26:04 +02:00
use crate::object::{Primitive, Value};
use crate::prelude::*;
2019-06-30 08:46:49 +02:00
pub fn value_to_toml_value(v: &Value) -> toml::Value {
match v {
Value::Primitive(Primitive::Boolean(b)) => toml::Value::Boolean(*b),
Value::Primitive(Primitive::Bytes(b)) => toml::Value::Integer(*b as i64),
Value::Primitive(Primitive::Date(d)) => toml::Value::String(d.to_string()),
Value::Primitive(Primitive::EndOfStream) => {
toml::Value::String("<End of Stream>".to_string())
}
Value::Primitive(Primitive::Float(f)) => toml::Value::Float(f.into_inner()),
Value::Primitive(Primitive::Int(i)) => toml::Value::Integer(*i),
Value::Primitive(Primitive::Nothing) => toml::Value::String("<Nothing>".to_string()),
Value::Primitive(Primitive::String(s)) => toml::Value::String(s.clone()),
Value::Filesystem => toml::Value::String("<Filesystem>".to_string()),
Value::List(l) => toml::Value::Array(l.iter().map(|x| value_to_toml_value(x)).collect()),
Value::Block(_) => toml::Value::String("<Block>".to_string()),
2019-07-04 07:11:56 +02:00
Value::Binary(b) => {
toml::Value::Array(b.iter().map(|x| toml::Value::Integer(*x as i64)).collect())
}
2019-06-30 08:46:49 +02:00
Value::Object(o) => {
let mut m = toml::map::Map::new();
for (k, v) in o.entries.iter() {
2019-07-03 19:37:09 +02:00
m.insert(k.clone(), value_to_toml_value(v));
2019-06-30 08:46:49 +02:00
}
toml::Value::Table(m)
}
}
}
2019-06-01 20:26:04 +02:00
pub fn to_toml(args: CommandArgs) -> Result<OutputStream, ShellError> {
let out = args.input;
2019-07-09 06:31:26 +02:00
let name_span = args.name_span;
2019-06-01 20:26:04 +02:00
Ok(out
.values
2019-07-09 06:31:26 +02:00
.map(move |a| {
2019-07-13 04:07:06 +02:00
match toml::to_string(&value_to_toml_value(&a)) {
2019-07-09 06:31:26 +02:00
Ok(val) => {
return ReturnSuccess::value(
Value::Primitive(Primitive::String(val)).spanned(name_span),
)
}
Err(err) => Err(ShellError::type_error(
2019-07-13 04:07:06 +02:00
"serializable to toml",
2019-07-09 06:31:26 +02:00
format!("{:?} - {:?}", a.type_name(), err).spanned(name_span),
)), // toml::Value::String(String) => {
// return ReturnSuccess::value(
// Value::Primitive(Primitive::String(x)).spanned(name_span),
// )
// }
// toml::Value::Integer(i64) => "Integer",
// toml::Value::Float(f64) => "Decimal",
// toml::Value::Boolean(bool) => "Boolean",
// toml::Value::Datetime(Datetime) => "Date",
// toml::Value::Array(Array) => "Array",
// toml::Value::Table(Table) => "Table",
2019-07-08 18:44:53 +02:00
}
2019-07-09 06:31:26 +02:00
// return Err(ShellError::type_error("String", ty.spanned(name_span)));
// Err(_) => Err(ShellError::maybe_labeled_error(
// "Can not convert to TOML string",
// "can not convert piped data to TOML string",
// name_span,
// )),
2019-06-16 01:03:49 +02:00
})
.to_output_stream())
2019-06-01 20:26:04 +02:00
}