to_document_utf8_lossy

Author: abr-egn

src/raw/document.rs

@@ -10,6 +10,7 @@ use crate::{
     error::{Error, Result},
     raw::{error::ErrorKind, serde::OwnedOrBorrowedRawDocument, RAW_DOCUMENT_NEWTYPE},
     DateTime,
+    RawBson,
     Timestamp,
 };
 
@@ -526,13 +527,24 @@ impl RawDocument {
 
     /// Copy this into a [`Document`], returning an error if invalid BSON is encountered.
     pub fn to_document(&self) -> Result<Document> {
-        self.as_ref().try_into()
+        self.try_into()
     }
 
     /// Copy this into a [`Document`], returning an error if invalid BSON is encountered.  Any
     /// invalid UTF-8 sequences will be replaced with the Unicode replacement character.
     pub fn to_document_utf8_lossy(&self) -> Result<Document> {
-        todo!()
+        self.iter_elements()
+            .map(|res| {
+                res.and_then(|e| {
+                    let key = e.key().to_owned();
+                    let raw_value: RawBson = match e.value_utf8_lossy()? {
+                        Some(l) => l.into(),
+                        None => e.value()?.to_raw_bson(),
+                    };
+                    Ok((key, raw_value.try_into()?))
+                })
+            })
+            .collect()
     }
 }
 

src/raw/iter.rs

@@ -455,3 +455,22 @@ pub(crate) struct Utf8LossyJavaScriptCodeWithScope<'a> {
     pub(crate) code: String,
     pub(crate) scope: &'a RawDocument,
 }
+
+impl<'a> From<Utf8LossyBson<'a>> for RawBson {
+    fn from(value: Utf8LossyBson<'a>) -> Self {
+        match value {
+            Utf8LossyBson::String(s) => RawBson::String(s),
+            Utf8LossyBson::JavaScriptCode(s) => RawBson::JavaScriptCode(s),
+            Utf8LossyBson::JavaScriptCodeWithScope(Utf8LossyJavaScriptCodeWithScope {
+                code,
+                scope,
+            }) => RawBson::JavaScriptCodeWithScope(super::RawJavaScriptCodeWithScope {
+                code,
+                scope: scope.to_raw_document_buf(),
+            }),
+            Utf8LossyBson::Symbol(s) => RawBson::Symbol(s),
+            Utf8LossyBson::DbPointer(p) => RawBson::DbPointer(p),
+            Utf8LossyBson::RegularExpression(r) => RawBson::RegularExpression(r),
+        }
+    }
+}