34 lines
1.3 KiB
Java
34 lines
1.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.common.ErrorCode;
|
|
import cn.wildfirechat.proto.ProtoConstants;
|
|
import cn.wildfirechat.proto.WFCMessage;
|
|
import io.moquette.spi.impl.Qos1PublishHandler;
|
|
import io.netty.buffer.ByteBuf;
|
|
import win.liyufan.im.IMTopic;
|
|
|
|
import static cn.wildfirechat.common.ErrorCode.ERROR_CODE_SUCCESS;
|
|
|
|
@Handler(IMTopic.GetApplicationTokenRequestTopic)
|
|
public class GetApplicationTokenHandler extends IMHandler<WFCMessage.AuthCodeRequest> {
|
|
@Override
|
|
public ErrorCode action(ByteBuf ackPayload, String clientID, String fromUser, ProtoConstants.RequestSourceType requestSourceType, WFCMessage.AuthCodeRequest request, Qos1PublishHandler.IMCallback callback) {
|
|
String authCode = m_messagesStore.getApplicationAuthCode(fromUser, request.getTargetId(), request.getType(), request.getHost());
|
|
if(authCode != null) {
|
|
byte[] data = WFCMessage.IDBuf.newBuilder().setId(authCode).build().toByteArray();
|
|
ackPayload.ensureWritable(data.length).writeBytes(data);
|
|
return ERROR_CODE_SUCCESS;
|
|
}
|
|
return ErrorCode.INVALID_PARAMETER;
|
|
|
|
}
|
|
}
|