iOS 环信 EaseUI 的优化

背景

去年曾为了项目的一个即时通信功能集成过环信，因为考虑开发的低成本，就直接使用了环信的 EaseUI

环信的文档提供了很多好友管理、群聊操作的API，但当你看到下图后，这些api直接可以无视掉

所以环信当作工具来使用即可，千万别用他的后台系统（我的公司安卓那边莫名的相信环信后台，白白浪费了近两个多星期）

为什么需要优化？

因为环信的EaseUI，是根据环信自己后台开发的，所以需要让EaseUI更好地跟我们自己的后台关联起来

红色圈出来的就是我接下来所做的优化的内容

ps:一开始我以为，在拉取好友列表的时候判断好，就基本没有大碍！但是后面经测试，在聊天的过程中，如果对方把你删除，聊天依旧可以继续，所以就需要对 EaseUI 的源码开刀。

具体流程

EaseUI 使用起来十分便捷，只需要自己创建一个继承于 EaseMessageViewController 的控制器即可

接着实现 EaseMessageViewControllerDataSource 该协议的

- (id<IMessageModel>)messageViewController:(EaseMessageViewController *)viewController
                           modelForMessage:(EMMessage *)message;

便可以对聊天内容的环信用户设置自定义头像和名字

接来下便开始修改他的源码:

第一步

打开 EaseUI 里面的 EaseMessageViewController.h 文件

在 EaseMessageViewControllerDelegate 下添加以下代码

/**
 * 是否需要发送这条消息  return false 将会阻止该消息的发送  - pfboy add
 *
 * @param viewController _
 * @param message 自己发送出去的消息
 * @param block 对发出去的数据的，进行校验，通过block的参数isDataRight返回数据的正确性
 *
 */
- (void) messageViewController:(EaseMessageViewController *)viewController willSendMessage:(EMMessage *)message  resultBlock:(void(^)(BOOL isDataRight)) block;

/**
 * 是否需要发送这条消息  return false 将会阻止该消息的发送  - pfboy add
 *
 * @param viewController _
 * @param message 接收到的数据
 * @param block 对接收到的数据，进行校验，通过block的参数isDataRight返回数据的正确性
 *
 */
- (void) messageViewController:(EaseMessageViewController *)viewController willReceiveMessage:(EMMessage *)message  resultBlock:(void(^)(BOOL isDataRight)) block;

这个是为了最后，我们可以在自己的 Controller 通过这两个方法，来判断发收信息的是否来自好友或者群聊

因为需要通过http请求来判断好友的真实性，所以使用block 进行控制

第二步

接着打开文件 EaseMessageViewController.m  并滑动到1721行

将原 _sendMessage 函数

- (void)_sendMessage:(EMMessage *)message
{
    if (self.conversation.type == EMConversationTypeGroupChat){
        message.chatType = EMChatTypeGroupChat;
    }
    else if (self.conversation.type == EMConversationTypeChatRoom){
        message.chatType = EMChatTypeChatRoom;
    }
    
    [self addMessageToDataSource:message
                        progress:nil];
    
    __weak typeof(self) weakself = self;
    [[EMClient sharedClient].chatManager sendMessage:message progress:nil completion:^(EMMessage *aMessage, EMError *aError) {
        if (!aError) {
            [weakself _refreshAfterSentMessage:aMessage];
        }
        else {
            [weakself.tableView reloadData];
        }
    }];
}

改成以下代码

/判断该消息是否需要截断  pfboy add
- (void) _sendCorrectMessage:(EMMessage *)message{
    __weak typeof(self) weakself = self;
    [[EMClient sharedClient].chatManager sendMessage:message progress:nil completion:^(EMMessage *aMessage, EMError *aError) {
        if (!aError) {
            [weakself _refreshAfterSentMessage:aMessage];
        }
        else {
            [weakself.tableView reloadData];
        }
    }];
}

// hook 这个地址
- (void) _sendMessage:(EMMessage *)message
{
    if (self.conversation.type == EMConversationTypeGroupChat){
        message.chatType = EMChatTypeGroupChat;
    }
    else if (self.conversation.type == EMConversationTypeChatRoom){
        message.chatType = EMChatTypeChatRoom;
    }
    
    [self addMessageToDataSource:message
                        progress:nil];
    
    //判断该消息是否需要截断  pfboy add
    if ([self.delegate respondsToSelector:@selector(messageViewController:willSendMessage:resultBlock:)]) {
        __weak typeof(self) weakself = self;
        [self.delegate messageViewController:self willSendMessage:message resultBlock:^(BOOL isDataRight) {
            if (isDataRight) {
                [weakself _sendCorrectMessage:message];
            }
        }];
    } else {
        // 如果没有实现该代理，则不会更改源码的流程
        [self _sendCorrectMessage:message];
    }
    
}

第三步

EaseMessageViewController.h 滑动到 1137 行将方法 statusButtonSelcted 替换成以下代码

- (void) _resendMessage:(id<IMessageModel>)model{
    __weak typeof(self) weakself = self;
    [[[EMClient sharedClient] chatManager] resendMessage:model.message progress:nil completion:^(EMMessage *message, EMError *error) {
        if (!error) {
            [weakself _refreshAfterSentMessage:message];
        }
        else {
            [weakself.tableView reloadData];
        }
    }];
    
    [self.tableView reloadData];
}

// hook 重发地址
- (void)statusButtonSelcted:(id<IMessageModel>)model withMessageCell:(EaseMessageCell*)messageCell
{
    if ((model.messageStatus != EMMessageStatusFailed) && (model.messageStatus != EMMessageStatusPending))
    {
        return;
    }
    //判断该消息是否需要截断  pfboy add
    if ([self.delegate respondsToSelector:@selector(messageViewController:willSendMessage:resultBlock:)]) {
        __weak typeof(self) weakself = self;
        [self.delegate messageViewController:self willSendMessage:model.message resultBlock:^(BOOL isDataRight) {
            if (isDataRight) {
                [weakself _resendMessage:model];
            }
        }];
    } else {
        // 如果没有实现该代理，则不会更改源码的流程
        [self _resendMessage:model];
    }
    
}

第四步

EaseMessageViewController.h 滑动到 1442 行，将 didReceiveMessages 替换成以下代码

- (void) _didReceiveMessages:(EMMessage *)message {
    [self addMessageToDataSource:message progress:nil];
    
    [self _sendHasReadResponseForMessages:@[message]
                                   isRead:NO];
    
    if ([self _shouldMarkMessageAsRead])
    {
        [self.conversation markMessageAsReadWithId:message.messageId error:nil];
    }
}
// hook 这里
- (void)didReceiveMessages:(NSArray *)aMessages
{
    for (EMMessage *message in aMessages) {
        if ([self.conversation.conversationId isEqualToString:message.conversationId]) {
            
            //判断该消息是否需要截断
            if ([self.delegate respondsToSelector:@selector(messageViewController:willReceiveMessage:resultBlock:)]) {
                __weak typeof(self) weakself = self;
                [self.delegate messageViewController:self willReceiveMessage:message resultBlock:^(BOOL isDataRight) {
                    //判断是否需要接受该数据
                    if (isDataRight) {
                        [weakself _didReceiveMessages:message];
                    }
                }];
            } else {
                // 如果没有实现该代理，则不会更改源码的流程
                [self _didReceiveMessages:message];
            }
        }
    }
}

通过以上四步，便可以在消息发送或接收前将消息截取下来，再通过protocol，返回到我们自己的控制器进行管理

最后在自己的 Controller 内，便可以将发送的消息与自己服务器的数据关联起来，实现对 EaseUI 的优化

// MARK: -EaseMessageViewControllerDelegate
func messageViewController(_ viewController: EaseMessageViewController!, willSend message: EMMessage!, resultBlock block: ((Bool) -> Void)!) {
     checkAccountCorrect(hx_account: message.conversationId) { (value) in
         block(value)
     }
}
    
func messageViewController(_ viewController: EaseMessageViewController!, willReceive message: EMMessage!, resultBlock block: ((Bool) -> Void)!) {
     checkAccountCorrect(hx_account: message.conversationId) { (value) in
         block(value)
     }
}

ps:方法 checkAccountCorrect 的内容主要与我们自己的服务器交互校验数据

本人能力有限，如果有什么更好的方法或者建议，欢迎指正