feat(swift): implement Document.toString helper

bitwalker · bitwalker · commit c91323f5c2e3 · 2022-10-06T15:05:23.000-04:00
diff --git a/crates/core/src/ffi/mod.rs b/crates/core/src/ffi/mod.rs
@@ -102,7 +102,7 @@ pub extern "C" fn document_empty() -> *mut dom::Document {
 #[export_name = "__liveview_native_core$Document$parse"]
 pub extern "C" fn document_parse<'a>(
     text: RustStr<'a>,
-    error: *mut support::RustString,
+    error: *mut RustString,
 ) -> support::RustResult {
     match dom::Document::parse(text.to_str()) {
         Ok(doc) => {
@@ -114,7 +114,7 @@ pub extern "C" fn document_parse<'a>(
         }
         Err(err) => {
             unsafe {
-                error.write(support::RustString::from_string(err.to_string()));
+                error.write(RustString::from_string(err.to_string()));
             }
             support::RustResult {
                 is_ok: false,
@@ -124,6 +124,12 @@ pub extern "C" fn document_parse<'a>(
     }
 }
 
+#[export_name = "__liveview_native_core$Document$to_string"]
+pub extern "C" fn document_to_string(doc: *mut dom::Document) -> RustString {
+    let doc = unsafe { &*doc };
+    RustString::from_string(doc.to_string())
+}
+
 #[export_name = "__liveview_native_core$Document$merge"]
 pub extern "C" fn document_merge(doc: *mut dom::Document, other: *const dom::Document) -> bool {
     let doc = unsafe { &mut *doc };
diff --git a/swift/LiveViewNative/Sources/LiveViewNative/LiveViewNative.swift b/swift/LiveViewNative/Sources/LiveViewNative/LiveViewNative.swift
@@ -65,6 +65,12 @@ public class Document {
                       }).get()
     }
 
+    /// Renders this document to a `String` for display and comparison
+    public func toString() -> String {
+        let str = RustString(__liveview_native_core$Document$to_string(self.repr))
+        return str.toString()
+    }
+
     /// Register a callback to be fired when a matching event occurs on this document.
     ///
     /// The given callback receives the document to which the event applies, as well as the
diff --git a/swift/LiveViewNative/Tests/LiveViewNativeTests/LiveViewNativeTests.swift b/swift/LiveViewNative/Tests/LiveViewNativeTests/LiveViewNativeTests.swift
@@ -23,6 +23,8 @@ final class LiveViewNativeTests: XCTestCase {
             XCTAssertEqual(ctx is MyContext, true)
             (ctx as! MyContext).didChange = true
         }
+        let rendered1 = doc1.toString()
+        XCTAssertEqual(input, rendered1)
 
         let updated = """
 <html lang="en">
@@ -36,9 +38,14 @@ final class LiveViewNativeTests: XCTestCase {
 </html>
 """
         let doc2 = try Document.parse(updated)
+        let rendered2 = doc2.toString()
+        XCTAssertEqual(updated, rendered2)
 
         doc1.merge(with: doc2)
 
         XCTAssertEqual(context.didChange, true)
+
+        let finalRender = doc1.toString()
+        XCTAssertEqual(finalRender, rendered2)
     }
 }
diff --git a/swift/LiveViewNativeCore/include/LiveViewNativeCore.h b/swift/LiveViewNativeCore/include/LiveViewNativeCore.h
@@ -40,6 +40,8 @@ typedef struct __Node {
   __NodeData data;
 } __Node;
 
+extern _RustString __liveview_native_core$Document$to_string(__Document doc);
+
 extern __Document __liveview_native_core$Document$empty();
 
 extern void __liveview_native_core$Document$drop(__Document doc);
