Files
featherVoice/TUIKit/TUIChat/BaseDataProvider/Base/TUIChatBaseDataProvider.m
2025-08-08 10:49:36 +08:00

259 lines
12 KiB
Objective-C

// Created by Tencent on 2023/06/09.
// Copyright © 2023 Tencent. All rights reserved.
@import ImSDK_Plus;
#import <objc/runtime.h>
#import <TUICore/NSDictionary+TUISafe.h>
#import <TUICore/TUICore.h>
#import <TUICore/TUIThemeManager.h>
#import "TUIChatBaseDataProvider.h"
#import "TUIMessageBaseDataProvider.h"
#import <TUICore/TUILogin.h>
#import "TUIChatDefine.h"
#define Input_SendBtn_Key @"Input_SendBtn_Key"
#define Input_SendBtn_Title @"Input_SendBtn_Title"
#define Input_SendBtn_ImageName @"Input_SendBtn_ImageName"
static NSArray *gCustomInputBtnInfo = nil;
@implementation TUIChatBaseDataProvider
+ (void)initialize {
[NSNotificationCenter.defaultCenter addObserver:self selector:@selector(onChangeLanguage) name:TUIChangeLanguageNotification object:nil];
}
+ (void)onChangeLanguage {
gCustomInputBtnInfo = nil;
}
+ (NSArray *)customInputBtnInfo {
if (gCustomInputBtnInfo == nil) {
gCustomInputBtnInfo = @[ @{
Input_SendBtn_Key : TUIInputMoreCellKey_Link,
Input_SendBtn_Title : TIMCommonLocalizableString(TUIKitMoreLink),
Input_SendBtn_ImageName : @"chat_more_link_img"
} ];
}
return gCustomInputBtnInfo;
}
- (void)getForwardMessageWithCellDatas:(NSArray<TUIMessageCellData *> *)uiMsgs
toTargets:(NSArray<TUIChatConversationModel *> *)targets
Merge:(BOOL)merge
ResultBlock:(void (^)(TUIChatConversationModel *targetConversation, NSArray<V2TIMMessage *> *msgs))resultBlock
fail:(nullable V2TIMFail)fail {
if (uiMsgs.count == 0) {
if (fail) {
fail(ERR_SVR_PROFILE_INVALID_PARAMETERS, @"uiMsgs is empty");
}
return;
}
dispatch_apply(targets.count, dispatch_get_global_queue(0, 0), ^(size_t index) {
TUIChatConversationModel *convCellData = targets[index];
NSMutableArray *tmpMsgs = [NSMutableArray array];
for (TUIMessageCellData *uiMsg in uiMsgs) {
V2TIMMessage *msg = uiMsg.innerMessage;
if (msg) {
[tmpMsgs addObject:msg];
}
}
NSArray *msgs = [NSArray arrayWithArray:tmpMsgs];
msgs = [msgs sortedArrayUsingComparator:^NSComparisonResult(V2TIMMessage *obj1, V2TIMMessage *obj2) {
if ([obj1.timestamp timeIntervalSince1970] == [obj2.timestamp timeIntervalSince1970]) {
return obj1.seq > obj2.seq;
} else {
return [obj1.timestamp compare:obj2.timestamp];
}
}];
if (!merge) {
NSMutableArray *forwardMsgs = [NSMutableArray array];
for (V2TIMMessage *msg in msgs) {
V2TIMMessage *forwardMessage = [V2TIMManager.sharedInstance createForwardMessage:msg];
if (forwardMessage) {
forwardMessage.isExcludedFromUnreadCount = [TUIConfig defaultConfig].isExcludedFromUnreadCount;
forwardMessage.isExcludedFromLastMessage = [TUIConfig defaultConfig].isExcludedFromLastMessage;
[forwardMsgs addObject:forwardMessage];
}
}
if (resultBlock) {
resultBlock(convCellData, forwardMsgs);
}
return;
}
@weakify(self);
NSString *loginUserId = [V2TIMManager.sharedInstance getLoginUser];
[V2TIMManager.sharedInstance getUsersInfo:@[ loginUserId ]
succ:^(NSArray<V2TIMUserFullInfo *> *infoList) {
@strongify(self);
NSString *myName = loginUserId;
if (infoList.firstObject.nickName.length > 0) {
myName = infoList.firstObject.nickName;
}
NSString *title = [self.delegate dataProvider:self mergeForwardTitleWithMyName:myName];
NSMutableArray *abstactList = [NSMutableArray array];
if (uiMsgs.count > 0) {
[abstactList addObject:[self abstractDisplayWithMessage:msgs[0]]];
}
if (uiMsgs.count > 1) {
[abstactList addObject:[self abstractDisplayWithMessage:msgs[1]]];
}
if (uiMsgs.count > 2) {
[abstactList addObject:[self abstractDisplayWithMessage:msgs[2]]];
}
NSString *compatibleText = TIMCommonLocalizableString(TUIKitRelayCompatibleText);
V2TIMMessage *mergeMessage = [V2TIMManager.sharedInstance createMergerMessage:msgs
title:title
abstractList:abstactList
compatibleText:compatibleText];
if (mergeMessage == nil) {
if (fail) {
fail(ERR_NO_SUCC_RESULT, @"failed to merge-forward");
}
return;
}
mergeMessage.isExcludedFromUnreadCount = [TUIConfig defaultConfig].isExcludedFromUnreadCount;
mergeMessage.isExcludedFromLastMessage = [TUIConfig defaultConfig].isExcludedFromLastMessage;
if (resultBlock) {
resultBlock(convCellData, @[ mergeMessage ]);
}
}
fail:fail];
});
}
- (NSString *)abstractDisplayWithMessage:(V2TIMMessage *)msg {
return nil;
}
@end
#pragma mark - TUIChatBaseDataProvider (IMSDK)
@implementation TUIChatBaseDataProvider (IMSDK)
+ (void)getTotalUnreadMessageCountWithSuccBlock:(void (^)(UInt64 totalCount))succ fail:(nullable V2TIMFail)fail {
[V2TIMManager.sharedInstance getTotalUnreadMessageCount:succ fail:fail];
}
+ (void)saveDraftWithConversationID:(NSString *)conversationId Text:(NSString *)text {
NSString *draft = text;
draft = [draft stringByTrimmingCharactersInSet:NSCharacterSet.whitespaceAndNewlineCharacterSet];
[[V2TIMManager sharedInstance] setConversationDraft:conversationId draftText:draft succ:nil fail:nil];
}
#pragma mark - C2C
+ (void)getFriendInfoWithUserId:(nullable NSString *)userID
SuccBlock:(void (^)(V2TIMFriendInfoResult *friendInfoResult))succ
failBlock:(nullable V2TIMFail)fail {
NSParameterAssert(userID);
if (fail && !userID) {
fail(ERR_INVALID_PARAMETERS, @"userID is nil");
return;
}
[[V2TIMManager sharedInstance] getFriendsInfo:@[ userID ]
succ:^(NSArray<V2TIMFriendInfoResult *> *resultList) {
V2TIMFriendInfoResult *result = resultList.firstObject;
succ(result);
}
fail:fail];
}
+ (void)getUserInfoWithUserId:(NSString *)userID SuccBlock:(void (^)(V2TIMUserFullInfo *userInfo))succ failBlock:(nullable V2TIMFail)fail {
NSParameterAssert(userID);
if (!userID) {
if (fail) {
fail(ERR_INVALID_PARAMETERS, @"userID is nil");
}
return;
}
[[V2TIMManager sharedInstance] getUsersInfo:@[ userID ]
succ:^(NSArray<V2TIMUserFullInfo *> *infoList) {
V2TIMUserFullInfo *info = infoList.firstObject;
if (succ) {
succ(info);
}
}
fail:fail];
}
#pragma mark - Group
+ (void)getGroupInfoWithGroupID:(NSString *)groupID SuccBlock:(void (^)(V2TIMGroupInfoResult *groupResult))succ failBlock:(nullable V2TIMFail)fail {
NSParameterAssert(groupID);
if (fail && !groupID) {
fail(ERR_INVALID_PARAMETERS, @"groupID is nil");
return;
}
[[V2TIMManager sharedInstance] getGroupsInfo:@[ groupID ]
succ:^(NSArray<V2TIMGroupInfoResult *> *groupResultList) {
V2TIMGroupInfoResult *result = groupResultList.firstObject;
if (result && result.resultCode == 0) {
if (succ) {
succ(result);
}
} else {
if (fail) {
fail(result.resultCode, result.resultMsg);
}
}
}
fail:fail];
}
+ (void)findMessages:(NSArray *)msgIDs callback:(void (^)(BOOL succ, NSString *error_message, NSArray *msgs))callback {
[V2TIMManager.sharedInstance findMessages:msgIDs
succ:^(NSArray<V2TIMMessage *> *msgs) {
if (callback) {
callback(YES, nil, msgs);
}
}
fail:^(int code, NSString *desc) {
if (callback) {
callback(NO, desc, @[]);
}
}];
}
+ (void)insertLocalTipsMessage:(NSString *)content chatID:(NSString *)chatID isGroup:(BOOL)isGroup {
NSDictionary *dic = @{
@"version" : @(1),
BussinessID : @"local_tips",
@"content" : content.length>0?content:@""
};
NSData *data = [NSJSONSerialization dataWithJSONObject:dic options:NSJSONWritingPrettyPrinted error:nil];
V2TIMMessage *msg = [[V2TIMManager sharedInstance] createCustomMessage:data];
if (msg == nil) {
return;
}
NSString *messageID = nil;
NSString *senderID = [TUILogin getUserID];
if (isGroup) {
NSString *groupID = chatID.length>0?chatID:@"";
messageID = [V2TIMManager.sharedInstance insertGroupMessageToLocalStorage:msg to:groupID sender:senderID succ:^{
NSDictionary *userInfo = @{@"message" : msg,@"needScrollToBottom":@"1"};
[[NSNotificationCenter defaultCenter] postNotificationName:TUIChatInsertMessageWithoutUpdateUINotification object:nil userInfo:userInfo];
} fail:^(int code, NSString *desc) {
}];
}
else {
NSString *userID = chatID.length>0?chatID:@"";
messageID = [V2TIMManager.sharedInstance insertC2CMessageToLocalStorage:msg to:userID sender:senderID succ:^{
NSDictionary *userInfo = @{@"message" : msg,@"needScrollToBottom":@"1"};
[[NSNotificationCenter defaultCenter] postNotificationName:TUIChatInsertMessageWithoutUpdateUINotification object:nil userInfo:userInfo];
} fail:^(int code, NSString *desc) {
}];
}
}
@end