import Combine import Foundation import GRDB import Martin enum ClientState: Equatable { enum ClientConnectionState { case connected case disconnected } case disabled case enabled(ClientConnectionState) } final class Client: ObservableObject { @Published private(set) var state: ClientState = .enabled(.disconnected) @Published private(set) var credentials: Credentials var rosters: [Roster] { get async { // Fetching only non-locally-deleted rosters and only for active credentials if credentials.isActive { let creds = credentials let rosters = try? await Database.shared.dbQueue.read { db in try Roster .filter(Column("bareJid") == creds.bareJid) .filter(Column("locallyDeleted") == false) .fetchAll(db) } return rosters ?? [] } else { return [] } } } private var connection: XMPPClient private var connectionCancellable: AnyCancellable? private var rosterManager = ClientMartinRosterManager() init(credentials: Credentials) { self.credentials = credentials state = credentials.isActive ? .enabled(.disconnected) : .disabled connection = Self.prepareConnection(credentials, rosterManager) connectionCancellable = connection.$state .sink { [weak self] state in guard let self = self else { return } guard self.credentials.isActive else { self.state = .disabled return } switch state { case .connected: self.state = .enabled(.connected) default: self.state = .enabled(.disconnected) } } } } extension Client { func addRoster(_ jid: String, name: String?, groups: [String]) async throws { _ = try await connection.module(.roster).addItem( jid: JID(jid), name: name, groups: groups ) } func deleteRoster(_ roster: Roster) async throws { _ = try await connection.module(.roster).removeItem(jid: JID(roster.contactBareJid)) } func connect() async { guard credentials.isActive, state == .enabled(.disconnected) else { return } try? await connection.loginAndWait() } func disconnect() { _ = connection.disconnect() } } extension Client { static func tryLogin(with credentials: Credentials) async throws -> Client { let client = Client(credentials: credentials) try await client.connection.loginAndWait() return client } } private extension Client { static func prepareConnection(_ credentials: Credentials, _ roster: RosterManager) -> XMPPClient { let client = XMPPClient() // register modules // core modules RFC 6120 client.modulesManager.register(StreamFeaturesModule()) client.modulesManager.register(SaslModule()) client.modulesManager.register(AuthModule()) client.modulesManager.register(SessionEstablishmentModule()) client.modulesManager.register(ResourceBinderModule()) client.modulesManager.register(DiscoveryModule(identity: .init(category: "client", type: "iOS", name: Const.appName))) // messaging modules RFC 6121 client.modulesManager.register(RosterModule(rosterManager: roster)) client.modulesManager.register(PresenceModule()) // client.modulesManager.register(PubSubModule()) // client.modulesManager.register(MessageModule(chatManager: manager)) // client.modulesManager.register(MessageArchiveManagementModule()) // client.modulesManager.register(MessageCarbonsModule()) // file transfer modules // client.modulesManager.register(HttpFileUploadModule()) // extensions client.modulesManager.register(SoftwareVersionModule()) client.modulesManager.register(PingModule()) client.connectionConfiguration.userJid = .init(credentials.bareJid) client.connectionConfiguration.credentials = .password(password: credentials.pass) // group chats // client.modulesManager.register(MucModule(roomManager: manager)) // channels // client.modulesManager.register(MixModule(channelManager: manager)) // add client to clients return client } }