43 lines
2.3 KiB
Java
43 lines
2.3 KiB
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 cn.wildfirechat.common.ErrorCode;
|
|
import win.liyufan.im.IMTopic;
|
|
|
|
import static cn.wildfirechat.common.ErrorCode.ERROR_CODE_SUCCESS;
|
|
import static win.liyufan.im.IMTopic.HandleFriendRequestTopic;
|
|
|
|
@Handler(IMTopic.AddFriendRequestTopic)
|
|
public class AddFriendHandler extends GroupHandler<WFCMessage.AddFriendRequest> {
|
|
@Override
|
|
public ErrorCode action(ByteBuf ackPayload, String clientID, String fromUser, ProtoConstants.RequestSourceType requestSourceType, WFCMessage.AddFriendRequest request, Qos1PublishHandler.IMCallback callback) {
|
|
long[] head = new long[1];
|
|
boolean isAdmin = requestSourceType == ProtoConstants.RequestSourceType.Request_From_Admin;
|
|
boolean isRobot = requestSourceType == ProtoConstants.RequestSourceType.Request_From_Robot;
|
|
ErrorCode errorCode = m_messagesStore.saveAddFriendRequest(fromUser, request, head, isAdmin, isRobot);
|
|
if (errorCode == ERROR_CODE_SUCCESS) {
|
|
WFCMessage.User user = m_messagesStore.getUserInfo(request.getTargetUid());
|
|
if (user != null && user.getType() == ProtoConstants.UserType.UserType_Normal) {
|
|
publisher.publishNotification(IMTopic.NotifyFriendRequestTopic, request.getTargetUid(), head[0], fromUser, request.getReason());
|
|
} else if(user != null && user.getType() == ProtoConstants.UserType.UserType_Robot) {
|
|
if(m_messagesStore.isRobotAutoAcceptFriendRequest()) {
|
|
WFCMessage.HandleFriendRequest handleFriendRequest = WFCMessage.HandleFriendRequest.newBuilder().setTargetUid(fromUser).setStatus(ProtoConstants.FriendRequestStatus.RequestStatus_Accepted).build();
|
|
mServer.onApiMessage(request.getTargetUid(), null, handleFriendRequest.toByteArray(), 0, fromUser, HandleFriendRequestTopic, requestSourceType);
|
|
}
|
|
}
|
|
}
|
|
return errorCode;
|
|
}
|
|
}
|