首次提交

TUIKit/TUIChat/BaseDataProvider/Base/TUIChatBaseDataProvider.m

@@ -0,0 +1,258 @@
+
+//  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