IM-server-new/broker/src/main/java/io/moquette/imhandler/HandleFriendRequestHandler.java

/*
 * This file is part of the Wildfire Chat package.
 * (c) Heavyrain2012 <[email protected]>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

package io.moquette.imhandler;

import cn.wildfirechat.proto.ProtoConstants;
import cn.wildfirechat.proto.WFCMessage;
import io.moquette.spi.impl.Qos1PublishHandler;
import io.netty.buffer.ByteBuf;
import common.cn.wildfirechat.ErrorCode;
import win.liyufan.im.IMTopic;
import win.liyufan.im.MessageShardingUtil;

import static common.cn.wildfirechat.ErrorCode.ERROR_CODE_SUCCESS;

@Handler(IMTopic.HandleFriendRequestTopic)
public class HandleFriendRequestHandler extends IMHandler<WFCMessage.HandleFriendRequest> {
    @Override
    public ErrorCode action(ByteBuf ackPayload, String clientID, String fromUser, boolean isAdmin, WFCMessage.HandleFriendRequest request, Qos1PublishHandler.IMCallback callback) {
            WFCMessage.Message.Builder builder = WFCMessage.Message.newBuilder();
            builder.setFromUser(request.getTargetUid());
            long[] heads = new long[2];
            ErrorCode errorCode = m_messagesStore.handleFriendRequest(fromUser, request, builder, heads, isAdmin);

            if (errorCode == ERROR_CODE_SUCCESS && !isAdmin) {
                long messageId = MessageShardingUtil.generateId();
                long timestamp = System.currentTimeMillis();
                builder.setMessageId(messageId);
                builder.setServerTimestamp(timestamp);
                saveAndPublish(request.getTargetUid(), null, builder.build());

                WFCMessage.MessageContent.Builder contentBuilder = WFCMessage.MessageContent.newBuilder().setType(90).setContent("以上是打招呼信息");
                builder = WFCMessage.Message.newBuilder();
                builder.setFromUser(request.getTargetUid());
                builder.setConversation(WFCMessage.Conversation.newBuilder().setTarget(fromUser).setLine(0).setType(ProtoConstants.ConversationType.ConversationType_Private).build());
                builder.setContent(contentBuilder);
                timestamp = System.currentTimeMillis();
                builder.setServerTimestamp(timestamp);

                messageId = MessageShardingUtil.generateId();
                builder.setMessageId(messageId);
                saveAndPublish(request.getTargetUid(), null, builder.build());

                contentBuilder.setContent("你们已经成为好友了，现在可以开始聊天了");
                builder.setContent(contentBuilder);
                messageId = MessageShardingUtil.generateId();
                builder.setMessageId(messageId);
                timestamp = System.currentTimeMillis();
                builder.setServerTimestamp(timestamp);
                saveAndPublish(request.getTargetUid(), null, builder.build());

                publisher.publishNotification(IMTopic.NotifyFriendTopic, request.getTargetUid(), heads[0]);
                publisher.publishNotification(IMTopic.NotifyFriendTopic, fromUser, heads[1]);
            }
            return errorCode;
    }
}