提交
This commit is contained in:
启星
458
TUIKit/TUISearch/BaseDataProvider/TUISearchDataProvider.m
Normal file
458
TUIKit/TUISearch/BaseDataProvider/TUISearchDataProvider.m
Normal file
@@ -0,0 +1,458 @@
|
||||
//
|
||||
// TUISearchDataProvider.m
|
||||
// Pods
|
||||
//
|
||||
// Created by harvy on 2020/12/28.
|
||||
// Copyright © 2023 Tencent. All rights reserved.
|
||||
//
|
||||
|
||||
#import "TUISearchDataProvider.h"
|
||||
#import <TIMCommon/TIMDefine.h>
|
||||
#import <TUICore/TUIThemeManager.h>
|
||||
#import "TUISearchGroupDataProvider.h"
|
||||
#import "TUISearchResultCellModel.h"
|
||||
#import <TUICore/NSString+TUIUtil.h>
|
||||
|
||||
TUISearchParamKey TUISearchChatHistoryParamKeyConversationId = @"TUISearchChatHistoryParamKeyConversationId";
|
||||
TUISearchParamKey TUISearchChatHistoryParamKeyCount = @"TUISearchChatHistoryParamKeyCount";
|
||||
TUISearchParamKey TUISearchChatHistoryParamKeyPage = @"TUISearchChatHistoryParamKeyPage";
|
||||
NSUInteger TUISearchDefaultPageSize = 20;
|
||||
|
||||
typedef void (^TUISearchResultCallback)(BOOL succ, NSString *__nullable errMsg, NSArray<TUISearchResultCellModel *> *__nullable results);
|
||||
|
||||
@interface TUISearchDataProvider ()
|
||||
|
||||
@property(nonatomic, strong) NSMutableDictionary<NSNumber *, NSArray<TUISearchResultCellModel *> *> *resultSet;
|
||||
|
||||
@end
|
||||
|
||||
@implementation TUISearchDataProvider
|
||||
|
||||
- (void)searchForKeyword:(NSString *)keyword forModules:(TUISearchResultModule)modules param:(NSDictionary<TUISearchParamKey, id> *__nullable)param {
|
||||
__weak typeof(self) weakSelf = self;
|
||||
if (![NSThread isMainThread]) {
|
||||
dispatch_async(dispatch_get_main_queue(), ^{
|
||||
[weakSelf searchForKeyword:keyword forModules:modules param:param];
|
||||
});
|
||||
return;
|
||||
}
|
||||
|
||||
if (keyword.length == 0) {
|
||||
[self.resultSet removeAllObjects];
|
||||
if ([self.delegate respondsToSelector:@selector(onSearchResults:forModules:)]) {
|
||||
[self.delegate onSearchResults:self.resultSet forModules:modules];
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
dispatch_group_t group = dispatch_group_create();
|
||||
BOOL request = NO;
|
||||
|
||||
// Contact
|
||||
if ((modules == TUISearchResultModuleAll) || (modules & TUISearchResultModuleContact)) {
|
||||
request = YES;
|
||||
dispatch_group_enter(group);
|
||||
[self searchContacts:keyword
|
||||
callback:^(BOOL succ, NSString *errMsg, NSArray *results) {
|
||||
if (succ && results) {
|
||||
dispatch_async(dispatch_get_main_queue(), ^{
|
||||
if (results.count) {
|
||||
[weakSelf.resultSet setObject:results forKey:@(TUISearchResultModuleContact)];
|
||||
} else {
|
||||
[weakSelf.resultSet removeObjectForKey:@(TUISearchResultModuleContact)];
|
||||
}
|
||||
dispatch_group_leave(group);
|
||||
});
|
||||
return;
|
||||
}
|
||||
dispatch_group_leave(group);
|
||||
}];
|
||||
}
|
||||
|
||||
// Group
|
||||
if ((modules == TUISearchResultModuleAll) || (modules & TUISearchResultModuleGroup)) {
|
||||
request = YES;
|
||||
dispatch_group_enter(group);
|
||||
[self searchGroups:keyword
|
||||
callback:^(BOOL succ, NSString *errMsg, NSArray *results) {
|
||||
if (succ && results) {
|
||||
dispatch_async(dispatch_get_main_queue(), ^{
|
||||
if (results.count) {
|
||||
[weakSelf.resultSet setObject:results forKey:@(TUISearchResultModuleGroup)];
|
||||
} else {
|
||||
[weakSelf.resultSet removeObjectForKey:@(TUISearchResultModuleGroup)];
|
||||
}
|
||||
dispatch_group_leave(group);
|
||||
});
|
||||
return;
|
||||
}
|
||||
dispatch_group_leave(group);
|
||||
}];
|
||||
}
|
||||
|
||||
// Chat history
|
||||
if ((modules == TUISearchResultModuleAll) || (modules & TUISearchResultModuleChatHistory)) {
|
||||
request = YES;
|
||||
dispatch_group_enter(group);
|
||||
[self searchChatHistory:keyword
|
||||
param:param
|
||||
callback:^(BOOL succ, NSString *_Nullable errMsg, NSArray<TUISearchResultCellModel *> *_Nullable results) {
|
||||
if (succ && results) {
|
||||
dispatch_async(dispatch_get_main_queue(), ^{
|
||||
if (results.count) {
|
||||
[weakSelf.resultSet setObject:results forKey:@(TUISearchResultModuleChatHistory)];
|
||||
} else {
|
||||
[weakSelf.resultSet removeObjectForKey:@(TUISearchResultModuleChatHistory)];
|
||||
}
|
||||
});
|
||||
}
|
||||
dispatch_group_leave(group);
|
||||
}];
|
||||
}
|
||||
|
||||
if (!request) {
|
||||
if ([self.delegate respondsToSelector:@selector(onSearchError:)]) {
|
||||
[self.delegate onSearchError:@"search module not exists"];
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
dispatch_group_notify(group, dispatch_get_main_queue(), ^{
|
||||
if ([self.delegate respondsToSelector:@selector(onSearchResults:forModules:)]) {
|
||||
[self.delegate onSearchResults:self.resultSet forModules:modules];
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
- (void)searchContacts:(NSString *)keyword callback:(TUISearchResultCallback)callback {
|
||||
if (keyword == nil || callback == nil) {
|
||||
if (callback) {
|
||||
callback(NO, @"invalid parameters, keyword is null", nil);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
V2TIMFriendSearchParam *param = [[V2TIMFriendSearchParam alloc] init];
|
||||
param.keywordList = @[ keyword ];
|
||||
param.isSearchUserID = YES;
|
||||
param.isSearchNickName = YES;
|
||||
param.isSearchRemark = YES;
|
||||
[V2TIMManager.sharedInstance searchFriends:param
|
||||
succ:^(NSArray<V2TIMFriendInfoResult *> *infoList) {
|
||||
NSMutableArray *arrayM = [NSMutableArray array];
|
||||
for (V2TIMFriendInfoResult *result in infoList) {
|
||||
TUISearchResultCellModel *cellModel = [[TUISearchResultCellModel alloc] init];
|
||||
NSString *title = result.friendInfo.friendRemark;
|
||||
if (title.length == 0) {
|
||||
title = result.friendInfo.userFullInfo.nickName;
|
||||
}
|
||||
if (title.length == 0) {
|
||||
title = result.friendInfo.userID;
|
||||
}
|
||||
|
||||
NSString *why = @"";
|
||||
if ([result.friendInfo.friendRemark.lowercaseString containsString:keyword.lowercaseString]) {
|
||||
why = result.friendInfo.friendRemark;
|
||||
} else if ([result.friendInfo.userFullInfo.nickName.lowercaseString containsString:keyword.lowercaseString]) {
|
||||
why = result.friendInfo.userFullInfo.nickName;
|
||||
} else if ([result.friendInfo.userID.lowercaseString containsString:keyword.lowercaseString]) {
|
||||
why = result.friendInfo.userID;
|
||||
}
|
||||
if (why.length) {
|
||||
if ([why isEqualToString:title]) {
|
||||
why = nil;
|
||||
} else {
|
||||
why = [NSString stringWithFormat:TIMCommonLocalizableString(TUIKitSearchResultMatchFormat), why];
|
||||
}
|
||||
}
|
||||
|
||||
cellModel.titleAttributeString = [TUISearchDataProvider attributeStringWithText:title key:keyword];
|
||||
cellModel.detailsAttributeString = [TUISearchDataProvider attributeStringWithText:why key:keyword];
|
||||
cellModel.groupID = nil;
|
||||
cellModel.avatarUrl = result.friendInfo.userFullInfo.faceURL;
|
||||
cellModel.avatarImage = DefaultAvatarImage;
|
||||
cellModel.context = result.friendInfo;
|
||||
[arrayM addObject:cellModel];
|
||||
}
|
||||
callback(YES, nil, arrayM);
|
||||
}
|
||||
fail:^(int code, NSString *desc) {
|
||||
callback(NO, desc, nil);
|
||||
}];
|
||||
}
|
||||
|
||||
- (void)searchGroups:(NSString *)keyword callback:(TUISearchResultCallback)callback {
|
||||
if (keyword == nil || callback == nil) {
|
||||
if (callback) {
|
||||
callback(NO, @"invalid parameters, keyword is null", nil);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
TUISearchGroupParam *param = [[TUISearchGroupParam alloc] init];
|
||||
param.keywordList = @[ keyword ];
|
||||
param.isSearchGroupID = YES;
|
||||
param.isSearchGroupName = YES;
|
||||
param.isSearchGroupMember = YES;
|
||||
param.isSearchMemberRemark = YES;
|
||||
param.isSearchMemberUserID = YES;
|
||||
param.isSearchMemberNickName = YES;
|
||||
param.isSearchMemberNameCard = YES;
|
||||
[TUISearchGroupDataProvider searchGroups:param
|
||||
succ:^(NSArray<TUISearchGroupResult *> *_Nonnull resultSet) {
|
||||
NSMutableArray *arrayM = [NSMutableArray array];
|
||||
for (TUISearchGroupResult *result in resultSet) {
|
||||
TUISearchResultCellModel *cellModel = [[TUISearchResultCellModel alloc] init];
|
||||
NSString *title = result.groupInfo.groupName;
|
||||
if (title.length == 0) {
|
||||
title = result.groupInfo.groupID;
|
||||
}
|
||||
cellModel.titleAttributeString = [TUISearchDataProvider attributeStringWithText:title key:keyword];
|
||||
cellModel.detailsAttributeString = nil;
|
||||
cellModel.groupID = result.groupInfo.groupID;
|
||||
cellModel.groupType = result.groupInfo.groupType;
|
||||
cellModel.avatarImage = DefaultGroupAvatarImageByGroupType(result.groupInfo.groupType);
|
||||
cellModel.avatarUrl = result.groupInfo.faceURL;
|
||||
cellModel.context = result.groupInfo;
|
||||
[arrayM addObject:cellModel];
|
||||
|
||||
if (result.matchField == TUISearchGroupMatchFieldGroupID) {
|
||||
NSString *text = [NSString stringWithFormat:TIMCommonLocalizableString(TUIKitSearchResultMatchGroupIDFormat), result.matchValue];
|
||||
cellModel.detailsAttributeString = [TUISearchDataProvider attributeStringWithText:text key:keyword];
|
||||
} else if (result.matchField == TUISearchGroupMatchFieldMember && result.matchMembers.count) {
|
||||
NSString *text = TIMCommonLocalizableString(TUIKitSearchResultMatchGroupMember);
|
||||
for (int i = 0; i < result.matchMembers.count; i++) {
|
||||
TUISearchGroupMemberMatchResult *memberResult = result.matchMembers[i];
|
||||
text = [text stringByAppendingString:memberResult.memberMatchValue];
|
||||
if (i < result.matchMembers.count - 1) {
|
||||
text = [text stringByAppendingString:@"、"];
|
||||
}
|
||||
}
|
||||
cellModel.detailsAttributeString = [TUISearchDataProvider attributeStringWithText:text key:keyword];
|
||||
}
|
||||
}
|
||||
callback(YES, nil, arrayM);
|
||||
}
|
||||
fail:^(NSInteger code, NSString *_Nonnull desc) {
|
||||
callback(NO, desc, nil);
|
||||
}];
|
||||
}
|
||||
|
||||
- (void)searchChatHistory:(NSString *)keyword param:(NSDictionary<TUISearchParamKey, id> *__nullable)paramters callback:(TUISearchResultCallback)callback {
|
||||
if (keyword == nil || callback == nil) {
|
||||
if (callback) {
|
||||
callback(NO, @"invalid parameters, keyword is null", nil);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
NSUInteger pageSize = TUISearchDefaultPageSize;
|
||||
NSUInteger pageIndex = 0;
|
||||
NSString *conversationID = nil;
|
||||
NSArray *allKeys = paramters.allKeys;
|
||||
BOOL displayWithConveration = YES;
|
||||
|
||||
if ([allKeys containsObject:TUISearchChatHistoryParamKeyCount]) {
|
||||
pageSize = [paramters[TUISearchChatHistoryParamKeyCount] integerValue];
|
||||
}
|
||||
if ([allKeys containsObject:TUISearchChatHistoryParamKeyPage]) {
|
||||
pageIndex = [paramters[TUISearchChatHistoryParamKeyPage] integerValue];
|
||||
}
|
||||
if ([allKeys containsObject:TUISearchChatHistoryParamKeyConversationId]) {
|
||||
conversationID = paramters[TUISearchChatHistoryParamKeyConversationId];
|
||||
displayWithConveration = NO;
|
||||
}
|
||||
|
||||
V2TIMMessageSearchParam *param = [[V2TIMMessageSearchParam alloc] init];
|
||||
param.keywordList = @[ keyword ];
|
||||
param.messageTypeList = nil;
|
||||
param.conversationID = conversationID;
|
||||
param.searchTimePosition = 0;
|
||||
param.searchTimePeriod = 0;
|
||||
param.pageIndex = pageIndex;
|
||||
param.pageSize = pageSize;
|
||||
[V2TIMManager.sharedInstance searchLocalMessages:param
|
||||
succ:^(V2TIMMessageSearchResult *searchResult) {
|
||||
if (searchResult.totalCount == 0) {
|
||||
if (callback) {
|
||||
callback(YES, nil, @[]);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
NSMutableArray *conversationIds = [NSMutableArray array];
|
||||
NSMutableDictionary *conversationInfoMap = [NSMutableDictionary dictionary];
|
||||
NSMutableDictionary *conversationMessageMap = [NSMutableDictionary dictionary];
|
||||
NSMutableDictionary *conversationCountMap = [NSMutableDictionary dictionary];
|
||||
NSArray<V2TIMMessageSearchResultItem *> *messageSearchResultItems = searchResult.messageSearchResultItems;
|
||||
for (V2TIMMessageSearchResultItem *searchItem in messageSearchResultItems) {
|
||||
NSString *conversationID = searchItem.conversationID;
|
||||
NSUInteger messageCount = searchItem.messageCount;
|
||||
NSArray<V2TIMMessage *> *messageList = searchItem.messageList ?: @[];
|
||||
if (conversationID.length == 0) {
|
||||
continue;
|
||||
}
|
||||
[conversationIds addObject:conversationID];
|
||||
conversationMessageMap[conversationID] = messageList;
|
||||
conversationCountMap[conversationID] = @(messageCount);
|
||||
}
|
||||
|
||||
if (conversationIds.count == 0) {
|
||||
if (callback) {
|
||||
callback(YES, nil, @[]);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
[V2TIMManager.sharedInstance getConversationList:conversationIds
|
||||
succ:^(NSArray<V2TIMConversation *> *list) {
|
||||
for (V2TIMConversation *conversation in list) {
|
||||
if (conversation.conversationID) {
|
||||
conversationInfoMap[conversation.conversationID] = conversation;
|
||||
}
|
||||
}
|
||||
NSMutableArray *arrayM = [NSMutableArray array];
|
||||
for (NSString *conversationId in conversationIds) {
|
||||
if (![conversationInfoMap.allKeys containsObject:conversationId]) {
|
||||
continue;
|
||||
}
|
||||
|
||||
V2TIMConversation *conv = conversationInfoMap[conversationId];
|
||||
NSArray *messageList = conversationMessageMap[conversationId];
|
||||
NSUInteger count = [conversationCountMap[conversationId] integerValue];
|
||||
if (displayWithConveration) {
|
||||
TUISearchResultCellModel *cellModel = [[TUISearchResultCellModel alloc] init];
|
||||
NSString *desc = [NSString stringWithFormat:TIMCommonLocalizableString(TUIKitSearchResultDisplayChatHistoryCountFormat), count];
|
||||
if (messageList.count == 1) {
|
||||
V2TIMMessage *firstMessage = messageList.firstObject;
|
||||
desc = [TUISearchDataProvider matchedTextForMessage:(V2TIMMessage *)firstMessage withKey:keyword];
|
||||
}
|
||||
cellModel.title = conv.showName;
|
||||
cellModel.detailsAttributeString = [TUISearchDataProvider attributeStringWithText:desc key:messageList.count == 1 ? keyword : nil];
|
||||
cellModel.groupID = conv.groupID;
|
||||
cellModel.avatarImage = conv.type == V2TIM_GROUP ? DefaultGroupAvatarImageByGroupType(conv.groupType) : DefaultAvatarImage;
|
||||
cellModel.groupType = conv.groupType;
|
||||
cellModel.avatarUrl = conv.faceUrl;
|
||||
cellModel.context = @{
|
||||
kSearchChatHistoryConversationId : conversationId,
|
||||
kSearchChatHistoryConverationInfo : conv,
|
||||
kSearchChatHistoryConversationMsgs : messageList
|
||||
};
|
||||
[arrayM addObject:cellModel];
|
||||
} else {
|
||||
for (V2TIMMessage *message in messageList) {
|
||||
TUISearchResultCellModel *cellModel = [[TUISearchResultCellModel alloc] init];
|
||||
cellModel.title = message.nickName ?: message.sender;
|
||||
NSString *desc = [TUISearchDataProvider matchedTextForMessage:message withKey:keyword];
|
||||
cellModel.detailsAttributeString = [TUISearchDataProvider attributeStringWithText:desc key:keyword];
|
||||
cellModel.groupID = conv.groupID;
|
||||
cellModel.groupType = conv.groupType;
|
||||
cellModel.avatarUrl = message.faceURL;
|
||||
cellModel.avatarImage = conv.type == V2TIM_GROUP ? DefaultGroupAvatarImageByGroupType(conv.groupType) : DefaultAvatarImage;
|
||||
cellModel.context = message;
|
||||
[arrayM addObject:cellModel];
|
||||
}
|
||||
}
|
||||
}
|
||||
TUISearchResultCellModel *lastCellModel = [arrayM lastObject];
|
||||
lastCellModel.hideSeparatorLine = YES;
|
||||
|
||||
if (callback) {
|
||||
callback(YES, nil, arrayM);
|
||||
}
|
||||
}
|
||||
fail:^(int code, NSString *desc) {
|
||||
if (callback) {
|
||||
callback(NO, desc, nil);
|
||||
}
|
||||
}];
|
||||
}
|
||||
fail:^(int code, NSString *desc) {
|
||||
if (callback) {
|
||||
callback(NO, desc, nil);
|
||||
}
|
||||
if (code == ERR_SDK_INTERFACE_NOT_SUPPORT) {
|
||||
[TUITool postUnsupportNotificationOfService:TIMCommonLocalizableString(TUIKitErrorUnsupportIntefaceSearch)];
|
||||
}
|
||||
}];
|
||||
}
|
||||
|
||||
+ (NSAttributedString *)attributeStringWithText:(NSString *)text key:(NSString *)key {
|
||||
if (text.length == 0) {
|
||||
return nil;
|
||||
}
|
||||
|
||||
if (key == nil || key.length == 0 || ![text.lowercaseString tui_containsString:key.lowercaseString]) {
|
||||
NSAttributedString *attributeString = [[NSAttributedString alloc] initWithString:text
|
||||
attributes:@{NSForegroundColorAttributeName : [UIColor darkGrayColor]}];
|
||||
return attributeString;
|
||||
}
|
||||
|
||||
NSMutableAttributedString *attr = [[NSMutableAttributedString alloc] initWithString:text
|
||||
attributes:@{NSForegroundColorAttributeName : [UIColor darkGrayColor]}];
|
||||
|
||||
NSUInteger loc = 0;
|
||||
NSUInteger len = text.length;
|
||||
while (len > 0) {
|
||||
NSRange range = [text.lowercaseString rangeOfString:key.lowercaseString options:NSCaseInsensitiveSearch range:NSMakeRange(loc, len)];
|
||||
if (range.length) {
|
||||
[attr addAttribute:NSForegroundColorAttributeName value:TIMCommonDynamicColor(@"primary_theme_color", @"#147AFF") range:range];
|
||||
loc = range.location + 1;
|
||||
len = text.length - loc;
|
||||
} else {
|
||||
len = 0;
|
||||
loc = 0;
|
||||
}
|
||||
}
|
||||
return [[NSAttributedString alloc] initWithAttributedString:attr];
|
||||
}
|
||||
|
||||
+ (NSString *)matchedTextForMessage:(V2TIMMessage *)msg withKey:(NSString *)key {
|
||||
if (key.length == 0) {
|
||||
return @"";
|
||||
}
|
||||
|
||||
if ((msg.elemType == V2TIM_ELEM_TYPE_TEXT) && [msg.textElem.text tui_containsString:key]) {
|
||||
return msg.textElem.text;
|
||||
} else if ((msg.elemType == V2TIM_ELEM_TYPE_IMAGE) && [msg.imageElem.path tui_containsString:key]) {
|
||||
return msg.imageElem.path;
|
||||
} else if ((msg.elemType == V2TIM_ELEM_TYPE_SOUND) && [msg.soundElem.path tui_containsString:key]) {
|
||||
return msg.soundElem.path;
|
||||
} else if (msg.elemType == V2TIM_ELEM_TYPE_VIDEO) {
|
||||
if ([msg.videoElem.videoPath tui_containsString:key]) {
|
||||
return msg.videoElem.videoPath;
|
||||
} else if ([msg.videoElem.snapshotPath tui_containsString:key]) {
|
||||
return msg.videoElem.snapshotPath;
|
||||
} else {
|
||||
return @"";
|
||||
}
|
||||
} else if ((msg.elemType == V2TIM_ELEM_TYPE_FILE) && [msg.fileElem.path tui_containsString:key]) {
|
||||
return msg.fileElem.path;
|
||||
} else if (msg.elemType == V2TIM_ELEM_TYPE_MERGER) {
|
||||
NSArray *abM = msg.mergerElem.abstractList;
|
||||
NSString *abs = @"";
|
||||
for (NSString *ab in abM) {
|
||||
abs = [abs stringByAppendingString:ab];
|
||||
abs = [abs stringByAppendingString:@","];
|
||||
}
|
||||
if ([msg.mergerElem.title tui_containsString:key]) {
|
||||
return msg.mergerElem.title;
|
||||
} else if ([abs tui_containsString:key]) {
|
||||
return abs;
|
||||
} else {
|
||||
return @"";
|
||||
}
|
||||
}
|
||||
|
||||
return @"";
|
||||
}
|
||||
|
||||
#pragma mark - Lazy
|
||||
- (NSMutableDictionary *)resultSet {
|
||||
if (_resultSet == nil) {
|
||||
_resultSet = [NSMutableDictionary dictionary];
|
||||
}
|
||||
return _resultSet;
|
||||
}
|
||||
|
||||
@end
|
||||
Reference in New Issue
Block a user