TIMSDK/iOS/TUIKit/TUIChat/DataProvider/TUIMessageDataProvider+MessageDeal.m

//
//  TUIMessageDataProvider+MessageDeal.m
//  TUIChat
//
//  Created by wyl on 2022/3/22.
//

#import "TUIMessageDataProvider+MessageDeal.h"
#import "TUIChatDataProvider.h"
#import "TUIMessageDataProvider.h"
#import "TUIMessageCellData.h"
#import "TUIImageMessageCellData.h"
@implementation TUIMessageDataProvider (MessageDeal)

- (void)loadOriginMessageFromReplyData:(TUIReplyMessageCellData *)replycellData dealCallback:(void(^)(void))callback
{
    if (replycellData.originMsgID.length == 0) {
        if (callback) {
            callback();
        }
        return;
    }
    
    @weakify(replycellData)
    [TUIChatDataProvider findMessages:@[replycellData.originMsgID] callback:^(BOOL succ, NSString * _Nonnull error_message, NSArray * _Nonnull msgs) {
        @strongify(replycellData)
        if (!succ) {
            replycellData.quoteData = [replycellData getQuoteData:nil];
            replycellData.originMessage = nil;
            if (callback) {
                callback();
            }
            return;
        }
        V2TIMMessage *originMessage = msgs.firstObject;
        if (originMessage == nil) {
            replycellData.quoteData = [replycellData getQuoteData:nil];
            if (callback) {
                callback();
            }
            return;
        }
        TUIMessageCellData *cellData = [TUIMessageDataProvider getCellData:originMessage];
        replycellData.originCellData = cellData;
        if ([cellData isKindOfClass:TUIImageMessageCellData.class]) {
            // 原始消息是图片
            TUIImageMessageCellData *imageData = (TUIImageMessageCellData *)cellData;
            [imageData downloadImage:TImage_Type_Thumb];
            replycellData.quoteData = [replycellData getQuoteData:imageData];
            replycellData.originMessage = originMessage;
            if (callback) {
                callback();
            }
        } else if ([cellData isKindOfClass:TUIVideoMessageCellData.class]) {
            // 原始消息是视频
            TUIVideoMessageCellData *videoData = (TUIVideoMessageCellData *)cellData;
            [videoData downloadThumb];
            replycellData.quoteData = [replycellData getQuoteData:videoData];
            replycellData.originMessage = originMessage;
            if (callback) {
                callback();
            }
        } else {
            replycellData.quoteData = [replycellData getQuoteData:cellData];
            replycellData.originMessage = originMessage;
            if (callback) {
                callback();
            }
        }
    }];
}
@end