element-ios/RiotSwiftUI/Modules/Spaces/MatrixItemChooser/Service/MatrixSDK/MatrixItemChooserRoomRestri...

86 lines
3.3 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
class MatrixItemChooserRoomRestrictedAllowedParentsDataSource: MatrixItemChooserDataSource {
private let roomId: String
private var allowedParentIds: [String] = []
var preselectedItemIds: Set<String>? {
Set(allowedParentIds)
}
init(roomId: String) {
self.roomId = roomId
}
func sections(with session: MXSession, completion: @escaping (Result<[MatrixListItemSectionData], Error>) -> Void) {
guard let room = session.room(withRoomId: roomId) else {
return
}
room.state { [weak self] state in
guard let self = self else { return }
let joinRuleEvent = state?.stateEvents(with: .roomJoinRules)?.last
let allowContent: [[String: String]] = joinRuleEvent?.wireContent[kMXJoinRulesContentKeyAllow] as? [[String: String]] ?? []
self.allowedParentIds = allowContent.compactMap { allowDictionnary in
guard let type = allowDictionnary[kMXJoinRulesContentKeyType], type == kMXEventTypeStringRoomMembership else {
return nil
}
return allowDictionnary[kMXJoinRulesContentKeyRoomId]
}
let ancestorsId = session.spaceService.ancestorsPerRoomId[self.roomId] ?? []
var sections = [
MatrixListItemSectionData(
title: VectorL10n.roomAccessSpaceChooserKnownSpacesSection(room.displayName ?? ""),
items: ancestorsId.compactMap { spaceId in
guard let space = session.spaceService.getSpace(withId: spaceId) else {
return nil
}
guard let room = space.room else {
return nil
}
return MatrixListItemData(mxRoom: room, spaceService: session.spaceService)
}.sorted { $0.displayName ?? "" < $1.displayName ?? "" }
)
]
var unknownParents = self.allowedParentIds
for roomId in ancestorsId {
if let index = unknownParents.firstIndex(of: roomId) {
unknownParents.remove(at: index)
}
}
if !unknownParents.isEmpty {
sections.append(MatrixListItemSectionData(
title: VectorL10n.roomAccessSpaceChooserOtherSpacesSection,
infoText: VectorL10n.roomAccessSpaceChooserOtherSpacesSectionInfo(room.displayName ?? ""),
items: unknownParents.compactMap { roomId in
MatrixListItemData(
id: roomId,
type: .space,
avatar: AvatarInput(mxContentUri: roomId, matrixItemId: roomId, displayName: roomId),
displayName: roomId,
detailText: nil
)
}
))
}
completion(Result(catching: {
sections
}))
}
}
}