123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373 |
- import UIKit
- import DcCore
- import Intents
- // MARK: - ChatListViewModel
- class ChatListViewModel: NSObject {
- var onChatListUpdate: VoidFunction?
- private var inBgSearch = false
- private var needsAnotherBgSearch = false
- enum ChatListSectionType {
- case chats
- case contacts
- case messages
- }
- private(set) public var isArchive: Bool
- private let dcContext: DcContext
- private(set) public var searchActive: Bool = false
- // if searchfield is empty we show default chat list
- private var showSearchResults: Bool {
- return searchActive && searchText.containsCharacters()
- }
- private var chatList: DcChatlist!
- // for search filtering
- private var searchText: String = ""
- private var searchResultChatList: DcChatlist?
- private var searchResultContactIds: [Int] = []
- private var searchResultMessageIds: [Int] = []
- // to manage sections dynamically
- private var searchResultsChatsSection: ChatListSectionType = .chats
- private var searchResultsContactsSection: ChatListSectionType = .contacts
- private var searchResultsMessagesSection: ChatListSectionType = .messages
- private var searchResultSections: [ChatListSectionType] = []
- init(dcContext: DcContext, isArchive: Bool) {
- self.isArchive = isArchive
- self.dcContext = dcContext
- super.init()
- updateChatList(notifyListener: true)
- }
- private func updateChatList(notifyListener: Bool) {
- var gclFlags: Int32 = 0
- if isArchive {
- gclFlags |= DC_GCL_ARCHIVED_ONLY
- } else if RelayHelper.sharedInstance.isForwarding() {
- gclFlags |= DC_GCL_FOR_FORWARDING
- }
- self.chatList = dcContext.getChatlist(flags: gclFlags, queryString: nil, queryId: 0)
- if notifyListener, let onChatListUpdate = onChatListUpdate {
- if Thread.isMainThread {
- onChatListUpdate()
- } else {
- DispatchQueue.main.async {
- onChatListUpdate()
- }
- }
- }
- }
- var numberOfSections: Int {
- if showSearchResults {
- return searchResultSections.count
- }
- return 1
- }
- func numberOfRowsIn(section: Int) -> Int {
- if showSearchResults {
- switch searchResultSections[section] {
- case .chats:
- return searchResultChatList?.length ?? 0
- case .contacts:
- return searchResultContactIds.count
- case .messages:
- return searchResultMessageIds.count
- }
- }
- return chatList.length
- }
- func cellDataFor(section: Int, row: Int) -> AvatarCellViewModel {
- if showSearchResults {
- switch searchResultSections[section] {
- case .chats:
- return makeChatCellViewModel(index: row, searchText: searchText)
- case .contacts:
- if row >= 0 && row < searchResultContactIds.count {
- return ContactCellViewModel.make(contactId: searchResultContactIds[row], searchText: searchText, dcContext: dcContext)
- } else {
- logger.warning("search: requested contact index \(row) not in range 0..\(searchResultContactIds.count)")
- }
- case .messages:
- if row >= 0 && row < searchResultMessageIds.count {
- return makeMessageCellViewModel(msgId: searchResultMessageIds[row])
- } else {
- logger.warning("search: requested message index \(row) not in range 0..\(searchResultMessageIds.count)")
- }
- }
- }
- return makeChatCellViewModel(index: row, searchText: "")
- }
- // only visible on search results
- func titleForHeaderIn(section: Int) -> String? {
- if showSearchResults {
- switch searchResultSections[section] {
- case .chats:
- return String.localized(stringID: "n_chats", count: numberOfRowsIn(section: section))
- case .contacts:
- return String.localized(stringID: "n_contacts", count: numberOfRowsIn(section: section))
- case .messages:
- let count = numberOfRowsIn(section: section)
- var ret = String.localized(stringID: "n_messages", count: count)
- if count==1000 {
- // a count of 1000 results may be limited, see documentation of dc_search_msgs()
- // (formatting may be "1.000" or "1,000", so just skip the first digit)
- ret = ret.replacingOccurrences(of: "000", with: "000+")
- }
- return ret
- }
- }
- return nil
- }
- func chatIdFor(section: Int, row: Int) -> Int? {
- let cellData = cellDataFor(section: section, row: row)
- switch cellData.type {
- case .chat(let data):
- return data.chatId
- case .contact:
- return nil
- case .profile:
- return nil
- }
- }
- func msgIdFor(row: Int) -> Int? {
- if showSearchResults {
- return nil
- }
- return chatList.getMsgId(index: row)
- }
- func refreshData() {
- updateChatList(notifyListener: true)
- }
- func beginSearch() {
- searchActive = true
- }
- func endSearch() {
- searchActive = false
- searchText = ""
- resetSearch()
- }
- var emptySearchText: String? {
- if searchActive && numberOfSections == 0 {
- return searchText
- }
- return nil
- }
- func deleteChat(chatId: Int) {
- dcContext.deleteChat(chatId: chatId)
- NotificationManager.removeNotificationsForChat(dcContext: dcContext, chatId: chatId)
- INInteraction.delete(with: ["\(dcContext.id).\(chatId)"])
- }
- func archiveChatToggle(chatId: Int) {
- let chat = dcContext.getChat(chatId: chatId)
- let isArchivedBefore = chat.isArchived
- dcContext.archiveChat(chatId: chatId, archive: !isArchivedBefore)
- if !isArchivedBefore {
- NotificationManager.removeNotificationsForChat(dcContext: dcContext, chatId: chatId)
- }
- updateChatList(notifyListener: false)
- }
- func pinChatToggle(chatId: Int) {
- let chat: DcChat = dcContext.getChat(chatId: chatId)
- let pinned = chat.visibility==DC_CHAT_VISIBILITY_PINNED
- self.dcContext.setChatVisibility(chatId: chatId, visibility: pinned ? DC_CHAT_VISIBILITY_NORMAL : DC_CHAT_VISIBILITY_PINNED)
- updateChatList(notifyListener: false)
- }
- }
- private extension ChatListViewModel {
- // MARK: - avatarCellViewModel factory
- func makeChatCellViewModel(index: Int, searchText: String) -> AvatarCellViewModel {
- let list: DcChatlist = searchResultChatList ?? chatList
- let chatId = list.getChatId(index: index)
- let summary = list.getSummary(index: index)
- let chat = dcContext.getChat(chatId: chatId)
- let unreadMessages = dcContext.getUnreadMessages(chatId: chatId)
- var chatTitleIndexes: [Int] = []
- if searchText.containsCharacters() {
- let chatName = chat.name
- chatTitleIndexes = chatName.containsExact(subSequence: searchText)
- }
- let viewModel = ChatCellViewModel(
- dcContext: dcContext,
- chatData: ChatCellData(
- chatId: chatId,
- highlightMsgId: nil,
- summary: summary,
- unreadMessages: unreadMessages
- ),
- titleHighlightIndexes: chatTitleIndexes
- )
- return viewModel
- }
- func makeMessageCellViewModel(msgId: Int) -> AvatarCellViewModel {
- let msg: DcMsg = dcContext.getMessage(id: msgId)
- let chatId: Int = msg.chatId
- let chat: DcChat = dcContext.getChat(chatId: chatId)
- if !chat.isValid {
- // chat might be deleted and searchResultMessageIds deprecated, so return a dummy view model
- // cleanup of the searchResultMessageIds will be done in message change event handling
- return ChatCellViewModel(
- dcContext: dcContext,
- chatData: ChatCellData(
- chatId: 0,
- highlightMsgId: 0,
- summary: DcLot(nil),
- unreadMessages: 0
- )
- )
- }
- let summary: DcLot = msg.summary(chat: chat)
- let unreadMessages = dcContext.getUnreadMessages(chatId: chatId)
- let viewModel = ChatCellViewModel(
- dcContext: dcContext,
- chatData: ChatCellData(
- chatId: chatId,
- highlightMsgId: msgId,
- summary: summary,
- unreadMessages: unreadMessages
- )
- )
- let subtitle = viewModel.subtitle
- viewModel.subtitleHighlightIndexes = subtitle.containsExact(subSequence: searchText)
- return viewModel
- }
- // MARK: - search
- private func updateSearchResultSections() {
- var sections: [ChatListSectionType] = []
- if let chatList = searchResultChatList, chatList.length > 0 {
- sections.append(searchResultsChatsSection)
- }
- if !searchResultContactIds.isEmpty {
- sections.append(searchResultsContactsSection)
- }
- if !searchResultMessageIds.isEmpty {
- sections.append(searchResultsMessagesSection)
- }
- searchResultSections = sections
- }
- private func resetSearch() {
- searchResultChatList = nil
- searchResultContactIds = []
- searchResultMessageIds = []
- updateSearchResultSections()
- }
- private func filterContentForSearchText(_ searchText: String) {
- if !searchText.isEmpty {
- filterAndUpdateList(searchText: searchText)
- } else {
- // when search input field empty we show default chatList
- resetSearch()
- }
- if let onChatListUpdate = onChatListUpdate {
- if Thread.isMainThread {
- onChatListUpdate()
- } else {
- DispatchQueue.main.async {
- onChatListUpdate()
- }
- }
- }
- }
- func filterAndUpdateList(searchText: String) {
- var overallCnt = 0
- // #1 chats with searchPattern in title bar
- searchResultChatList = dcContext.getChatlist(flags: DC_GCL_NO_SPECIALS, queryString: searchText, queryId: 0)
- if let chatlist = searchResultChatList {
- overallCnt += chatlist.length
- }
- // #2 contacts with searchPattern in name or in email
- if searchText != self.searchText && overallCnt > 0 {
- logger.info("... skipping getContacts and searchMessages, more recent search pending")
- searchResultContactIds = []
- searchResultMessageIds = []
- updateSearchResultSections()
- return
- }
- searchResultContactIds = dcContext.getContacts(flags: DC_GCL_ADD_SELF, queryString: searchText)
- overallCnt += searchResultContactIds.count
- // #3 messages with searchPattern (filtered by dc_core)
- if searchText != self.searchText && overallCnt > 0 {
- logger.info("... skipping searchMessages, more recent search pending")
- searchResultMessageIds = []
- updateSearchResultSections()
- return
- }
- if searchText.count <= 1 {
- logger.info("... skipping searchMessages, string too short")
- searchResultMessageIds = []
- updateSearchResultSections()
- return
- }
- searchResultMessageIds = dcContext.searchMessages(searchText: searchText)
- updateSearchResultSections()
- }
- }
- // MARK: UISearchResultUpdating
- extension ChatListViewModel: UISearchResultsUpdating {
- func updateSearchResults(for searchController: UISearchController) {
- self.searchText = searchController.searchBar.text ?? ""
- if inBgSearch {
- needsAnotherBgSearch = true
- logger.info("... search call debounced")
- } else {
- inBgSearch = true
- DispatchQueue.global(qos: .userInteractive).async { [weak self] in
- usleep(100000)
- self?.needsAnotherBgSearch = false
- self?.filterContentForSearchText(self?.searchText ?? "")
- while self?.needsAnotherBgSearch != false {
- usleep(100000)
- self?.needsAnotherBgSearch = false
- logger.info("... executing debounced search call")
- self?.filterContentForSearchText(self?.searchText ?? "")
- }
- self?.inBgSearch = false
- }
- }
- }
- }
|