another.im-ios/AnotherIM/AppData/Model/Chat.swift

import Foundation
import GRDB

enum ConversationType: Int, Codable, DatabaseValueConvertible {
    case chat = 0
    case room = 1
    case channel = 2
}

struct Chat: DBStorable {
    static let databaseTableName = "chats"

    var id: String
    var account: String
    var participant: String
    var type: ConversationType
    var encrypted: Bool
}

extension Chat: Equatable {}

extension Chat {
    func fetchRoster() async throws -> Roster {
        try await Database.shared.dbQueue.read { db in
            guard
                let roster = try Roster
                    .filter(Column("bareJid") == account && Column("contactBareJid") == participant)
                    .fetchOne(db)
            else {
                throw AppError.rosterNotFound
            }
            return roster
        }
    }

    func setEncrypted(_ encrypted: Bool) async throws {
        try await Database.shared.dbQueue.write { db in
            var chat = self
            chat.encrypted = encrypted
            try chat.update(db)
        }
    }
}