conversations-classic-ios/ConversationsClassic/AppCore/Middlewares/DatabaseMiddleware.swift

import Combine
import Foundation
import GRDB

// swiftlint:disable:next type_body_length
final class DatabaseMiddleware {
    static let shared = DatabaseMiddleware()
    private let database = Database.shared
    private var cancellables: Set<AnyCancellable> = []
    private var conversationCancellables: Set<AnyCancellable> = []

    private init() {
        // Database changes
        ValueObservation
            .tracking(Roster.fetchAll)
            .publisher(in: database._db, scheduling: .immediate)
            .sink { _ in
                // Handle completion
            } receiveValue: { rosters in
                DispatchQueue.main.async {
                    store.dispatch(.databaseAction(.storedRostersLoaded(rosters: rosters)))
                }
            }
            .store(in: &cancellables)
        ValueObservation
            .tracking(Chat.fetchAll)
            .publisher(in: database._db, scheduling: .immediate)
            .sink { _ in
                // Handle completion
            } receiveValue: { chats in
                DispatchQueue.main.async {
                    store.dispatch(.databaseAction(.storedChatsLoaded(chats: chats)))
                }
            }
            .store(in: &cancellables)
    }

    // swiftlint:disable:next function_body_length
    func middleware(state _: AppState, action: AppAction) -> AnyPublisher<AppAction, Never> {
        switch action {
        // MARK: Accounts
        case .startAction(.loadStoredAccounts):
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.databaseAction(.loadingStoredAccountsFailed)))
                        return
                    }
                    do {
                        try database._db.read { db in
                            let accounts = try Account.fetchAll(db)
                            promise(.success(.databaseAction(.storedAccountsLoaded(accounts: accounts))))
                        }
                    } catch {
                        promise(.success(.databaseAction(.loadingStoredAccountsFailed)))
                    }
                }
            }
            .eraseToAnyPublisher()

        case .accountsAction(.makeAccountPermanent(let account)):
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.databaseAction(.updateAccountFailed)))
                        return
                    }
                    do {
                        try database._db.write { db in
                            // make permanent and store to database
                            var acc = account
                            acc.isTemp = false
                            try acc.insert(db)

                            // Re-Fetch all accounts
                            let accounts = try Account.fetchAll(db)

                            // Use the accounts
                            promise(.success(.databaseAction(.storedAccountsLoaded(accounts: accounts))))
                        }
                    } catch {
                        promise(.success(.databaseAction(.updateAccountFailed)))
                    }
                }
            }
            .eraseToAnyPublisher()

        // MARK: Rosters
        case .rostersAction(.markRosterAsLocallyDeleted(let ownerJID, let contactJID)):
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.rostersAction(.rosterDeletingFailed(reason: L10n.Global.Error.genericDbError))))
                        return
                    }
                    do {
                        _ = try database._db.write { db in
                            try Roster
                                .filter(Column("bareJid") == ownerJID)
                                .filter(Column("contactBareJid") == contactJID)
                                .updateAll(db, Column("locallyDeleted").set(to: true))
                        }
                        promise(.success(.empty))
                    } catch {
                        promise(.success(.rostersAction(.rosterDeletingFailed(reason: L10n.Global.Error.genericDbError))))
                    }
                }
            }
            .eraseToAnyPublisher()

        case .rostersAction(.unmarkRosterAsLocallyDeleted(let ownerJID, let contactJID)):
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.rostersAction(.rosterDeletingFailed(reason: L10n.Global.Error.genericDbError))))
                        return
                    }
                    do {
                        _ = try database._db.write { db in
                            try Roster
                                .filter(Column("bareJid") == ownerJID)
                                .filter(Column("contactBareJid") == contactJID)
                                .updateAll(db, Column("locallyDeleted").set(to: false))
                        }
                        promise(.success(.empty))
                    } catch {
                        promise(.success(.rostersAction(.rosterDeletingFailed(reason: L10n.Global.Error.genericDbError))))
                    }
                }
            }
            .eraseToAnyPublisher()

        // MARK: Chats
        case .chatsAction(.createNewChat(let accountJid, let participantJid)):
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.chatsAction(.chatCreationFailed(reason: L10n.Global.Error.genericDbError))))
                        return
                    }
                    do {
                        try database._db.write { db in
                            let chat = Chat(
                                id: UUID().uuidString,
                                account: accountJid,
                                participant: participantJid,
                                type: .chat
                            )
                            try chat.insert(db)
                            promise(.success(.chatsAction(.chatCreated(chat: chat))))
                        }
                    } catch {
                        promise(.success(.chatsAction(.chatCreationFailed(reason: L10n.Global.Error.genericDbError))))
                    }
                }
            }
            .eraseToAnyPublisher()

        // MARK: Conversation and messages
        case .conversationAction(.makeConversationActive(let chat, _)):
            subscribeToMessages(chat: chat)
            return Empty().eraseToAnyPublisher()

        case .xmppAction(.xmppMessageReceived(let message)):
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.databaseAction(.storeMessageFailed(reason: L10n.Global.Error.genericDbError))))
                        return
                    }
                    guard message.contentType != .typing, message.body != nil else {
                        promise(.success(.empty))
                        return
                    }
                    do {
                        try database._db.write { db in
                            try message.insert(db)
                        }
                        promise(.success(.empty))
                    } catch {
                        promise(.success(.databaseAction(.storeMessageFailed(reason: error.localizedDescription))))
                    }
                }
            }
            .eraseToAnyPublisher()

        case .conversationAction(.sendMessage(let from, let to, let body)):
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.databaseAction(.storeMessageFailed(reason: L10n.Global.Error.genericDbError))))
                        return
                    }
                    do {
                        let message = Message(
                            id: UUID().uuidString,
                            type: .chat,
                            contentType: .text,
                            from: from,
                            to: to,
                            body: body,
                            subject: nil,
                            thread: nil,
                            oobUrl: nil,
                            date: Date(),
                            pending: true,
                            sentError: false
                        )
                        try database._db.write { db in
                            try message.insert(db)
                        }
                        promise(.success(.xmppAction(.xmppMessageSent(message))))
                    } catch {
                        promise(.success(.databaseAction(.storeMessageFailed(reason: error.localizedDescription))))
                    }
                }
            }
            .eraseToAnyPublisher()

        case .xmppAction(.xmppMessageSendSuccess(let msgId)):
            // mark message as pending false and sentError false
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.databaseAction(.storeMessageFailed(reason: L10n.Global.Error.genericDbError))))
                        return
                    }
                    do {
                        _ = try database._db.write { db in
                            try Message
                                .filter(Column("id") == msgId)
                                .updateAll(db, Column("pending").set(to: false), Column("sentError").set(to: false))
                        }
                        promise(.success(.empty))
                    } catch {
                        promise(.success(.databaseAction(.storeMessageFailed(reason: error.localizedDescription)))
                        )
                    }
                }
            }
            .eraseToAnyPublisher()

        case .xmppAction(.xmppMessageSendFailed(let msgId)):
            // mark message as pending false and sentError true
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.databaseAction(.storeMessageFailed(reason: L10n.Global.Error.genericDbError))))
                        return
                    }
                    do {
                        _ = try database._db.write { db in
                            try Message
                                .filter(Column("id") == msgId)
                                .updateAll(db, Column("pending").set(to: false), Column("sentError").set(to: true))
                        }
                        promise(.success(.empty))
                    } catch {
                        promise(.success(.databaseAction(.storeMessageFailed(reason: error.localizedDescription))))
                    }
                }
            }
            .eraseToAnyPublisher()

        // MARK: Attachments
        case .fileAction(.downloadingAttachmentFileFailed(let id, _)):
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.databaseAction(.updateAttachmentFailed(id: id, reason: L10n.Global.Error.genericDbError)))
                        )
                        return
                    }
                    do {
                        _ = try database._db.write { db in
                            try Message
                                .filter(Column("id") == id)
                                .updateAll(db, Column("attachmentDownloadFailed").set(to: true))
                        }
                        promise(.success(.empty))
                    } catch {
                        promise(.success(.databaseAction(.updateAttachmentFailed(id: id, reason: error.localizedDescription)))
                        )
                    }
                }
            }
            .eraseToAnyPublisher()

        case .fileAction(.attachmentFileDownloaded(let id, let localUrl)):
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.databaseAction(.updateAttachmentFailed(id: id, reason: L10n.Global.Error.genericDbError)))
                        )
                        return
                    }
                    do {
                        _ = try database._db.write { db in
                            try Message
                                .filter(Column("id") == id)
                                .updateAll(db, Column("attachmentLocalPath").set(to: localUrl))
                        }
                        promise(.success(.empty))
                    } catch {
                        promise(.success(.databaseAction(.updateAttachmentFailed(id: id, reason: error.localizedDescription)))
                        )
                    }
                }
            }
            .eraseToAnyPublisher()

        case .fileAction(.attachmentThumbnailCreated(let id, let thumbnailUrl)):
            return Future<AppAction, Never> { promise in
                Task(priority: .background) { [weak self] in
                    guard let database = self?.database else {
                        promise(.success(.databaseAction(.updateAttachmentFailed(id: id, reason: L10n.Global.Error.genericDbError)))
                        )
                        return
                    }
                    do {
                        _ = try database._db.write { db in
                            try Message
                                .filter(Column("id") == id)
                                .updateAll(db, Column("attachmentThumbnailPath").set(to: thumbnailUrl))
                        }
                        promise(.success(.empty))
                    } catch {
                        promise(.success(.databaseAction(.updateAttachmentFailed(id: id, reason: error.localizedDescription)))
                        )
                    }
                }
            }
            .eraseToAnyPublisher()

        default:
            return Empty().eraseToAnyPublisher()
        }
    }
}

private extension DatabaseMiddleware {
    func subscribeToMessages(chat: Chat) {
        conversationCancellables = []
        ValueObservation
            .tracking(
                Message
                    .filter(
                        (Column("to") == chat.account && Column("from") == chat.participant) ||
                            (Column("from") == chat.account && Column("to") == chat.participant)
                    )
                    .order(Column("date").desc)
                    .fetchAll
            )
            .publisher(in: database._db, scheduling: .immediate)
            .sink { _ in
            } receiveValue: { messages in
                // messages
                DispatchQueue.main.async {
                    store.dispatch(.conversationAction(.messagesUpdated(messages: messages)))
                }
            }
            .store(in: &conversationCancellables)
    }
}