Add JSON data function (#2)

0xLeif · Jun 10, 2023 · a1ad092 · a1ad092
1 parent 6ddb9ec
commit a1ad092
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/Sources/Cache/JSON/JSON.swift b/Sources/Cache/JSON/JSON.swift
@@ -71,6 +71,21 @@ public struct JSON<Key: RawRepresentable & Hashable>: Cacheable where Key.RawVal
         }
     }
 
+    /// Returns JSON data.
+    ///
+    /// - Throws: Errors are from `JSONSerialization.data(withJSONObject:)`
+    public func data() throws -> Data {
+        var stringKeyedValues: [String: Any] = [:]
+
+        for (key, value) in allValues {
+            stringKeyedValues[key.rawValue] = value
+        }
+
+        return try JSONSerialization.data(
+            withJSONObject: stringKeyedValues
+        )
+    }
+
     /**
     Retrieves a nested JSON object within the current object.
 

diff --git a/Tests/CacheTests/JSONTests.swift b/Tests/CacheTests/JSONTests.swift
@@ -24,6 +24,13 @@ final class JSONTests: XCTestCase {
         XCTAssertEqual(json.get(.text), "Hello, World!")
         XCTAssertEqual(json.get(.count), 27)
         XCTAssertEqual(json.get(.isError), false)
+
+        let data = try json.data()
+        let loadedJSON: JSON<Key> = JSON(data: data)
+
+        XCTAssertEqual(loadedJSON.get(.text, as: String.self), json.get(.text))
+        XCTAssertEqual(loadedJSON.get(.count, as: Int.self), json.get(.count))
+        XCTAssertEqual(loadedJSON.get(.isError, as: Bool.self), json.get(.isError))
     }
 
     func testInitArray() throws {