use crate::commands::WholeStreamCommand; use crate::prelude::*; use bson::{encode_document, oid::ObjectId, spec::BinarySubtype, Bson, Document}; use nu_errors::{CoerceInto, ShellError}; use nu_protocol::{ Dictionary, Primitive, ReturnSuccess, Signature, SpannedTypeName, UnspannedPathMember, UntaggedValue, Value, }; use std::convert::TryInto; pub struct ToBSON; impl WholeStreamCommand for ToBSON { fn name(&self) -> &str { "to-bson" } fn signature(&self) -> Signature { Signature::build("to-bson") } fn usage(&self) -> &str { "Convert table into .bson text." } fn run( &self, args: CommandArgs, registry: &CommandRegistry, ) -> Result { to_bson(args, registry) } fn is_binary(&self) -> bool { true } } pub fn value_to_bson_value(v: &Value) -> Result { Ok(match &v.value { UntaggedValue::Primitive(Primitive::Boolean(b)) => Bson::Boolean(*b), // FIXME: What about really big decimals? UntaggedValue::Primitive(Primitive::Bytes(decimal)) => Bson::FloatingPoint( (decimal) .to_f64() .expect("Unimplemented BUG: What about big decimals?"), ), UntaggedValue::Primitive(Primitive::Duration(secs)) => Bson::I64(*secs as i64), UntaggedValue::Primitive(Primitive::Date(d)) => Bson::UtcDatetime(*d), UntaggedValue::Primitive(Primitive::EndOfStream) => Bson::Null, UntaggedValue::Primitive(Primitive::BeginningOfStream) => Bson::Null, UntaggedValue::Primitive(Primitive::Decimal(d)) => { Bson::FloatingPoint(d.to_f64().ok_or_else(|| { ShellError::labeled_error( "Could not convert value to decimal", "could not convert to decimal", &v.tag, ) })?) } UntaggedValue::Primitive(Primitive::Int(i)) => { Bson::I64(i.tagged(&v.tag).coerce_into("converting to BSON")?) } UntaggedValue::Primitive(Primitive::Nothing) => Bson::Null, UntaggedValue::Primitive(Primitive::String(s)) => Bson::String(s.clone()), UntaggedValue::Primitive(Primitive::Line(s)) => Bson::String(s.clone()), UntaggedValue::Primitive(Primitive::ColumnPath(path)) => Bson::Array( path.iter() .map(|x| match &x.unspanned { UnspannedPathMember::String(string) => Ok(Bson::String(string.clone())), UnspannedPathMember::Int(int) => Ok(Bson::I64( int.tagged(&v.tag).coerce_into("converting to BSON")?, )), }) .collect::, ShellError>>()?, ), UntaggedValue::Primitive(Primitive::Pattern(p)) => Bson::String(p.clone()), UntaggedValue::Primitive(Primitive::Path(s)) => Bson::String(s.display().to_string()), UntaggedValue::Table(l) => Bson::Array( l.iter() .map(|x| value_to_bson_value(x)) .collect::>()?, ), UntaggedValue::Block(_) | UntaggedValue::Primitive(Primitive::Range(_)) => Bson::Null, UntaggedValue::Error(e) => return Err(e.clone()), UntaggedValue::Primitive(Primitive::Binary(b)) => { Bson::Binary(BinarySubtype::Generic, b.clone()) } UntaggedValue::Row(o) => object_value_to_bson(o)?, }) } // object_value_to_bson handles all Objects, even those that correspond to special // types (things like regex or javascript code). fn object_value_to_bson(o: &Dictionary) -> Result { let mut it = o.entries.iter(); if it.len() > 2 { return generic_object_value_to_bson(o); } match it.next() { Some((regex, tagged_regex_value)) if regex == "$regex" => match it.next() { Some((options, tagged_opts_value)) if options == "$options" => { let r: Result = tagged_regex_value.try_into(); let opts: Result = tagged_opts_value.try_into(); match (r, opts) { (Ok(r), Ok(opts)) => Ok(Bson::RegExp(r, opts)), _ => generic_object_value_to_bson(o), } } _ => generic_object_value_to_bson(o), }, Some((javascript, tagged_javascript_value)) if javascript == "$javascript" => { match it.next() { Some((scope, tagged_scope_value)) if scope == "$scope" => { let js: Result = tagged_javascript_value.try_into(); let s: Result<&Dictionary, _> = tagged_scope_value.try_into(); match (js, s) { (Ok(js), Ok(s)) => { if let Bson::Document(doc) = object_value_to_bson(s)? { Ok(Bson::JavaScriptCodeWithScope(js, doc)) } else { generic_object_value_to_bson(o) } } _ => generic_object_value_to_bson(o), } } None => { let js: Result = tagged_javascript_value.try_into(); match js { Err(_) => generic_object_value_to_bson(o), Ok(v) => Ok(Bson::JavaScriptCode(v)), } } _ => generic_object_value_to_bson(o), } } Some((timestamp, tagged_timestamp_value)) if timestamp == "$timestamp" => { let ts: Result = tagged_timestamp_value.try_into(); if let Ok(ts) = ts { Ok(Bson::TimeStamp(ts)) } else { generic_object_value_to_bson(o) } } Some((binary_subtype, tagged_binary_subtype_value)) if binary_subtype == "$binary_subtype" => { match it.next() { Some((binary, tagged_bin_value)) if binary == "$binary" => { let bst = get_binary_subtype(tagged_binary_subtype_value); let bin: Result, _> = tagged_bin_value.try_into(); match (bin, bst) { (Ok(bin), Ok(v)) => Ok(Bson::Binary(v, bin)), _ => generic_object_value_to_bson(o), } } _ => generic_object_value_to_bson(o), } } Some((object_id, tagged_object_id_value)) if object_id == "$object_id" => { let obj_id: Result = tagged_object_id_value.try_into(); if let Ok(obj_id) = obj_id { let obj_id = ObjectId::with_string(&obj_id); if let Ok(obj_id) = obj_id { Ok(Bson::ObjectId(obj_id)) } else { generic_object_value_to_bson(o) } } else { generic_object_value_to_bson(o) } } Some((symbol, tagged_symbol_value)) if symbol == "$symbol" => { let sym: Result = tagged_symbol_value.try_into(); if let Ok(sym) = sym { Ok(Bson::Symbol(sym)) } else { generic_object_value_to_bson(o) } } _ => generic_object_value_to_bson(o), } } fn get_binary_subtype(tagged_value: &Value) -> Result { match &tagged_value.value { UntaggedValue::Primitive(Primitive::String(s)) => Ok(match s.as_ref() { "generic" => BinarySubtype::Generic, "function" => BinarySubtype::Function, "binary_old" => BinarySubtype::BinaryOld, "uuid_old" => BinarySubtype::UuidOld, "uuid" => BinarySubtype::Uuid, "md5" => BinarySubtype::Md5, _ => unreachable!(), }), UntaggedValue::Primitive(Primitive::Int(i)) => Ok(BinarySubtype::UserDefined( i.tagged(&tagged_value.tag) .coerce_into("converting to BSON binary subtype")?, )), _ => Err(ShellError::type_error( "bson binary", tagged_value.spanned_type_name(), )), } } // generic_object_value_bson handles any Object that does not // correspond to a special bson type (things like regex or javascript code). fn generic_object_value_to_bson(o: &Dictionary) -> Result { let mut doc = Document::new(); for (k, v) in o.entries.iter() { doc.insert(k.clone(), value_to_bson_value(v)?); } Ok(Bson::Document(doc)) } fn shell_encode_document(writer: &mut Vec, doc: Document, tag: Tag) -> Result<(), ShellError> { match encode_document(writer, &doc) { Err(e) => Err(ShellError::labeled_error( format!("Failed to encode document due to: {:?}", e), "requires BSON-compatible document", tag, )), _ => Ok(()), } } fn bson_value_to_bytes(bson: Bson, tag: Tag) -> Result, ShellError> { let mut out = Vec::new(); match bson { Bson::Array(a) => { for v in a.into_iter() { match v { Bson::Document(d) => shell_encode_document(&mut out, d, tag.clone())?, _ => { return Err(ShellError::labeled_error( format!("All top level values must be Documents, got {:?}", v), "requires BSON-compatible document", &tag, )) } } } } Bson::Document(d) => shell_encode_document(&mut out, d, tag)?, _ => { return Err(ShellError::labeled_error( format!("All top level values must be Documents, got {:?}", bson), "requires BSON-compatible document", tag, )) } } Ok(out) } fn to_bson(args: CommandArgs, registry: &CommandRegistry) -> Result { let args = args.evaluate_once(registry)?; let name_tag = args.name_tag(); let name_span = name_tag.span; let stream = async_stream! { let input: Vec = args.input.values.collect().await; let to_process_input = if input.len() > 1 { let tag = input[0].tag.clone(); vec![Value { value: UntaggedValue::Table(input), tag } ] } else if input.len() == 1 { input } else { vec![] }; for value in to_process_input { match value_to_bson_value(&value) { Ok(bson_value) => { let value_span = value.tag.span; match bson_value_to_bytes(bson_value, name_tag.clone()) { Ok(x) => yield ReturnSuccess::value( UntaggedValue::binary(x).into_value(&name_tag), ), _ => yield Err(ShellError::labeled_error_with_secondary( "Expected a table with BSON-compatible structure from pipeline", "requires BSON-compatible input", name_span, "originates from here".to_string(), value_span, )), } } _ => yield Err(ShellError::labeled_error( "Expected a table with BSON-compatible structure from pipeline", "requires BSON-compatible input", &name_tag)) } } }; Ok(stream.to_output_stream()) }