Last active
January 31, 2024 12:49
-
-
Save rnapier/d91175be2f737aca7ad476759948666e to your computer and use it in GitHub Desktop.
Generic JSON Decodable
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import Foundation | |
@dynamicMemberLookup | |
enum JSON: Codable, CustomStringConvertible { | |
var description: String { | |
switch self { | |
case .string(let string): return "\"\(string)\"" | |
case .number(let double): | |
if let int = Int(exactly: double) { | |
return "\(int)" | |
} else { | |
return "\(double)" | |
} | |
case .object(let object): | |
let keyValues = object | |
.map { (key, value) in "\"\(key)\": \(value)" } | |
.joined(separator: ",") | |
return "{\(keyValues)}" | |
case .array(let array): | |
return "\(array)" | |
case .bool(let bool): | |
return "\(bool)" | |
case .null: | |
return "null" | |
} | |
} | |
var isEmpty: Bool { | |
switch self { | |
case .string(let string): return string.isEmpty | |
case .object(let object): return object.isEmpty | |
case .array(let array): return array.isEmpty | |
case .null: return true | |
case .number, .bool: return false | |
} | |
} | |
struct Key: CodingKey, Hashable, CustomStringConvertible { | |
var description: String { | |
return stringValue | |
} | |
let stringValue: String | |
init(_ string: String) { self.stringValue = string } | |
init?(stringValue: String) { self.init(stringValue) } | |
var intValue: Int? { return nil } | |
init?(intValue: Int) { return nil } | |
} | |
case string(String) | |
case number(Double) // FIXME: Split Int and Double | |
case object([Key: JSON]) | |
case array([JSON]) | |
case bool(Bool) | |
case null | |
init(from decoder: Decoder) throws { | |
if let string = try? decoder.singleValueContainer().decode(String.self) { self = .string(string) } | |
else if let number = try? decoder.singleValueContainer().decode(Double.self) { self = .number(number) } | |
else if let object = try? decoder.container(keyedBy: Key.self) { | |
var result: [Key: JSON] = [:] | |
for key in object.allKeys { | |
result[key] = (try? object.decode(JSON.self, forKey: key)) ?? .null | |
} | |
self = .object(result) | |
} | |
else if var array = try? decoder.unkeyedContainer() { | |
var result: [JSON] = [] | |
for _ in 0..<(array.count ?? 0) { | |
result.append(try array.decode(JSON.self)) | |
} | |
self = .array(result) | |
} | |
else if let bool = try? decoder.singleValueContainer().decode(Bool.self) { self = .bool(bool) } | |
else if let isNull = try? decoder.singleValueContainer().decodeNil(), isNull { self = .null } | |
else { throw DecodingError.dataCorrupted(DecodingError.Context(codingPath: [], | |
debugDescription: "Unknown JSON type")) } | |
} | |
func encode(to encoder: Encoder) throws { | |
switch self { | |
case .string(let string): | |
var container = encoder.singleValueContainer() | |
try container.encode(string) | |
case .number(let number): | |
var container = encoder.singleValueContainer() | |
try container.encode(number) | |
case .bool(let bool): | |
var container = encoder.singleValueContainer() | |
try container.encode(bool) | |
case .object(let object): | |
var container = encoder.container(keyedBy: Key.self) | |
for (key, value) in object { | |
try container.encode(value, forKey: key) | |
} | |
case .array(let array): | |
var container = encoder.unkeyedContainer() | |
for value in array { | |
try container.encode(value) | |
} | |
case .null: | |
var container = encoder.singleValueContainer() | |
try container.encodeNil() | |
} | |
} | |
var objectValue: [String: JSON]? { | |
switch self { | |
case .object(let object): | |
let mapped: [String: JSON] = Dictionary(uniqueKeysWithValues: | |
object.map { (key, value) in (key.stringValue, value) }) | |
return mapped | |
default: return nil | |
} | |
} | |
var arrayValue: [JSON]? { | |
switch self { | |
case .array(let array): return array | |
default: return nil | |
} | |
} | |
subscript(key: String) -> JSON? { | |
guard let jsonKey = Key(stringValue: key), | |
case .object(let object) = self, | |
let value = object[jsonKey] | |
else { return nil } | |
return value | |
} | |
var stringValue: String? { | |
switch self { | |
case .string(let string): return string | |
default: return nil | |
} | |
} | |
var doubleValue: Double? { | |
switch self { | |
case .number(let number): return number | |
default: return nil | |
} | |
} | |
var intValue: Int? { | |
switch self { | |
case .number(let number): return Int(number) | |
default: return nil | |
} | |
} | |
subscript(index: Int) -> JSON? { | |
switch self { | |
case .array(let array): return array[index] | |
default: return nil | |
} | |
} | |
var boolValue: Bool? { | |
switch self { | |
case .bool(let bool): return bool | |
default: return nil | |
} | |
} | |
var anyValue: Any? { | |
switch self { | |
case .string(let string): return string | |
case .number(let number): | |
if let int = Int(exactly: number) { return int } | |
else { return number } | |
case .bool(let bool): return bool | |
case .object(let object): | |
return Dictionary(uniqueKeysWithValues: | |
object.compactMap { (key, value) -> (String, Any)? in | |
if let nonNilValue = value.anyValue { | |
return (key.stringValue, nonNilValue) | |
} else { return nil } | |
}) | |
case .array(let array): | |
return array.compactMap{ $0.anyValue } | |
case .null: | |
return nil | |
} | |
} | |
var dictionaryValue: [String: Any]? { | |
return anyValue as? [String: Any] | |
} | |
subscript(dynamicMember member: String) -> JSON { | |
return self[member] ?? .null | |
} | |
} | |
extension JSON { | |
init(_ value: Any) throws { | |
if let string = value as? String { self = .string(string) } | |
else if let number = value as? NSNumber { self = .number(number.doubleValue) } | |
else if let object = value as? [String: Any] { | |
var result: [Key: JSON] = [:] | |
for (key, subvalue) in object { | |
result[Key(key)] = try JSON(subvalue) | |
} | |
self = .object(result) | |
} | |
else if let array = value as? [Any] { | |
self = .array(try array.map(JSON.init)) | |
} | |
else if let bool = value as? Bool { self = .bool(bool) } | |
else { | |
throw EncodingError.invalidValue(value, EncodingError.Context(codingPath: [], | |
debugDescription: "Cannot encode value")) | |
} | |
} | |
} | |
extension JSONEncoder { | |
func stringEncode<T>(_ value: T) throws -> String where T : Encodable { | |
// JSONEncoder promises to always return UTF-8 | |
return String(data: try self.encode(value), encoding: .utf8)! | |
} | |
} |
Is there a way to import all of this into a project without it getting stuck "indexing - processing files"? I'm trying to use JSON as an Encodable type so I can encode nested objects with variants in them.
Yes. I don't see any problem importing this into a project.
Xcode has many bugs and gets hung very often. I don't see any particular problem with this gist.
Maybe nice to add helpers like:
static func object(_ key: String, _ value: JSON) -> Self { Self.object([Key.init(key): value]) }
and
static func array(_ ints: Int...) -> Self { .array(ints.map { JSON.number(Double($0)) }) }
which allows maybe a little more readability for .object, e.g.: JSON.object("key", .array(1,2,3))
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Is there a way to import all of this into a project without it getting stuck "indexing - processing files"? I'm trying to use JSON as an Encodable type so I can encode nested objects with variants in them.