//
//  ChatManager.m
//  EaseMob-Example
//
//  Created by zzg on 2018/7/11.
//  Copyright © 2018年. All rights reserved.
//

#import "ChatManager.h"
#import <Hyphenate/Hyphenate.h>
#import "NSString+Util.h"
#import "NSObject+Util.h"
#import "Constant.h"

@implementation ChatManager

RCT_EXPORT_MODULE();

RCT_EXPORT_METHOD(getConversation:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *conversationId = [allParams objectForKey:@"conversationId"];
    EMConversationType type = [[allParams objectForKey:@"type"] intValue];
    BOOL aIfCreate = [[allParams objectForKey:@"ifCreate"] boolValue];
    EMConversation *conversation = [[EMClient sharedClient].chatManager getConversation:conversationId type:type createIfNotExist:aIfCreate];
    resolve([conversation objectToJSONString]);
}

RCT_EXPORT_METHOD(getAllConversations:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSArray *conversations = [[EMClient sharedClient].chatManager getAllConversations];
    NSMutableArray *dicArray = [NSMutableArray array];
    [conversations enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        [dicArray addObject:[obj objectToDictionary]];
    }];
    resolve(JSONSTRING(dicArray));
}

RCT_EXPORT_METHOD(deleteConversation:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *conversationId = [allParams objectForKey:@"conversationId"];
    BOOL isDeleteMessages = [[allParams objectForKey:@"ifClearAllMessage"] boolValue];
    [[EMClient sharedClient].chatManager deleteConversation:conversationId isDeleteMessages:isDeleteMessages completion:^(NSString *aConversationId, EMError *error) {
        if (!error) {
            resolve(@"{}");
        } else {
            reject([NSString stringWithFormat:@"%ld", (NSInteger)error.code], error.errorDescription, nil);
        }
    }];
}

RCT_EXPORT_METHOD(sendMessage:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *conversationId = [allParams objectForKey:@"conversationId"];
    EMChatType chatType = [[allParams objectForKey:@"chatType"] intValue];
    NSString *from = [[EMClient sharedClient] currentUsername];
    NSString *to = [allParams objectForKey:@"to"];
    EMMessageBodyType messageType = [[allParams objectForKey:@"messageType"] intValue];
    NSDictionary *messageExt = [allParams objectForKey:@"messageExt"];
    NSDictionary *bodyDic = [allParams objectForKey:@"body"];
    EMMessageBody *body;
    switch (messageType) {
        case EMMessageBodyTypeText: {
            body = [[EMTextMessageBody alloc] initWithText:bodyDic[@"text"] ];
        }
            break;
        case EMMessageBodyTypeImage: {
            NSString *path = bodyDic[@"path"];
            NSData *imageData = [NSData dataWithContentsOfFile:path];
            body = [[EMImageMessageBody alloc] initWithData:imageData displayName:@"image"];
        }
            break;
        case EMMessageBodyTypeLocation: {
            double latitude = [bodyDic[@"latitude"] doubleValue];
            double longitude = [bodyDic[@"longitude"] doubleValue];
            NSString *address = bodyDic[@"address"];
            body = [[EMLocationMessageBody alloc] initWithLatitude:latitude longitude:longitude address:address];
        }
            break;
        case EMMessageBodyTypeVoice: {
            NSString *path = bodyDic[@"path"];
            int duration = [bodyDic[@"duration"] intValue];
            EMVoiceMessageBody *voiceBody = [[EMVoiceMessageBody alloc] initWithLocalPath:path displayName:@"audio"];
            voiceBody.duration = duration;
            body = voiceBody;
        }
            break;
        case EMMessageBodyTypeVideo: {
            NSString *path = bodyDic[@"path"];
            body = [[EMVideoMessageBody alloc] initWithLocalPath:path displayName:@"video"];
        }
            break;
        case EMMessageBodyTypeFile: {
            NSString *path = bodyDic[@"path"];
            body = [[EMFileMessageBody alloc] initWithLocalPath:path displayName:@"file"];
        }
            break;
        case EMMessageBodyTypeCmd: {
            NSString *action = bodyDic[@"action"];
            body = [[EMCmdMessageBody alloc] initWithAction:action];
        }
            break;
        default:
            break;
    }
    EMMessage *message = [[EMMessage alloc] initWithConversationID:conversationId from:from to:to body:body ext:messageExt];
    message.chatType = chatType;
    [[EMClient sharedClient].chatManager sendMessage:message progress:nil completion:^(EMMessage *message, EMError *error) {
        resolve([message objectToJSONString]);
    }];
}

RCT_EXPORT_METHOD(insertMessage:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *conversationId = [allParams objectForKey:@"conversationId"];
    NSInteger chatType = [[allParams objectForKey:@"chatType"] intValue];
    NSString *fromParam = [allParams objectForKey:@"from"];
    NSString *from;
    if (fromParam) {
        from = fromParam;
    } else {
        from = [[EMClient sharedClient] currentUsername];
    }
    NSString *to = [allParams objectForKey:@"to"];
    EMMessageBodyType messageType = [[allParams objectForKey:@"messageType"] intValue];
    NSDictionary *messageExt = [allParams objectForKey:@"messageExt"];
    NSDictionary *bodyDic = [allParams objectForKey:@"body"];
    long long timestamp = [[allParams objectForKey:@"timestamp"] longLongValue];
    long long localTime = [[allParams objectForKey:@"localTime"] longLongValue];
    EMMessageBody *body;
    BOOL needDownload = NO;
    switch (messageType) {
        case EMMessageBodyTypeText: {
            body = [[EMTextMessageBody alloc] initWithText:[((NSObject *)bodyDic[@"text"]) description]];
        }
            break;
        case EMMessageBodyTypeImage: {
            EMImageMessageBody *imageBody = [[EMImageMessageBody alloc] initWithData:nil displayName:@"image"];
            imageBody.remotePath = bodyDic[@"remotePath"];
            imageBody.secretKey = bodyDic[@"secretKey"];
            body = imageBody;
            needDownload = YES;
        }
            break;
        case EMMessageBodyTypeLocation: {
            double latitude = [bodyDic[@"latitude"] doubleValue];
            double longitude = [bodyDic[@"longitude"] doubleValue];
            NSString *address = bodyDic[@"address"];
            body = [[EMLocationMessageBody alloc] initWithLatitude:latitude longitude:longitude address:address];
        }
            break;
        case EMMessageBodyTypeVoice: {
            EMVoiceMessageBody *voiceBody = [[EMVoiceMessageBody alloc] initWithData:nil displayName:@"audio"];
            voiceBody.remotePath = bodyDic[@"remotePath"];
            voiceBody.secretKey = bodyDic[@"secretKey"];
            voiceBody.duration = [bodyDic[@"duration"] intValue];
            body = voiceBody;
            needDownload = YES;
        }
            break;
        case EMMessageBodyTypeVideo: {
            EMVideoMessageBody *videoBody = [[EMVideoMessageBody alloc] initWithData:nil displayName:@"video"];
            videoBody.remotePath = bodyDic[@"remotePath"];
            videoBody.secretKey = bodyDic[@"secretKey"];
            body = videoBody;
            needDownload = YES;
        }
            break;
        case EMMessageBodyTypeFile: {
            EMFileMessageBody *fileBody = [[EMFileMessageBody alloc] initWithData:nil displayName:@"file"];
            fileBody.remotePath = bodyDic[@"remotePath"];
            fileBody.secretKey = bodyDic[@"secretKey"];
            body = fileBody;
            needDownload = YES;
        }
            break;
        default:
            break;
    }
    EMMessage *message = [[EMMessage alloc] initWithConversationID:conversationId from:from to:to body:body ext:messageExt];
    message.chatType = chatType;
    message.timestamp = timestamp;
    message.localTime = localTime;
    message.status = EMMessageStatusSucceed;
    message.direction = [[allParams objectForKey:@"direction"] intValue];
    if (needDownload) {
        [[EMClient sharedClient].chatManager downloadMessageAttachment:message progress:nil completion:^(EMMessage *result, EMError *error) {
            EMConversation *conversation = [[EMClient sharedClient].chatManager getConversation:conversationId type:chatType createIfNotExist:YES];
            [conversation insertMessage:result error:nil];
            resolve([result objectToJSONString]);
        }];
    } else {
        EMConversation *conversation = [[EMClient sharedClient].chatManager getConversation:conversationId type:chatType createIfNotExist:YES];
        [conversation insertMessage:message error:nil];
        resolve([message objectToJSONString]);
    }
}

RCT_EXPORT_METHOD(loadMessages:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *conversationId = [allParams objectForKey:@"conversationId"];
    EMConversationType type = [[allParams objectForKey:@"chatType"] intValue];
    EMConversation *conversation = [[EMClient sharedClient].chatManager getConversation:conversationId type:type createIfNotExist:YES];
    NSString *fromId = [allParams objectForKey:@"fromId"];
    int count = [[allParams objectForKey:@"count"] intValue];
    EMMessageSearchDirection searchDirection = [[allParams objectForKey:@"searchDirection"] intValue];
    [[EMClient sharedClient].chatManager asyncFetchHistoryMessagesFromServer:conversation.conversationId conversationType:type startMessageId:fromId pageSize:count completion:^(EMCursorResult *aResult, EMError *aError) {
        [conversation loadMessagesStartFromId:fromId count:count searchDirection:searchDirection completion:^(NSArray *aMessages, EMError *error) {
            NSMutableArray *dicArray = [NSMutableArray array];
            [aMessages enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
                [dicArray addObject:[obj objectToDictionary]];
            }];
            if (!error) {
                resolve([dicArray objectToJSONString]);
            } else {
                reject([NSString stringWithFormat:@"%ld", (NSInteger)error.code], error.errorDescription, nil);
            }
        }];
    }];
}

RCT_EXPORT_METHOD(fetchHistoryMessagesFromServer:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *conversationId = [allParams objectForKey:@"conversationId"];
    EMConversationType type = [[allParams objectForKey:@"chatType"] intValue];
    EMConversation *conversation = [[EMClient sharedClient].chatManager getConversation:conversationId type:type createIfNotExist:YES];
    NSString *fromId = [allParams objectForKey:@"fromId"];
    int count = [[allParams objectForKey:@"count"] intValue];
    [[EMClient sharedClient].chatManager asyncFetchHistoryMessagesFromServer:conversation.conversationId conversationType:type startMessageId:fromId pageSize:count completion:^(EMCursorResult *aResult, EMError *aError) {
        if (!aError) {
            resolve([@{} objectToJSONString]);
        } else {
            reject([NSString stringWithFormat:@"%ld", (NSInteger)aError.code], aError.errorDescription, nil);
        }
    }];
}

RCT_EXPORT_METHOD(recallMessage:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *conversationId = [allParams objectForKey:@"conversationId"];
    EMConversationType type = [[allParams objectForKey:@"chatType"] intValue];
    EMConversation *conversation = [[EMClient sharedClient].chatManager getConversation:conversationId type:type createIfNotExist:NO];
    NSString *messageId = [allParams objectForKey:@"messageId"];
    EMMessage *message = [conversation loadMessageWithId:messageId error:nil];
    [[EMClient sharedClient].chatManager recallMessage:message completion:^(EMMessage *aMessage, EMError *aError) {
        if (!aError) {
            resolve([aMessage objectToJSONString]);
        } else {
            reject([NSString stringWithFormat:@"%ld", (NSInteger)aError.code], aError.errorDescription, nil);
        }
    }];
}

RCT_EXPORT_METHOD(deleteMessage:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *conversationId = [allParams objectForKey:@"conversationId"];
    EMConversationType type = [[allParams objectForKey:@"chatType"] intValue];
    EMConversation *conversation = [[EMClient sharedClient].chatManager getConversation:conversationId type:type createIfNotExist:NO];
    NSString *messageId = [allParams objectForKey:@"messageId"];
    EMError *error = nil;
    [conversation deleteMessageWithId:messageId error:&error];
    if (!error) {
        resolve(@"{}");
    } else {
        reject([NSString stringWithFormat:@"%ld", (NSInteger)error.code], error.errorDescription, nil);
    }
}

RCT_EXPORT_METHOD(markAllMessagesAsRead:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *conversationId = [allParams objectForKey:@"conversationId"];
    EMConversationType type = [[allParams objectForKey:@"chatType"] intValue];
    EMConversation *conversation = [[EMClient sharedClient].chatManager getConversation:conversationId type:type createIfNotExist:NO];
    EMError *error = nil;
    [conversation markAllMessagesAsRead:&error];
    if (!error) {
        resolve(@"{}");
    } else {
        reject([NSString stringWithFormat:@"%ld", (NSInteger)error.code], error.errorDescription, nil);
    }
}

RCT_EXPORT_METHOD(deleteAllMessages:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *conversationId = [allParams objectForKey:@"conversationId"];
    EMConversationType type = [[allParams objectForKey:@"chatType"] intValue];
    EMConversation *conversation = [[EMClient sharedClient].chatManager getConversation:conversationId type:type createIfNotExist:NO];
    EMError *error = nil;
    [conversation deleteAllMessages:&error];
    if (!error) {
        resolve(@"{}");
    } else {
        reject([NSString stringWithFormat:@"%ld", (NSInteger)error.code], error.errorDescription, nil);
    }
}

RCT_EXPORT_METHOD(updateMessageExt:(NSString *)params
                  resolver:(RCTPromiseResolveBlock)resolve
                  rejecter:(RCTPromiseRejectBlock)reject) {
    NSDictionary *allParams = [params jsonStringToDictionary];
    NSString *messageId = [allParams objectForKey:@"messageId"];
    NSDictionary *ext = [allParams objectForKey:@"ext"];
    EMMessage *message = [[EMClient sharedClient].chatManager getMessageWithMessageId:messageId];
    NSMutableDictionary *extM = message.ext.mutableCopy;
    [extM addEntriesFromDictionary:ext];
    message.ext = extM.copy;
    [[EMClient sharedClient].chatManager updateMessage:message completion:^(EMMessage *aMessage, EMError *aError) {
        if (!aError) {
            resolve(@"{}");
        } else {
            reject([NSString stringWithFormat:@"%ld", (NSInteger)aError.code], aError.errorDescription, nil);
        }
    }];
}

@end