提交

TUIKit/TUIConversation/BaseDataProvider/TUIFoldConversationListBaseDataProvider.m

@@ -0,0 +1,197 @@
+//
+//  TUIFoldConversationListBaseDataProvider.m
+//  TUIConversation
+//
+//  Created by wyl on 2022/7/21.
+//  Copyright © 2023 Tencent. All rights reserved.
+//
+
+#import "TUIFoldConversationListBaseDataProvider.h"
+#import <ImSDK_Plus/ImSDK_Plus.h>
+#import <TIMCommon/TIMDefine.h>
+#import <TUICore/TUICore.h>
+#import <TUICore/TUILogin.h>
+#import "TUIConversationCellData.h"
+
+@interface TUIFoldConversationListBaseDataProvider (private)
+@property(nonatomic, assign, getter=isLastPage) BOOL lastPage;
+
+- (BOOL)filteConversation:(V2TIMConversation *)conv;
+- (TUIConversationCellData *)cellDataForConversation:(V2TIMConversation *)conversation;
+- (void)sortDataList:(NSMutableArray<TUIConversationCellData *> *)dataList;
+- (void)handleHideConversation:(TUIConversationCellData *)conversation;
+- (void)handleInsertConversationList:(NSArray<TUIConversationCellData *> *)conversationList;
+- (void)handleUpdateConversationList:(NSArray<TUIConversationCellData *> *)conversationList
+                           positions:(NSMutableDictionary<NSString *, NSNumber *> *)positionMaps;
+- (TUIConversationCellData *)cellDataOfGroupID:(NSString *)groupID;
+- (void)handleRemoveConversation:(TUIConversationCellData *)conversation;
+- (void)updateMarkUnreadCount;
+
+@end
+
+@implementation TUIFoldConversationListBaseDataProvider
+
+- (void)loadNexPageConversations {
+    if (self.isLastPage) {
+        return;
+    }
+    @weakify(self);
+    V2TIMConversationListFilter *filter = [[V2TIMConversationListFilter alloc] init];
+    filter.type = V2TIM_GROUP;
+    filter.markType = V2TIM_CONVERSATION_MARK_TYPE_FOLD;
+
+    [V2TIMManager.sharedInstance getConversationListByFilter:filter
+        nextSeq:self.pageIndex
+        count:(int)self.pageSize
+        succ:^(NSArray<V2TIMConversation *> *list, uint64_t nextSeq, BOOL isFinished) {
+          @strongify(self);
+          self.pageIndex = nextSeq;
+          self.lastPage = isFinished;
+          [self preprocess:list];
+        }
+        fail:^(int code, NSString *desc) {
+          @strongify(self);
+          self.lastPage = YES;
+        }];
+}
+- (void)preprocess:(NSArray<V2TIMConversation *> *)v2Convs {
+    if (!NSThread.isMainThread) {
+        __weak typeof(self) weakSelf = self;
+        dispatch_async(dispatch_get_main_queue(), ^{
+          [weakSelf preprocess:v2Convs];
+        });
+        return;
+    }
+
+    NSMutableDictionary<NSString *, NSNumber *> *conversationMap = [NSMutableDictionary dictionary];
+    for (TUIConversationCellData *item in self.conversationList) {
+        if (item.conversationID) {
+            [conversationMap setObject:@([self.conversationList indexOfObject:item]) forKey:item.conversationID];
+        }
+    }
+
+    NSMutableArray *duplicateDataList = [NSMutableArray array];
+    NSMutableArray *addedDataList = [NSMutableArray array];
+    NSMutableArray *markHideDataList = [NSMutableArray array];
+    NSMutableArray *needHideByCancelMarkFoldDataList = [NSMutableArray array];
+
+    for (V2TIMConversation *conv in v2Convs) {
+        if ([self filteConversation:conv]) {
+            continue;
+        }
+
+        if ([TUIConversationCellData isMarkedByHideType:conv.markList]) {
+            if ([TUIConversationCellData isMarkedByHideType:conv.markList]) {
+                [markHideDataList addObject:[self cellDataForConversation:conv]];
+            }
+            continue;
+        }
+
+        TUIConversationCellData *cellData = [self cellDataForConversation:conv];
+        if ([TUIConversationCellData isMarkedByFoldType:conv.markList]) {
+            if ([conversationMap objectForKey:cellData.conversationID]) {
+                [duplicateDataList addObject:cellData];
+            } else {
+                [addedDataList addObject:cellData];
+            }
+        } else {
+            // If not marked as folded, this conversation is the data that needs to be removed from the folded list
+            if ([conversationMap objectForKey:cellData.conversationID]) {
+                [needHideByCancelMarkFoldDataList addObject:cellData];
+            }
+        }
+    }
+
+    if (duplicateDataList.count) {
+        [self sortDataList:duplicateDataList];
+        [self handleUpdateConversationList:duplicateDataList positions:conversationMap];
+    }
+
+    if (addedDataList.count) {
+        [self sortDataList:addedDataList];
+        [self handleInsertConversationList:addedDataList];
+    }
+
+    if (markHideDataList.count) {
+        [self sortDataList:markHideDataList];
+        NSMutableArray *pRemoveCellUIList = [NSMutableArray array];
+        NSMutableDictionary<NSString *, TUIConversationCellData *> *pMarkHideDataMap = [NSMutableDictionary dictionary];
+        for (TUIConversationCellData *item in markHideDataList) {
+            if (item.conversationID) {
+                [pRemoveCellUIList addObject:item];
+                [pMarkHideDataMap setObject:item forKey:item.conversationID];
+            }
+        }
+        for (TUIConversationCellData *item in self.conversationList) {
+            if ([pMarkHideDataMap objectForKey:item.conversationID]) {
+                [pRemoveCellUIList addObject:item];
+            }
+        }
+        for (TUIConversationCellData *item in pRemoveCellUIList) {
+            [self handleHideConversation:item];
+        }
+    }
+
+    if (needHideByCancelMarkFoldDataList.count) {
+        [self sortDataList:needHideByCancelMarkFoldDataList];
+        NSMutableArray *pRemoveCellUIList = [NSMutableArray array];
+        NSMutableDictionary<NSString *, TUIConversationCellData *> *pMarkCancelFoldDataMap = [NSMutableDictionary dictionary];
+        for (TUIConversationCellData *item in needHideByCancelMarkFoldDataList) {
+            if (item.conversationID) {
+                [pRemoveCellUIList addObject:item];
+                [pMarkCancelFoldDataMap setObject:item forKey:item.conversationID];
+            }
+        }
+        for (TUIConversationCellData *item in self.conversationList) {
+            if ([pMarkCancelFoldDataMap objectForKey:item.conversationID]) {
+                [pRemoveCellUIList addObject:item];
+            }
+        }
+        for (TUIConversationCellData *item in pRemoveCellUIList) {
+            [self handleHideConversation:item];
+        }
+    }
+}
+
+- (void)handleRemoveConversation:(TUIConversationCellData *)conversation {
+    NSInteger index = [self.conversationList indexOfObject:conversation];
+    [self.conversationList removeObject:conversation];
+    if (self.delegate && [self.delegate respondsToSelector:@selector(deleteConversationAtIndexPaths:)]) {
+        [self.delegate deleteConversationAtIndexPaths:@[ [NSIndexPath indexPathForRow:index inSection:0] ]];
+    }
+
+    @weakify(self);
+    void (^deleteAction)(void) = ^{
+      [[V2TIMManager sharedInstance] deleteConversation:conversation.conversationID
+                                                   succ:^{
+                                                     @strongify(self);
+                                                     [self updateMarkUnreadCount];
+                                                   }
+                                                   fail:nil];
+    };
+
+    [V2TIMManager.sharedInstance markConversation:@[ conversation.conversationID ]
+        markType:@(V2TIM_CONVERSATION_MARK_TYPE_FOLD)
+        enableMark:NO
+        succ:^(NSArray<V2TIMConversationOperationResult *> *result) {
+          if (deleteAction) {
+              deleteAction();
+          }
+        }
+        fail:^(int code, NSString *desc) {
+          if (deleteAction) {
+              deleteAction();
+          }
+        }];
+
+    [self.needRemoveConversationList addObject:conversation.conversationID];
+}
+
+- (NSMutableArray *)needRemoveConversationList {
+    if (!_needRemoveConversationList) {
+        _needRemoveConversationList = [NSMutableArray array];
+    }
+    return _needRemoveConversationList;
+}
+
+@end