linphone-iphone/Linphone/Utils/MagicSearchSingleton.swift

/*
 * Copyright (c) 2010-2023 Belledonne Communications SARL.
 *
 * This file is part of linphone-iphone
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

import linphonesw
import Combine
import SwiftUI

final class MagicSearchSingleton: ObservableObject {
	
	static let shared = MagicSearchSingleton()
	private var coreContext = CoreContext.shared
	private var contactsManager = ContactsManager.shared
	
	private var magicSearch: MagicSearch?
	
	var currentFilter: String = ""
	var previousFilter: String?
	
	var needUpdateLastSearchContacts = false
	
	private var limitSearchToLinphoneAccounts = true
	
	@Published var allContact = false
	let allContactKey = "all_contact"
	
	var domainDefaultAccount = ""
	
	var searchDelegate: MagicSearchDelegate?
    
    private var contactLoadedDebounceWorkItem: DispatchWorkItem?
    
    let nativeAddressBookFriendList = "Native address-book"
    let linphoneAddressBookFriendList = "Linphone address-book"
    let tempRemoteAddressBookFriendList = "TempRemoteDirectoryContacts address-book"
	
	@Published var isLoading = false
	
	func destroyMagicSearch() {
		magicSearch = nil
	}
	
	private init() {
		let preferences = UserDefaults.standard
		if preferences.object(forKey: allContactKey) == nil {
			preferences.set(allContact, forKey: allContactKey)
		} else {
			allContact = preferences.bool(forKey: allContactKey)
		}
		
		coreContext.doOnCoreQueue { core in
			self.domainDefaultAccount = (core.defaultAccount?.params?.domain?.contains("sip.linphone.org") == true) ? (core.defaultAccount?.params?.domain ?? "") : "*"
			
			self.magicSearch = try? core.createMagicSearch()
			
			guard let magicSearch = self.magicSearch else {
				return
			}
			
			magicSearch.limitedSearch = false
			
			self.searchDelegate = MagicSearchDelegateStub(onSearchResultsReceived: { (magicSearch: MagicSearch) in
				print("[MagicSearchSingleton] [onSearchResultsReceived] Received search results")
				self.needUpdateLastSearchContacts = true
				
				var lastSearchFriend: [SearchResult] = []
				var lastSearchSuggestions: [SearchResult] = []
				
				magicSearch.lastSearch.forEach { searchResult in
                    if searchResult.friend != nil && (searchResult.friend?.friendList?.displayName == self.nativeAddressBookFriendList || searchResult.friend?.friendList?.displayName == self.linphoneAddressBookFriendList || searchResult.friend?.friendList?.displayName == self.tempRemoteAddressBookFriendList) {
						if let address = searchResult.address,
						   !lastSearchFriend.contains(where: { $0.address?.weakEqual(address2: address) ?? false }) {
							lastSearchFriend.append(searchResult)
						} else if let phoneNumber = searchResult.phoneNumber,
								  !lastSearchFriend.contains(where: { $0.phoneNumber == phoneNumber }) {
							lastSearchFriend.append(searchResult)
						}
					} else if searchResult.friend != nil && (searchResult.hasSourceFlag(source: .RemoteCardDAV) || searchResult.friend?.friendList?.type == .CardDAV || searchResult.hasSourceFlag(source: .LdapServers)) {
						lastSearchFriend.append(searchResult)
					} else {
						lastSearchSuggestions.append(searchResult)
					}
				}
				
				lastSearchSuggestions.sort(by: {
					($0.address?.asStringUriOnly() ?? "") < ($1.address?.asStringUriOnly() ?? "")
				})
				
				if let defaultAccount = core.defaultAccount, let contactAddress = defaultAccount.contactAddress {
					lastSearchSuggestions.removeAll {
						$0.address?.weakEqual(address2: contactAddress) ?? false
					}
				}
				
				let sortedLastSearch = lastSearchFriend.sorted {
					let name1 = $0.friend?.name?.lowercased()
						.folding(options: .diacriticInsensitive, locale: .current) ?? ""
					let name2 = $1.friend?.name?.lowercased()
						.folding(options: .diacriticInsensitive, locale: .current) ?? ""
					return name1 < name2
				}
				
				var addedAvatarListModel: [ContactAvatarModel] = []
				sortedLastSearch.forEach { searchResult in
					if searchResult.friend != nil {
                        if (searchResult.friend?.friendList?.displayName == self.nativeAddressBookFriendList || searchResult.friend?.friendList?.displayName == self.linphoneAddressBookFriendList || searchResult.friend?.friendList?.displayName == self.tempRemoteAddressBookFriendList) {
                            addedAvatarListModel.append(
                                ContactAvatarModel(
                                    friend: searchResult.friend!,
                                    name: searchResult.friend?.name ?? "",
                                    address: searchResult.friend?.address?.clone()?.asStringUriOnly() ?? "",
                                    withPresence: true
                                )
                            )
						} else if searchResult.hasSourceFlag(source: .RemoteCardDAV) || searchResult.friend?.friendList?.type == .CardDAV {
							addedAvatarListModel.append(
								ContactAvatarModel(
									friend: searchResult.friend!,
									name: searchResult.friend?.name ?? "",
									address: searchResult.friend?.address?.clone()?.asStringUriOnly() ?? "",
									withPresence: true
								)
							)
						} else if searchResult.hasSourceFlag(source: .LdapServers) {
							addedAvatarListModel.append(
								ContactAvatarModel(
									friend: searchResult.friend!,
									name: searchResult.friend?.name ?? "",
									address: searchResult.friend?.address?.clone()?.asStringUriOnly() ?? "",
									withPresence: false
								)
							)
						}
					}
				}
				
				self.contactsManager.avatarListModel.forEach { contactAvatarModel in
					contactAvatarModel.removeFriendDelegate()
				}
                
                self.updateContacts(sortedLastSearch: sortedLastSearch, lastSearchSuggestions: lastSearchSuggestions, addedAvatarListModel: addedAvatarListModel)
			})
			
			magicSearch.addDelegate(delegate: self.searchDelegate!)
		}
	}
	
	func changeAllContact(allContactBool: Bool) {
		let preferences = UserDefaults.standard
		
		allContact = allContactBool
		preferences.set(allContact, forKey: allContactKey)
	}
    
    func updateContacts(
        sortedLastSearch: [SearchResult],
        lastSearchSuggestions: [SearchResult],
        addedAvatarListModel: [ContactAvatarModel]
    ) {
        DispatchQueue.main.async {			
			if SharedMainViewModel.shared.displayedFriend != nil {
				if let avatarModel = addedAvatarListModel.first(where: { $0.address == SharedMainViewModel.shared.displayedFriend?.address }) {
					SharedMainViewModel.shared.displayedFriend = avatarModel
				}
			}
			
            self.contactsManager.lastSearch = sortedLastSearch
            self.contactsManager.lastSearchSuggestions = lastSearchSuggestions
            
            self.contactsManager.avatarListModel.removeAll()
            self.contactsManager.avatarListModel += addedAvatarListModel

            // Cancel previous debounce task
            self.contactLoadedDebounceWorkItem?.cancel()

            // Schedule new debounce task
            let workItem = DispatchWorkItem {
                NotificationCenter.default.post(name: NSNotification.Name("ContactLoaded"), object: nil)
            }
			
			self.isLoading = false

            self.contactLoadedDebounceWorkItem = workItem
            DispatchQueue.main.asyncAfter(deadline: .now() + 0.5, execute: workItem)
        }
    }
	
	func searchForContacts() {
		coreContext.doOnCoreQueue { _ in
			DispatchQueue.main.async {
				self.isLoading = true
			}
			
			var needResetCache = false
			
			if let oldFilter = self.previousFilter {
				if oldFilter.count > self.currentFilter.count || oldFilter != self.currentFilter {
					needResetCache = true
				}
			}
			
			self.previousFilter = self.currentFilter
			
			guard let magicSearch = self.magicSearch else {
				return
			}
			
			if needResetCache {
				magicSearch.resetSearchCache()
			}
			
			magicSearch.getContactsListAsync(
				filter: self.currentFilter,
				domain: self.allContact ? "" : self.domainDefaultAccount,
				sourceFlags: MagicSearch.Source.All.rawValue, //MagicSearch.Source.Friends.rawValue | MagicSearch.Source.LdapServers.rawValue | MagicSearch.Source.RemoteCardDAV.rawValue,
				aggregation: MagicSearch.Aggregation.Friend
			)
		}
	}
}