MiMO_iOS/TUIKit/TUIChat/BaseDataProvider/Base/TUIChatBaseDataProvider.m

//  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