UserDefaults+Codable - handy extension for Codable models storing in UserDefaults

UserDefaults+Codable.swift

import Foundation

private let jsonDecoder: JSONDecoder = {
    let decoder = JSONDecoder()
    if #available(iOS 10.0, *) {
        decoder.dateDecodingStrategy = JSONDecoder.DateDecodingStrategy.iso8601
    }
    return decoder
}()

private let jsonEncoder: JSONEncoder = {
    let encoder = JSONEncoder()
    if #available(iOS 10.0, *) {
        encoder.dateEncodingStrategy = JSONEncoder.DateEncodingStrategy.iso8601
    }
    return encoder
}()

private
struct Container<ObjectType: Codable>: Codable {
    let object: ObjectType
}

extension UserDefaults {
    
    func set<ObjectType: Codable>( object: ObjectType?, forKey defaultName: String) {
        
        guard let object = object else {
            self.removeObject(forKey: defaultName)
            return
        }
        
        let container = Container(object: object)
        do {
            let data = try jsonEncoder.encode(container)
            self.set(data, forKey: defaultName)
            
        } catch {
            assertionFailure(error.localizedDescription)
            // ErrorLoggerService.logWithTrace(error: error)
        }
    }
    
    func get<ObjectType: Codable>(forKey defaultName: String, type: ObjectType.Type) -> ObjectType? {
        guard let data = self.data(forKey: defaultName) else {
            return nil
        }
        
        do {
            let container = try jsonDecoder.decode(Container<ObjectType>.self, from: data)
            return container.object
        } catch {
            assertionFailure(error.localizedDescription)
            // ErrorLoggerService.logWithTrace(error: error)
        }
        
        return nil
    }
    
}