UserDefaultsStorage.swift

protocol Storable {
    func get<T: Codable>(key: String) -> T?
    func remove(for key: String)
    func add<T: Codable>(object: T, key: String)
    func update<T: Codable>(object: T, key: String)
}

class UserDefaultsStorage: Storable {
    
    static let shared = UserDefaultsStorage()
    
    let storage = UserDefaults.standard
    
    func add<T: Codable>(object: T, key: String) {
        do {
            let encoded = try JSONEncoder().encode(object)
            storage.set(encoded, forKey: key)
        } catch {
        }
    }
    
    func remove(for key: String) {
        storage.removeObject(forKey: key)
    }
    
    func update<T: Codable>(object: T, key: String) {
        remove(for: key)
        add(object: object, key: key)
    }
    
    func get<T: Codable>(key: String) -> T? {
        if let object = storage.object(forKey: key) as? Data {
            return try? JSONDecoder().decode(T.self, from: object)
        }
        return nil
    }
    
}