From 1f2bfe8df42bd1c544b10a3e89b6018deb726a54 Mon Sep 17 00:00:00 2001 From: Matthew Chen Date: Wed, 10 Oct 2018 15:39:00 -0400 Subject: [PATCH] Fix breakage in UD manager. --- SignalServiceKit/src/Messages/UD/OWSUDManager.swift | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/SignalServiceKit/src/Messages/UD/OWSUDManager.swift b/SignalServiceKit/src/Messages/UD/OWSUDManager.swift index 68a701175..6110ceb2b 100644 --- a/SignalServiceKit/src/Messages/UD/OWSUDManager.swift +++ b/SignalServiceKit/src/Messages/UD/OWSUDManager.swift @@ -150,7 +150,10 @@ public class OWSUDManagerImpl: NSObject, OWSUDManager { } } - guard let existingValue = dbConnection.object(forKey: recipientId, inCollection: kUnidentifiedAccessCollection) as? UnidentifiedAccessMode else { + guard let existingRawValue = dbConnection.object(forKey: recipientId, inCollection: kUnidentifiedAccessCollection) as? Int else { + return .unknown + } + guard let existingValue = UnidentifiedAccessMode(rawValue: existingRawValue) else { return .unknown } return existingValue @@ -158,7 +161,7 @@ public class OWSUDManagerImpl: NSObject, OWSUDManager { @objc public func setUnidentifiedAccessMode(_ mode: UnidentifiedAccessMode, recipientId: String) { - dbConnection.setObject(mode, forKey: recipientId, inCollection: kUnidentifiedAccessCollection) + dbConnection.setObject(mode.rawValue as Int, forKey: recipientId, inCollection: kUnidentifiedAccessCollection) } // Returns the UD access key for a given recipient