|
|
|
|
@ -3,11 +3,12 @@ import PromiseKit
|
|
|
|
|
|
|
|
|
|
@objc(LKClosedGroupPoller)
|
|
|
|
|
public final class ClosedGroupPoller : NSObject {
|
|
|
|
|
private var isPolling = false
|
|
|
|
|
private var timer: Timer?
|
|
|
|
|
private var isPolling: [String:Bool] = [:]
|
|
|
|
|
private var timers: [String:Timer] = [:]
|
|
|
|
|
|
|
|
|
|
// MARK: Settings
|
|
|
|
|
private static let pollInterval: TimeInterval = 3
|
|
|
|
|
private static let minPollInterval: Double = 4
|
|
|
|
|
private static let maxPollInterval: Double = 2 * 60
|
|
|
|
|
|
|
|
|
|
// MARK: Error
|
|
|
|
|
private enum Error : LocalizedError {
|
|
|
|
|
@ -22,65 +23,117 @@ public final class ClosedGroupPoller : NSObject {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// MARK: Initialization
|
|
|
|
|
public static let shared = ClosedGroupPoller()
|
|
|
|
|
|
|
|
|
|
private override init() { }
|
|
|
|
|
|
|
|
|
|
// MARK: Public API
|
|
|
|
|
@objc public func startIfNeeded() {
|
|
|
|
|
@objc public func start() {
|
|
|
|
|
#if DEBUG
|
|
|
|
|
assert(Thread.current.isMainThread) // Timers don't do well on background queues
|
|
|
|
|
#endif
|
|
|
|
|
guard !isPolling else { return }
|
|
|
|
|
isPolling = true
|
|
|
|
|
timer = Timer.scheduledTimer(withTimeInterval: ClosedGroupPoller.pollInterval, repeats: true) { [weak self] _ in
|
|
|
|
|
let _ = self?.poll()
|
|
|
|
|
}
|
|
|
|
|
let storage = SNMessagingKitConfiguration.shared.storage
|
|
|
|
|
let allGroupPublicKeys = storage.getUserClosedGroupPublicKeys()
|
|
|
|
|
allGroupPublicKeys.forEach { startPolling(for: $0) }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public func pollOnce() -> [Promise<Void>] {
|
|
|
|
|
guard !isPolling else { return [] }
|
|
|
|
|
isPolling = true
|
|
|
|
|
return poll()
|
|
|
|
|
public func startPolling(for groupPublicKey: String) {
|
|
|
|
|
guard !isPolling(for: groupPublicKey) else { return }
|
|
|
|
|
setUpPolling(for: groupPublicKey)
|
|
|
|
|
isPolling[groupPublicKey] = true
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@objc public func stop() {
|
|
|
|
|
isPolling = false
|
|
|
|
|
timer?.invalidate()
|
|
|
|
|
let storage = SNMessagingKitConfiguration.shared.storage
|
|
|
|
|
let allGroupPublicKeys = storage.getUserClosedGroupPublicKeys()
|
|
|
|
|
allGroupPublicKeys.forEach { stopPolling(for: $0) }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public func stopPolling(for groupPublicKey: String) {
|
|
|
|
|
timers[groupPublicKey]?.invalidate()
|
|
|
|
|
isPolling[groupPublicKey] = false
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// MARK: Private API
|
|
|
|
|
private func poll() -> [Promise<Void>] {
|
|
|
|
|
guard isPolling else { return [] }
|
|
|
|
|
let publicKeys = Storage.shared.getUserClosedGroupPublicKeys()
|
|
|
|
|
return publicKeys.map { publicKey in
|
|
|
|
|
let promise = SnodeAPI.getSwarm(for: publicKey).then2 { [weak self] swarm -> Promise<[JSON]> in
|
|
|
|
|
// randomElement() uses the system's default random generator, which is cryptographically secure
|
|
|
|
|
guard let snode = swarm.randomElement() else { return Promise(error: Error.insufficientSnodes) }
|
|
|
|
|
guard let self = self, self.isPolling else { return Promise(error: Error.pollingCanceled) }
|
|
|
|
|
return SnodeAPI.getRawMessages(from: snode, associatedWith: publicKey).map2 {
|
|
|
|
|
SnodeAPI.parseRawMessagesResponse($0, from: snode, associatedWith: publicKey)
|
|
|
|
|
}
|
|
|
|
|
private func setUpPolling(for groupPublicKey: String) {
|
|
|
|
|
poll(groupPublicKey).done2 { [weak self] _ in
|
|
|
|
|
DispatchQueue.main.async { // Timers don't do well on background queues
|
|
|
|
|
self?.pollRecursively(groupPublicKey)
|
|
|
|
|
}
|
|
|
|
|
promise.done2 { [weak self] messages in
|
|
|
|
|
guard let self = self, self.isPolling else { return }
|
|
|
|
|
if !messages.isEmpty {
|
|
|
|
|
SNLog("Received \(messages.count) new message(s) in closed group with public key: \(publicKey).")
|
|
|
|
|
}.catch2 { [weak self] error in
|
|
|
|
|
// The error is logged in poll(_:)
|
|
|
|
|
DispatchQueue.main.async { // Timers don't do well on background queues
|
|
|
|
|
self?.pollRecursively(groupPublicKey)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private func pollRecursively(_ groupPublicKey: String) {
|
|
|
|
|
let groupID = LKGroupUtilities.getEncodedClosedGroupIDAsData(groupPublicKey)
|
|
|
|
|
guard isPolling(for: groupPublicKey),
|
|
|
|
|
let thread = TSGroupThread.fetch(uniqueId: TSGroupThread.threadId(fromGroupId: groupID)) else { return }
|
|
|
|
|
// Get the received date of the last message in the thread. If we don't have any messages yet, pick some
|
|
|
|
|
// reasonable fake time interval to use instead.
|
|
|
|
|
let lastMessageDate =
|
|
|
|
|
(thread.numberOfInteractions() > 0) ? thread.lastInteraction.receivedAtDate() : Date().addingTimeInterval(-5 * 60)
|
|
|
|
|
let timeSinceLastMessage = Date().timeIntervalSince(lastMessageDate)
|
|
|
|
|
let minPollInterval = ClosedGroupPoller.minPollInterval
|
|
|
|
|
let limit: Double = 12 * 60 * 60
|
|
|
|
|
let a = (ClosedGroupPoller.maxPollInterval - minPollInterval) / limit
|
|
|
|
|
let nextPollInterval = a * min(timeSinceLastMessage, limit) + minPollInterval
|
|
|
|
|
SNLog("Next poll interval for closed group with public key: \(groupPublicKey) is \(nextPollInterval) s.")
|
|
|
|
|
timers[groupPublicKey] = Timer.scheduledTimer(withTimeInterval: nextPollInterval, repeats: false) { [weak self] timer in
|
|
|
|
|
timer.invalidate()
|
|
|
|
|
self?.poll(groupPublicKey).done2 { _ in
|
|
|
|
|
DispatchQueue.main.async { // Timers don't do well on background queues
|
|
|
|
|
self?.pollRecursively(groupPublicKey)
|
|
|
|
|
}
|
|
|
|
|
messages.forEach { json in
|
|
|
|
|
guard let envelope = SNProtoEnvelope.from(json) else { return }
|
|
|
|
|
do {
|
|
|
|
|
let data = try envelope.serializedData()
|
|
|
|
|
let job = MessageReceiveJob(data: data, isBackgroundPoll: false)
|
|
|
|
|
SNMessagingKitConfiguration.shared.storage.write { transaction in
|
|
|
|
|
SessionMessagingKit.JobQueue.shared.add(job, using: transaction)
|
|
|
|
|
}
|
|
|
|
|
} catch {
|
|
|
|
|
SNLog("Failed to deserialize envelope due to error: \(error).")
|
|
|
|
|
}
|
|
|
|
|
}.catch2 { error in
|
|
|
|
|
// The error is logged in poll(_:)
|
|
|
|
|
DispatchQueue.main.async { // Timers don't do well on background queues
|
|
|
|
|
self?.pollRecursively(groupPublicKey)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
promise.catch2 { error in
|
|
|
|
|
SNLog("Polling failed for closed group with public key: \(publicKey) due to error: \(error).")
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private func poll(_ groupPublicKey: String) -> Promise<Void> {
|
|
|
|
|
guard isPolling(for: groupPublicKey) else { return Promise.value(()) }
|
|
|
|
|
let promise = SnodeAPI.getSwarm(for: groupPublicKey).then2 { [weak self] swarm -> Promise<[JSON]> in
|
|
|
|
|
// randomElement() uses the system's default random generator, which is cryptographically secure
|
|
|
|
|
guard let snode = swarm.randomElement() else { return Promise(error: Error.insufficientSnodes) }
|
|
|
|
|
guard let self = self, self.isPolling(for: groupPublicKey) else { return Promise(error: Error.pollingCanceled) }
|
|
|
|
|
return SnodeAPI.getRawMessages(from: snode, associatedWith: groupPublicKey).map2 {
|
|
|
|
|
SnodeAPI.parseRawMessagesResponse($0, from: snode, associatedWith: groupPublicKey)
|
|
|
|
|
}
|
|
|
|
|
promise.retainUntilComplete()
|
|
|
|
|
return promise.map { _ in }
|
|
|
|
|
}
|
|
|
|
|
promise.done2 { [weak self] rawMessages in
|
|
|
|
|
guard let self = self, self.isPolling(for: groupPublicKey) else { return }
|
|
|
|
|
if !rawMessages.isEmpty {
|
|
|
|
|
SNLog("Received \(rawMessages.count) new message(s) in closed group with public key: \(groupPublicKey).")
|
|
|
|
|
}
|
|
|
|
|
rawMessages.forEach { json in
|
|
|
|
|
guard let envelope = SNProtoEnvelope.from(json) else { return }
|
|
|
|
|
do {
|
|
|
|
|
let data = try envelope.serializedData()
|
|
|
|
|
let job = MessageReceiveJob(data: data, isBackgroundPoll: false)
|
|
|
|
|
SNMessagingKitConfiguration.shared.storage.write { transaction in
|
|
|
|
|
SessionMessagingKit.JobQueue.shared.add(job, using: transaction)
|
|
|
|
|
}
|
|
|
|
|
} catch {
|
|
|
|
|
SNLog("Failed to deserialize envelope due to error: \(error).")
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
promise.catch2 { error in
|
|
|
|
|
SNLog("Polling failed for closed group with public key: \(groupPublicKey) due to error: \(error).")
|
|
|
|
|
}
|
|
|
|
|
return promise.map { _ in }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// MARK: Convenience
|
|
|
|
|
private func isPolling(for groupPublicKey: String) -> Bool {
|
|
|
|
|
return isPolling[groupPublicKey] ?? false
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
Niels Andriesse
4 years ago
committed by
GitHub