113 lines
4.2 KiB
Swift
113 lines
4.2 KiB
Swift
import Combine
|
|
import Foundation
|
|
import GRDB
|
|
|
|
@MainActor
|
|
final class RostersStore: ObservableObject {
|
|
@Published private(set) var rosters: [Roster] = []
|
|
@Published private(set) var locallyDeletedRosters: [Roster] = []
|
|
|
|
private var cancellable: AnyCancellable?
|
|
|
|
init(clientsPublisher: Published<[Client]>.Publisher) {
|
|
subscribeToClientsStore(clientsPublisher: clientsPublisher)
|
|
}
|
|
|
|
private func subscribeToClientsStore(clientsPublisher: Published<[Client]>.Publisher) {
|
|
let rostersPublisher = ValueObservation.tracking(Roster.fetchAll)
|
|
.publisher(in: Database.shared.dbQueue)
|
|
.receive(on: DispatchQueue.main)
|
|
.catch { _ in Just([]) }
|
|
|
|
cancellable = clientsPublisher
|
|
.map { $0.filter { $0.state != .disabled } } // look rosters only for enabled clients
|
|
.flatMap { clients in
|
|
Publishers.MergeMany(clients.map { $0.$state })
|
|
.prepend(clients.map { $0.state })
|
|
.collect()
|
|
}
|
|
.combineLatest(rostersPublisher)
|
|
.sink { [weak self] clientStates, rosters in
|
|
self?.handleUpdates(clientStates: clientStates, rosters: rosters)
|
|
}
|
|
}
|
|
|
|
private func handleUpdates(clientStates: [ClientState], rosters: [Roster]) {
|
|
self.rosters = rosters.filter { !$0.locallyDeleted }
|
|
locallyDeletedRosters = rosters.filter { $0.locallyDeleted }
|
|
print("Client States: \(clientStates.count), Rosters: \(rosters.count)")
|
|
}
|
|
}
|
|
|
|
extension RostersStore {
|
|
enum RosterAddResult {
|
|
case success
|
|
case connectionError
|
|
case serverError
|
|
}
|
|
|
|
func addRoster(ownerJid: String, contactJID: String, name _: String?, groups _: [String]) async -> RosterAddResult {
|
|
// check if such roster was already exists or locally deleted
|
|
if var exists = rosters.first(where: { $0.bareJid == ownerJid && $0.contactBareJid == contactJID }), exists.locallyDeleted {
|
|
try? await exists.setLocallyDeleted(false)
|
|
return .success
|
|
}
|
|
|
|
// add new roster
|
|
// check that client is enabled and connected
|
|
guard let client = ClientsStore.shared.clients.first(where: { $0.credentials.bareJid == ownerJid }), client.state == .enabled(.connected) else {
|
|
return .connectionError
|
|
}
|
|
|
|
// add roster
|
|
do {
|
|
try await client.addRoster(contactJID, name: nil, groups: [])
|
|
return .success
|
|
} catch {
|
|
return .serverError
|
|
}
|
|
|
|
// guard let client = clientsStore.clients.first(where: { $0.credentials == ownerCredentials }), client.state == .enabled(.connected) else {
|
|
// router.showAlert(
|
|
// .alert,
|
|
// title: L10n.Global.Error.title,
|
|
// subtitle: L10n.Contacts.Add.connectionError
|
|
// ) {
|
|
// Button(L10n.Global.ok, role: .cancel) {}
|
|
// }
|
|
// return
|
|
// }
|
|
//
|
|
// router.showModal {
|
|
// LoadingScreen()
|
|
// }
|
|
//
|
|
// do {
|
|
// try await client.addRoster(contactJID, name: nil, groups: [])
|
|
// } catch {
|
|
// router.showAlert(
|
|
// .alert,
|
|
// title: L10n.Global.Error.title,
|
|
// subtitle: L10n.Contacts.Add.serverError
|
|
// ) {
|
|
// Button(L10n.Global.ok, role: .cancel) {}
|
|
// }
|
|
// return
|
|
// }
|
|
//
|
|
// router.dismissModal()
|
|
// router.dismissScreen()
|
|
|
|
// client.addRoster(jid: )
|
|
|
|
// guard let ownerAccount else { return }
|
|
// if let exists = store.state.rostersState.rosters.first(where: { $0.bareJid == ownerAccount.bareJid && $0.contactBareJid == contactJID }), exists.locallyDeleted {
|
|
// store.dispatch(.rostersAction(.unmarkRosterAsLocallyDeleted(ownerJID: ownerAccount.bareJid, contactJID: contactJID)))
|
|
// isPresented = false
|
|
// } else {
|
|
// isShowingLoader = true
|
|
// store.dispatch(.rostersAction(.addRoster(ownerJID: ownerAccount.bareJid, contactJID: contactJID, name: nil, groups: [])))
|
|
// }
|
|
}
|
|
}
|