deltachat-ios/deltachat-ios/ViewModel/ContactCellViewModel.swift

import Foundation
import DcCore

protocol AvatarCellViewModel {
    var dcContext: DcContext { get }
    var type: CellModel { get }
    var title: String { get }
    var titleHighlightIndexes: [Int] { get }
    var subtitle: String { get }
    var subtitleHighlightIndexes: [Int] { get }
}

enum CellModel {
    case contact(ContactCellData)
    case chat(ChatCellData)
    case profile
}

struct ContactCellData {
    let contactId: Int
    let chatId: Int?
}

struct ChatCellData {
    let chatId: Int
    let highlightMsgId: Int?
    let summary: DcLot
    let unreadMessages: Int
}

class ContactCellViewModel: AvatarCellViewModel {

    private let contact: DcContact
    let dcContext: DcContext

    var type: CellModel
    var title: String {
        return contact.displayName
    }
    var subtitle: String {
        return contact.email
    }

    var avartarTitle: String {
        return DcUtils.getInitials(inputName: title)
    }

    var titleHighlightIndexes: [Int]
    var subtitleHighlightIndexes: [Int]

    init(dcContext: DcContext, contactData: ContactCellData, titleHighlightIndexes: [Int] = [], subtitleHighlightIndexes: [Int] = []) {
        type = CellModel.contact(contactData)
        self.titleHighlightIndexes = titleHighlightIndexes
        self.subtitleHighlightIndexes = subtitleHighlightIndexes
        self.contact = dcContext.getContact(id: contactData.contactId)
        self.dcContext = dcContext
    }
}

class ProfileViewModel: AvatarCellViewModel {

    let dcContext: DcContext
    var type: CellModel {
        return CellModel.profile
    }

    var title: String

    private let contact: DcContact

    var titleHighlightIndexes: [Int] {
        return []
    }

    var subtitle: String

    var subtitleHighlightIndexes: [Int] {
        return []
    }

    init(context: DcContext) {
        self.dcContext = context
        contact = context.getContact(id: Int(DC_CONTACT_ID_SELF))
        title = context.displayname ?? String.localized("pref_your_name")
        subtitle = context.addr ?? ""
    }
}

class ChatCellViewModel: AvatarCellViewModel {

    private let chat: DcChat
    let dcContext: DcContext

    private var summary: DcLot

    var type: CellModel
    var title: String {
        return chat.name
    }

    var subtitle: String {
        let result1 = summary.text1 ?? ""
        let result2 = summary.text2 ?? ""
        let result: String
        if !result1.isEmpty, !result2.isEmpty {
            result = "\(result1): \(result2)"
        } else {
            result = "\(result1)\(result2)"
        }
        return result
    }

    var titleHighlightIndexes: [Int]
    var subtitleHighlightIndexes: [Int]

    init(dcContext: DcContext, chatData: ChatCellData, titleHighlightIndexes: [Int] = [], subtitleHighlightIndexes: [Int] = []) {
        self.type = CellModel.chat(chatData)
        self.titleHighlightIndexes = titleHighlightIndexes
        self.subtitleHighlightIndexes = subtitleHighlightIndexes
        self.summary = chatData.summary
        self.chat = dcContext.getChat(chatId: chatData.chatId)
        self.dcContext = dcContext
    }
}

extension ContactCellViewModel {
    static func make(contactId: Int, searchText: String?, dcContext: DcContext) -> ContactCellViewModel {
        let contact = dcContext.getContact(id: contactId)
        let nameIndexes = contact.displayName.containsExact(subSequence: searchText)
        let emailIndexes = contact.email.containsExact(subSequence: searchText)
        let chatId: Int? = dcContext.getChatIdByContactIdOld(contactId)
            // contact contains searchText
        let viewModel = ContactCellViewModel(
            dcContext: dcContext,
            contactData: ContactCellData(
                contactId: contactId,
                chatId: chatId
            ),
            titleHighlightIndexes: nameIndexes,
            subtitleHighlightIndexes: emailIndexes
        )
        return viewModel
    }
}