140 lines
4.5 KiB
Swift
140 lines
4.5 KiB
Swift
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
|
|
}
|
|
}
|