129 lines
4.4 KiB
Swift
129 lines
4.4 KiB
Swift
//
|
|
// Copyright (c) 2019 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
@objc(OWSBlockListCacheDelegate)
|
|
public protocol BlockListCacheDelegate: class {
|
|
func blockListCacheDidUpdate(_ blocklistCache: BlockListCache)
|
|
}
|
|
|
|
/// A performant cache for which contacts/groups are blocked.
|
|
///
|
|
/// The source of truth for which contacts and groups are blocked is the `blockingManager`, but because
|
|
/// those accessors are made to be thread safe, they can be slow in tight loops, e.g. when rendering table
|
|
/// view cells.
|
|
///
|
|
/// Typically you'll want to create a Cache, update it to the latest state while simultaneously being informed
|
|
/// of any future changes to block list state.
|
|
///
|
|
/// class SomeViewController: BlockListCacheDelegate {
|
|
/// let blockListCache = BlockListCache()
|
|
/// func viewDidLoad() {
|
|
/// super.viewDidLoad()
|
|
/// blockListCache.startObservingAndSyncState(delegate: self)
|
|
/// self.updateAnyViewsWhichDepondOnBlockListCache()
|
|
/// }
|
|
///
|
|
/// func blockListCacheDidUpdate(_ blocklistCache: BlockListCache) {
|
|
/// self.updateAnyViewsWhichDepondOnBlockListCache()
|
|
/// }
|
|
///
|
|
/// ...
|
|
/// }
|
|
///
|
|
@objc(OWSBlockListCache)
|
|
public class BlockListCache: NSObject {
|
|
|
|
private var blockedPhoneNumbers: Set<String> = Set()
|
|
private var blockedUUIDs: Set<String> = Set()
|
|
private var blockedGroupIds: Set<Data> = Set()
|
|
private let serialQueue: DispatchQueue = DispatchQueue(label: "BlockListCache")
|
|
weak var delegate: BlockListCacheDelegate?
|
|
|
|
private var blockingManager: OWSBlockingManager {
|
|
return OWSBlockingManager.shared()
|
|
}
|
|
|
|
/// Generally something which wants to use this cache wants to do 3 things
|
|
/// 1. get the cache on the latest state
|
|
/// 2. update the cache whenever the blockingManager's state changes
|
|
/// 3. be notified when the cache updates
|
|
/// This method does all three.
|
|
@objc
|
|
public func startObservingAndSyncState(delegate: BlockListCacheDelegate) {
|
|
self.delegate = delegate
|
|
NotificationCenter.default.addObserver(self,
|
|
selector: #selector(blockListDidChange),
|
|
name: ._BlockListDidChange,
|
|
object: nil)
|
|
updateWithoutNotifyingDelegate()
|
|
}
|
|
|
|
// MARK: -
|
|
|
|
@objc
|
|
func blockListDidChange() {
|
|
self.update()
|
|
}
|
|
|
|
@objc(isAddressBlocked:)
|
|
public func isBlocked(address: SignalServiceAddress) -> Bool {
|
|
return serialQueue.sync {
|
|
var blocked = false
|
|
if let phoneNumber = address.phoneNumber {
|
|
blocked = blockedPhoneNumbers.contains(phoneNumber)
|
|
}
|
|
if !blocked, let uuidString = address.uuidString {
|
|
blocked = blockedUUIDs.contains(uuidString)
|
|
}
|
|
return blocked
|
|
}
|
|
}
|
|
|
|
@objc(isGroupIdBlocked:)
|
|
public func isBlocked(groupId: Data) -> Bool {
|
|
return serialQueue.sync {
|
|
blockedGroupIds.contains(groupId)
|
|
}
|
|
}
|
|
|
|
@objc(isThreadBlocked:)
|
|
public func isBlocked(thread: TSThread) -> Bool {
|
|
switch thread {
|
|
case let contactThread as TSContactThread:
|
|
return isBlocked(address: contactThread.contactAddress)
|
|
case let groupThread as TSGroupThread:
|
|
return isBlocked(groupId: groupThread.groupModel.groupId)
|
|
default:
|
|
owsFailDebug("\(self.logTag) in \(#function) unexpected thread type: \(type(of: thread))")
|
|
return false
|
|
}
|
|
}
|
|
|
|
// MARK: -
|
|
|
|
public func update() {
|
|
updateWithoutNotifyingDelegate()
|
|
DispatchQueue.main.async {
|
|
self.delegate?.blockListCacheDidUpdate(self)
|
|
}
|
|
}
|
|
|
|
private func updateWithoutNotifyingDelegate() {
|
|
let blockedPhoneNumbers = Set(blockingManager.blockedPhoneNumbers)
|
|
let blockedUUIDs = Set(blockingManager.blockedUUIDs)
|
|
let blockedGroupIds = Set(blockingManager.blockedGroupIds)
|
|
update(blockedPhoneNumbers: blockedPhoneNumbers, blockedUUIDs: blockedUUIDs, blockedGroupIds: blockedGroupIds)
|
|
}
|
|
|
|
private func update(blockedPhoneNumbers: Set<String>, blockedUUIDs: Set<String>, blockedGroupIds: Set<Data>) {
|
|
serialQueue.sync {
|
|
self.blockedPhoneNumbers = blockedPhoneNumbers
|
|
self.blockedUUIDs = blockedUUIDs
|
|
self.blockedGroupIds = blockedGroupIds
|
|
}
|
|
}
|
|
}
|