feature(SabyJSON): add raw

ab180 · Aug 25, 2023 · 201ba2d · 201ba2d
1 parent 8fd8ce8
commit 201ba2d
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/Source/JSON/Modify/JSONUnwrap.swift b/Source/JSON/Modify/JSONUnwrap.swift
@@ -32,6 +32,19 @@ extension JSON {
         guard case .array(let value) = self else { return nil }
         return value
     }
+
+    public var raw: Any? {
+        if case .string(let value) = self { return value }
+        if case .number(let value) = self { return value }
+        if case .boolean(let value) = self { return value }
+        if case .object(let value) = self {
+            return value.mapValues { $0.raw }
+        }
+        if case .array(let value) = self {
+            return value.map { $0.raw }
+        }
+        return nil
+    }
 }
 
 extension JSON {

diff --git a/Test/JSON/Modify/JSONUnwrapTest.swift b/Test/JSON/Modify/JSONUnwrapTest.swift
@@ -34,6 +34,25 @@ final class JSONUnwrapTest: XCTestCase {
         XCTAssertEqual(json.rawArray, [])
     }
 
+    func test__raw() {
+        let json = JSON.from([
+            "a": "a",
+            "b": 1,
+            "c": [
+                "a": "a",
+                "b": 1,
+                "c": true
+            ],
+            "d": [
+                "a",
+                1,
+                true
+            ],
+            "e": nil
+        ])
+        XCTAssertEqual(try JSON.from(unsafe: json.raw), json)
+    }
+
     func test__is_string() {
         let json = JSON.from("string")
         XCTAssertTrue(json.isString)