79 lines
2.8 KiB
Swift
79 lines
2.8 KiB
Swift
//
|
|
// Copyright 2021-2024 New Vector Ltd.
|
|
//
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
// Please see LICENSE in the repository root for full details.
|
|
//
|
|
|
|
import Foundation
|
|
import UIKit
|
|
|
|
@objcMembers
|
|
public class ThreadDataSource: RoomDataSource {
|
|
|
|
private typealias ThreadID = String
|
|
|
|
/// Map of cached data sources. Keys are thread identifiers.
|
|
private static var dataSourceCache: [ThreadID: ThreadDataSource] = [:]
|
|
|
|
public override func finalizeInitialization() {
|
|
super.finalizeInitialization()
|
|
showReadMarker = true
|
|
showBubbleReceipts = true
|
|
showTypingRow = false
|
|
|
|
NotificationCenter.default.addObserver(self,
|
|
selector: #selector(didReceiveMemoryWarning),
|
|
name: UIApplication.didReceiveMemoryWarningNotification,
|
|
object: nil)
|
|
|
|
NotificationCenter.default.addObserver(self,
|
|
selector: #selector(didLeaveRoom(_:)),
|
|
name: .mxSessionDidLeaveRoom,
|
|
object: nil)
|
|
}
|
|
|
|
public override class func load(withRoomId roomId: String!,
|
|
initialEventId: String!,
|
|
threadId: String!,
|
|
andMatrixSession mxSession: MXSession!,
|
|
onComplete: ((Any?) -> Void)!) {
|
|
if let dataSource = dataSourceCache[threadId] {
|
|
onComplete(dataSource)
|
|
return
|
|
}
|
|
super.load(withRoomId: roomId, initialEventId: initialEventId, threadId: threadId, andMatrixSession: mxSession) { dataSource in
|
|
if let dataSource = dataSource as? ThreadDataSource {
|
|
Self.dataSourceCache[threadId] = dataSource
|
|
}
|
|
onComplete(dataSource)
|
|
}
|
|
}
|
|
|
|
@objc
|
|
private func didReceiveMemoryWarning() {
|
|
MXLog.debug("[ThreadDataSource] didReceiveMemoryWarning. Will reload not active data sources in cache.")
|
|
|
|
Self.dataSourceCache.forEach {
|
|
if $1.delegate == nil {
|
|
$1.reload()
|
|
}
|
|
}
|
|
}
|
|
|
|
@objc
|
|
private func didLeaveRoom(_ notification: Notification) {
|
|
MXLog.debug("[ThreadDataSource] didReceiveMemoryWarning. Will reload not active data sources in cache.")
|
|
|
|
guard let session = notification.object as? MXSession,
|
|
session == mxSession,
|
|
let roomId = notification.userInfo?[kMXSessionNotificationRoomIdKey] as? String else {
|
|
return
|
|
}
|
|
|
|
let threadIds = Array(Self.dataSourceCache.filter { $1.roomId == roomId }.keys)
|
|
threadIds.forEach { Self.dataSourceCache.removeValue(forKey: $0) }
|
|
}
|
|
|
|
}
|