another.im-ios/ConversationsClassic/AppData/Client/Client+MartinMessages.swift

import Combine
import Foundation
import GRDB
import Martin

final class ClientMartinMessagesManager {
    private var cancellables: Set<AnyCancellable> = []

    init(_ xmppConnection: XMPPClient) {
        xmppConnection.module(MessageModule.self).messagesPublisher
            .sink { [weak self] message in
                self?.handleMessage(message)
            }
            .store(in: &cancellables)
    }

    private func handleMessage(_ received: Martin.MessageModule.MessageReceived) {
        let message = received.message
        let chat = received.chat
        #if DEBUG
            print("---")
            print("Message received: \(received)")
            print("Chat: \(chat)")
            print("---")
        #endif

        // Try to recognize if message is omemo-encoded and decode it
        // if let omemo = OMEMO.decode(message: message, chat: chat) {
        //     // If it is omemo-encoded, save it and return
        //     saveMessage(omemo)
        //     return
        // }

        if let msg = Message.map(message) {
            Task {
                do {
                    try await msg.save()
                } catch {
                    logIt(.error, "Error saving message: \(error)")
                }
            }
        }
    }
}