//
//  TUIChatModifyMessageHelper.m
//  TUIChat
//
//  Created by wyl on 2022/6/13.
//

#import "TUIChatModifyMessageHelper.h"
#import "TUITool.h"
#import "TUICloudCustomDataTypeCenter.h"
#import "TUILogin.h"

#define RETRY_MIN_TIME 500
#define RETRY_MAX_TIME 3000

#define TIMES_CONTROL 3

@interface TUIChatModifyMessageObj : NSObject

@property (nonatomic, assign) NSInteger time;

@property (nonatomic, copy) NSString *msgID;

@property (nonatomic, strong) V2TIMMessage *msg;
//react
@property (nonatomic, copy) NSString *reactEmoji;
//reply
@property (nonatomic, strong) NSDictionary *simpleCurrentContent;
//revoke
@property (nonatomic, copy) NSString *revokeMsgID;

@end

@implementation TUIChatModifyMessageObj
- (instancetype)init {
    if (self = [super init]) {
        self.time = 0;
    }
    return self;
}

- (V2TIMMessage *)resolveOriginCloudCustomData:(V2TIMMessage *)rootMsg {
    if (self.reactEmoji) {
        return  [self.class resolveOriginCloudCustomData:rootMsg reactEmoji:self.reactEmoji];
    }
    if (self.simpleCurrentContent) {
        return [self.class resolveOriginCloudCustomData:rootMsg simpleCurrentContent:self.simpleCurrentContent];
    }
    if (self.revokeMsgID) {
        return [self.class resolveOriginCloudCustomData:rootMsg revokeMsgID:self.revokeMsgID];
    }
    return rootMsg;
}
//react
+ (V2TIMMessage *)resolveOriginCloudCustomData:(V2TIMMessage *)rootMsg  reactEmoji:(NSString *)emojiName{
    NSMutableDictionary *resultDic = [[NSMutableDictionary alloc] initWithCapacity:5];

    if (rootMsg.cloudCustomData) {
        NSDictionary * originDic = [TUITool jsonData2Dictionary:rootMsg.cloudCustomData];
        if (originDic && [originDic isKindOfClass:[NSDictionary class]]) {
            [resultDic addEntriesFromDictionary:originDic];
        }
    }

    NSDictionary *orignMessageReactDic = resultDic[@"messageReact"];
    TUIReactModelMessageReact *welcome  = [[TUIReactModelMessageReact alloc] init];
    
    NSString *loginUser = [TUILogin getUserID]?:@"";
    
    if (orignMessageReactDic &&[orignMessageReactDic isKindOfClass:NSDictionary.class]) {
        [welcome applyWithDic:orignMessageReactDic emojiName:emojiName loginUser:loginUser];
    }
    else {
        welcome.version = @"1";
        TUIReactModelReacts * react = [[TUIReactModelReacts alloc] init];
        react.emojiKey = emojiName;
        react.emojiIdArray =  [NSMutableArray arrayWithArray:@[loginUser]];
        welcome.reacts = [NSMutableArray arrayWithCapacity:3];
        [welcome.reacts addObject:react];
    }
    [resultDic setValue:[welcome descriptionDic] forKey:@"messageReact"];
    
    NSData *data = [TUITool dictionary2JsonData:resultDic];
    if (data) {
        rootMsg.cloudCustomData = data;
    }
    return  rootMsg;
}

//Input
+ (V2TIMMessage *)resolveOriginCloudCustomData:(V2TIMMessage *)rootMsg  simpleCurrentContent:(NSDictionary *)simpleCurrentContent  {
    NSMutableDictionary *mudic = [[NSMutableDictionary alloc] initWithCapacity:5];
    NSMutableArray *replies = [[NSMutableArray alloc] initWithCapacity:5];
    NSMutableDictionary *messageReplies = [[NSMutableDictionary alloc] initWithCapacity:5];

    if (rootMsg.cloudCustomData) {
        NSDictionary * originDic = [TUITool jsonData2Dictionary:rootMsg.cloudCustomData];
        if (originDic && [originDic isKindOfClass:[NSDictionary class]]) {
            [messageReplies addEntriesFromDictionary:originDic];
        }
        NSArray * messageRepliesArray = originDic[@"messageReplies"][@"replies"];
        if (messageRepliesArray && [messageRepliesArray isKindOfClass:NSArray.class] && messageRepliesArray.count>0) {
            [replies addObjectsFromArray:messageRepliesArray];
        }
    }
    [replies addObject:simpleCurrentContent];
    [mudic setValue:replies forKey:@"replies"];
    

    [messageReplies setValue:mudic forKey:@"messageReplies"];
    [messageReplies setValue:@"1" forKey:@"version"];

    NSData *data = [TUITool dictionary2JsonData:messageReplies];
    if (data) {
        rootMsg.cloudCustomData = data;
    }
    
    return rootMsg;
}

//revoke

+ (V2TIMMessage *)resolveOriginCloudCustomData:(V2TIMMessage *)rootMsg  revokeMsgID:(NSString *)revokeMsgId  {
    NSMutableDictionary *mudic = [[NSMutableDictionary alloc] initWithCapacity:5];
    NSMutableArray *replies = [[NSMutableArray alloc] initWithCapacity:5];
    NSMutableDictionary *messageReplies = [[NSMutableDictionary alloc] initWithCapacity:5];

    if (rootMsg.cloudCustomData) {
        NSDictionary * originDic = [TUITool jsonData2Dictionary:rootMsg.cloudCustomData];
        if (originDic && [originDic isKindOfClass:[NSDictionary class]]) {
            [messageReplies addEntriesFromDictionary:originDic];
            if (originDic[@"messageReplies"] && [originDic[@"messageReplies"] isKindOfClass:[NSDictionary class]]) {
                NSArray * messageRepliesArray = originDic[@"messageReplies"][@"replies"];
                if (messageRepliesArray && [messageRepliesArray isKindOfClass:NSArray.class] && messageRepliesArray.count>0) {
                    [replies addObjectsFromArray:messageRepliesArray];
                }
            }
        }
        
    }
    
    NSMutableArray *filterReplies = [[NSMutableArray alloc] initWithCapacity:5];
    [replies enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        if (obj && [obj isKindOfClass:NSDictionary.class]) {
            NSDictionary *  dic = (NSDictionary *)obj;
            if (IS_NOT_EMPTY_NSSTRING(dic[@"messageID"]) ) {
                NSString *messageID = dic[@"messageID"];
                if (![messageID isEqualToString:revokeMsgId] ) {
                    [filterReplies addObject:dic];
                }
            }
        }
    }];

    [mudic setValue:filterReplies forKey:@"replies"];
    

    [messageReplies setValue:mudic forKey:@"messageReplies"];
    [messageReplies setValue:@"1" forKey:@"version"];

    NSData *data = [TUITool dictionary2JsonData:messageReplies];
    if (data) {
        rootMsg.cloudCustomData = data;
    }
    
    return rootMsg;
}
@end

@interface ModifyCustomOperation : NSOperation

@property (nonatomic, strong) TUIChatModifyMessageObj *obj;

@property (nonatomic, copy) void (^SuccessBlock)(void);

@property (nonatomic, copy) void (^FailedBlock)(int code, NSString * desc, V2TIMMessage *msg);


@property (nonatomic, assign) BOOL bees_executing;

@property (nonatomic, assign) BOOL bees_finished;

@end

@implementation ModifyCustomOperation



- (void)dealloc {
    NSLog(@"operation-------dealloc");
}

- (void)start {
    
    if (self.isCancelled)  {
        [self completeOperation];
        return;
    }
    
    __block V2TIMMessage *resolveMsg = nil;

    __weak typeof(self) weakSelf = self;

    self.bees_executing = YES;

    resolveMsg = [self.obj resolveOriginCloudCustomData:self.obj.msg];

    [[V2TIMManager sharedInstance] modifyMessage:resolveMsg completion:^(int code, NSString *desc, V2TIMMessage *msg) {
        __strong typeof(weakSelf)strongSelf = weakSelf;
        if (code !=0) {
            if (strongSelf.FailedBlock) {
                strongSelf.FailedBlock(code, desc, msg);
            }
        }
        else {
            if (strongSelf.SuccessBlock) {
                strongSelf.SuccessBlock();
            }
        }
        
        [strongSelf completeOperation];
    }];
}

- (void)completeOperation {
  self.bees_executing = NO;
  self.bees_finished = YES;
}


- (void)cancel {
  [super cancel];
  [self completeOperation];
}

#pragma mark - settter and getter
- (void)setBees_executing:(BOOL)bees_executing {
  [self willChangeValueForKey:@"isExecuting"];
  _bees_executing = bees_executing;
  [self didChangeValueForKey:@"isExecuting"];
}

- (void)setBees_finished:(BOOL)bees_finished {
  [self willChangeValueForKey:@"isFinished"];
  _bees_finished = bees_finished;
  [self didChangeValueForKey:@"isFinished"];
}

- (BOOL)isExecuting {
  return self.bees_executing;
}

- (BOOL)isFinished {
  return self.bees_finished;
}

@end

@interface TUIChatModifyMessageHelper() <V2TIMAdvancedMsgListener>

@property (nonatomic, strong) NSMutableDictionary<NSString *, TUIChatModifyMessageObj *> *modifyMessageHelperMap;

@property (nonatomic, strong) NSOperationQueue *queue;

@end

@implementation TUIChatModifyMessageHelper


+ (TUIChatModifyMessageHelper *)defaultHelper
{
    static dispatch_once_t onceToken;
    static TUIChatModifyMessageHelper *config;
    dispatch_once(&onceToken, ^{
        config = [[TUIChatModifyMessageHelper alloc] init];
    });
    return config;
}

- (id)init
{
    self = [super init];
    if(self){
        [self registerTUIKitNotification];
        self.modifyMessageHelperMap = [NSMutableDictionary dictionaryWithCapacity:10];
        [self setupOpQueue];
    }
    return self;
}
- (void)setupOpQueue {
    self.queue = [[NSOperationQueue alloc]init];
    self.queue.maxConcurrentOperationCount = 1;
}

- (void)registerTUIKitNotification {
    [[V2TIMManager sharedInstance] addAdvancedMsgListener:self];
}

- (void)onRecvMessageModified:(V2TIMMessage *)msg {
    NSString *msgID = msg.msgID;
    TUIChatModifyMessageObj *obj = self.modifyMessageHelperMap[msgID];
    if (obj && [obj isKindOfClass:[TUIChatModifyMessageObj class]] ) {
        //update;
        obj.msg = msg;
    }
}
#pragma mark - public

- (void)modifyMessage:(V2TIMMessage *)msg
           reactEmoji:(NSString *)emojiName {
    [self modifyMessage:msg reactEmoji:emojiName simpleCurrentContent:nil revokeMsgID:nil timeControl:0];
}

- (void)modifyMessage:(V2TIMMessage *)msg
 simpleCurrentContent:(NSDictionary *)simpleCurrentContent {
    [self modifyMessage:msg reactEmoji:nil simpleCurrentContent:simpleCurrentContent revokeMsgID:nil timeControl:0];
}

- (void)modifyMessage:(V2TIMMessage *)msg
          revokeMsgID:(NSString *)revokeMsgID {
    [self modifyMessage:msg reactEmoji:nil simpleCurrentContent:nil revokeMsgID:revokeMsgID timeControl:0];
}


#pragma mark - private
- (void)modifyMessage:(V2TIMMessage *)msg
           reactEmoji:(NSString *)emojiName
 simpleCurrentContent:(NSDictionary *)simpleCurrentContent
          revokeMsgID:(NSString *)revokeMsgID
          timeControl:(NSInteger)time {
    
    NSString *msgID = msg.msgID;
    if (!IS_NOT_EMPTY_NSSTRING(msgID)) {
        return;
    }

    TUIChatModifyMessageObj * obj = [[TUIChatModifyMessageObj alloc] init];
    obj.msgID = msgID;
    obj.msg = msg;
    obj.time = time;
    if (emojiName) {
        obj.reactEmoji = emojiName;
    }
    if (simpleCurrentContent) {
        obj.simpleCurrentContent = simpleCurrentContent;
    }
    if (revokeMsgID) {
        obj.revokeMsgID = revokeMsgID;
    }
    
    [self.modifyMessageHelperMap setObject:obj forKey:obj.msgID];

    __weak typeof(self) weakSelf = self;

    ModifyCustomOperation *modifyop = [[ModifyCustomOperation alloc] init];
    modifyop.obj = obj;
    modifyop.SuccessBlock = ^{
        __strong typeof(weakSelf)strongSelf = weakSelf;
        [strongSelf.modifyMessageHelperMap removeObjectForKey:msgID];
    };

    modifyop.FailedBlock = ^(int code, NSString *desc, V2TIMMessage *msg) {
        __strong typeof(weakSelf)strongSelf = weakSelf;
        if (obj.time <= TIMES_CONTROL) {
            int delay;
            delay = [self getRandomNumber:RETRY_MIN_TIME to:RETRY_MAX_TIME];
            
            TUIChatModifyMessageObj *obj = strongSelf.modifyMessageHelperMap[msgID];
            //update
            obj.msg = msg;
            obj.time = obj.time + 1;
            
            dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(delay * NSEC_PER_MSEC)), dispatch_get_main_queue(), ^{
                TUIChatModifyMessageObj *obj = strongSelf.modifyMessageHelperMap[msgID];
                if (obj && [obj isKindOfClass:[TUIChatModifyMessageObj class]] ) {
                    [strongSelf modifyMessage:obj.msg reactEmoji:obj.reactEmoji simpleCurrentContent:obj.simpleCurrentContent revokeMsgID: obj.revokeMsgID timeControl:obj.time];
                }
            });
        }
        else {
            
            [strongSelf.modifyMessageHelperMap removeObjectForKey:msgID];
        }
    };
    
    if (!modifyop.isCancelled) {
        [self.queue addOperation:modifyop];
    }
    
}

- (int)getRandomNumber:(int)from to:(int)to {
    return (int)(from + (arc4random() % (to - from + 1)));
}

@end
