Merge LokiAPI+Convenience & LokiAPI

SignalServiceKit/src/Loki/API/LokiAPI+Convenience.swift

@@ -1,49 +0,0 @@
-import PromiseKit
-
-internal extension LokiAPI {
-
-    private static let receivedMessageHashValuesKey = "receivedMessageHashValuesKey"
-    private static let receivedMessageHashValuesCollection = "receivedMessageHashValuesCollection"
-
-    internal static func getLastMessageHashValue(for target: LokiAPITarget) -> String? {
-        var result: String? = nil
-        // Uses a read/write connection because getting the last message hash value also removes expired messages as needed
-        // TODO: This shouldn't be the case; a getter shouldn't have an unexpected side effect
-        storage.dbReadWriteConnection.readWrite { transaction in
-            result = storage.getLastMessageHash(forServiceNode: target.address, transaction: transaction)
-        }
-        return result
-    }
-
-    internal static func setLastMessageHashValue(for target: LokiAPITarget, hashValue: String, expirationDate: UInt64) {
-        storage.dbReadWriteConnection.readWrite { transaction in
-            storage.setLastMessageHash(forServiceNode: target.address, hash: hashValue, expiresAt: expirationDate, transaction: transaction)
-        }
-    }
-
-    internal static func getReceivedMessageHashValues() -> Set<String>? {
-        var result: Set<String>? = nil
-        storage.dbReadConnection.read { transaction in
-            result = transaction.object(forKey: receivedMessageHashValuesKey, inCollection: receivedMessageHashValuesCollection) as! Set<String>?
-        }
-        return result
-    }
-
-    internal static func setReceivedMessageHashValues(to receivedMessageHashValues: Set<String>) {
-        storage.dbReadWriteConnection.readWrite { transaction in
-            transaction.setObject(receivedMessageHashValues, forKey: receivedMessageHashValuesKey, inCollection: receivedMessageHashValuesCollection)
-        }
-    }
-}
-
-internal extension Promise {
-    
-    internal func recoveringNetworkErrorsIfNeeded() -> Promise<T> {
-        return recover() { error -> Promise<T> in
-            switch error {
-            case NetworkManagerError.taskError(_, let underlyingError): throw underlyingError
-            default: throw error
-            }
-        }
-    }
-}

SignalServiceKit/src/Loki/API/LokiAPI.swift

@@ -173,4 +173,51 @@ public final class LokiAPI : NSObject {
             return envelope
         }
     }
+
+    // MARK: Caching
+    private static let receivedMessageHashValuesKey = "receivedMessageHashValuesKey"
+    private static let receivedMessageHashValuesCollection = "receivedMessageHashValuesCollection"
+
+    private static func getLastMessageHashValue(for target: LokiAPITarget) -> String? {
+        var result: String? = nil
+        // Uses a read/write connection because getting the last message hash value also removes expired messages as needed
+        // TODO: This shouldn't be the case; a getter shouldn't have an unexpected side effect
+        storage.dbReadWriteConnection.readWrite { transaction in
+            result = storage.getLastMessageHash(forServiceNode: target.address, transaction: transaction)
+        }
+        return result
+    }
+
+    private static func setLastMessageHashValue(for target: LokiAPITarget, hashValue: String, expirationDate: UInt64) {
+        storage.dbReadWriteConnection.readWrite { transaction in
+            storage.setLastMessageHash(forServiceNode: target.address, hash: hashValue, expiresAt: expirationDate, transaction: transaction)
+        }
+    }
+
+    private static func getReceivedMessageHashValues() -> Set<String>? {
+        var result: Set<String>? = nil
+        storage.dbReadConnection.read { transaction in
+            result = transaction.object(forKey: receivedMessageHashValuesKey, inCollection: receivedMessageHashValuesCollection) as! Set<String>?
+        }
+        return result
+    }
+
+    private static func setReceivedMessageHashValues(to receivedMessageHashValues: Set<String>) {
+        storage.dbReadWriteConnection.readWrite { transaction in
+            transaction.setObject(receivedMessageHashValues, forKey: receivedMessageHashValuesKey, inCollection: receivedMessageHashValuesCollection)
+        }
+    }
+}
+
+// MARK: Error Handling
+private extension Promise {
+
+    fileprivate func recoveringNetworkErrorsIfNeeded() -> Promise<T> {
+        return recover() { error -> Promise<T> in
+            switch error {
+            case NetworkManagerError.taskError(_, let underlyingError): throw underlyingError
+            default: throw error
+            }
+        }
+    }
 }